loan-backend/src/main/java/com/loan/system/controller/wechat/UploadController.java

package com.loan.system.controller.wechat;

import cn.hutool.core.util.StrUtil;
import com.loan.system.config.FileUploadConfig;
import com.loan.system.domain.entity.Document;
import com.loan.system.domain.enums.ExceptionEnum;
import com.loan.system.domain.enums.StepEnum;
import com.loan.system.domain.pojo.Result;
import com.loan.system.domain.vo.DocumentVO;
import com.loan.system.domain.vo.StepVO;
import com.loan.system.exception.DescribeException;
import com.loan.system.service.DocumentService;
import com.loan.system.service.StepService;
import com.loan.system.utils.ResultUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.commons.io.FilenameUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.util.ObjectUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.StandardCopyOption;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.*;

import static com.loan.system.domain.enums.ExceptionEnum.STEP_HAS_NOT_PROCESS;

@RestController
@RequestMapping("/wechat/file")
@Api(tags = "文件处理接口")
public class UploadController {
    @Autowired
    private DocumentService fileService;
    @Autowired
    private FileUploadConfig config;
    @Autowired
    private StepService stepService;

    /** 上传文件（支持任意类型，只要在白名单内） */
    @PostMapping("/upload/{caseId}/{type}")
    @ApiOperation("文件上传")//若合同不是统一提交，则合同1，合同2，合同3
    @PreAuthorize("@pms.hasAnyRoles('LEAD_SALES','ASSIST_SALES')")
    public Result uploadFile(@PathVariable("caseId") Long caseId,@PathVariable("type") String fileType,@RequestParam(required = false) MultipartFile file,@RequestParam Map<String , String>  isDelete) throws IOException{
        Iterator<Map.Entry<String, String>> iterator = isDelete.entrySet().iterator();
        while (iterator.hasNext()) {
            Map.Entry<String, String> entry = iterator.next();
            Long id = Long.parseLong(entry.getKey());
            Integer value = Integer.parseInt(entry.getValue());
            Document document = fileService.findById(id);
            if (ObjectUtils.isEmpty( document))
                break;

            if( value!= 0 ){
                System.out.println("删除文件"+id);
                fileService.deleteFileById(id);
                //删除本地文件
               deleteFile(document.getFilePath());
            }
        }

        if (file == null)
            return ResultUtil.success("success");

        String originalName = file.getOriginalFilename();
        String ext = FilenameUtils.getExtension(originalName).toLowerCase(Locale.ROOT);

        // 从配置中读取允许的扩展名
        Set<String> allowed = config.getAllowedExtensions();
        if (!allowed.contains(ext)) {
            ResultUtil.error(ExceptionEnum.FILE_UPLOAD_TYPE_NOT_DEFINED);
        }

        String newFileName = String.format("%d-%s-%s.%s", caseId, fileType, UUID.randomUUID(), ext);;
        // 生成新的文件名

        // 创建目录
        File uploadDir = new File(config.getUploadDir(), caseId + "/" + fileType);
        System.out.println(uploadDir);
        System.out.println(newFileName);
        if (!uploadDir.exists() && !uploadDir.mkdirs()) {
            return ResultUtil.error(ExceptionEnum.DIRECTORY_CREATE_ERROR);
        }

        // 使用NIO复制流，稳定性更好
        File destFile = new File(uploadDir, newFileName);
        try (InputStream in = file.getInputStream()) {
            System.out.println(destFile.toPath());
            Files.copy(in, destFile.toPath(), StandardCopyOption.REPLACE_EXISTING);
        }

        Document document = new Document();
        document.setCaseId(caseId);
        document.setFileName(newFileName);
        document.setOriginName(originalName);
        document.setFilePath(destFile.getPath());
        document.setFileSize(file.getSize());
        document.setDocType(ext);
        document.setDictType(fileType); // 简单转换为整数
        document.setIsDelete( false);
        document.setCreateTime(LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
        document.setUpdateTime(LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
        DocumentVO documentVO = fileService.uploadFile(document);

        return ResultUtil.success("上传成功",documentVO);
    }

    private void deleteFile(String filePath) {
        if (filePath == null || filePath.isEmpty()) {
            return;
        }

        try {
            Path path = Paths.get(filePath);
            boolean deleted = Files.deleteIfExists(path);
            if (deleted) {
                System.out.println("文件删除成功: " + filePath);
            } else {
                System.out.println("文件不存在或已被删除: " + filePath);
            }
        } catch (IOException e) {
            System.err.println("删除文件时发生错误: " + e.getMessage());
            e.printStackTrace();
        }
    }


    /** 下载文件 */
    @GetMapping("/download/{fileName}")
    @ApiOperation("文件下载")
    public void downloadFile(@PathVariable String fileName, @RequestParam String caseId, @RequestParam String fileType, HttpServletResponse response)throws  IOException{
        String path = config.getUploadDir() + caseId + "/" + fileType + "/" ;
        File file = new File(path, fileName);
        if (!file.exists()) {
            response.sendError(HttpServletResponse.SC_NOT_FOUND, "文件不存在");
            return;
        }

        response.setContentType(Files.probeContentType(file.toPath()));
        response.setHeader("Content-Disposition",
                "attachment; filename=" + URLEncoder.encode(file.getName(), "UTF-8"));

        try (InputStream in = new FileInputStream(file);
             OutputStream out = response.getOutputStream()) {
            byte[] buffer = new byte[8192];
            int len;
            while ((len = in.read(buffer)) != -1) {
                out.write(buffer, 0, len);
            }
        }
    }

    /** 删除文件 */
    @DeleteMapping("/delete/{caseId}/{type}")
    @ApiOperation("文件删除")
    public Result deleteFileByFileName(@RequestParam String fileName ,@PathVariable Long caseId, @PathVariable String fileType) {
        String newUploadDir = config.getUploadDir() + caseId + "/" + fileType;
        File file = new File(newUploadDir, fileName);

        if (!file.exists())
            return ResultUtil.error(ExceptionEnum.FILE_NOT_EXIST);

        boolean isDelete = file.delete();
        if (!isDelete)
            return ResultUtil.error(ExceptionEnum.FILE_DELETE_ERROR);

        return ResultUtil.success("文件删除成功");
    }

    @GetMapping("/{caseId}")
    @ApiOperation("获取文件列表(type可选)")
    public Result getFileList(@PathVariable Long caseId, @RequestParam(required = false) String type){
        List<DocumentVO> files = fileService.findByCaseId(caseId);

        if(StrUtil.isBlank( type))
            return ResultUtil.success("success",files);

        List<DocumentVO> fileList = new ArrayList<>();
        for(DocumentVO file : files){
            if(file.getDictType().equals( type))
                fileList.add(file);
        }

        return ResultUtil.success("success",fileList);
    }


}