merge from main

aws · Jun 14, 2024 · 5ceabb7 · 5ceabb7
1 parent b0e0b9f
commit 5ceabb7
Show file tree

Hide file tree

Showing 421 changed files with 16,284 additions and 3,519 deletions.
diff --git a/...rc/main/java/software/amazon/smithy/aws/go/codegen/customization/AwsEndpointBuiltins.java b/...rc/main/java/software/amazon/smithy/aws/go/codegen/customization/AwsEndpointBuiltins.java
@@ -5,10 +5,13 @@
 import software.amazon.smithy.go.codegen.GoDelegator;
 import software.amazon.smithy.go.codegen.GoSettings;
 import software.amazon.smithy.go.codegen.GoWriter;
+import software.amazon.smithy.go.codegen.SmithyGoTypes;
 import software.amazon.smithy.go.codegen.integration.GoIntegration;
 import software.amazon.smithy.go.codegen.integration.RuntimeClientPlugin;
 import software.amazon.smithy.model.Model;
+import software.amazon.smithy.rulesengine.traits.EndpointRuleSetTrait;
 import software.amazon.smithy.utils.ListUtils;
+import software.amazon.smithy.utils.MapUtils;
 
 import java.util.List;
 
@@ -36,6 +39,8 @@ public class AwsEndpointBuiltins implements GoIntegration {
  goTemplate("$T(options.UseARNRegion)", SdkGoTypes.Aws.Bool);
  private static final GoWriter.Writable BindAwsS3DisableMultiRegionAccessPoints =
  goTemplate("$T(options.DisableMultiRegionAccessPoints)", SdkGoTypes.Aws.Bool);
+ private static final GoWriter.Writable BindAccountID =
+ goTemplate("resolveAccountID(getIdentity(ctx), options.AccountIDEndpointMode)");
 
  @Override
  public List<RuntimeClientPlugin> getClientPlugins() {
@@ -49,12 +54,38 @@ public List<RuntimeClientPlugin> getClientPlugins() {
  .addEndpointBuiltinBinding("AWS::S3::UseArnRegion", BindAwsS3UseArnRegion)
  .addEndpointBuiltinBinding("AWS::S3::DisableMultiRegionAccessPoints", BindAwsS3DisableMultiRegionAccessPoints)
  .addEndpointBuiltinBinding("AWS::S3Control::UseArnRegion", BindAwsS3UseArnRegion)
+ .addEndpointBuiltinBinding("AWS::Auth::AccountId", BindAccountID)
  .build());
  }
 
  @Override
  public void writeAdditionalFiles(GoSettings settings, Model model, SymbolProvider symbolProvider, GoDelegator goDelegator) {
  goDelegator.useFileWriter("endpoints.go", settings.getModuleName(), builtinBindingSource());
+ if (!settings.getService(model).hasTrait(EndpointRuleSetTrait.class)) {
+ return;
+ }
+ goDelegator.useShapeWriter(settings.getService(model), goTemplate("""
+ func resolveAccountID(identity $auth:T, mode $accountIDEndpointMode:T) *string {
+ if mode == $aidModeDisabled:T {
+ return nil
+ }
+ 
+ if ca, ok := identity.(*$credentialsAdapter:T); ok && ca.Credentials.AccountID != "" {
+ return $string:T(ca.Credentials.AccountID)
+ }
+ 
+ return nil
+ }
+ """,
+ MapUtils.of(
+ "auth", SmithyGoTypes.Auth.Identity,
+ "accountIDEndpointMode", SdkGoTypes.Aws.AccountIDEndpointMode,
+ "aidModeUnset", SdkGoTypes.Aws.AccountIDEndpointModeUnset,
+ "aidModeDisabled", SdkGoTypes.Aws.AccountIDEndpointModeDisabled,
+ "credentialsAdapter", SdkGoTypes.Internal.Auth.Smithy.CredentialsAdapter,
+ "string", SdkGoTypes.Aws.String
+ )
+ ));
  }
 
  private GoWriter.Writable builtinBindingSource() {

diff --git a/...n/resources/META-INF/services/software.amazon.smithy.go.codegen.integration.GoIntegration b/...n/resources/META-INF/services/software.amazon.smithy.go.codegen.integration.GoIntegration
@@ -76,4 +76,5 @@ software.amazon.smithy.aws.go.codegen.customization.CloudFrontKVSSigV4a
 software.amazon.smithy.aws.go.codegen.customization.BackfillProtocolTestServiceTrait
 software.amazon.smithy.go.codegen.integration.MiddlewareStackSnapshotTests
 software.amazon.smithy.aws.go.codegen.customization.s3.S3ExpiresShapeCustomization
-software.amazon.smithy.aws.go.codegen.ClockSkewGenerator
+software.amazon.smithy.aws.go.codegen.ClockSkewGenerator
+software.amazon.smithy.aws.go.codegen.customization.AccountIDEndpointRouting
diff --git a/feature/dynamodbstreams/attributevalue/go_module_metadata.go b/feature/dynamodbstreams/attributevalue/go_module_metadata.go
diff --git a/internal/protocoltest/awsrestjson/auth.go b/internal/protocoltest/awsrestjson/auth.go
diff --git a/internal/protocoltest/ec2query/auth.go b/internal/protocoltest/ec2query/auth.go
diff --git a/internal/protocoltest/jsonrpc10/auth.go b/internal/protocoltest/jsonrpc10/auth.go
diff --git a/internal/protocoltest/query/auth.go b/internal/protocoltest/query/auth.go
diff --git a/internal/protocoltest/restxml/auth.go b/internal/protocoltest/restxml/auth.go
diff --git a/internal/protocoltest/restxmlwithnamespace/api_client.go b/internal/protocoltest/restxmlwithnamespace/api_client.go
diff --git a/internal/protocoltest/restxmlwithnamespace/auth.go b/internal/protocoltest/restxmlwithnamespace/auth.go
diff --git a/internal/protocoltest/smithyrpcv2cbor/api_client.go b/internal/protocoltest/smithyrpcv2cbor/api_client.go
diff --git a/internal/protocoltest/smithyrpcv2cbor/auth.go b/internal/protocoltest/smithyrpcv2cbor/auth.go
diff --git a/internal/protocoltest/smithyrpcv2cbor/options.go b/internal/protocoltest/smithyrpcv2cbor/options.go
diff --git a/service/accessanalyzer/api_client.go b/service/accessanalyzer/api_client.go