jetbrains.buildServer.util.amazon.AWSClients.java Source code

Introduction

Here is the source code for jetbrains.buildServer.util.amazon.AWSClients.java

Source

/*
 * Copyright 2000-2016 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License 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.
 */

package jetbrains.buildServer.util.amazon;

import com.amazonaws.AmazonWebServiceClient;
import com.amazonaws.ClientConfiguration;
import com.amazonaws.auth.AWSCredentials;
import com.amazonaws.auth.AWSSessionCredentials;
import com.amazonaws.auth.BasicAWSCredentials;
import com.amazonaws.auth.BasicSessionCredentials;
import com.amazonaws.regions.Region;
import com.amazonaws.services.elasticbeanstalk.AWSElasticBeanstalkClient;
import com.amazonaws.services.securitytoken.AWSSecurityTokenServiceClient;
import com.amazonaws.services.securitytoken.model.AssumeRoleRequest;
import com.amazonaws.services.securitytoken.model.Credentials;
import jetbrains.buildServer.util.StringUtil;
import jetbrains.buildServer.version.ServerVersionHolder;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

public class AWSClients {

    @Nullable
    private final AWSCredentials myCredentials;
    @NotNull
    private final Region myRegion;
    @NotNull
    private final ClientConfiguration myClientConfiguration;

    private AWSClients(@Nullable AWSCredentials credentials, @NotNull String region) {
        myCredentials = credentials;
        myRegion = AWSRegions.getRegion(region);
        myClientConfiguration = createClientConfiguration();
    }

    @NotNull
    public static AWSClients fromExistingCredentials(@NotNull AWSCredentials credentials, @NotNull String region) {
        return new AWSClients(credentials, region);
    }

    @NotNull
    public static AWSClients fromDefaultCredentialProviderChain(@NotNull String region) {
        return new AWSClients(null, region);
    }

    @NotNull
    public static AWSClients fromBasicCredentials(@NotNull String accessKeyId, @NotNull String secretAccessKey,
            @NotNull String region) {
        return fromExistingCredentials(new BasicAWSCredentials(accessKeyId, secretAccessKey), region);
    }

    @NotNull
    public AWSElasticBeanstalkClient createElasticBeanstalkClient() {
        return withRegion(myCredentials == null ? new AWSElasticBeanstalkClient(myClientConfiguration)
                : new AWSElasticBeanstalkClient(myCredentials, myClientConfiguration));
    }

    @NotNull
    public AWSSecurityTokenServiceClient createSecurityTokenServiceClient() {
        return myCredentials == null ? new AWSSecurityTokenServiceClient(myClientConfiguration)
                : new AWSSecurityTokenServiceClient(myCredentials, myClientConfiguration);
    }

    @NotNull
    public String getRegion() {
        return myRegion.getName();
    }

    @NotNull
    private <T extends AmazonWebServiceClient> T withRegion(@NotNull T client) {
        return client.withRegion(myRegion);
    }

    @NotNull
    public AWSSessionCredentials createSessionCredentials(@NotNull String iamRoleARN, @Nullable String externalID,
            @NotNull String sessionName, int sessionDuration) throws AWSException {
        final AssumeRoleRequest assumeRoleRequest = new AssumeRoleRequest().withRoleArn(iamRoleARN)
                .withRoleSessionName(sessionName).withDurationSeconds(sessionDuration);
        if (StringUtil.isNotEmpty(externalID))
            assumeRoleRequest.setExternalId(externalID);
        try {
            final Credentials credentials = createSecurityTokenServiceClient().assumeRole(assumeRoleRequest)
                    .getCredentials();
            return new BasicSessionCredentials(credentials.getAccessKeyId(), credentials.getSecretAccessKey(),
                    credentials.getSessionToken());
        } catch (Exception e) {
            throw new AWSException(e);
        }
    }

    public static final String UNSUPPORTED_SESSION_NAME_CHARS = "[^\\w+=,.@-]";
    public static final int MAX_SESSION_NAME_LENGTH = 64;

    @NotNull
    public static String patchSessionName(@NotNull String sessionName) {
        return StringUtil.truncateStringValue(sessionName.replaceAll(UNSUPPORTED_SESSION_NAME_CHARS, "_"),
                MAX_SESSION_NAME_LENGTH);
    }

    @NotNull
    private static ClientConfiguration createClientConfiguration() {
        return new ClientConfiguration()
                .withUserAgent("JetBrains TeamCity " + ServerVersionHolder.getVersion().getDisplayVersion());
    }
}