package com.gkhy.safePlatform.minioDemo.service;

import com.gkhy.safePlatform.commons.enums.ResultCodes;
import com.gkhy.safePlatform.commons.vo.ResultVO;
import com.gkhy.safePlatform.minioDemo.config.MinioConfig;
import com.gkhy.safePlatform.minioDemo.enums.ModuleTypeEnums;
import com.gkhy.safePlatform.minioDemo.utils.FileUtil;
import io.minio.GetPresignedObjectUrlArgs;
import io.minio.MinioClient;
import io.minio.errors.*;
import io.minio.http.Method;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.text.SimpleDateFormat;
import java.util.*;

import static com.gkhy.safePlatform.minioDemo.enums.ModuleTypeEnums.*;

@Service
public class FileService {

    @Resource
    private MinioConfig minioConfig;

    @Autowired
    private MinioClient minioClient;

    @Resource
    private FileUtil fileUtil;

    /**
     * 文件下载
     * @param fileName
     * @param response
     */
    public void downloadFile(String fileName, HttpServletResponse response) {
        fileUtil.downloadFile(fileName,response);
    }

    /**
     * 文件上传
     * @param file
     * @param moduleType
     * @return
     */
    public ResultVO<String> uploadFile(MultipartFile file, Integer moduleType) {
        String url = "";
        String moduleName = initModuleName(moduleType);

        // 文件名转换
        String time = new SimpleDateFormat("yyyyMMdd").format(new Date());
        String fileSuffixName = Objects.requireNonNull(file.getOriginalFilename()).substring(file.getOriginalFilename().lastIndexOf(".")+1);
        String fileName = moduleName + "_" + time + "_" + initUUID()+"."+fileSuffixName;
        try {
            url=fileUtil.upload(file, fileName);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return new ResultVO<>(ResultCodes.OK, url);
    }

    /**
     * 模块名称转换
     */
    private String initModuleName(Integer moduleType){
        String moduleName = "";
        ModuleTypeEnums moduleTypeEnums = ModuleTypeEnums.getReviewStatus(moduleType);
        assert moduleTypeEnums != null;
        switch (moduleTypeEnums) {
            case EMERGENCY:
                moduleName = EMERGENCY.getModuleName();
                break;
            case EQUIPMENT:
                moduleName = EQUIPMENT.getModuleName();
                break;
            case GOAL_MANAGE:
                moduleName = GOAL_MANAGE.getModuleName();
                break;
            case INCIDENT_MANAGE:
                moduleName = INCIDENT_MANAGE.getModuleName();
                break;
            default:
                break;
        }
        return moduleName;
    }

    /**
     * 生成8位UUID
     */
    private String initUUID() {
        String[] chars = new String[]{"a", "b", "c", "d", "e", "f",
                "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s",
                "t", "u", "v", "w", "x", "y", "z", "0", "1", "2", "3", "4", "5",
                "6", "7", "8", "9", "A", "B", "C", "D", "E", "F", "G", "H", "I",
                "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V",
                "W", "X", "Y", "Z"};
        StringBuffer shortBuffer = new StringBuffer();
        String uuid = UUID.randomUUID().toString().replace("-", "");
        for (int i = 0; i < 8; i++) {
            String str = uuid.substring(i * 4, i * 4 + 4);
            int x = Integer.parseInt(str, 16);
            shortBuffer.append(chars[x % 0x3E]);
        }
        return shortBuffer.toString();
    }





    public String getPutUrl(String objName) {
        String url = null;
        try {
            Map<String, String> reqParams = new HashMap<>();
            reqParams.put("response-content-type", "application/json");
            url = minioClient.getPresignedObjectUrl(GetPresignedObjectUrlArgs.builder()
                    .method(Method.PUT)
                    .bucket(minioConfig.getBucketName())
                    .object(objName)
                    .expiry(60 * 2)
                    .extraQueryParams(reqParams)
                    .build());
        } catch (ErrorResponseException e) {
            e.printStackTrace();
        } catch (InsufficientDataException e) {
            e.printStackTrace();
        } catch (InternalException e) {
            e.printStackTrace();
        } catch (InvalidKeyException e) {
            e.printStackTrace();
        } catch (InvalidResponseException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        } catch (XmlParserException e) {
            e.printStackTrace();
        } catch (ServerException e) {
            e.printStackTrace();
        }
        return url;
    }


}