This is an automated email from the ASF dual-hosted git repository. acosentino pushed a commit to branch main in repository https://gitbox.apache.org/repos/asf/camel.git
commit 6f229d71a1cebcd5ea29349b8938bf1f037637d2 Author: Andrea Cosentino <anco...@gmail.com> AuthorDate: Mon Jun 19 14:31:58 2023 +0200 CAMEL-19159 - Camel-AWS: Support Profile Credential provider as configuration - AWS STS Signed-off-by: Andrea Cosentino <anco...@gmail.com> --- .../aws2/sts/client/STS2ClientFactory.java | 10 ++- .../client/impl/STS2ClientIAMProfileOptimized.java | 94 ++++++++++++++++++++++ 2 files changed, 102 insertions(+), 2 deletions(-) diff --git a/components/camel-aws/camel-aws2-sts/src/main/java/org/apache/camel/component/aws2/sts/client/STS2ClientFactory.java b/components/camel-aws/camel-aws2-sts/src/main/java/org/apache/camel/component/aws2/sts/client/STS2ClientFactory.java index b50c8f87565..53bcdc58da8 100644 --- a/components/camel-aws/camel-aws2-sts/src/main/java/org/apache/camel/component/aws2/sts/client/STS2ClientFactory.java +++ b/components/camel-aws/camel-aws2-sts/src/main/java/org/apache/camel/component/aws2/sts/client/STS2ClientFactory.java @@ -18,6 +18,7 @@ package org.apache.camel.component.aws2.sts.client; import org.apache.camel.component.aws2.sts.STS2Configuration; import org.apache.camel.component.aws2.sts.client.impl.STS2ClientIAMOptimized; +import org.apache.camel.component.aws2.sts.client.impl.STS2ClientIAMProfileOptimized; import org.apache.camel.component.aws2.sts.client.impl.STS2ClientStandardImpl; /** @@ -35,7 +36,12 @@ public final class STS2ClientFactory { * @return StsClient */ public static STS2InternalClient getStsClient(STS2Configuration configuration) { - return Boolean.TRUE.equals(configuration.isUseDefaultCredentialsProvider()) - ? new STS2ClientIAMOptimized(configuration) : new STS2ClientStandardImpl(configuration); + if (Boolean.TRUE.equals(configuration.isUseDefaultCredentialsProvider())) { + return new STS2ClientIAMOptimized(configuration); + } else if (Boolean.TRUE.equals(configuration.isUseProfileCredentialsProvider())) { + return new STS2ClientIAMProfileOptimized(configuration); + } else { + return new STS2ClientStandardImpl(configuration); + } } } diff --git a/components/camel-aws/camel-aws2-sts/src/main/java/org/apache/camel/component/aws2/sts/client/impl/STS2ClientIAMProfileOptimized.java b/components/camel-aws/camel-aws2-sts/src/main/java/org/apache/camel/component/aws2/sts/client/impl/STS2ClientIAMProfileOptimized.java new file mode 100644 index 00000000000..a7b6d0b5711 --- /dev/null +++ b/components/camel-aws/camel-aws2-sts/src/main/java/org/apache/camel/component/aws2/sts/client/impl/STS2ClientIAMProfileOptimized.java @@ -0,0 +1,94 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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 org.apache.camel.component.aws2.sts.client.impl; + +import org.apache.camel.component.aws2.sts.STS2Configuration; +import org.apache.camel.component.aws2.sts.client.STS2InternalClient; +import org.apache.camel.util.ObjectHelper; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import software.amazon.awssdk.auth.credentials.ProfileCredentialsProvider; +import software.amazon.awssdk.http.SdkHttpClient; +import software.amazon.awssdk.http.SdkHttpConfigurationOption; +import software.amazon.awssdk.http.apache.ApacheHttpClient; +import software.amazon.awssdk.http.apache.ProxyConfiguration; +import software.amazon.awssdk.regions.Region; +import software.amazon.awssdk.services.sts.StsClient; +import software.amazon.awssdk.services.sts.StsClientBuilder; +import software.amazon.awssdk.utils.AttributeMap; + +import java.net.URI; + +/** + * Manage an AWS STS client for all users to use. This implementation is for remote instances to manage the credentials + * on their own (eliminating credential rotations) + */ +public class STS2ClientIAMProfileOptimized implements STS2InternalClient { + private static final Logger LOG = LoggerFactory.getLogger(STS2ClientIAMProfileOptimized.class); + private STS2Configuration configuration; + + /** + * Constructor that uses the config file. + */ + public STS2ClientIAMProfileOptimized(STS2Configuration configuration) { + LOG.trace("Creating an AWS STS client for working on AWS Services"); + this.configuration = configuration; + } + + /** + * Getting the STS aws client that is used. + * + * @return Amazon STS Client. + */ + @Override + public StsClient getStsClient() { + StsClient client = null; + StsClientBuilder clientBuilder = StsClient.builder(); + ProxyConfiguration.Builder proxyConfig = null; + ApacheHttpClient.Builder httpClientBuilder = null; + + if (ObjectHelper.isNotEmpty(configuration.getProxyHost()) && ObjectHelper.isNotEmpty(configuration.getProxyPort())) { + proxyConfig = ProxyConfiguration.builder(); + URI proxyEndpoint = URI.create(configuration.getProxyProtocol() + "://" + configuration.getProxyHost() + ":" + + configuration.getProxyPort()); + proxyConfig.endpoint(proxyEndpoint); + httpClientBuilder = ApacheHttpClient.builder().proxyConfiguration(proxyConfig.build()); + clientBuilder = clientBuilder.httpClientBuilder(httpClientBuilder); + } + if (configuration.getProfileCredentialsName() != null) { + clientBuilder = clientBuilder.httpClientBuilder(httpClientBuilder) + .credentialsProvider(ProfileCredentialsProvider.create(configuration.getProfileCredentialsName())); + } + if (ObjectHelper.isNotEmpty(configuration.getRegion())) { + clientBuilder = clientBuilder.region(Region.of(configuration.getRegion())); + } + if (configuration.isOverrideEndpoint()) { + clientBuilder.endpointOverride(URI.create(configuration.getUriEndpointOverride())); + } + if (configuration.isTrustAllCertificates()) { + SdkHttpClient ahc = ApacheHttpClient.builder().buildWithDefaults(AttributeMap + .builder() + .put( + SdkHttpConfigurationOption.TRUST_ALL_CERTIFICATES, + Boolean.TRUE) + .build()); + clientBuilder.httpClient(ahc); + } + client = clientBuilder.build(); + return client; + } +}