package com.gkhy.safePlatform.minioDemo.utils;

import com.gkhy.safePlatform.minioDemo.config.MinioConfig;
import com.sun.org.apache.regexp.internal.RE;
import io.minio.*;
import io.minio.MinioClient;
import io.minio.errors.*;
import io.minio.http.Method;
import lombok.SneakyThrows;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;

@Configuration
public class FileUtil {

    @Resource
    private MinioConfig minioConfig;

    @Autowired
    private MinioClient minioClient;


    /**
     * 文件下载
     * @param fileName
     * @param response
     */
    public void downloadFile(String fileName, HttpServletResponse response) {
        try {
            InputStream file = minioClient.getObject(GetObjectArgs.builder()
                    .bucket(minioConfig.getBucketName())
                    .object(fileName)
                    .build());
            String filename = new String(fileName.getBytes("ISO8859-1"), StandardCharsets.UTF_8);
            response.setHeader("Content-Disposition", "attachment;filename=" + filename);
            ServletOutputStream servletOutputStream = response.getOutputStream();
            int len;
            byte[] buffer = new byte[1024];
            while ((len = file.read(buffer)) > 0) {
                servletOutputStream.write(buffer, 0, len);
            }
            servletOutputStream.flush();
            file.close();
            servletOutputStream.close();
        } catch (ErrorResponseException e) {
            e.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }


    /**
     * 文件上传
     *
     * @param file       文件
     * @param fileName   存储的文件名称
     */
    public String upload(MultipartFile file, String fileName) {
        String url;
        try {
            PutObjectArgs objectArgs = PutObjectArgs.builder()
                    .bucket(minioConfig.getBucketName())
                    .object(fileName)
                    .stream(file.getInputStream(), file.getSize(), -1)
                    .contentType(file.getContentType())
                    .build();
            //文件名称相同会覆盖
            minioClient.putObject(objectArgs);
            url = getFileUrl(fileName);
        } catch (Exception e) {
            e.printStackTrace();
            url = "";
        }
        return url;
    }

    /**
     * 文件访问路径
     *
     * @param objectName 存储桶里的对象名称
     * expiry :链接失效时间，秒为单位
     * @return
     */
    public String getFileUrl(String objectName) {
        String url = null;
        try {
            url = minioClient.getPresignedObjectUrl(GetPresignedObjectUrlArgs.builder()
                    .method(Method.GET)
                    .bucket(minioConfig.getBucketName())
                    .object(objectName)
                    .expiry(60).build());
        } catch (ErrorResponseException | InsufficientDataException | InternalException | InvalidKeyException | InvalidResponseException | IOException | NoSuchAlgorithmException | XmlParserException | ServerException e) {
            e.printStackTrace();
        }
        return url;
    }


    /**
     * 检查存储桶是否存在
     *
     * @return
     */
    @SneakyThrows
    public boolean bucketCheck() {
        boolean flag = minioClient.bucketExists(BucketExistsArgs.builder()
                .bucket(minioConfig.getBucketName()).build());
        return flag;
    }

}