From fc4159eaffe15f155b08d6761a030e44be2d289d Mon Sep 17 00:00:00 2001 From: Isaiah Vita Date: Mon, 17 Oct 2022 16:24:25 +0000 Subject: [PATCH] [issues-1787] add AssumeRole op to presign generate list This resolves customer issue 1787 by adding the AssumeRole operation to the presign autogeneration list for codegen on the aws sdk side --- .../AwsHttpPresignURLClientGenerator.java | 4 +++- service/sts/api_op_AssumeRole.go | 24 +++++++++++++++++++ 2 files changed, 27 insertions(+), 1 deletion(-) diff --git a/codegen/smithy-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsHttpPresignURLClientGenerator.java b/codegen/smithy-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsHttpPresignURLClientGenerator.java index 2422a7410c7..7d13ae09467 100644 --- a/codegen/smithy-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsHttpPresignURLClientGenerator.java +++ b/codegen/smithy-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsHttpPresignURLClientGenerator.java @@ -100,7 +100,9 @@ public class AwsHttpPresignURLClientGenerator implements GoIntegration { ShapeId.from("com.amazonaws.s3#DeleteBucket") ), ShapeId.from("com.amazonaws.sts#AWSSecurityTokenServiceV20110615"), SetUtils.of( - ShapeId.from("com.amazonaws.sts#GetCallerIdentity")) + ShapeId.from("com.amazonaws.sts#GetCallerIdentity"), + ShapeId.from("com.amazonaws.sts#AssumeRole") + ) ); // map of service to list of operations for which presignedURL client and operation should diff --git a/service/sts/api_op_AssumeRole.go b/service/sts/api_op_AssumeRole.go index 7d00b6bd7fa..9ae61e72c3c 100644 --- a/service/sts/api_op_AssumeRole.go +++ b/service/sts/api_op_AssumeRole.go @@ -415,3 +415,27 @@ func newServiceMetadataMiddleware_opAssumeRole(region string) *awsmiddleware.Reg OperationName: "AssumeRole", } } + +// PresignAssumeRole is used to generate a presigned HTTP Request which contains +// presigned URL, signed headers and HTTP method used. +func (c *PresignClient) PresignAssumeRole(ctx context.Context, params *AssumeRoleInput, optFns ...func(*PresignOptions)) (*v4.PresignedHTTPRequest, error) { + if params == nil { + params = &AssumeRoleInput{} + } + options := c.options.copy() + for _, fn := range optFns { + fn(&options) + } + clientOptFns := append(options.ClientOptions, withNopHTTPClientAPIOption) + + result, _, err := c.client.invokeOperation(ctx, "AssumeRole", params, clientOptFns, + c.client.addOperationAssumeRoleMiddlewares, + presignConverter(options).convertToPresignMiddleware, + ) + if err != nil { + return nil, err + } + + out := result.(*v4.PresignedHTTPRequest) + return out, nil +}