Track smithy kotlin changes

Author: 0marperez

codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/customization/flexiblechecksums/FlexibleChecksumsRequest.kt

@@ -81,6 +81,8 @@ class FlexibleChecksumsRequest : KotlinIntegration {
         }
 
         override fun render(ctx: ProtocolGenerator.GenerationContext, op: OperationShape, writer: KotlinWriter) {
+            val inputSymbol = ctx.symbolProvider.toSymbol(ctx.model.expectShape(op.inputShape))
+
             val httpChecksumTrait = op.getTrait<HttpChecksumTrait>()!!
 
             val requestAlgorithmMember = ctx.model.expectShape<StructureShape>(op.input.get())
@@ -91,13 +93,14 @@ class FlexibleChecksumsRequest : KotlinIntegration {
             val requestChecksumRequired = httpChecksumTrait.isRequestChecksumRequired
 
             writer.withBlock(
-                "op.interceptors.add(#T(",
+                "op.interceptors.add(#T<#T>(",
                 "))",
                 RuntimeTypes.HttpClient.Interceptors.FlexibleChecksumsRequestInterceptor,
+                inputSymbol,
             ) {
-                writer.write("requestChecksumRequired = #L,", requestChecksumRequired)
-                writer.write("requestChecksumCalculation = config.requestChecksumCalculation,")
-                writer.write("userSelectedChecksumAlgorithm = input.#L?.value,", requestAlgorithmMemberName)
+                writer.write("#L,", requestChecksumRequired)
+                writer.write("config.requestChecksumCalculation,")
+                writer.write("input.#L?.value,", requestAlgorithmMemberName)
             }
         }
     }

codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/customization/flexiblechecksums/FlexibleChecksumsResponse.kt

@@ -18,6 +18,13 @@ import software.amazon.smithy.model.Model
 import software.amazon.smithy.model.shapes.OperationShape
 import software.amazon.smithy.model.shapes.StructureShape
 
+/**
+ * AWS services that can return composite checksums in their responses
+ */
+private val compositeChecksumServices = setOf(
+    "s3",
+)
+
 /**
  * Adds a middleware which validates checksums returned in responses if the user has opted-in.
  */
@@ -75,19 +82,23 @@ class FlexibleChecksumsResponse : KotlinIntegration {
         }
 
         override fun render(ctx: ProtocolGenerator.GenerationContext, op: OperationShape, writer: KotlinWriter) {
+            val inputSymbol = ctx.symbolProvider.toSymbol(ctx.model.expectShape(op.inputShape))
+
             val httpChecksumTrait = op.getTrait<HttpChecksumTrait>()!!
             val requestValidationModeMember = ctx.model.expectShape<StructureShape>(op.input.get())
                 .members()
                 .first { it.memberName == httpChecksumTrait.requestValidationModeMember.get() }
             val requestValidationModeMemberName = ctx.symbolProvider.toMemberName(requestValidationModeMember)
 
             writer.withBlock(
-                "op.interceptors.add(#T(",
+                "op.interceptors.add(#T<#T>(",
                 "))",
                 RuntimeTypes.HttpClient.Interceptors.FlexibleChecksumsResponseInterceptor,
+                inputSymbol,
             ) {
-                writer.write("responseValidationRequired = input.#L?.value == \"ENABLED\",", requestValidationModeMemberName)
-                writer.write("responseChecksumValidation = config.responseChecksumValidation,")
+                writer.write("input.#L?.value == \"ENABLED\",", requestValidationModeMemberName)
+                writer.write("config.responseChecksumValidation,")
+                writer.write("#L", compositeChecksumServices.contains(ctx.settings.sdkId))
             }
         }
     }