package com.chinaztt.mes.plan.util;


import com.amazonaws.util.IOUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.util.FileSystemUtils;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.PostConstruct;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;

/**
 * 保存文件工具
 */
@Slf4j
@Component
public class FileSaveUtil {

    // 取yml中的路径 + /
    public static String FILE_PATH;

    public static String[] ALLOWED;

    @Value("${file.path}")
    private String file;

    @Value("${file.allowed}")
    private String[] allowed;

    @PostConstruct
    public void getFile() {
        FILE_PATH = this.file;
    }

    @PostConstruct
    public void getAllowed(){
        ALLOWED = this.allowed;
    }
    /**
     * 存储文件主函数
     * @param file 文件二进制流
     * @return 返回文件名称用于存储数据库
     */
    public static String StoreFile(MultipartFile file) {
        String originalFilename = file.getOriginalFilename();
        // 生成随机名称：时间_随机6位数字
        String FileName = System.currentTimeMillis() + "_" + MyUtil.getNumber(6);
        String suffix = null;
        if (originalFilename != null) {
            suffix = originalFilename.substring(originalFilename.lastIndexOf("."));
            // 如果后缀名不通过抛出异常
            if (!isFileAllowed(suffix)){
                throw  new RuntimeException(suffix);
            }
        }
        // 名称拼接
        String fileName = FileName + suffix;
        // 进行存储
        try {
            storeFileWithFileName(file.getBytes(), fileName);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        return fileName;
    }

    public static Boolean DeleteFile(String fileName) {
        return FileSystemUtils.deleteRecursively(new File(FILE_PATH + "/" + fileName));
    }
    /**
     * 存储文件函数
     * @param content 文件二进制流
     * @param fileName 文件名称
     */
    private static void storeFileWithFileName(byte[] content, String fileName) {
        // 存储路径
        String path = FILE_PATH + File.separatorChar;
        // 目录不存在则创建
        File file = new File(path);
        if (!file.exists()) {
            file.mkdirs();
        }
        // 开始存储
        try (FileOutputStream os = new FileOutputStream(path + fileName);
             ByteArrayInputStream is = new ByteArrayInputStream(content)) {
             IOUtils.copy(is, os);
        } catch (IOException e) {
            MyUtil.PrintLog("存储文件异常：" + e);
        }
    }

    /**
     * 判断文件是否被允许上传
     *
     * @param fileName 文件名
     * @return 允许true, 否则false
     */
    private static boolean isFileAllowed(String fileName) {
        // 获取后缀名
        String suffixName = fileName.substring(fileName.lastIndexOf(".") + 1).toLowerCase();
        for (String allow : ALLOWED) {
            if (allow.equals(suffixName)) {
                return true;
            }
        }
        return false;
    }
}