feat: apply clock skew interceptor (#1065)

Author: lauzadis

.changes/8665c18c-d742-4212-b920-6fea43f70c2c.json

@@ -0,0 +1,5 @@
+{
+    "id": "8665c18c-d742-4212-b920-6fea43f70c2c",
+    "type": "feature",
+    "description": "Detect and automatically correct clock skew to prevent signing errors"
+}

codegen/smithy-aws-kotlin-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/ServiceClientCompanionObjectWriter.kt

@@ -4,10 +4,7 @@
  */
 package aws.sdk.kotlin.codegen
 
-import software.amazon.smithy.kotlin.codegen.core.KotlinWriter
-import software.amazon.smithy.kotlin.codegen.core.RuntimeTypes
-import software.amazon.smithy.kotlin.codegen.core.getContextValue
-import software.amazon.smithy.kotlin.codegen.core.withBlock
+import software.amazon.smithy.kotlin.codegen.core.*
 import software.amazon.smithy.kotlin.codegen.integration.SectionWriter
 import software.amazon.smithy.kotlin.codegen.lang.KotlinTypes
 import software.amazon.smithy.kotlin.codegen.rendering.ServiceClientGenerator
@@ -49,8 +46,8 @@ class ServiceClientCompanionObjectWriter(
             RuntimeTypes.Core.Utils.LazyAsyncValue,
             AwsRuntimeTypes.Config.Profile.AwsSharedConfig,
         ) {
+            declareSection(ServiceClientGenerator.Sections.FinalizeConfig)
             writeResolveEndpointUrl()
-
             extendFinalizeConfig?.let {
                 write("")
                 it()

codegen/smithy-aws-kotlin-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/customization/ClockSkew.kt

@@ -0,0 +1,24 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+package aws.sdk.kotlin.codegen.customization
+
+import software.amazon.smithy.kotlin.codegen.core.RuntimeTypes
+import software.amazon.smithy.kotlin.codegen.integration.KotlinIntegration
+import software.amazon.smithy.kotlin.codegen.integration.SectionWriter
+import software.amazon.smithy.kotlin.codegen.integration.SectionWriterBinding
+import software.amazon.smithy.kotlin.codegen.rendering.ServiceClientGenerator
+
+/**
+ * Adds a section writer which applies an interceptor that detects and corrects clock skew
+ */
+class ClockSkew : KotlinIntegration {
+    override val sectionWriters: List<SectionWriterBinding>
+        get() = listOf(SectionWriterBinding(ServiceClientGenerator.Sections.FinalizeConfig, clockSkewSectionWriter))
+
+    private val clockSkewSectionWriter = SectionWriter { writer, _ ->
+        val interceptorSymbol = RuntimeTypes.AwsProtocolCore.ClockSkewInterceptor
+        writer.write("builder.config.interceptors.add(0, #T())", interceptorSymbol)
+    }
+}

codegen/smithy-aws-kotlin-codegen/src/main/resources/META-INF/services/software.amazon.smithy.kotlin.codegen.integration.KotlinIntegration

@@ -29,4 +29,5 @@ aws.sdk.kotlin.codegen.customization.s3control.HostPrefixFilter
 aws.sdk.kotlin.codegen.customization.s3control.ClientConfigIntegration
 aws.sdk.kotlin.codegen.protocols.endpoints.BindAwsEndpointBuiltins
 aws.sdk.kotlin.codegen.customization.s3.HostPrefixRequestRouteFilter
-aws.sdk.kotlin.codegen.customization.s3.UnwrappedXmlOutputIntegration
+aws.sdk.kotlin.codegen.customization.s3.UnwrappedXmlOutputIntegration
+aws.sdk.kotlin.codegen.customization.ClockSkew

gradle/libs.versions.toml

@@ -7,7 +7,7 @@ coroutines-version = "1.7.3"
 atomicfu-version = "0.19.0"
 
 # smithy-kotlin codegen and runtime are versioned together
-smithy-kotlin-version = "0.27.7"
+smithy-kotlin-version = "0.27.8-SNAPSHOT"
 
 # codegen
 smithy-version = "1.39.0"