Adds accountId support to process credentials provider (#4332)

* Adds accountId support to process credentials provider

* Updates codegen tests from upstream branch to take account ID into account

Author: cenedhryn

codegen/src/test/resources/software/amazon/awssdk/codegen/poet/auth/scheme/query-endpoint-auth-params-auth-scheme-default-params-without-allowlist.java

@@ -33,6 +33,9 @@ public final class DefaultQueryAuthSchemeParams implements QueryAuthSchemeParams
 
     private final Boolean useFIPSEndpoint;
 
+
+    private final String awsAccountId;
+
     private final String endpointId;
 
     private final Boolean defaultTrueParam;
@@ -54,6 +57,7 @@ private DefaultQueryAuthSchemeParams(Builder builder) {
         this.region = builder.region;
         this.useDualStackEndpoint = builder.useDualStackEndpoint;
         this.useFIPSEndpoint = builder.useFIPSEndpoint;
+        this.awsAccountId = builder.awsAccountId;
         this.endpointId = builder.endpointId;
         this.defaultTrueParam = Validate.paramNotNull(builder.defaultTrueParam, "defaultTrueParam");
         this.defaultStringParam = Validate.paramNotNull(builder.defaultStringParam, "defaultStringParam");
@@ -88,6 +92,11 @@ public Boolean useFipsEndpoint() {
         return useFIPSEndpoint;
     }
 
+    @Override
+    public String awsAccountId() {
+        return awsAccountId;
+    }
+
     @Override
     public String endpointId() {
         return endpointId;
@@ -143,6 +152,8 @@ private static final class Builder implements QueryAuthSchemeParams.Builder, Que
 
         private Boolean useFIPSEndpoint;
 
+        private String awsAccountId;
+
         private String endpointId;
 
         private Boolean defaultTrueParam = true;
@@ -167,6 +178,7 @@ private static final class Builder implements QueryAuthSchemeParams.Builder, Que
             this.region = params.region;
             this.useDualStackEndpoint = params.useDualStackEndpoint;
             this.useFIPSEndpoint = params.useFIPSEndpoint;
+            this.awsAccountId = params.awsAccountId;
             this.endpointId = params.endpointId;
             this.defaultTrueParam = params.defaultTrueParam;
             this.defaultStringParam = params.defaultStringParam;
@@ -201,6 +213,12 @@ public Builder useFipsEndpoint(Boolean useFIPSEndpoint) {
             return this;
         }
 
+        @Override
+        public Builder awsAccountId(String awsAccountId) {
+            this.awsAccountId = awsAccountId;
+            return this;
+        }
+
         @Override
         public Builder endpointId(String endpointId) {
             this.endpointId = endpointId;

codegen/src/test/resources/software/amazon/awssdk/codegen/poet/auth/scheme/query-endpoint-auth-params-auth-scheme-endpoint-provider-without-allowlist.java

@@ -43,7 +43,7 @@ public static QueryAuthSchemeProvider create() {
     public List<AuthSchemeOption> resolveAuthScheme(QueryAuthSchemeParams params) {
         QueryEndpointParams endpointParameters = QueryEndpointParams.builder().region(params.region())
                                                                     .useDualStackEndpoint(params.useDualStackEndpoint()).useFipsEndpoint(params.useFipsEndpoint())
-                                                                    .endpointId(params.endpointId()).defaultTrueParam(params.defaultTrueParam())
+                                                                    .awsAccountId(params.awsAccountId()).endpointId(params.endpointId()).defaultTrueParam(params.defaultTrueParam())
                                                                     .defaultStringParam(params.defaultStringParam()).deprecatedParam(params.deprecatedParam())
                                                                     .booleanContextParam(params.booleanContextParam()).stringContextParam(params.stringContextParam())
                                                                     .operationContextParam(params.operationContextParam()).build();

codegen/src/test/resources/software/amazon/awssdk/codegen/poet/auth/scheme/query-endpoint-auth-params-auth-scheme-params-without-allowlist.java

@@ -49,6 +49,8 @@ static Builder builder() {
 
     Boolean useFipsEndpoint();
 
+    String awsAccountId();
+
     String endpointId();
 
     /**
@@ -90,6 +92,8 @@ interface Builder extends CopyableBuilder<Builder, QueryAuthSchemeParams> {
 
         Builder useFipsEndpoint(Boolean useFIPSEndpoint);
 
+        Builder awsAccountId(String awsAccountId);
+
         Builder endpointId(String endpointId);
 
         /**

codegen/src/test/resources/software/amazon/awssdk/codegen/poet/auth/scheme/query-endpoint-auth-params-without-allowlist-auth-scheme-interceptor.java

@@ -90,6 +90,7 @@ private QueryAuthSchemeParams authSchemeParams(SdkRequest request, ExecutionAttr
         builder.region(endpointParams.region());
         builder.useDualStackEndpoint(endpointParams.useDualStackEndpoint());
         builder.useFipsEndpoint(endpointParams.useFipsEndpoint());
+        builder.awsAccountId(endpointParams.awsAccountId());
         builder.endpointId(endpointParams.endpointId());
         builder.defaultTrueParam(endpointParams.defaultTrueParam());
         builder.defaultStringParam(endpointParams.defaultStringParam());

codegen/src/test/resources/software/amazon/awssdk/codegen/poet/rules/endpoint-parameters.java

@@ -158,6 +158,7 @@ private BuilderImpl(QueryEndpointParams builder) {
             this.region = builder.region;
             this.useDualStackEndpoint = builder.useDualStackEndpoint;
             this.useFIPSEndpoint = builder.useFIPSEndpoint;
+            this.awsAccountId = builder.awsAccountId;
             this.endpointId = builder.endpointId;
             this.defaultTrueParam = builder.defaultTrueParam;
             this.defaultStringParam = builder.defaultStringParam;

core/auth/src/main/java/software/amazon/awssdk/auth/credentials/ProcessCredentialsProvider.java

@@ -176,6 +176,7 @@ private AwsCredentials credentials(JsonNode credentialsJson) {
         String accessKeyId = getText(credentialsJson, "AccessKeyId");
         String secretAccessKey = getText(credentialsJson, "SecretAccessKey");
         String sessionToken = getText(credentialsJson, "SessionToken");
+        String accountId = getText(credentialsJson, "AccountId");
 
         Validate.notEmpty(accessKeyId, "AccessKeyId cannot be empty.");
         Validate.notEmpty(secretAccessKey, "SecretAccessKey cannot be empty.");
@@ -185,11 +186,14 @@ private AwsCredentials credentials(JsonNode credentialsJson) {
                                     .accessKeyId(accessKeyId)
                                     .secretAccessKey(secretAccessKey)
                                     .sessionToken(sessionToken)
+                                    .expirationTime(credentialExpirationTime(credentialsJson))
+                                    .accountId(accountId)
                                     .providerName(PROVIDER_NAME)
                                     .build() :
                AwsBasicCredentials.builder()
                                   .accessKeyId(accessKeyId)
                                   .secretAccessKey(secretAccessKey)
+                                  .accountId(accountId)
                                   .providerName(PROVIDER_NAME)
                                   .build();
     }