package com.gci.boot.common.util;

import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.Date;
import java.util.LinkedList;
import java.util.List;
import java.util.Optional;

import cn.hutool.core.convert.Convert;
import com.gci.boot.constant.CommonConstant;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import com.gci.boot.exception.GciBootException;
import com.gci.boot.modules.vehicles.vo.AttachmentDetailVO;

import cn.hutool.core.date.DatePattern;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.io.FileTypeUtil;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.util.RandomUtil;
import cn.hutool.core.util.StrUtil;
import io.minio.BucketExistsArgs;
import io.minio.CopyObjectArgs;
import io.minio.CopySource;
import io.minio.GetBucketPolicyArgs;
import io.minio.GetObjectArgs;
import io.minio.GetPresignedObjectUrlArgs;
import io.minio.ListObjectsArgs;
import io.minio.MakeBucketArgs;
import io.minio.MinioClient;
import io.minio.ObjectWriteResponse;
import io.minio.PutObjectArgs;
import io.minio.RemoveBucketArgs;
import io.minio.RemoveObjectArgs;
import io.minio.Result;
import io.minio.StatObjectArgs;
import io.minio.UploadObjectArgs;
import io.minio.http.Method;
import io.minio.messages.Bucket;
import io.minio.messages.DeleteObject;
import io.minio.messages.Item;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import sun.misc.BASE64Decoder;

import javax.servlet.http.HttpServletRequest;

/**
 * MinIO工具类
 *
 */
@Slf4j
@Component
public class MinioUtils {
    @Autowired
    private MinioClient minioClient;
    @Value("${minio.bucketName}")
    private String bucketName;
    @Value("${minio.fileUrl}")
    private String fileUrl;
    @Value("${minio.minioUrl}")
    private String minioUrl;
    private final String bizPath = "common";
    private final String contentType = "application/octet-stream";
    private final String noStorage = "Storage backend has reached its minimum free disk threshold. Please delete a few objects to proceed.";


    /******************************  Operate Bucket Start  ******************************/

    /**
     * 启动SpringBoot容器的时候初始化Bucket
     * 如果没有Bucket则创建
     *
     * @param bucketName
     */
    @SneakyThrows(Exception.class)
    private void createBucket(String bucketName) {
        if (!bucketExists(bucketName)) {
            minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucketName).build());
        }
    }

    /**
     * 判断Bucket是否存在，true：存在，false：不存在
     *
     * @param bucketName
     * @return
     */
    @SneakyThrows(Exception.class)
    public boolean bucketExists(String bucketName) {
        return minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build());
    }

    /**
     * 获得Bucket的策略
     *
     * @param bucketName
     * @return
     */
    @SneakyThrows(Exception.class)
    public String getBucketPolicy(String bucketName) {
        return minioClient.getBucketPolicy(GetBucketPolicyArgs
                .builder()
                .bucket(bucketName)
                .build());
    }

    /**
     * 获得所有Bucket列表
     *
     * @return
     */
    @SneakyThrows(Exception.class)
    public List<Bucket> getAllBuckets() {
        return minioClient.listBuckets();
    }

    /**
     * 根据bucketName获取其相关信息
     *
     * @param bucketName
     * @return
     */
    @SneakyThrows(Exception.class)
    public Optional<Bucket> getBucket(String bucketName) {
        return getAllBuckets().stream().filter(b -> b.name().equals(bucketName)).findFirst();
    }

    /**
     * 根据bucketName删除Bucket，true：删除成功； false：删除失败，文件或已不存在
     *
     * @param bucketName
     * @throws Exception
     */
    @SneakyThrows(Exception.class)
    public void removeBucket(String bucketName) {
        minioClient.removeBucket(RemoveBucketArgs.builder().bucket(bucketName).build());
    }

    /******************************  Operate Bucket End  ******************************/


    /******************************  Operate Files Start  ******************************/

    /**
     * 判断文件是否存在
     *
     * @param bucketName
     * @param objectName
     * @return
     */
    public boolean isObjectExist(String bucketName, String objectName) {
        boolean exist = true;
        try {
            minioClient.statObject(StatObjectArgs.builder().bucket(bucketName).object(objectName).build());
        } catch (Exception e) {
            log.error("[Minio工具类]>>>> 判断文件是否存在, 异常：", e);
            exist = false;
        }
        return exist;
    }

    /**
     * 判断文件夹是否存在
     *
     * @param bucketName
     * @param objectName
     * @return
     */
    public boolean isFolderExist(String bucketName, String objectName) {
        boolean exist = false;
        try {
            Iterable<Result<Item>> results = minioClient.listObjects(
                    ListObjectsArgs.builder().bucket(bucketName).prefix(objectName).recursive(false).build());
            for (Result<Item> result : results) {
                Item item = result.get();
                if (item.isDir() && objectName.equals(item.objectName())) {
                    exist = true;
                }
            }
        } catch (Exception e) {
            log.error("[Minio工具类]>>>> 判断文件夹是否存在，异常：", e);
            exist = false;
        }
        return exist;
    }

    /**
     * 根据文件前置查询文件
     *
     * @param bucketName 存储桶
     * @param prefix     前缀
     * @param recursive  是否使用递归查询
     * @return MinioItem 列表
     */
    @SneakyThrows(Exception.class)
    public List<Item> getAllObjectsByPrefix(String bucketName,
                                            String prefix,
                                            boolean recursive) {
        List<Item> list = new ArrayList<>();
        Iterable<Result<Item>> objectsIterator = minioClient.listObjects(
                ListObjectsArgs.builder().bucket(bucketName).prefix(prefix).recursive(recursive).build());
        if (objectsIterator != null) {
            for (Result<Item> o : objectsIterator) {
                Item item = o.get();
                list.add(item);
            }
        }
        return list;
    }

    /**
     * 获取文件流
     *
     * @param bucketName 存储桶
     * @param objectName 文件名
     * @return 二进制流
     */
    @SneakyThrows(Exception.class)
    public InputStream getObject(String bucketName, String objectName) {
        return minioClient.getObject(
                GetObjectArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .build());
    }

    /**
     * 断点下载
     *
     * @param bucketName 存储桶
     * @param objectName 文件名称
     * @param offset     起始字节的位置
     * @param length     要读取的长度
     * @return 二进制流
     */
    @SneakyThrows(Exception.class)
    public InputStream getObject(String bucketName, String objectName, long offset, long length) {
        return minioClient.getObject(
                GetObjectArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .offset(offset)
                        .length(length)
                        .build());
    }

    /**
     * 获取路径下文件列表
     *
     * @param bucketName 存储桶
     * @param prefix     文件名称
     * @param recursive  是否递归查找，false：模拟文件夹结构查找
     * @return 二进制流
     */
    public Iterable<Result<Item>> listObjects(String bucketName, String prefix, boolean recursive) {
        return minioClient.listObjects(
                ListObjectsArgs.builder()
                        .bucket(bucketName)
                        .prefix(prefix)
                        .recursive(recursive)
                        .build());
    }

    /**
     * 使用MultipartFile进行文件上传
     *
     * @param bucketName  存储桶
     * @param file        文件名
     * @param objectName  对象名
     * @param contentType 类型
     * @return
     */
    @SneakyThrows(Exception.class)
    public ObjectWriteResponse uploadFile(String bucketName, MultipartFile file, String objectName, String contentType) {
        InputStream inputStream = file.getInputStream();
        return minioClient.putObject(
                PutObjectArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .contentType(contentType)
                        .stream(inputStream, inputStream.available(), -1)
                        .build());
    }

    /**
     * 使用MultipartFile进行文件上传
     *
     * @param file  文件
     * @param fileSuffixList 需要校验的文件列表
     * @return
     */
    @SneakyThrows(Exception.class)
    public String uploadFileValid(MultipartFile file,List<String> fileSuffixList) {
        createBucket(bucketName);

        InputStream inputStream = file.getInputStream();
        BufferedInputStream bufferedInputStream = IoUtil.toBuffered(inputStream);

        bufferedInputStream.mark(0);
        String fileContentType = file.getContentType();
        String type = FileTypeUtil.getType(bufferedInputStream,file.getOriginalFilename());
        if (!fileSuffixList.contains(type)) {
            bufferedInputStream.close();
            throw new GciBootException("上传失败，文件类型不在"+fileSuffixList.toString()+"中！");
        }
        bufferedInputStream.reset();

        String name = file.getOriginalFilename();
        int typeIndex = name.lastIndexOf(".");
        String fileType = "";
        if(typeIndex>0) {
        	fileType = name.substring(typeIndex);
        	name = name.substring(0, typeIndex);
        }
        String s = RandomUtil.randomNumbers(10);
        String afterBizPath = bizPath + StrUtil.SLASH + new DateTime().toString(DatePattern.NORM_MONTH_PATTERN);
        String objectName = afterBizPath + "/" + name + "_" + s + fileType;
        try {
            minioClient.putObject(
                    PutObjectArgs.builder()
                            .bucket(bucketName)
                            .object(objectName)
                            .contentType(fileContentType)
                            .stream(bufferedInputStream, bufferedInputStream.available(), -1)
                            .build());
        } catch (Exception e) {
            e.printStackTrace();
            log.error("文件上传报错：{}",e.getMessage());
            if (e.getMessage().equals(noStorage)) {
                throw new GciBootException("minio内存不足，无法上传！请联系管理员！");
            } else {
                throw new GciBootException("minio上传出现未知异常，请联系管理员！");
            }
        } finally {
            bufferedInputStream.close();
        }
        return getRelFileUrl() + bucketName + "/" + objectName;
    }
    
    /**
     * 使用MultipartFile进行文件上传
     *
     * @param file  文件
     * @param fileSuffixList 需要校验的文件列表
     * @return
     */
    @SneakyThrows(Exception.class)
    public String uploadFileValidNotIn(MultipartFile file,List<String> fileSuffixList) {
        createBucket(bucketName);

        InputStream inputStream = file.getInputStream();
        BufferedInputStream bufferedInputStream = IoUtil.toBuffered(inputStream);

        bufferedInputStream.mark(0);
        String fileContentType = file.getContentType();
        String type = FileTypeUtil.getType(bufferedInputStream,file.getOriginalFilename());
        if (fileSuffixList.contains(type)) {
            bufferedInputStream.close();
            throw new GciBootException("上传失败，文件类型不能在"+fileSuffixList.toString()+"中！");
        }
        bufferedInputStream.reset();

        String name = file.getOriginalFilename();
        int typeIndex = name.lastIndexOf(".");
        String fileType = "";
        if(typeIndex>0) {
        	fileType = name.substring(typeIndex);
        	name = name.substring(0, typeIndex);
        }
        String s = RandomUtil.randomNumbers(10);
        String afterBizPath = bizPath + StrUtil.SLASH + new DateTime().toString(DatePattern.NORM_MONTH_PATTERN);
        String objectName = afterBizPath + "/" + name + "_" + s + fileType;

        try {
            minioClient.putObject(
                    PutObjectArgs.builder()
                            .bucket(bucketName)
                            .object(objectName)
                            .contentType(fileContentType)
                            .stream(bufferedInputStream, bufferedInputStream.available(), -1)
                            .build());
        } catch (Exception e) {
            e.printStackTrace();
            log.error("文件上传报错：{}",e.getMessage());
            if (e.getMessage().equals(noStorage)) {
                throw new GciBootException("minio内存不足，无法上传！请联系管理员！");
            } else {
                throw new GciBootException("minio上传出现未知异常，请联系管理员！");
            }
        } finally {
            bufferedInputStream.close();
        }
        return getRelFileUrl() + bucketName + "/" + objectName;
    }

    /**
     * 使用MultipartFile进行文件上传
     *
     * @param file  文件
     * @param fileSuffixList 需要校验的文件列表
     * @return
     */
    @SneakyThrows(Exception.class)
    public AttachmentDetailVO uploadFileV2Valid(MultipartFile file, List<String> fileSuffixList) {
        createBucket(bucketName);

        InputStream inputStream = file.getInputStream();
        BufferedInputStream bufferedInputStream = IoUtil.toBuffered(inputStream);

        bufferedInputStream.mark(0);
        String fileContentType = file.getContentType();
        String type = FileTypeUtil.getType(bufferedInputStream,file.getOriginalFilename());
        if (!fileSuffixList.contains(type)) {
            bufferedInputStream.close();
            throw new GciBootException("上传失败，文件类型不在"+fileSuffixList.toString()+"中！");
        }
        bufferedInputStream.reset();

        String name = file.getOriginalFilename();
        int typeIndex = name.lastIndexOf(".");
        String fileType = "";
        if(typeIndex>0) {
        	fileType = name.substring(typeIndex);
        	name = name.substring(0, typeIndex);
        }
        String s = RandomUtil.randomNumbers(10);
        String afterBizPath = bizPath + StrUtil.SLASH + new DateTime().toString(DatePattern.NORM_MONTH_PATTERN);
        String objectName = afterBizPath + "/" + name + "_" + s + fileType;

        try {
            minioClient.putObject(
                    PutObjectArgs.builder()
                            .bucket(bucketName)
                            .object(objectName)
                            .contentType(fileContentType)
                            .stream(bufferedInputStream, bufferedInputStream.available(), -1)
                            .build());
        } catch (Exception e) {
            e.printStackTrace();
            log.error("文件上传报错：{}",e.getMessage());
            if (e.getMessage().equals(noStorage)) {
                throw new GciBootException("minio内存不足，无法上传！请联系管理员！");
            } else {
                throw new GciBootException("minio上传出现未知异常，请联系管理员！");
            }
        } finally {
            bufferedInputStream.close();
        }

        AttachmentDetailVO vo = new AttachmentDetailVO();
        vo.setFileUrl(getRelFileUrl() + bucketName + "/" + objectName);
        vo.setFileName(name);
        vo.setFileType(type);
        return vo;
    }

    /**
     * 图片上传
     * @param bucketName
     * @param imageBase64
     * @param imageName
     * @return
     */
    public ObjectWriteResponse uploadImage(String bucketName, String imageBase64, String imageName) {
        if (!StrUtil.isEmpty(imageBase64)) {
            InputStream in = base64ToInputStream(imageBase64);
            String newName = System.currentTimeMillis() + "_" + imageName + ".jpg";
            String year = String.valueOf(new Date().getYear());
            String month = String.valueOf(new Date().getMonth());
            return uploadFile(bucketName, year + "/" + month + "/" + newName, in);

        }
        return null;
    }

    public static InputStream base64ToInputStream(String base64) {
        ByteArrayInputStream stream = null;
        try {
            byte[] bytes = new BASE64Decoder().decodeBuffer(base64.trim());
            stream = new ByteArrayInputStream(bytes);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return stream;
    }


    /**
     * 上传本地文件
     *
     * @param bucketName 存储桶
     * @param objectName 对象名称
     * @param fileName   本地文件路径
     * @return
     */
    @SneakyThrows(Exception.class)
    public ObjectWriteResponse uploadFile(String bucketName, String objectName, String fileName) {
        return minioClient.uploadObject(
                UploadObjectArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .filename(fileName)
                        .build());
    }

    /**
     * 通过流上传文件
     *
     * @param bucketName  存储桶
     * @param objectName  文件对象
     * @param inputStream 文件流
     * @return
     */
    @SneakyThrows(Exception.class)
    public ObjectWriteResponse uploadFile(String bucketName, String objectName, InputStream inputStream) {
        return minioClient.putObject(
                PutObjectArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .stream(inputStream, inputStream.available(), -1)
                        .build());
    }

    /**
     * 创建文件夹或目录
     *
     * @param bucketName 存储桶
     * @param objectName 目录路径
     * @return
     */
    @SneakyThrows(Exception.class)
    public ObjectWriteResponse createDir(String bucketName, String objectName) {
        return minioClient.putObject(
                PutObjectArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .stream(new ByteArrayInputStream(new byte[]{}), 0, -1)
                        .build());
    }

    /**
     * 获取文件信息, 如果抛出异常则说明文件不存在
     *
     * @param bucketName 存储桶
     * @param objectName 文件名称
     * @return
     */
    @SneakyThrows(Exception.class)
    public String getFileStatusInfo(String bucketName, String objectName) {
        return minioClient.statObject(
                StatObjectArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .build()).toString();
    }

    /**
     * 拷贝文件
     *
     * @param bucketName    存储桶
     * @param objectName    文件名
     * @param srcBucketName 目标存储桶
     * @param srcObjectName 目标文件名
     */
    @SneakyThrows(Exception.class)
    public ObjectWriteResponse copyFile(String bucketName, String objectName, String srcBucketName, String srcObjectName) {
        return minioClient.copyObject(
                CopyObjectArgs.builder()
                        .source(CopySource.builder().bucket(bucketName).object(objectName).build())
                        .bucket(srcBucketName)
                        .object(srcObjectName)
                        .build());
    }

    /**
     * 删除文件
     *
     * @param bucketName 存储桶
     * @param objectName 文件名称
     */
    @SneakyThrows(Exception.class)
    public void removeFile(String bucketName, String objectName) {
        minioClient.removeObject(
                RemoveObjectArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .build());
    }

    /**
     * 批量删除文件
     *
     * @param bucketName 存储桶
     * @param keys       需要删除的文件列表
     * @return
     */
    public void removeFiles(String bucketName, List<String> keys) {
        List<DeleteObject> objects = new LinkedList<>();
        keys.forEach(s -> {
            objects.add(new DeleteObject(s));
            try {
                removeFile(bucketName, s);
            } catch (Exception e) {
                log.error("[Minio工具类]>>>> 批量删除文件，异常：", e);
            }
        });
    }

    /**
     * 获取文件外链
     *
     * @param bucketName 存储桶
     * @param objectName 文件名
     * @param expires    过期时间 <=7 秒 （外链有效时间（单位：秒））
     * @return url
     */
    @SneakyThrows(Exception.class)
    public String getPresignedObjectUrl(String bucketName, String objectName, Integer expires) {
        GetPresignedObjectUrlArgs args = GetPresignedObjectUrlArgs.builder().expiry(expires).bucket(bucketName).object(objectName).build();
        return minioClient.getPresignedObjectUrl(args);
    }

    /**
     * 获得文件外链
     *
     * @param bucketName
     * @param objectName
     * @return url
     */
    @SneakyThrows(Exception.class)
    public String getPresignedObjectUrl(String bucketName, String objectName) {
        GetPresignedObjectUrlArgs args = GetPresignedObjectUrlArgs.builder()
                .bucket(bucketName)
                .object(objectName)
                .method(Method.GET).build();
        return minioClient.getPresignedObjectUrl(args);
    }

    /**
     * 将URLDecoder编码转成UTF8
     *
     * @param str
     * @return
     * @throws UnsupportedEncodingException
     */
    public String getUtf8ByURLDecoder(String str) throws UnsupportedEncodingException {
        String url = str.replaceAll("%(?![0-9a-fA-F]{2})", "%25");
        return URLDecoder.decode(url, "UTF-8");
    }
    

    /**
     * @param file  文件
     * @param fileSuffixList 需要校验的文件列表
     * @return
     */
    @SneakyThrows(Exception.class)
    public String uploadFileValid(File file,List<String> fileSuffixList) {
        createBucket(bucketName);

        InputStream inputStream = new FileInputStream(file);
        BufferedInputStream bufferedInputStream = IoUtil.toBuffered(inputStream);

        bufferedInputStream.mark(0);
        String type = FileTypeUtil.getType(bufferedInputStream,file.getName());
        if (!fileSuffixList.contains(type)) {
            bufferedInputStream.close();
            throw new GciBootException("上传失败，文件类型不在"+fileSuffixList.toString()+"中！");
        }
        bufferedInputStream.reset();

        String name = file.getName();
        int typeIndex = name.lastIndexOf(".");
        String fileType = "";
        if(typeIndex>0) {
        	fileType = name.substring(typeIndex);
        	name = name.substring(0, typeIndex);
        }
        String s = RandomUtil.randomNumbers(10);
        String afterBizPath = bizPath + StrUtil.SLASH + new DateTime().toString(DatePattern.NORM_MONTH_PATTERN);
        String objectName = afterBizPath + "/" + name + "_" + s + fileType;
        try {
            minioClient.putObject(
                    PutObjectArgs.builder()
                            .bucket(bucketName)
                            .object(objectName)
                            //.contentType(type)
                            .stream(bufferedInputStream, bufferedInputStream.available(), -1)
                            .build());
        } catch (Exception e) {
            e.printStackTrace();
            log.error("文件上传报错：{}",e.getMessage());
            if (e.getMessage().equals(noStorage)) {
                throw new GciBootException("minio内存不足，无法上传！请联系管理员！");
            } else {
                throw new GciBootException("minio上传出现未知异常，请联系管理员！");
            }
        } finally {
            bufferedInputStream.close();
        }
        return getRelFileUrl() + bucketName + "/" + objectName;
    }
    
    public void checkType(MultipartFile file,List<String> fileSuffixList) {
    	boolean nopass = false;
		try {
			InputStream inputStream = file.getInputStream();
	        BufferedInputStream bufferedInputStream = IoUtil.toBuffered(inputStream);

	        bufferedInputStream.mark(0);
	        String type = FileTypeUtil.getType(bufferedInputStream,file.getName());
	        if (!fileSuffixList.contains(type)) {
	        	nopass = true;
	        }
	        bufferedInputStream.close();
	        
		} catch (Exception e) {
			throw new GciBootException("文件上传失败！");
		}
		if(nopass) {
			throw new GciBootException("上传失败，文件类型不在"+fileSuffixList.toString()+"中！");
		}
    }

    private String getRelFileUrl() {
        String inFileUrl;
        HttpServletRequest request = SpringContextUtil.getHttpServletRequest();
        String header = request.getHeader(CommonConstant.IS_INNER);
        if (Convert.toBool(header,false)) {
            inFileUrl = minioUrl;
        } else {
            inFileUrl = fileUrl;
        }
        return inFileUrl;
    }
}