package com.clou.inteface.domain.web.user;

import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

/**
 * 文件上傳
 * @author Fly
 *
 */
@RestController
public class FileUpload {
 
 /**
 * 用戶管理 -> 業(yè)務(wù)層
 */
 @Autowired
 private SUserService sUserService;
 
 /**
 * 文件上傳根目錄(在Spring的application.yml的配置文件中配置):<br>
 * web:
 * upload-path: （jar包所在目錄）/resources/static/
 */
 @Value("${web.upload-path}")
 private String webUploadPath;
 
 /**
 * ResultVo是一個(gè)對(duì)象，包含：
 * private int errorCode;
 * private String errorMsg;
 * private Integer total;
 * private Object data;
 */

 /**
 * 基于用戶標(biāo)識(shí)的頭像上傳
 * @param file 圖片
 * @param userId 用戶標(biāo)識(shí)
 * @return
 */
 @PostMapping(value = "/fileUpload", consumes = MediaType.MULTIPART_FORM_DATA_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
 public ResultVo fileUpload(@RequestParam("file") MultipartFile file, @RequestParam("userId") Integer userId) {
 ResultVo resultVo = new ResultVo();
 if (!file.isEmpty()) {
 if (file.getContentType().contains("image")) {
 try {
  String temp = "images" + File.separator + "upload" + File.separator;
  // 獲取圖片的文件名
  String fileName = file.getOriginalFilename();
  // 獲取圖片的擴(kuò)展名
  String extensionName = StringUtils.substringAfter(fileName, ".");
  // 新的圖片文件名 = 獲取時(shí)間戳+"."圖片擴(kuò)展名
  String newFileName = String.valueOf(System.currentTimeMillis()) + "." + extensionName;
  // 數(shù)據(jù)庫保存的目錄
  String datdDirectory = temp.concat(String.valueOf(userId)).concat(File.separator);
  // 文件路徑
  String filePath = webUploadPath.concat(datdDirectory);

  File dest = new File(filePath, newFileName);
  if (!dest.getParentFile().exists()) {
  dest.getParentFile().mkdirs();
  }
  // 判斷是否有舊頭像，如果有就先刪除舊頭像，再上傳
  SUser userInfo = sUserService.findUserInfo(userId.toString());
  if (StringUtils.isNotBlank(userInfo.getUserHead())) {
  String oldFilePath = webUploadPath.concat(userInfo.getUserHead());
  File oldFile = new File(oldFilePath);
  if (oldFile.exists()) {
  oldFile.delete();
  }
  }
  // 上傳到指定目錄
  file.transferTo(dest);

  // 將圖片流轉(zhuǎn)換進(jìn)行BASE64加碼
  //BASE64Encoder encoder = new BASE64Encoder();
  //String data = encoder.encode(file.getBytes());

  // 將反斜杠轉(zhuǎn)換為正斜杠
  String data = datdDirectory.replaceAll("\\\\", "/") + newFileName;
  Map<String, Object> resultMap = new HashMap<>();
  resultMap.put("file", data);
  resultVo.setData(resultMap);
  resultVo.setError(1, "上傳成功!");
 } catch (IOException e) {
  resultVo.setError(0, "上傳失敗!");
 }
 } else {
 resultVo.setError(0, "上傳的文件不是圖片類型，請(qǐng)重新上傳!");
 }
 return resultVo;
 } else {
 resultVo.setError(0, "上傳失敗，請(qǐng)選擇要上傳的圖片!");
 return resultVo;
 }
 }

}