package com.fzzy.eoms.file.controller;

import com.fzzy.eoms.file.CoreFileService;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.file.FileUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;
import java.util.Date;

/**
 * @program: eoms
 * @description: 文件操作
 * @author: 0range
 * @create: 2021-01-05 17:05
 **/
@Slf4j
@RestController
@RequestMapping("/file")
public class FileController {


    @Resource
    private CoreFileService fileService;

    /**
     * 附件下载
     * @param fileId
     * @param fileDate
     * @return
     * @throws Exception
     */
    @GetMapping("/download")
    public void download(@RequestParam(value = "id", required = true) String fileId,
                         @RequestParam(value = "date", required = true) String fileDate,
                         HttpServletResponse response) throws Exception {

        log.info("-----download-----fileId={},fileDate={}", fileId, fileDate);

        // 将字符串转换为 long
        long timestamp = Long.parseLong(fileDate);
        Date date = new Date(timestamp);

        String storePath = fileService.getStoragePath(date);

        String filePath = storePath + fileId;
        response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);
        FileUtils.setAttachmentResponseHeader(response, fileId);
        FileUtils.writeBytes(filePath, response.getOutputStream());
    }

    /**
     * 附件预览，仅图片和PDF类支持预览
     * @param fileId
     * @param fileDate
     * @return
     * @throws Exception
     */
    @GetMapping("/preview")
    public void preview(@RequestParam(value = "id", required = true) String fileId,
                        @RequestParam(value = "date", required = true) String fileDate,
                        HttpServletResponse response){

        log.info("-----preview-----fileId={},fileDate={}", fileId, fileDate);

        // 将字符串转换为 long
        long timestamp = Long.parseLong(fileDate);
        Date date = new Date(timestamp);

        String storePath = fileService.getStoragePath(date);

        String filePath = storePath + fileId;

        File file = new File(filePath);
        response.setHeader("Content-Disposition", "inline; filename*=UTF-8''" + fileId);
        ServletOutputStream os = null;
        try {
            os = response.getOutputStream();
            os.write(org.apache.commons.io.FileUtils.readFileToByteArray(file));
            os.flush();
        } catch (Exception e) {
            log.error("-----preview-----fileId={},fileDate={},error={}", fileId, fileDate, e.getMessage());
            log.error("preview error ",e);
        } finally {
            if (os != null) {
                try {
                    os.close();
                } catch (Exception e) {
                    log.error("-----preview-----fileId={},fileDate={},error={}", fileId, fileDate, e.getMessage());
                    log.error("preview error ",e);
                }
            }
        }
    }



}