package com.zzg.web.controller.common;

import cn.hutool.core.lang.Dict;
import cn.hutool.core.text.StrPool;
import com.zzg.common.core.domain.AjaxResult;
import com.zzg.system.service.system.IFileService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.util.Arrays;
import java.util.List;

/**
 * 文件上传
 */
@RestController
@Api(tags = "文件上传,下载")
@RequestMapping("/file")
public class FileController {
    private static final Logger log = LoggerFactory.getLogger(FileController.class);

    @Resource
    private IFileService fileService;
    @Value("${ruoyi.upload}")
    String landexp;

    @ApiOperation(value = "文件上传", notes = "返回文件的名字,下载文件时可以通过文件的名字下载！")
    @PostMapping("/public/upload")
    public List<Dict> publicUpload(@RequestParam("file") MultipartFile[] file, HttpServletRequest req) {
//        return fileService.upload2Temp(Arrays.asList(file), req);
        List<MultipartFile> fileList = Arrays.asList(file);
        if (fileList.size() > 10) {
            throw new RuntimeException("一次最多上传10份文件");
        }
        return fileService.upload2Temp(fileList, req);
    }


    @GetMapping("/public/download")
    @ApiOperation(value = "文件下载", notes = "单文件")
    public AjaxResult download(@RequestParam String filename, HttpServletResponse response) throws IOException, InterruptedException {
        File absFile = new File(landexp + filename);
        if (absFile.exists()) {
            //协议文件名称没有下划线，只有upload路径--pload/征地拆迁住房安置协议-许肖肖-204125.pdf
            filename = filename.substring(filename.lastIndexOf(StrPool.SLASH) + 1);
            //不同文件
            filename = filename.substring(filename.indexOf(StrPool.UNDERLINE) + 1);
            response.setContentType("application/octet-stream;charset=utf-8");
            response.addHeader("Content-Disposition", "attachment; filename=" +
                    URLEncoder.encode(filename, "UTF-8"));
            FileInputStream fileInputStream = new FileInputStream(absFile);
            int len = 0;
            byte[] buffer = new byte[1024];
            OutputStream outputStream = response.getOutputStream();
            while ((len = fileInputStream.read(buffer)) > 0) {
                outputStream.write(buffer, 0, len);
            }
            fileInputStream.close();
            outputStream.flush();
            outputStream.close();
            return AjaxResult.success();
        } else {
            log.error("下载文件地址：" + filename);
            return AjaxResult.error("文件不存在！");
        }
    }

}