package com.xkrs.controller;

import com.xkrs.common.encapsulation.PromptMessageEnum;
import com.xkrs.dao.FilePathDao;
import com.xkrs.model.entity.FilePath;
import com.xkrs.utils.FileFastDfs;
import com.xkrs.utils.FileUtil;
import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import java.util.Map;

import static com.xkrs.common.encapsulation.OutputEncapsulation.outputEncapsulationObject;
import static com.xkrs.utils.FileUtil.getUploadInfo;

/**
 * @author XinYi Song
 * 文件上传
 */
@RestController
public class FileController {

    @Resource
    private FileFastDfs fileFastDfs;

    @Resource
    private FilePathDao filePathDao;

    /**
     * 上传多张图片
     * @param files
     * @return
     */
    @PostMapping("/uploadFileMore")
    public String uploadFileMore(MultipartFile[] files) throws Exception {
        Locale locale = LocaleContextHolder.getLocale();
        if (null == files || files.length == 0){
            return outputEncapsulationObject(PromptMessageEnum.PARAM_NULL, "result or fileType is null", locale);
        }
        List list = new ArrayList();
        for (MultipartFile file :
                files) {
            boolean m = FileUtil.checkFileSize(file.getSize(), 100, "M");
            if(m == false){
                return outputEncapsulationObject(PromptMessageEnum.DATA_WRONG, "图片大小不能超过100M", locale);
            }
            File file1 = fileFastDfs.multipartFileToFile(file);
            String dir = "fire_point";
            String info = fileFastDfs.uploadFile(file1, dir);
            Map<String, String> map = getUploadInfo(info);
            file1.delete();
            FilePath filePath = new FilePath();
            filePath.setFilePath(map.get("path"));
            filePath.setFileName(map.get("fileName"));
            filePath.setSize(map.get("size"));
            filePath.setMtime(map.get("mtime"));
            filePath.setUrl(map.get("url"));
            filePath.setMd5(map.get("md5"));
            filePath.setScene(map.get("scene"));
            filePathDao.save(filePath);
        }
        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, "上传成功！", locale);
    }

    /**
     * 上传单张图片
     * @param file
     * @return
     */
    @PostMapping("/uploadFile")
    public String uploadFile(MultipartFile file) throws Exception {
        Locale locale = LocaleContextHolder.getLocale();
        if (null == file){
            return outputEncapsulationObject(PromptMessageEnum.PARAM_NULL, "result or fileType is null", locale);
        }
        boolean m = FileUtil.checkFileSize(file.getSize(), 100, "M");
        if(m == false){
            return outputEncapsulationObject(PromptMessageEnum.DATA_WRONG, "图片大小不能超过100M", locale);
        }
        File file1 = fileFastDfs.multipartFileToFile(file);
        String dir = "fire_point";
        String info = fileFastDfs.uploadFile(file1, dir);
        Map<String, String> map = getUploadInfo(info);
        file1.delete();
        FilePath filePath = new FilePath();
        filePath.setFilePath(map.get("path"));
        filePath.setFileName(map.get("fileName"));
        filePath.setSize(map.get("size"));
        filePath.setMtime(map.get("mtime"));
        filePath.setUrl(map.get("url"));
        filePath.setMd5(map.get("md5"));
        filePath.setScene(map.get("scene"));
        filePathDao.save(filePath);
        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, "上传成功！", locale);
    }

    /**
     * 查询火点图片路径
     * @return
     */
    @GetMapping("/findPath")
    public String findPath(){
        Locale locale = LocaleContextHolder.getLocale();
        List<Map<String, String>> path = filePathDao.findPath();
        if(path == null || path.size() == 0){
            return outputEncapsulationObject(PromptMessageEnum.DATA_NONE, "还未上传任何图片信息！", locale);
        }
        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, path, locale);
    }
}