package com.nuvole.util;

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.StrUtil;
import com.amazonaws.auth.AWSCredentialsProvider;
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.AmazonS3ClientBuilder;
import com.amazonaws.services.s3.model.*;
import org.springframework.core.io.DefaultResourceLoader;
import org.springframework.core.io.ResourceLoader;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.net.URL;
import java.time.Instant;
import java.util.Date;
import java.util.Properties;

/**
 * @author ChenLong
 * @version 1.0
 * @ClassName SFtpUtil
 * @date 2019/7/24 20:53
 */
public class OSSUtil {

    private static String bucketName = null;
    private static String endpoint = null;
    private static String endpointNet = null;
    private static String region = null;
    private static String accessKey = null;
    private static String secretKey = null;

    private static AmazonS3 amazonS3NetClient = null;

    public static Properties getYdOssProp() {
        Properties props = new Properties();
        InputStream in = null;
        try {
            ResourceLoader resourceLoader = new DefaultResourceLoader();
            in = resourceLoader.getResource("ydOss.properties").getInputStream();
            props.load(in);
            return props;
        } catch (IOException e) {
            e.printStackTrace();
            return props;
        } finally {
            ResourceUtil.safeClose(in);
        }
    }
//    private static AmazonS3 client = null;

    public static AmazonS3 ossInit() {
        Properties props = getYdOssProp();
        bucketName = props.getProperty("ydOss.bucketName");
        endpoint = props.getProperty("ydOss.endpoint");
        region = props.getProperty("ydOss.region");
        accessKey = props.getProperty("ydOss.accessKey");
        secretKey = props.getProperty("ydOss.secretKey");
        // 创建 AmazonS3 实例。
        AwsClientBuilder.EndpointConfiguration endpointConfiguration = new AwsClientBuilder.EndpointConfiguration(endpoint, region);
        BasicAWSCredentials credentials = new BasicAWSCredentials(accessKey, secretKey);
        AWSCredentialsProvider credentialsProvider = new AWSStaticCredentialsProvider(credentials);
        AmazonS3 client = AmazonS3ClientBuilder.standard()
                .withEndpointConfiguration(endpointConfiguration)
                .withCredentials(credentialsProvider).build();
        return client;
    }

    /**
     * 获取外网oss地址
     *
     * @return
     */
    public static AmazonS3 ossNetInit() {
        if (amazonS3NetClient != null) {
            return amazonS3NetClient;
        }
        Properties props = getYdOssProp();
        bucketName = props.getProperty("ydOss.bucketName");
        endpointNet = props.getProperty("ydOss.endpointNet");
        region = props.getProperty("ydOss.region");
        accessKey = props.getProperty("ydOss.accessKey");
        secretKey = props.getProperty("ydOss.secretKey");
        // 创建 AmazonS3 实例。
        AwsClientBuilder.EndpointConfiguration endpointConfiguration = new AwsClientBuilder.EndpointConfiguration(endpointNet, region);
        BasicAWSCredentials credentials = new BasicAWSCredentials(accessKey, secretKey);
        AWSCredentialsProvider credentialsProvider = new AWSStaticCredentialsProvider(credentials);
        amazonS3NetClient = AmazonS3ClientBuilder.standard()
                .withEndpointConfiguration(endpointConfiguration)
                .withCredentials(credentialsProvider).build();
        return amazonS3NetClient;
    }

    // 上传文件。
    public static void upload(AmazonS3 client, String fileName, InputStream inputStream, String targetPath) {
        if (targetPath.startsWith("/")) {
            targetPath = targetPath.substring(1);
        }
        String objectName = targetPath + "/" + fileName;
        objectName = objectName.replace("//", "/");
        mkdirFolder(client, targetPath);

        ObjectMetadata objectMetadata = new ObjectMetadata();
        try {
            // 设置对象（Object）大小。
            objectMetadata.setContentLength(inputStream.available());
            // 设置上传内容类型。
//            objectMetadata.setContentType("text/plain");
        } catch (IOException e) {
            e.printStackTrace();
        }

        // 上传同时设置访问权限，例如'Private'私有权限。
        PutObjectRequest request = new PutObjectRequest(bucketName, objectName, inputStream, objectMetadata);
        request.setCannedAcl(CannedAccessControlList.Private);
        PutObjectResult putObjectResult = client.putObject(request);
//        PutObjectResult putObjectResult = client.putObject(bucketName, objectName, inputStream, null);
        if (putObjectResult == null) {
            return;
        }
        try {
//            return getShareUrl(objectName);
            return;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return;
    }


    public static void upload(AmazonS3 client, String sourceFilePath, String targetFilePath, boolean isDelSourceFile) {
        String targetPath = targetFilePath.substring(0, targetFilePath.lastIndexOf("/"));
        System.out.println("上传 本地路径" + sourceFilePath+" 上传到 " + targetFilePath);
        File file = new File(sourceFilePath);
        FileInputStream fileInputStream = null;
        try {
            fileInputStream = new FileInputStream(file);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            return;
        }
        upload(client, file.getName(), fileInputStream, targetPath);

        if (isDelSourceFile) {
            FileUtil.del(sourceFilePath);
        }
        return;
    }

    public static void upload(AmazonS3 client, String FileName, BufferedImage bufferedImage, String targetPath)
            throws IOException {
        if (FileName.lastIndexOf(".") <= -1) {
            return;
        }
        String ext = FileName.substring(FileName.lastIndexOf(".") + 1);
        String sourceFilePath = System.getProperty("user.dir") + ResourceUtil.createFileName(ext);
        File tmpFile = FileUtil.touch(sourceFilePath);
        ImageIO.write(bufferedImage, "png", tmpFile);
        System.out.println("targetPath" + targetPath);
        upload(client, sourceFilePath, targetPath + FileName, true);
    }

    public static void download(AmazonS3 client, String sourceFilePath, String targetFilePath) {
        if (StrUtil.isEmpty(sourceFilePath)){
            return;
        }
        if (sourceFilePath.startsWith("/")){
            sourceFilePath = sourceFilePath.substring(1);
        }
        // 将文件（Object）下载到文件中，并返回文件（Object）的元数据。
        GetObjectRequest request = new GetObjectRequest(bucketName, sourceFilePath);
        ObjectMetadata meta = client.getObject(request, FileUtil.touch(targetFilePath));
    }

    /**
     * 创建文件夹
     *
     * @return
     */
    private static boolean mkdirFolder(AmazonS3 client, String directory) {
        String folder = String.copyValueOf(directory.toCharArray());
        if (directory.startsWith("/")) {
            folder = folder.substring(1);
        }
        if (!folder.endsWith("/")) {
            folder = folder + "/";
        }
        PutObjectResult putObjectResult = client.putObject(bucketName, folder, new ByteArrayInputStream(new byte[0]), null);
        return putObjectResult != null;
    }

    /**
     * 获取对象的分享链接
     *
     * @param objectName
     * @return
     */
    public static String getShareUrl(String objectName) {
        ossNetInit();
        // 生成可预览的外链。
        GeneratePresignedUrlRequest request = new GeneratePresignedUrlRequest(bucketName, objectName);

        // 设置过期时间，当到达该时间点时，URL 就会过期，其他人不再能访问该对象（Object）。
        Date expiration = new Date();
        // 设置 30 分钟后过期。
        long expTimeMillis = Instant.now().toEpochMilli();
        expTimeMillis += 1000 * 60 * 10;
        expiration.setTime(expTimeMillis);

        request.setExpiration(expiration);

        // 设置返回头
        // 设置为 "inline" 时在浏览器中展示，设置为 "attachment" 时以文件形式下载。
        // 此外设置为 "attachment;filename=\"filename.jpg\"" ，还可以让下载的文件名字重命名为 "filename.jpg"。
        ResponseHeaderOverrides headerOverrides = new ResponseHeaderOverrides();
        headerOverrides.setContentDisposition("inline");
        request.setResponseHeaders(headerOverrides);

        URL url = amazonS3NetClient.generatePresignedUrl(request);
        return url.toString();
    }

    public static void close(AmazonS3 client) {
        if (client != null) {
            client.shutdown();
        }
    }

    public static void main(String[] args) {
        // 创建 AmazonS3 实例。
        endpoint = "https://eos-beijing-7.cmecloud.cn";
        region = "beijing7";
        accessKey = "EDVFWAO1INQKHGJ78MK0";
        secretKey = "rQSMQ02h6oQPMaZyXR85XxLVhZh92jrwXFRkX6FX";
        bucketName = "ecosphere";
        AwsClientBuilder.EndpointConfiguration endpointConfiguration = new AwsClientBuilder.EndpointConfiguration(endpoint, region);
        BasicAWSCredentials credentials = new BasicAWSCredentials(accessKey, secretKey);
        AWSCredentialsProvider credentialsProvider = new AWSStaticCredentialsProvider(credentials);
        AmazonS3 amazonS3 = AmazonS3ClientBuilder.standard()
                .withEndpointConfiguration(endpointConfiguration)
                .withCredentials(credentialsProvider).build();
        OSSUtil.download(amazonS3, "/ecosphereBase/2024/1/3/5894af65566a44a791baa4246bd5910a.csv", "C:\\Users\\cy\\Desktop\\新建文件夹\\b.csv");
        OSSUtil.close(amazonS3);
    }


}