package com.common.aws.util;


import lombok.extern.slf4j.Slf4j;
import org.springframework.web.multipart.MultipartFile;
import software.amazon.awssdk.auth.credentials.InstanceProfileCredentialsProvider;
import software.amazon.awssdk.core.ResponseBytes;
import software.amazon.awssdk.core.sync.RequestBody;
import software.amazon.awssdk.services.s3.S3Client;
import software.amazon.awssdk.services.s3.model.*;
import sun.misc.BASE64Encoder;

import java.util.ArrayList;
import java.util.List;


/**
 * @author Ben Pi
 * @date 2022-01-21
 */
@Slf4j
public class S3Util {



    // 创建s3对象
    private static final S3Client s3Client = S3Client.builder()
            .credentialsProvider(InstanceProfileCredentialsProvider.builder().build())
            .build();


//    public static final S3Client getS3ClientUsingEC2Instance(){
//        return S3Client.builder()
//        .credentialsProvider(InstanceProfileCredentialsProvider.builder().build())
//        .build();
//    }

    /**
     * 上传文件
     *
     * @param key 上传路径
     */
    public static String uploadFile(MultipartFile file,String key,String bucketName) {
        try {
            log.info("文件上传key:{}",key);
            // 设置文件上传对象
            PutObjectRequest putObjectRequest = PutObjectRequest.builder().bucket(bucketName).key(key).build();
            // 上传文件
            PutObjectResponse putObjectResponse = s3Client.putObject(putObjectRequest, RequestBody.fromBytes(file.getBytes()));
            if (null != putObjectResponse) {
                log.info("文件上传成功:{}",key);
                return key;
            }
            return null;
        } catch (Exception e) {
            log.error("upload file error",e);
        }
        return null;
    }

    /**
     * 文件下载
     * @param key 文件的key
     * @return
     */
    public static ResponseBytes<GetObjectResponse> downloadFile(String key,String bucketName){
        GetObjectRequest objectRequest = GetObjectRequest
                    .builder()
                    .key(key)
                    .bucket(bucketName)
                    .build();
        ResponseBytes<GetObjectResponse> objectBytes = s3Client.getObjectAsBytes(objectRequest);
        return objectBytes;
    }


    /**
     * s3文件预览
     * @param key
     */
    public static String  DownloadFromS3(String key,String bucketName) {
        BASE64Encoder encoder = new BASE64Encoder();
        GetObjectRequest objectRequest = GetObjectRequest
                .builder()
                .key(key)
                .bucket(bucketName)
                .build();
        ResponseBytes<GetObjectResponse> objectBytes = s3Client.getObjectAsBytes(objectRequest);
        byte[] data = objectBytes.asByteArray();
        // 转换成base64文件流,用于前端数据解析
        return encoder.encodeBuffer(data).trim();
    }


    public static String getURL(String keyName,String bucketName) {
        String url = null;
        try {
            GetUrlRequest request = GetUrlRequest.builder()
                    .bucket(bucketName)
                    .key(keyName)
                    .build();
            url =s3Client.utilities().getUrl(request).toString();
            return url;
        } catch (S3Exception e) {
            e.awsErrorDetails().errorMessage();
        }
        return url;
    }


    public static DeleteObjectResponse deleteByKey(String key,String bucketName){
        DeleteObjectRequest build = DeleteObjectRequest
                .builder()
                .key(key)
                .bucket(bucketName)
                .build();
        return s3Client.deleteObject(build);
    }

    public static void deleteByKeys(List<String> keys, String bucketName){
        ArrayList<ObjectIdentifier> to_delete = new ArrayList<ObjectIdentifier>();
        for (String key : keys) {
            to_delete.add(ObjectIdentifier.builder().key(key).build());
        }
        try{
        DeleteObjectsRequest build = DeleteObjectsRequest.builder()
                .bucket(bucketName)
                .delete(Delete.builder().objects(to_delete).build())
                .build();
         s3Client.deleteObjects(build);
        }catch (S3Exception e){
            log.warn("AWS S3 删除多对象异常:"+e);
            throw e;
        }
    }

}