package com.moral.api.utils;

import com.moral.api.pojo.enums.FileType;

/**
 * @ClassName: FileTypeUtils
 * @Description:
 * @Author: fan
 * @Date: 2021/2/1 13:41
 * @Version: 1.0
 **/
public class FileTypeUtils {
    /**
     * 图片后缀
     */
    private static final String SUFFIX_JPE = ".jpe";
    private static final String SUFFIX_JPG = ".jpg";
    private static final String SUFFIX_PNG = ".png";
    private static final String SUFFIX_JPEG = ".jpeg";
    private static final String SUFFIX_BMP = ".bmp";
    private static final String SUFFIX_GIF = ".gif";

    /**
     * word
     */
    public static final String SUFFIX_DOC = ".doc";
    public static final String SUFFIX_DOCX = ".docx";

    /**
     * excel
     */
    private static final String SUFFIX_XLSX = ".xlsx";
    private static final String SUFFIX_XLS = ".xls";

    /**
     * pdf
     */
    public static final String SUFFIX_PDF = ".pdf";

    /**
     * 压缩包
     */
    private static final String SUFFIX_ZIP = ".zip";
    private static final String SUFFIX_RAR = ".rar";
    private static final String SUFFIX_JAR = ".jar";
    private static final String SUFFIX_GZIP = ".gzip";

    private static final String SUFFIX_TXT = ".txt";

    /**
     * ppt
     */
    public static final String SUFFIX_PPT = ".ppt";
    public static final String SUFFIX_PPTX = ".pptx";


    public static final String SUFFIX_MP4 = ".mp4";

    /**
     * @Description 获取文件类型
     * @author fanhq
     * @date 2020/4/20 15:35
     * @param suffix
     * @return java.lang.Integer   1. 图片  * 2. word  * 3. excel * 4. pdf  * 5. 压缩包
     */
    public static FileType getFileType(String suffix) {
        switch (suffix) {
            case SUFFIX_JPE:
            case SUFFIX_JPG:
            case SUFFIX_PNG:
            case SUFFIX_JPEG:
            case SUFFIX_BMP:
                return FileType.PICTURE;
            case SUFFIX_DOC:
            case SUFFIX_DOCX:
                return FileType.WORD;
            case SUFFIX_XLSX:
            case SUFFIX_XLS:
                return FileType.EXCEL;
            case SUFFIX_PDF:
                return FileType.PDF;
            case SUFFIX_ZIP:
            case SUFFIX_RAR:
            case SUFFIX_JAR:
                return FileType.ZIP;
            default:
                return FileType.NON;
        }
    }

    /**
     * 根据文件类型转换响应头类型
     * @param type
     * @return
     */
    public static String convertHeaderType(String type) {
        switch (type){
            case SUFFIX_JPE:
            case SUFFIX_JPG:
            case SUFFIX_JPEG:
                return "image/jpeg";
            case SUFFIX_PNG:
                return "image/png";
            case SUFFIX_GIF:
                return "image/gif";
            case SUFFIX_ZIP:
                return "application/zip";
            case SUFFIX_GZIP:
                return "application/gzip";
            case SUFFIX_DOC:
            case SUFFIX_DOCX:
                return "application/msword";
            case SUFFIX_XLSX:
            case SUFFIX_XLS:
                return "application/x-xls";
            case SUFFIX_PDF:
                return "application/pdf";
            case SUFFIX_TXT:
                return "text/plain";
            case SUFFIX_MP4:
                return "video/mp4";
            default:
                return "application/octet-stream";
        }
    }
}