AWSClientFactory.java Source code

Introduction

Here is the source code for AWSClientFactory.java

Source

/*
 *     Copyright 2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 *     Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance with the License.
 *     A copy of the License is located at
 *
 *         http://aws.amazon.com/apache2.0/
 *
 *     or in the "license" file accompanying this file.
 *     This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *     See the License for the specific language governing permissions and limitations under the License.
 */

import com.amazonaws.ClientConfiguration;
import com.amazonaws.auth.AWSCredentials;
import com.amazonaws.auth.BasicAWSCredentials;
import com.amazonaws.services.codebuild.AWSCodeBuildClient;
import com.amazonaws.services.codebuild.model.InvalidInputException;
import com.amazonaws.services.logs.AWSLogsClient;
import com.amazonaws.services.s3.AmazonS3Client;

public class AWSClientFactory {

    private final String proxyHost;
    private final String proxyPort;
    private final String awsAccessKey;
    private final String awsSecretKey;
    private final String region;

    private ClientConfiguration clientConfig;
    private AWSCredentials awsCredentials;

    public AWSClientFactory(String proxyHost, String proxyPort, String awsAccessKey, String awsSecretKey,
            String region) throws InvalidInputException {

        Validation.checkAWSClientFactoryConfig(proxyHost, proxyPort, awsAccessKey, awsSecretKey);

        this.proxyHost = proxyHost;
        this.proxyPort = proxyPort;
        this.awsAccessKey = awsAccessKey;
        this.awsSecretKey = awsSecretKey;
        this.region = region;

        clientConfig = new ClientConfiguration();
        clientConfig.setUserAgentPrefix("CodeBuild-Jenkins-Plugin"); //tags all calls made from Jenkins plugin.
        clientConfig.setProxyHost(this.proxyHost);
        if (Validation.parseInt(this.proxyPort) != null) {
            clientConfig.setProxyPort(Validation.parseInt(proxyPort));
        }

        awsCredentials = new BasicAWSCredentials(this.awsAccessKey, this.awsSecretKey);
    }

    public AWSCodeBuildClient getCodeBuildClient() throws InvalidInputException {
        AWSCodeBuildClient client = new AWSCodeBuildClient(awsCredentials, clientConfig);
        client.setEndpoint("https://codebuild." + region + ".amazonaws.com");
        return client;
    }

    public AmazonS3Client getS3Client() throws InvalidInputException {
        return new AmazonS3Client(awsCredentials, clientConfig);
    }

    public AWSLogsClient getCloudWatchLogsClient() throws InvalidInputException {
        return new AWSLogsClient(awsCredentials, clientConfig);
    }
}