com.aventstack.chaintest.storage.AWSS3Client Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of chaintest-core Show documentation
Core Java client library for ChainTest framework
The newest version!
package com.aventstack.chaintest.storage;

import com.aventstack.chaintest.domain.Embed;
import com.aventstack.chaintest.domain.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import software.amazon.awssdk.auth.credentials.DefaultCredentialsProvider;
import software.amazon.awssdk.core.sync.RequestBody;
import software.amazon.awssdk.http.apache.ApacheHttpClient;
import software.amazon.awssdk.services.s3.S3Client;
import software.amazon.awssdk.services.s3.model.CreateBucketRequest;
import software.amazon.awssdk.services.s3.model.HeadBucketRequest;
import software.amazon.awssdk.services.s3.model.PutObjectRequest;
import software.amazon.awssdk.services.s3.model.S3Exception;

import java.io.File;
import java.util.Base64;
import java.util.Map;

/**
 * StorageService provides the ability to upload files to a storage service
 *
 * Further reading:
 * 
 *   
 *       
 *           Default credentials provider chain
 *   
 *   
 *       
 *       Environment variables
 *   
 * 
 */
public class AWSS3Client implements StorageService {

    private static final Logger log = LoggerFactory.getLogger(AWSS3Client.class);
    private static final String AWS_ACCESS_KEY_ID = "AWS_ACCESS_KEY_ID";
    private static final String AWS_SECRET_ACCESS_KEY = "AWS_SECRET_ACCESS_KEY";
    private static final String AWS_DEFAULT_REGION = "AWS_DEFAULT_REGION";

    private S3Client _client;
    private String _bucket;
    private String _prefix;

    @Override
    public boolean create(final Map config) {
        try {
            _client = S3Client.builder()
                    .httpClientBuilder(ApacheHttpClient.builder())
                    .credentialsProvider(DefaultCredentialsProvider.builder().build())
                    .build();
            _bucket = config.getOrDefault(STORAGE_CONTAINER_NAME, DEFAULT_CONTAINER_NAME);
            _bucket = _bucket.isBlank() ? DEFAULT_CONTAINER_NAME : _bucket;
            log.info("S3 client ready, using bucket: {}", _bucket);
            return true;
        } catch (final Exception e) {
            log.error("Failed to create AWS S3 client", e);
            return false;
        }
    }

    public void withPrefix(final String prefix) {
        _prefix = prefix;
    }

    private void createBucket(final String bucketName) {
        try {
            _client.createBucket(CreateBucketRequest
                    .builder()
                    .bucket(bucketName)
                    .build());
            log.debug("Creating bucket: {}", bucketName);
            _client.waiter().waitUntilBucketExists(HeadBucketRequest.builder()
                    .bucket(bucketName)
                    .build());
            log.debug("{} is ready.", bucketName);
        } catch (final S3Exception e) {
            log.error("Failed to create bucket", e);
        }
    }

    @Override
    public String upload(final Test test, final String key, final byte[] data) {
        final String prefixKey = getPrefixKey(_prefix, key);
        _client.putObject(PutObjectRequest.builder().bucket(_bucket).key(prefixKey)
                        .build(),
                RequestBody.fromBytes(data));
        return getUrl(test, prefixKey);
    }

    private String getUrl(final Test test, final String key) {
        return _client.utilities().getUrl(builder -> builder.bucket(_bucket).key(key)).toExternalForm();
    }

    @Override
    public String upload(final Test test, String key, String base64) {
        final byte[] data = Base64.getDecoder().decode(base64.getBytes());
        return upload(test, key, data);
    }

    @Override
    public String upload(final Test test, final String key, final File file) {
        final String prefixKey = getPrefixKey(_prefix, key);
        _client.putObject(PutObjectRequest.builder().bucket(_bucket).key(prefixKey)
                        .build(),
                RequestBody.fromFile(file));
        return getUrl(test, prefixKey);
    }

    @Override
    public void upload(final Test test, Embed embed) {
        String url = null;
        if (null != embed.getBytes()) {
            url = upload(test, embed.getName(), embed.getBytes());
        } else if (null != embed.getBase64() && !embed.getBase64().isBlank()) {
            url = upload(test, embed.getName(), embed.getBase64());
        } else if (null != embed.getFile()) {
            url = upload(test, embed.getName(), embed.getFile());
        } else {
            log.error("Unable to upload Embed to Azure Blob Storage. Source missing");
        }
        embed.setUrl(url);
    }

    @Override
    public void close() {
        _client.close();
    }

}