package com.ld.igds.util;

import com.ld.igds.data.ConfigData;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.io.ClassPathResource;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.io.FileOutputStream;
import java.io.*;
import java.net.URL;

/**
 * @author chen
 */
@Slf4j
@Component
public class FilesUtil {

    @Resource
    private ConfigData configData;

    /**
     * 根据url地址下载保存图片
     *
     * @param imgUrl  图片地址,如：https://ip:port/imgUrl
     * @param path    图片保存路径,如：F:/IGDS/COMMON/
     * @param imgName 图片名称,如：img1.png
     */
    public static void downLoadImg(String imgUrl, String path, String imgName) {
        URL url = null;
        //创建一个输入流
        InputStream in = null;
        //创建输出流
        OutputStream os = null;
        try {
            url = new URL(imgUrl);
            in = new DataInputStream(url.openStream());
            os = new FileOutputStream(new File(path + imgName));

            ByteArrayOutputStream output = new ByteArrayOutputStream();

            byte[] buffer = new byte[1024];
            int length;
            while ((length = in.read(buffer)) > 0) {
                output.write(buffer, 0, length);
            }
            os.write(output.toByteArray());
            in.close();
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }


    /**
     * 验证static/img/下有没有当前文件 有返回true，没有返回false
     *
     * @param fileName
     * @return
     */
    public static boolean checkImgFile(String fileName) {
        if (null == fileName) return false;

        try {
            ClassPathResource classPathResource = new ClassPathResource("static/img/" + fileName);
            log.debug("---获取IMG下面文件路径-{}", classPathResource.getURL().getPath());
            return classPathResource.exists();
        } catch (Exception e) {
            log.warn("---获取IMG下面文件路径失败:{}", e.getMessage());
            return false;
        }
    }


    /**
     * 根据文件路径+文件名创建
     *
     * @param path
     * @return
     */
    public static boolean createFile(String path) {
        if (path.contains(".")) {
            int partition1 = path.lastIndexOf("/");
            if (partition1 == -1) {
                partition1 = path.lastIndexOf("\\");
            }
            // 纯文件名
            String fileName = path.substring(partition1 + 1);
            String filePath = path.substring(0, partition1);
            return createNewFile(fileName, filePath);
        } else {
            return createCatalog(path);
        }
    }

    /**
     * 根据路径和文件名称创建
     *
     * @param fileName
     * @param filePath
     * @return
     */
    public static boolean createNewFile(String fileName, String filePath) {
        if (createCatalog(filePath)) {
            File f = new File(filePath, fileName);
            try {
                return f.createNewFile();
            } catch (IOException e) {
                new RuntimeException("创建文件失败");
                e.printStackTrace();
            }
        }
        return false;

    }

    /**
     * 创建文件夹
     *
     * @param path
     * @return
     */
    private static boolean createCatalog(String path) {
        File f = new File(path);
        if (!f.exists()) {
            return f.mkdirs();
        }
        return true;
    }


    /**
     * 根据组织编码获取当前用户的企业LOGO
     *
     * @param companyId
     * @return
     */
    public static String getLogoByCompanyId(String companyId) {
        String logoName = "logo-" + companyId + ".png";
        boolean isExists = FilesUtil.checkImgFile(logoName);
        if (isExists) return logoName;
        return "logo-default.png";
    }

    /**
     * 根据用户所在组织获取系统首页系统名称
     *
     * @param companyId
     * @return
     */
    public static String getLogoTitleByCompanyId(String companyId) {
        String logoName = "logo-title-" + companyId + ".png";
        boolean isExists = FilesUtil.checkImgFile(logoName);
        if (isExists) return logoName;
        return "logo-title-default.png";
    }


    /**
     * 根据组织编码获取当前用户的企业库区鸟瞰图
     *
     * @param companyId
     * @return
     */
    public static String getAerialByCompanyId(String companyId) {
        String logoName = "aerial-" + companyId + ".png";
        boolean isExists = FilesUtil.checkImgFile(logoName);
        if (isExists) return logoName;
        return "aerial-default.png";
    }

    /**
     * 根据当前人所在分库ＩＤ获取鸟瞰图
     *
     * @param deptId
     * @return
     */
    public static String getAerialByDeptId(String deptId) {
        String logoName = "aerial-" + deptId + ".png";
        boolean isExists = FilesUtil.checkImgFile(logoName);
        if (isExists) return logoName;
        return "aerial-default.png";
    }


    /**
     * 返回项目临时目录
     *
     * @param companyId
     * @return
     */
    public String getTempPath(String companyId) {
        String path = configData.getFilePath() + "TEMP/" + companyId;

        createCatalog(path);

        return path;
    }

    /**
     * 获取系统配置文件夹
     *
     * @return
     */
    public String getConfPath() {
        return configData.getFilePath() + "CONF";
    }
}