package com.ruoyi.common.utils.file;

import com.ruoyi.common.config.RuoYiConfig;
import com.ruoyi.common.constant.Constants;
import com.ruoyi.common.core.redis.RedisCache;
import com.ruoyi.common.utils.RandomUtils;
import com.ruoyi.common.utils.ServletUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.common.utils.ZipUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.List;
import java.util.concurrent.TimeUnit;

/**
 * 导出下载文件
 */
public class DownloadExportUtil {
    protected static final Logger logger = LoggerFactory.getLogger(DownloadExportUtil.class);
    public enum ExprotStatus{
        XZZ("0","下载中"),
        FQXZ("1","放弃下载"),
        XZYC("-1","数据导出失败"),
        ;
        private String status;
        private String desc;

        private ExprotStatus(String status,String desc){
            this.status=status;
            this.desc = desc;
        }
        public String getStatus() {
            return status;
        }

        public String getDesc() {
            return desc;
        }
    }
    /**
     * 获取导出数据key
     * @param key
     * @return
     */
    public static String getExprotKey( String key){
        return Constants.EXPORT_PROFILE+":"+key;
    }
    /**
     * 清除导出任务
     * @param redisCache
     * @param key
     * @return
     */
    public static void clearDownload(RedisCache redisCache, String key) {
        deleteDownloadFile(redisCache,key, ExprotStatus.FQXZ.status);
    }
    /**
     * 删除下载的文件
     * @param key
     * @param value 设置的值,空直接删除
     */
    public static void deleteDownloadFile(RedisCache redisCache,String key,String value){
        try {
            String val =redisCache.getCacheObject(Constants.EXPORT_PROFILE+":"+key);
            if(value==null) {
                redisCache.deleteObject(Constants.EXPORT_PROFILE  + ":" + key);
            }else{
                redisCache.setCacheObject(Constants.EXPORT_PROFILE  + ":" + key,value,10,TimeUnit.MINUTES);
            }
            if(StringUtils.isEmpty(val)) {
                String filePath = RuoYiConfig.getDownloadPath() + val;
                FileUtils.deleteFile(filePath);
            }
        }catch (Exception e) {
            logger.error("异常",e);
        }
    }
    /**
     * 设置下载的文件
     * @param key
     */
    public static void setDownloadFile(RedisCache redisCache, String key,String fileName){
        try {
            redisCache.setCacheObject(Constants.EXPORT_PROFILE+":"+key,fileName,10, TimeUnit.MINUTES);
        }catch (Exception e) {
            logger.error("异常",e);
        }
    }
    public static String setZipFile(String fileName,List<String> fileNames){
        if(fileNames.size()==0){
            return fileName;
        }
        fileName = fileName.replace("xlsx","zip");
        ZipUtils.generatorZipFile(fileName,fileNames,true);
        return fileName;
    }

    /**
     * 设置下载的文件
     * @param exportKey
     */
    public static String setDownloadFile(RedisCache redisCache,  String exportKey, String fileName, List<String> fileNames){
        try {
            //直接生成一个表格
            if(fileNames.size()==0){
                //设置下载文件
                setDownloadFile(redisCache,  exportKey, fileName);
                return fileName;
            }

            //region 压缩处理
            fileName=fileName.substring(0,fileName.lastIndexOf("."))+".zip";
            ZipUtils.generatorZipFile(fileName,fileNames,true);
            setDownloadFile(redisCache,  exportKey, fileName);
            //endregion
        }catch (Exception e) {
            logger.error("异常",e);
        }
        return fileName;
    }

    /**
     * 获取下载文件地址
     * @param redisCache
     * @param key
     * @return
     */
    public static String getDownloadFileName(RedisCache redisCache, String key){
        try {
            String redisKey = getExprotKey(key);
            return redisCache.getCacheObject(redisKey);
        }catch (Exception e) {
            logger.error("异常",e);
            return null;
        }
    }
    /**
     * 下载文件
     * @param fileName
     * @param delete
     */
    public static void fileDownload(String fileName, Boolean delete)
    {
        try
        {
            HttpServletRequest request=  ServletUtils.getRequest();;
            HttpServletResponse response=ServletUtils.getResponse();
            if (!FileUtils.isValidFilename(fileName))
            {
                throw new Exception(StringUtils.format("文件名称({})非法，不允许下载。 ", fileName));
            }
            String num = RandomUtils.random(2);
            String realFileName = fileName.substring(fileName.indexOf("_") + 1).replace(".",num+".");
            String filePath = RuoYiConfig.getDownloadPath() + fileName;

            response.setCharacterEncoding("utf-8");
            response.setContentType("multipart/form-data");
            response.setHeader("Content-Disposition",
                    "attachment;fileName=" + FileUtils.setFileDownloadHeader(request, realFileName));
            FileUtils.writeBytes(filePath, response.getOutputStream());
            if (delete)
            {
                FileUtils.deleteFile(filePath);
            }
        }
        catch (Exception e)
        {
            logger.error("下载文件失败", e);
        }
    }

}