feat: expose presign / presignURL methods in Models.swift to service client object

Sources/Services/AWSPolly/PollyClient.swift

@@ -481,3 +481,23 @@ extension PollyClient: PollyClientProtocol {
     }
 
 }
+
+extension PollyClient {
+    public func presignURLSynthesizeSpeech(input: SynthesizeSpeechInput, expiration: Foundation.TimeInterval) async throws -> ClientRuntime.URL {
+        let presignedURL = try await input.presignURL(config: config, expiration: expiration)
+        guard let presignedURL else {
+            throw ClientError.unknownError("Returned URL from input.presignURL() was nil.")
+        }
+        return presignedURL
+    }
+}
+
+extension PollyClient {
+    public func presignSynthesizeSpeech(input: SynthesizeSpeechInput, expiration: Foundation.TimeInterval) async throws -> ClientRuntime.SdkHttpRequest {
+        let presignedRequest = try await input.presign(config: config, expiration: expiration)
+        guard let presignedRequest else {
+            throw ClientError.unknownError("Returned request from input.presign() was nil.")
+        }
+        return presignedRequest
+    }
+}

Sources/Services/AWSS3/S3Client.swift

@@ -5023,3 +5023,53 @@ extension S3Client: S3ClientProtocol {
     }
 
 }
+
+extension S3Client {
+    public func presignURLGetObject(input: GetObjectInput, expiration: Foundation.TimeInterval) async throws -> ClientRuntime.URL {
+        let presignedURL = try await input.presignURL(config: config, expiration: expiration)
+        guard let presignedURL else {
+            throw ClientError.unknownError("Returned URL from input.presignURL() was nil.")
+        }
+        return presignedURL
+    }
+}
+
+extension S3Client {
+    public func presignURLPutObject(input: PutObjectInput, expiration: Foundation.TimeInterval) async throws -> ClientRuntime.URL {
+        let presignedURL = try await input.presignURL(config: config, expiration: expiration)
+        guard let presignedURL else {
+            throw ClientError.unknownError("Returned URL from input.presignURL() was nil.")
+        }
+        return presignedURL
+    }
+}
+
+extension S3Client {
+    public func presignGetObject(input: GetObjectInput, expiration: Foundation.TimeInterval) async throws -> ClientRuntime.SdkHttpRequest {
+        let presignedRequest = try await input.presign(config: config, expiration: expiration)
+        guard let presignedRequest else {
+            throw ClientError.unknownError("Returned request from input.presign() was nil.")
+        }
+        return presignedRequest
+    }
+}
+
+extension S3Client {
+    public func presignPutObject(input: PutObjectInput, expiration: Foundation.TimeInterval) async throws -> ClientRuntime.SdkHttpRequest {
+        let presignedRequest = try await input.presign(config: config, expiration: expiration)
+        guard let presignedRequest else {
+            throw ClientError.unknownError("Returned request from input.presign() was nil.")
+        }
+        return presignedRequest
+    }
+}
+
+extension S3Client {
+    public func presignUploadPart(input: UploadPartInput, expiration: Foundation.TimeInterval) async throws -> ClientRuntime.SdkHttpRequest {
+        let presignedRequest = try await input.presign(config: config, expiration: expiration)
+        guard let presignedRequest else {
+            throw ClientError.unknownError("Returned request from input.presign() was nil.")
+        }
+        return presignedRequest
+    }
+}

Sources/Services/AWSSTS/STSClient.swift

@@ -462,3 +462,13 @@ extension STSClient: STSClientProtocol {
     }
 
 }
+
+extension STSClient {
+    public func presignGetCallerIdentity(input: GetCallerIdentityInput, expiration: Foundation.TimeInterval) async throws -> ClientRuntime.SdkHttpRequest {
+        let presignedRequest = try await input.presign(config: config, expiration: expiration)
+        guard let presignedRequest else {
+            throw ClientError.unknownError("Returned request from input.presign() was nil.")
+        }
+        return presignedRequest
+    }
+}

codegen/smithy-aws-swift-codegen/src/main/kotlin/software/amazon/smithy/aws/swift/codegen/PresignerGenerator.kt

@@ -23,6 +23,7 @@ import software.amazon.smithy.swift.codegen.middleware.MiddlewareExecutionGenera
 import software.amazon.smithy.swift.codegen.middleware.MiddlewareStep
 import software.amazon.smithy.swift.codegen.middleware.OperationMiddleware
 import software.amazon.smithy.swift.codegen.model.expectShape
+import software.amazon.smithy.swift.codegen.model.toUpperCamelCase
 
 data class PresignableOperation(
     val serviceId: String,
@@ -50,6 +51,11 @@ class PresignerGenerator : SwiftIntegration {
                 var serviceConfig = AWSServiceConfig(writer, protoCtx)
                 renderPresigner(writer, ctx, delegator, op, inputType, serviceConfig)
             }
+            // Expose presign-request as a method for service client object
+            val symbol = protoCtx.symbolProvider.toSymbol(protoCtx.service)
+            protoCtx.delegator.useFileWriter("./${ctx.settings.moduleName}/${symbol.name}.swift") { writer ->
+                renderPresignAPIInServiceClient(writer, symbol.name, op, inputType)
+            }
         }
     }
 
@@ -105,6 +111,27 @@ class PresignerGenerator : SwiftIntegration {
         }
     }
 
+    private fun renderPresignAPIInServiceClient(
+        writer: SwiftWriter,
+        clientName: String,
+        op: OperationShape,
+        inputType: String
+    ) {
+        writer.apply {
+            openBlock("extension $clientName {", "}") {
+                val params = listOf("input: $inputType", "expiration: Foundation.TimeInterval")
+                val returnType = "ClientRuntime.SdkHttpRequest"
+                openBlock("public func presign${op.toUpperCamelCase()}(${params.joinToString()}) async throws -> $returnType {", "}") {
+                    write("let presignedRequest = try await input.presign(config: config, expiration: expiration)")
+                    openBlock("guard let presignedRequest else {", "}") {
+                        write("throw ClientError.unknownError(\"Returned request from input.presign() was nil.\")")
+                    }
+                    write("return presignedRequest")
+                }
+            }
+        }
+    }
+
     private fun resolveOperationMiddleware(protocolGenerator: ProtocolGenerator, op: OperationShape, ctx: CodegenContext): OperationMiddleware {
         val operationMiddlewareCopy = protocolGenerator.operationMiddleware.clone()
         operationMiddlewareCopy.removeMiddleware(op, MiddlewareStep.FINALIZESTEP, "AWSSigningMiddleware")

codegen/smithy-aws-swift-codegen/src/main/kotlin/software/amazon/smithy/aws/swift/codegen/customization/presignable/PresignableUrlIntegration.kt

@@ -31,6 +31,7 @@ import software.amazon.smithy.swift.codegen.middleware.MiddlewareExecutionGenera
 import software.amazon.smithy.swift.codegen.middleware.MiddlewareStep
 import software.amazon.smithy.swift.codegen.middleware.OperationMiddleware
 import software.amazon.smithy.swift.codegen.model.expectShape
+import software.amazon.smithy.swift.codegen.model.toUpperCamelCase
 
 internal val PRESIGNABLE_URL_OPERATIONS: Map<String, Set<String>> = mapOf(
     "com.amazonaws.polly#Parrot_v1" to setOf(
@@ -70,6 +71,11 @@ class PresignableUrlIntegration(private val presignedOperations: Map<String, Set
                 val serviceConfig = AWSServiceConfig(writer, protocolGenerationContext)
                 renderPresigner(writer, ctx, delegator, op, inputType, serviceConfig)
             }
+            // Expose presign-URL as a method for service client object
+            val symbol = protocolGenerationContext.symbolProvider.toSymbol(protocolGenerationContext.service)
+            protocolGenerationContext.delegator.useFileWriter("./${ctx.settings.moduleName}/${symbol.name}.swift") { writer ->
+                renderPresignURLAPIInServiceClient(writer, symbol.name, op, inputType)
+            }
             when (presignableOperation.operationId) {
                 "com.amazonaws.s3#GetObject", "com.amazonaws.polly#SynthesizeSpeech" -> {
                     renderMiddlewareClassForQueryString(ctx, delegator, op)
@@ -144,6 +150,27 @@ class PresignableUrlIntegration(private val presignedOperations: Map<String, Set
         }
     }
 
+    private fun renderPresignURLAPIInServiceClient(
+        writer: SwiftWriter,
+        clientName: String,
+        op: OperationShape,
+        inputType: String
+    ) {
+        writer.apply {
+            openBlock("extension $clientName {", "}") {
+                val params = listOf("input: $inputType", "expiration: Foundation.TimeInterval")
+                val returnType = "ClientRuntime.URL"
+                openBlock("public func presignURL${op.toUpperCamelCase()}(${params.joinToString()}) async throws -> $returnType {", "}") {
+                    write("let presignedURL = try await input.presignURL(config: config, expiration: expiration)")
+                    openBlock("guard let presignedURL else {", "}") {
+                        write("throw ClientError.unknownError(\"Returned URL from input.presignURL() was nil.\")")
+                    }
+                    write("return presignedURL")
+                }
+            }
+        }
+    }
+
     private fun resolveOperationMiddleware(protocolGenerator: ProtocolGenerator, context: ProtocolGenerator.GenerationContext, op: OperationShape): OperationMiddleware {
         val inputSymbol = MiddlewareShapeUtils.inputSymbol(context.symbolProvider, context.model, op)
         val operationMiddlewareCopy = protocolGenerator.operationMiddleware.clone()