package com.heeexy.example.controller;

import com.alibaba.fastjson.JSONObject;
import com.heeexy.example.util.CommonUtil;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import javax.servlet.http.HttpServletRequest;
import java.io.*;
import java.util.*;

@RestController
@RequestMapping("/common")
public class UploadController {
 //設置上傳文件夾
 File uploadPath = null;

 //單文件上傳
 @PostMapping("/upload")
 public JSONObject upload(@RequestParam(value = "file", required = false)MultipartFile file,HttpServletRequest request) throws Exception{
 //定義返回客戶端json對象
 JSONObject returnData = new JSONObject();
 //定義處理流對象
 BufferedOutputStream out = null;

 //將request對象轉成JSONObject對象
 JSONObject jsonObject = CommonUtil.request2Json(request);
 //驗證必填字段
 CommonUtil.hasAllRequired(jsonObject,"user_id,equi_id,upload_type");

 //獲取當前用戶id
 String user_id = jsonObject.getString("user_id");
 //獲取當前設備id
 String equi_id = jsonObject.getString("equi_id");
 //獲取上傳文件的類型 1:巡檢 2:維保
 String upload_type = jsonObject.getString("upload_type");

 //判斷上傳文件類型并設置前置路徑
 File uploadPath = null;
 String basePath = "/root/img";   //基礎文件上傳路徑
 String inspection = "/inspection";  //巡檢文件夾路徑
 String maintenance = "/maintenance";  //維保文件夾路徑

 switch (upload_type){
  case "1":
  uploadPath = new File(basePath+inspection);
  break;
  case "2":
  uploadPath = new File(basePath+maintenance);
  break;
  default:
  uploadPath = new File(basePath);
 }
 //判斷服務器上傳文件夾是否存在
 if(!uploadPath.exists()){
  uploadPath.mkdirs();
 }
 //判斷上傳的文件是否為空
 if (file!=null) {
  //獲取上傳文件后綴
  String houzhui = file.getOriginalFilename().split("\\.")[1];
  //拼接上傳文件保存路徑(當前用戶id+設備id+時間戳.后綴名)
  File fil = new File(uploadPath+"/"+user_id+equi_id+new Date().getTime()+"."+houzhui);
  try {
  //將上傳文件保存到服務器上傳文件夾目錄下
  out = new BufferedOutputStream(new FileOutputStream(fil));
  out.write(file.getBytes());
  out.flush();
  out.close();
  //返回上傳文件的訪問路徑 getAbsolutePath()返回文件上傳的絕對路徑
  returnData.put("message",fil.getName());
  } catch (FileNotFoundException e) {
  e.printStackTrace();
  returnData.put("message", "文件上傳失敗:" + e.getMessage());
  } catch (IOException e) {
  e.printStackTrace();
  returnData.put("message", "文件上傳失敗:" + e.getMessage());
  }finally {
  //關閉處理流
  if(out!=null){out.close();}
  }
 } else {
  returnData.put("message", "文件上傳失敗,文件為空");
 }
 return CommonUtil.successJson(returnData);
 }

 //多文件上傳
 @PostMapping("/batchUpload")
 public JSONObject handleFileUpload(HttpServletRequest request) throws Exception{
 //定義返回客戶端json對象
 JSONObject returnData = new JSONObject();
 //定義處理流對象,處理文件上傳
 BufferedOutputStream stream = null;
 //定義map存儲返回結果集
 Map<String,String> returnfileMap = new HashMap<String, String>();

 //獲取前端上傳的文件列表
 List<MultipartFile> files = ((MultipartHttpServletRequest) request).getFiles("file");
 MultipartFile file = null;

 //將request對象轉成JSONObject對象
 JSONObject jsonObject = CommonUtil.request2Json(request);
 //驗證必填字段,用戶id,設備id,上傳文件類型
 CommonUtil.hasAllRequired(jsonObject,"user_id,equi_id,upload_type");

 //獲取當前用戶id
 String user_id = jsonObject.getString("user_id");
 //獲取當前設備id
 String equi_id = jsonObject.getString("equi_id");
 //獲取上傳文件的類型 1:巡檢 2:維保
 String upload_type = jsonObject.getString("upload_type");

 //判斷上傳文件類型并設置前置路徑
 File uploadPath = null;
 String basePath = "/root/img"; //基礎文件上傳路徑
 String inspection = "/inspection"; //巡檢文件夾路徑
 String maintenance = "/maintenance"; //維保文件夾路徑

 switch (upload_type){
  case "1":
  uploadPath = new File(basePath+inspection);
  break;
  case "2":
  uploadPath = new File(basePath+maintenance);
  break;
  default:
  uploadPath = new File(basePath);
 }
 //判斷服務器上傳文件夾是否存在
 if(!uploadPath.exists()){
  uploadPath.mkdirs();
 }

 //遍歷客戶端上傳文件列表
 for (int i = 0; i < files.size(); ++i) {
  //獲取到每個文件
  file = files.get(i);
  try {
   //獲取上傳文件后綴
   String houzhui = file.getOriginalFilename().split("\\.")[1];
   //拼接上傳文件保存在服務器的路徑(當前用戶id+設備id+時間戳.后綴名)
   File fil = new File(uploadPath+"/"+user_id+equi_id+new Date().getTime()+"."+houzhui);
   //將上傳文件保存到服務器上傳文件夾目錄下
   byte[] bytes = file.getBytes();
   stream = new BufferedOutputStream(new FileOutputStream(fil));
   stream.write(bytes);
   stream.close();
   //每成功上傳一個文件,將上傳文件名作為key,服務器保存路徑作為value存入returnfileMap中
   switch (upload_type){
   case "1":
    returnfileMap.put(file.getOriginalFilename(),inspection+"/"+fil.getName());
    break;
   case "2":
    returnfileMap.put(file.getOriginalFilename(),maintenance+"/"+fil.getName());
    break;
   }
  } catch (Exception e) {
   stream = null;
   //保存上傳失敗的文件信息,將上傳文件名作為key,value值為"fail",存入returnfileMap中
   returnfileMap.put(file.getOriginalFilename(),"fail");
  }finally {
   //關閉處理流
   if(stream!=null){stream.close();}
  }
 }
 //返回returnfileMap集合到客戶端
 returnData.put("message",returnfileMap);
 return CommonUtil.successJson(returnData);
 }
 }