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 {
 
 /**
 * 用户管理 -> 业务层
 */
 @Autowired
 private SUserService sUserService;
 
 /**
 * 文件上传根目录(在Spring的application.yml的配置文件中配置):<br>
 * web:
 *  upload-path: （jar包所在目录）/resources/static/
 */
 @Value("${web.upload-path}")
 private String webUploadPath;
 
 /**
 * ResultVo是一个对象，包含：
 * private int errorCode;
 * private String errorMsg;
 * private Integer total;
 * private Object data;
 */

 /**
 * 基于用户标识的头像上传
 * @param file 图片
 * @param userId 用户标识
 * @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();
   // 获取图片的扩展名
   String extensionName = StringUtils.substringAfter(fileName, ".");
   // 新的图片文件名 = 获取时间戳+"."图片扩展名
   String newFileName = String.valueOf(System.currentTimeMillis()) + "." + extensionName;
   // 数据库保存的目录
   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);

   // 将图片流转换进行BASE64加码
   //BASE64Encoder encoder = new BASE64Encoder();
   //String data = encoder.encode(file.getBytes());

   // 将反斜杠转换为正斜杠
   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, "上传的文件不是图片类型，请重新上传!");
  }
  return resultVo;
 } else {
  resultVo.setError(0, "上传失败，请选择要上传的图片!");
  return resultVo;
 }
 }

}