package com.tievd.jyz.util;

import com.amazonaws.ClientConfiguration;
import com.amazonaws.HttpMethod;
import com.amazonaws.Protocol;
import com.amazonaws.SDKGlobalConfiguration;
import com.amazonaws.auth.AWSStaticCredentialsProvider;
import com.amazonaws.auth.BasicAWSCredentials;
import com.amazonaws.client.builder.AwsClientBuilder;
import com.amazonaws.services.s3.AmazonS3;
import com.amazonaws.services.s3.model.GeneratePresignedUrlRequest;
import com.amazonaws.services.s3.model.ObjectMetadata;
import com.amazonaws.services.s3.model.S3Object;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import javax.annotation.PostConstruct;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.net.URL;
import java.util.Date;

/**
 * S3协议存储处理工具类
 * @author Wang
 *
 */
@Component
public class S3Util {

    public final String HTTPS_STR = "HTTPS";

    @Value("${init.minio.minio_name}")
    public String accessKey;
    @Value("${init.minio.minio_pass}")
    public String secretKey;
    @Value("${init.minio.bucketName}")
    public String bucketName;
    @Value("${init.minio.minio_url}")
    public String serviceEndpoint;
    @Value("${init.minio.ext_minio_url:${init.minio.minio_url}}")
    public String extServiceEndpoint;
    @Value("${init.minio.device_minio_url:${init.minio.ext_minio_url:${init.minio.minio_url}}}")
    public String deviceServiceEndpoint;

    @Value("${init.minio.pathStyle:true}")
    public boolean isPathStyle;

    private static AmazonS3 conn = null;

    /**
     * 上传文件
     * @param bytes
     * @param key s3 key名
     * @param bucketName s3桶名
     * @return
     * @throws Exception
     */
    public String uploadObject(byte[] bytes, String key, String bucketName) throws Exception {
        try(InputStream inputStream = new ByteArrayInputStream(bytes);){
            ObjectMetadata metadata = new ObjectMetadata();
            metadata.setContentLength(inputStream.available());
            conn.putObject(bucketName, key, inputStream, metadata);
            return key;
        }
    }

    /**
     * 上传文件
     * @param bytes 数据
     * @param fileName 文件名
     * @return
     * @throws Exception
     */
    public String uploadObject(byte[] bytes,String fileName) throws Exception {
        try(InputStream inputStream = new ByteArrayInputStream(bytes);){
            ObjectMetadata metadata = new ObjectMetadata();
            metadata.setContentLength(inputStream.available());
            String key = fileName.substring(0, fileName.lastIndexOf(".")) + "_" + System.currentTimeMillis() + fileName.substring(fileName.lastIndexOf("."));
            conn.putObject(bucketName, key, inputStream, metadata);
            return key;
        }
    }


    /**
     * 根据文件名称进行删除
     * @param bucketName
     * @param key
     * @return
     */
    public void delFile(String bucketName, String key){
        try {
            if (conn.doesBucketExistV2(bucketName) == false) {
                return;
            }
            conn.deleteObject(bucketName, key);
        }catch (Exception ex){
            ex.printStackTrace();
        }
    }

    /**
     * 下载文件
     * @param key
     * @param bucketName
     * @return
     * @throws Exception
     */
    public S3Object download(String key, String bucketName) throws Exception {
        S3Object obj = conn.getObject(bucketName, key);
        return obj;
    }

    /**
     * 生成url
     * @param bucketName
     * @param key
     * @param expires 单位：秒
     * @return
     */
    public String createUrl(String bucketName, String key, Integer expires){
        if(StringUtils.isEmpty(bucketName)){
            return key;
        }
        try {
            GeneratePresignedUrlRequest urlRequest = new GeneratePresignedUrlRequest(bucketName, key, HttpMethod.GET);
            urlRequest.setExpiration(new Date(System.currentTimeMillis()+expires*1000));
            URL url = conn.generatePresignedUrl(urlRequest);
            return url.toString();
        }catch (Exception e){
            e.printStackTrace();
            return "";
        }catch (NoSuchFieldError noSuchFieldError){
            noSuchFieldError.printStackTrace();
            return "";
        }
    }

    /**
     * 创建桶
     * @param bucketName
     */
    public void creatBucket(String bucketName) {
    	conn.createBucket(bucketName);
    }

    @PostConstruct
    public void creatConnect() {
        System.setProperty(SDKGlobalConfiguration.DISABLE_CERT_CHECKING_SYSTEM_PROPERTY, "true");
        System.setProperty(SDKGlobalConfiguration.ENABLE_S3_SIGV4_SYSTEM_PROPERTY,
                "true");
        if (SDKGlobalConfiguration.isCertCheckingDisabled()) {
            System.out.println("Cert checking is disabled");
        }
        ClientConfiguration config = new ClientConfiguration();
        config.setSignerOverride("S3SignerType");
        config.setProtocol(Protocol.HTTP);
        if(isHttps(serviceEndpoint)) {
            config.setProtocol(Protocol.HTTPS);
        }
        config.setConnectionTimeout(30*1000);
        config.setMaxConnections(200);
        BasicAWSCredentials credentials = new BasicAWSCredentials(accessKey, secretKey);
        AwsClientBuilder.EndpointConfiguration endpointConfiguration
                = new AwsClientBuilder.EndpointConfiguration(serviceEndpoint, "");
        conn = com.amazonaws.services.s3.AmazonS3ClientBuilder.standard()
                .withClientConfiguration(config)
                .withCredentials(new AWSStaticCredentialsProvider(credentials))
                .withEndpointConfiguration(endpointConfiguration)
                .withPathStyleAccessEnabled(isPathStyle)
                .build();

    }

    /**
     * 是否是https请求
     * @param serviceEndpoint
     * @return
     */
    private boolean isHttps(String serviceEndpoint){
        String tmpServiceEndpoint = serviceEndpoint.toUpperCase();
        if(tmpServiceEndpoint.startsWith(HTTPS_STR)){
            return true;
        }
        return false;
    }

    public String getBucketName(){
        return this.bucketName;
    }
}