Consolidate fluent client builder implementations

aws/sdk-codegen/src/main/kotlin/software/amazon/smithy/rustsdk/AwsCodegenDecorator.kt

@@ -20,7 +20,7 @@ val DECORATORS = listOf(
     SigV4SigningDecorator(),
     RetryPolicyDecorator(),
     IntegrationTestDecorator(),
-    FluentClientDecorator(),
+    AwsFluentClientDecorator(),
     CrateLicenseDecorator(),
 
     // Service specific decorators

aws/sdk-codegen/src/main/kotlin/software/amazon/smithy/rustsdk/AwsFluentClientDecorator.kt

@@ -0,0 +1,120 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+
+package software.amazon.smithy.rustsdk
+
+import software.amazon.smithy.rust.codegen.rustlang.Attribute
+import software.amazon.smithy.rust.codegen.rustlang.CargoDependency
+import software.amazon.smithy.rust.codegen.rustlang.Feature
+import software.amazon.smithy.rust.codegen.rustlang.RustMetadata
+import software.amazon.smithy.rust.codegen.rustlang.RustModule
+import software.amazon.smithy.rust.codegen.rustlang.RustWriter
+import software.amazon.smithy.rust.codegen.rustlang.asType
+import software.amazon.smithy.rust.codegen.rustlang.rust
+import software.amazon.smithy.rust.codegen.rustlang.rustBlock
+import software.amazon.smithy.rust.codegen.rustlang.rustTemplate
+import software.amazon.smithy.rust.codegen.rustlang.writable
+import software.amazon.smithy.rust.codegen.smithy.RuntimeConfig
+import software.amazon.smithy.rust.codegen.smithy.RuntimeType
+import software.amazon.smithy.rust.codegen.smithy.RustCrate
+import software.amazon.smithy.rust.codegen.smithy.customize.RustCodegenDecorator
+import software.amazon.smithy.rust.codegen.smithy.generators.ClientGenerics
+import software.amazon.smithy.rust.codegen.smithy.generators.FluentClientGenerator
+import software.amazon.smithy.rust.codegen.smithy.generators.LibRsCustomization
+import software.amazon.smithy.rust.codegen.smithy.generators.LibRsSection
+import software.amazon.smithy.rust.codegen.smithy.generators.ProtocolConfig
+
+private class Types(runtimeConfig: RuntimeConfig) {
+    private val smithyClientDep = CargoDependency.SmithyClient(runtimeConfig).copy(optional = true)
+    private val awsHyperDep = runtimeConfig.awsRuntimeDependency("aws-hyper").copy(optional = true)
+
+    val awsHyper = awsHyperDep.asType()
+
+    val AwsMiddleware = RuntimeType("AwsMiddleware", awsHyperDep, "aws_hyper")
+    val DynConnector = RuntimeType("DynConnector", smithyClientDep, "smithy_client::erase")
+    val Standard = RuntimeType("Standard", smithyClientDep, "smithy_client::retry")
+}
+
+class AwsFluentClientDecorator : RustCodegenDecorator {
+    override val name: String = "FluentClient"
+    override val order: Byte = 0
+
+    override fun extras(protocolConfig: ProtocolConfig, rustCrate: RustCrate) {
+        val types = Types(protocolConfig.runtimeConfig)
+        val module = RustMetadata(additionalAttributes = listOf(Attribute.Cfg.feature("client")), public = true)
+        rustCrate.withModule(RustModule("client", module)) { writer ->
+            FluentClientGenerator(
+                protocolConfig,
+                includeSmithyGenericClientDocs = false,
+                generics = ClientGenerics(
+                    connectorDefault = "#{AwsFluentClient_DynConnector}",
+                    middlewareDefault = "#{AwsFluentClient_AwsMiddleware}",
+                    retryDefault = "#{AwsFluentClient_Standard}",
+                    codegenScope = listOf(
+                        "AwsFluentClient_AwsMiddleware" to types.AwsMiddleware,
+                        "AwsFluentClient_DynConnector" to types.DynConnector,
+                        "AwsFluentClient_Standard" to types.Standard,
+                    )
+                ),
+            ).render(writer)
+            AwsFluentClientExtensions(types).render(writer)
+        }
+        val awsHyper = "aws-hyper"
+        rustCrate.addFeature(Feature("client", true, listOf(awsHyper, "smithy-client")))
+        rustCrate.addFeature(Feature("rustls", default = true, listOf("$awsHyper/rustls")))
+        rustCrate.addFeature(Feature("native-tls", default = false, listOf("$awsHyper/native-tls")))
+    }
+
+    override fun libRsCustomizations(
+        protocolConfig: ProtocolConfig,
+        baseCustomizations: List<LibRsCustomization>
+    ): List<LibRsCustomization> {
+        return baseCustomizations + object : LibRsCustomization() {
+            override fun section(section: LibRsSection) = when (section) {
+                is LibRsSection.Body -> writable {
+                    Attribute.Cfg.feature("client").render(this)
+                    rust("pub use client::Client;")
+                }
+                else -> emptySection
+            }
+        }
+    }
+}
+
+private class AwsFluentClientExtensions(private val types: Types) {
+    fun render(writer: RustWriter) {
+        writer.rustBlock("impl<C> Client<C, aws_hyper::AwsMiddleware, smithy_client::retry::Standard>") {
+            rustTemplate(
+                """
+                pub fn from_conf_conn(conf: crate::Config, conn: C) -> Self {
+                    let client = #{aws_hyper}::Client::new(conn);
+                    Self { handle: std::sync::Arc::new(Handle { client, conf }) }
+                }
+                """,
+                "aws_hyper" to types.awsHyper,
+            )
+        }
+        writer.rustBlock("impl Client<smithy_client::erase::DynConnector, aws_hyper::AwsMiddleware, smithy_client::retry::Standard>") {
+            rustTemplate(
+                """
+                ##[cfg(any(feature = "rustls", feature = "native-tls"))]
+                pub async fn from_env() -> Self {
+                    // backwards compatibility shim
+                    use aws_types::region::ProvideRegion;
+                    let region = aws_types::region::default_provider().region().await;
+                    Self::from_conf(crate::Config::builder().region(region).build())
+                }
+
+                ##[cfg(any(feature = "rustls", feature = "native-tls"))]
+                pub fn from_conf(conf: crate::Config) -> Self {
+                    let client = #{aws_hyper}::Client::https();
+                    Self { handle: std::sync::Arc::new(Handle { client, conf }) }
+                }
+                """,
+                "aws_hyper" to types.awsHyper,
+            )
+        }
+    }
+}