ANTAI-WLTMS-API/src/main/java/com/steerinfo/dil/controller/SystemFileController.java

package com.steerinfo.dil.controller;

import com.google.common.collect.Iterators;
import com.steerinfo.dil.mapper.SystemFileMapper;
import com.steerinfo.dil.util.FtpFileUtil;
import com.steerinfo.dil.util.IDutils;
import com.steerinfo.framework.controller.BaseRESTfulController;
import com.steerinfo.framework.controller.RESTfulResult;
import com.steerinfo.framework.service.pagehelper.PageList;
import com.steerinfo.framework.utils.collection.ListUtils;
import com.steerinfo.dil.model.SystemFile;
import com.steerinfo.dil.service.ISystemFileService;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.apache.commons.lang3.StringUtils;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.MultiValueMap;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartRequest;

import javax.annotation.Resource;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.*;
import java.math.BigDecimal;
import java.util.List;

/**
 * SystemFile RESTful接口：
 *
 * @author generator
 * @version 1.0-SNAPSHORT 2023-10-30 09:19
 * 类描述
 * 修订历史：
 * 日期：2023-10-30
 * 作者：generator
 * 参考：
 * 描述：SystemFile RESTful接口
 * @Copyright 湖南视拓信息技术股份有限公司. All rights reserved.
 * @see null
 */
@RestController
@RequestMapping("/${api.version}/systemfiles")
public class SystemFileController extends BaseRESTfulController {

    @Autowired
    ISystemFileService systemFileService;


    @Resource
    SystemFileMapper systemFileMapper;


    @Autowired
    private FtpFileUtil ftpFileUtil;

    //同一个信息包含多个上传文件信息
    @PostMapping("/insertFile")
    public RESTfulResult insertFile(String fileuuid, @ModelAttribute MultipartFile[] file) {
        String filenames = "";
        for (int i = 0; i < file.length; i++) {
            filenames += file[i].getOriginalFilename() + ";";
        }
        String filesid = "";
        if (file != null) {
            for (int i = 0; i < file.length; i++) {
                try {
                    //获取系统时间
                    SimpleDateFormat simpleDateFormat = new SimpleDateFormat("/yyyy/MM/dd");
                    //获取文件名
                    String oldName = file[i].getOriginalFilename();
                    //取当前时间的长整形值包含毫秒
                    String newName = IDutils.getImageName();
                    //重新命名文件
                    newName = newName + oldName.substring(oldName.lastIndexOf("."));
                    String filePath = simpleDateFormat.format(new Date());
                    //获取输入流
                    InputStream inputStream = file[i].getInputStream();
                    boolean result = ftpFileUtil.uploadToFtp(inputStream, filePath, newName, false);
                    inputStream.close();
                    if (result) {
                        SystemFile uploadFile = new SystemFile();
                        uploadFile.setFilename(oldName);
                        uploadFile.setFilepath(filePath + "/" + newName);
                        uploadFile.setId(fileuuid);
                        SystemFile modela = systemFileService.add(uploadFile);
                        if (modela != null) {
                            filesid += "," + modela.getId();
                        }
                    } else {
                        return failed(null, "上传文件失败");
                    }
                } catch (Exception e) {
                    e.getMessage();
                }
            }
            return success(filesid);

        } else {
            return failed();
        }

    }

    @PostMapping("/insertFiles2")
    public RESTfulResult insertFiles2(@RequestParam String fileuuid, @ModelAttribute MultipartFile[] file) {
        String filenames = "";
        for (int i = 0; i < file.length; i++) {
            filenames += file[i].getOriginalFilename() + ";";
        }
        String filesid = "";

        String[] fileuuids = fileuuid.split(",");
        if (file != null) {
            for (int i = 0; i < file.length; i++) {
                try {
                    //获取系统时间
                    SimpleDateFormat simpleDateFormat = new SimpleDateFormat("/yyyy/MM/dd");
                    //获取文件名
                    String oldName = file[i].getOriginalFilename();
                    //取当前时间的长整形值包含毫秒
                    String newName = IDutils.getImageName();
                    //重新命名文件
                    newName = newName + oldName.substring(oldName.lastIndexOf("."));
                    String filePath = simpleDateFormat.format(new Date());
                    //获取输入流
                    InputStream inputStream = file[i].getInputStream();
                    boolean result = ftpFileUtil.uploadToFtp(inputStream, filePath, newName, false);
                    inputStream.close();
                    if (result) {
                        SystemFile uploadFile = new SystemFile();
                        uploadFile.setFilename(oldName);
                        uploadFile.setFilepath(filePath + "/" + newName);
                        uploadFile.setId(fileuuids[i]);
                        SystemFile modela = systemFileService.add(uploadFile);
                        if (modela != null) {
                            filesid += "," + modela.getId();
                        }
                    } else {
                        return failed(null, "上传文件失败");
                    }
                } catch (Exception e) {
                    e.getMessage();
                }
            }
            return success(filesid);

        } else {
            return failed();
        }

    }

    @PostMapping("/insertFiles")
    public RESTfulResult insertFiles(String[] fileUuids, String[] fileNames, String[] fileTypes, Map<Object, MultipartFile> FileList, MultipartRequest request) throws Exception {


        List<MultipartFile> files = new ArrayList<>();
        int index = -1;
        for (int j = 0; j < fileTypes.length; j++) {
            for (int i = 0; i < 21; i++) {
                MultipartFile file = request.getFile("file" + j + "" + i);
                if (file != null) {
                    files.add(file);
                    if (index < 0) {
                        index = i;
                    }
                }
            }
        }

        if (files.size() == 0) {
            return failed("传输失败");
        }
        for (int i = 0; i < fileUuids.length; i++) {
            //获取系统时间
            SimpleDateFormat simpleDateFormat = new SimpleDateFormat("/yyyy/MM/dd");
            //获取文件名
            String oldName = fileNames[i].toString();
            //取当前时间的长整形值包含毫秒
            String newName = IDutils.getImageName();
            //重新命名文件

            String filePath = simpleDateFormat.format(new Date()) + "/" + fileTypes[i].toString();
            //获取输入流
            String fileTypeName = files.get(i).getOriginalFilename();
            fileTypeName = fileTypeName.substring(fileTypeName.lastIndexOf("."));
            newName = oldName + "(" + newName + ")" + fileTypeName;
            InputStream inputStream = files.get(i).getInputStream();
            boolean result = ftpFileUtil.uploadToFtp(inputStream, filePath, newName, false);
            inputStream.close();
            if (result) {
                SystemFile uploadFile = new SystemFile();
                uploadFile.setFilename(oldName);
                uploadFile.setFilepath(filePath + "/" + newName);
                uploadFile.setId(fileUuids[i]);
                SystemFile modela = systemFileService.add(uploadFile);
                if (modela != null) {
                    fileUuids[i] += "," + modela.getId();
                }
            } else {
                return failed(null, "上传文件失败");
            }
        }

        return success();

    }

    @PostMapping("/previewfile")
    public RESTfulResult previewfile(@RequestBody HashMap parmas) {
        String filename = parmas.get("filename").toString();
        String filepath = parmas.get("filepath").toString();
        if (filename == null || filename.isEmpty()) {
            return failed("该图片不存在！");
        }
        if (filepath == null || filepath.isEmpty()) {
            return failed("该图片地址不存在！");
        }
        try {
            String result = ftpFileUtil.downloadFile(filename, filepath);
            return success(result);
        } catch (IOException e) {
            e.getMessage();
            return failed();
        }
    }

    @PostMapping("/previewfile2")
    public RESTfulResult previewfile2(@RequestBody HashMap parmas) {
        try {
            if (parmas.get("id")==null) {
                return failed(1,"当前详单没有图片");
            }
        SystemFile value = systemFileMapper.selectByPrimaryKey(parmas.get("id").toString());
        if (value==null){
            return failed(1,"该文件不存在！请查验文件是否被清除或已损坏！");
        }
        String fileName = value.getFilename();
        String filepath = value.getFilepath();
        if (fileName == null || fileName.isEmpty()) {
            return failed(1,"该文件不存在！");
        }
        if (filepath == null || filepath.isEmpty()) {
            return failed(1,"该文件地址不存在！");
        }
            String result = ftpFileUtil.downloadFile(fileName, filepath);
            return success(result);
        } catch (IOException e) {
            e.getMessage();
            return failed();
        }

    }

    @PostMapping("/updateFile")
    public RESTfulResult updateFile(String alternateFields1, @ModelAttribute MultipartFile[] file) {
        systemFileService.delete(alternateFields1);
        String filenames = "";
        for (int i = 0; i < file.length; i++) {
            filenames += file[i].getOriginalFilename() + ";";
        }
        String filesid = "";
        if (file != null) {
            for (int i = 0; i < file.length; i++) {
                try {

                    //获取系统时间
                    SimpleDateFormat simpleDateFormat = new SimpleDateFormat("/yyyy/MM/dd");
                    //获取文件名
                    String oldName = file[i].getOriginalFilename();
                    //取当前时间的长整形值包含毫秒
                    String newName = IDutils.getImageName();
                    //重新命名文件
                    newName = newName + oldName.substring(oldName.lastIndexOf("."));
                    String filePath = simpleDateFormat.format(new Date());
                    //获取输入流
                    InputStream inputStream = file[i].getInputStream();
                    boolean result = ftpFileUtil.uploadToFtp(inputStream, filePath, newName, false);
                    inputStream.close();
                    if (result) {
                        SystemFile uploadFile = new SystemFile();
                        uploadFile.setFilename(oldName);
                        uploadFile.setFilepath(filePath + "/" + newName);
                        uploadFile.setId(alternateFields1);
                        SystemFile modela = systemFileService.add(uploadFile);
                        if (modela != null) {
                            filesid += "," + modela.getId();
                        }
                    } else {
                        return failed(null, "上传文件失败");
                    }
                } catch (Exception e) {
                    e.getMessage();
                }
            }
            return success(filesid);

        } else {
            return failed();
        }

    }

    @PostMapping("/previewfileList")
    public RESTfulResult previewfileList(@RequestBody HashMap map) {
        String uuid = map.get("uuid").toString();
        String[] uuidList = uuid.split(";");
        //遍历数组，去找文件名称和路径
        List<Map<String, Object>> mapList = systemFileMapper.getFileInfo(uuidList);
        List<Map<String, Object>> resultList = new ArrayList<>();
        for (Map<String, Object> parmas : mapList) {
            String filename = parmas.get("FILENAME").toString();
            String filepath = parmas.get("FILEPATH").toString();
            if (filename == null || filename.isEmpty()) {
                return failed("该图片不存在！");
            }
            if (filepath == null || filepath.isEmpty()) {
                return failed("该图片地址不存在！");
            }
            try {
                Map<String, Object> result = ftpFileUtil.downloadFileNew(filename, filepath);
                //如果是返回base64,则给这个;网络路径则给netUrl
                result.put("dataType","base");
                resultList.add(result);
            } catch (IOException e) {
                e.getMessage();
                continue;
            }
        }
        return success(resultList);

    }


    @PostMapping("/insertFilesReal")
    public RESTfulResult insertFilesReal(MultipartRequest request, String[] uuidTypes) throws Exception {
        //List<MultipartFile> files  = new ArrayList<>();
        //获取多个文件
        MultiValueMap<String, MultipartFile> multiFileMap = request.getMultiFileMap();
        List<MultipartFile> files = multiFileMap.get("file");
        System.out.println(uuidTypes + "uuidTypes");
        System.out.println(request);
        if (files.size() == 0) {
            return failed("传输失败");
        }
        int index = 0;
        for (MultipartFile file : files) {
            String uuidType = uuidTypes[index];
            String fileMediaType = file.getContentType();
            String newName = uuidType.split(";")[0];

            String uuid = uuidType.split(";")[0];
            String fileType = uuidType.split(";")[1];
            InputStream inputStream = file.getInputStream();
            if (fileMediaType != null && fileMediaType.contains("image")) {
                newName = uuidType.split(";")[0] + "." +fileMediaType.split("/")[1];
                ////如果是图片
                //BufferedImage originalImage = ImageIO.read(inputStream);
                //// 确保原始图片有效且可以读取
                //if (originalImage != null) {
                //    // 添加水印逻辑
                //    inputStream = addWatermark(originalImage);
                //    // 保存或返回带有水印的图片
                //    // ...
                //}
            }
            SimpleDateFormat simpleDateFormat = new SimpleDateFormat("/yyyy/MM/dd");
            String filePath = "/" + fileType  + simpleDateFormat.format(new Date());
            boolean result = ftpFileUtil.uploadToFtp(inputStream, filePath, newName, false);
            inputStream.close();
            if (result) {
                SystemFile uploadFile = new SystemFile();
                uploadFile.setFilename(newName);
                uploadFile.setFilepath(filePath + "/" + newName);
                uploadFile.setId(uuid);
                SystemFile modela = systemFileService.add(uploadFile);
                if (modela != null) {
                    uuid += "," + modela.getId();
                }
            } else {
                return failed(null, "上传文件失败");
            }
            index++;
        }
        return success();
}

    //private InputStream addWatermark(BufferedImage originalImage) {
    //    int width = originalImage.getWidth();
    //    int height = originalImage.getHeight();
    //
    //    //BufferedImage watermarkedImage = Scalr.resize(originalImage, width, height, Scalr.Mode.FIT_EXACT);
    //
    //    Graphics2D graphics = watermarkedImage.createGraphics();
    //
    //    // 设置水印样式
    //    graphics.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
    //    Font font = new Font("Arial", Font.BOLD, 30);
    //    graphics.setFont(font);
    //    Color color = new Color(255, 0, 0, 128);
    //    graphics.setColor(color);
    //
    //    // 添加水印文字
    //    graphics.drawString("Your Watermark Text", 10, height - 40); // 示例位置
    //
    //    graphics.dispose();
    //
    //    // 将带水印的图片转为InputStream
    //    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    //    ImageIO.write(watermarkedImage, "jpg", baos);
    //    InputStream watermarkedStream = new ByteArrayInputStream(baos.toByteArray());
    //
    //    return watermarkedStream;
    //}
}