add preprocessor to rename the KMS service namespace

.github/workflows/ci.yml

@@ -194,4 +194,5 @@ jobs:
       - name: Submit Dependencies
         uses: scalacenter/sbt-dependency-submission@v2
         with:
+          modules-ignore: smithy4s-preprocessors_2.12
           configs-ignore: test scala-tool scala-doc-tool test-internal

.mergify.yml

@@ -14,3 +14,11 @@ pull_request_rules:
   - status-success=Build and Test (ubuntu-latest, 3, temurin@8)
   actions:
     merge: {}
+- name: Label smithy4s-preprocessors PRs
+  conditions:
+  - files~=^smithy4s-preprocessors/
+  actions:
+    label:
+      add:
+      - smithy4s-preprocessors
+      remove: []

META-INF/MANIFEST.MF

@@ -0,0 +1,10 @@
+Manifest-Version: 1.0
+Licenses: Apache-2.0 (Apache License 2.0)
+URL: http://github.com/disneystreaming/aws-sdk-specs
+Implementation-Version: 2023.23.30
+Description: Jar containing smithy spec for kms
+Implementation-Vendor: com.disneystreaming.smithy
+Implementation-Title: aws-kms-spec
+Created-By: Mill 0.11.1
+Tool: Mill-0.11.1
+

META-INF/smithy/manifest

@@ -0,0 +1 @@
+kms.smithy

build.sbt

@@ -19,6 +19,20 @@ ThisBuild / startYear := Option(2018)
 ThisBuild / tlBaseVersion := "0.4"
 ThisBuild / tlJdkRelease := Some(8)
 
+lazy val `smithy4s-preprocessors` = project
+  .in(file("smithy4s-preprocessors"))
+  .settings(
+    scalaVersion := "2.12.13", // 2.12 to match what SBT uses
+    scalacOptions -= "-source:future",
+    libraryDependencies ++= {
+      Seq(
+        "org.typelevel" %% "cats-core" % "2.10.0",
+        "software.amazon.smithy" % "smithy-build" % smithy4s.codegen.BuildInfo.smithyVersion,
+      )
+    },
+  )
+  .enablePlugins(NoPublishPlugin)
+
 lazy val `secure-config` = (project in file("."))
   .settings(
     libraryDependencies ++= {
@@ -30,8 +44,10 @@ lazy val `secure-config` = (project in file("."))
         "org.typelevel" %% "mouse" % "1.3.1",
       )
     },
-    smithy4sAwsSpecs ++= Seq(AWS.kms), // TODO can we put this into its own package to avoid clashing with generated code downstream?
+    smithy4sAwsSpecs ++= Seq(AWS.kms),
     scalacOptions += "-Wconf:src=src_managed/.*:s",
+    Compile / smithy4sModelTransformers += "com.dwolla.smithy.ShadeNamespace",
+    Compile / smithy4sAllDependenciesAsJars += (`smithy4s-preprocessors` / Compile / packageBin).value,
   )
   .enablePlugins(
     Smithy4sCodegenPlugin,

smithy-build.json

@@ -0,0 +1,16 @@
+{
+  "version": "1.0",
+  "projections": {
+    "exampleProjection": {
+      "transforms": [
+        {
+          "name": "flattenNamespaces",
+          "args": {
+            "namespace": "com.dwolla.amazonaws.kms",
+            "service": "com.amazonaws.kms#KMS"
+          }
+        }
+      ]
+    }
+  }
+}

smithy4s-preprocessors/src/main/resources/META-INF/services/software.amazon.smithy.build.ProjectionTransformer

@@ -0,0 +1 @@
+com.dwolla.smithy.ShadeNamespace

smithy4s-preprocessors/src/main/scala/com/dwolla/smithy/ShadeNamespace.scala

@@ -0,0 +1,27 @@
+package com.dwolla.smithy
+
+import cats.syntax.all._
+import software.amazon.smithy.build.{ProjectionTransformer, TransformContext}
+import software.amazon.smithy.model.Model
+import software.amazon.smithy.model.shapes.Shape
+
+import java.util.stream.Collectors
+import scala.collection.JavaConverters._
+
+class ShadeNamespace extends ProjectionTransformer {
+  override def getName: String = "com.dwolla.smithy.ShadeNamespace"
+
+  private val namespacesToRename: Set[String] = Set("com.amazonaws.kms")
+
+  override def transform(context: TransformContext): Model = {
+    val renames =
+      context.getModel
+        .shapes().collect(Collectors.toList[Shape]).asScala.toList
+        .map(_.getId)
+        .filter(id => namespacesToRename.contains(id.getNamespace))
+        .fproduct(id => id.withNamespace(s"com.dwolla.smithy_shaded.${id.getNamespace}"))
+        .toMap.asJava
+
+    context.getTransformer.renameShapes(context.getModel, renames)
+  }
+}

src/main/scala/com/dwolla/config/package.scala

@@ -2,7 +2,7 @@ package com.dwolla
 
 import cats.effect.*
 import cats.syntax.all.*
-import com.amazonaws.kms.{CiphertextType, KMS}
+import com.dwolla.smithy_shaded.com.amazonaws.kms.{CiphertextType, KMS}
 import fs2.compression.Compression
 import monix.newtypes.NewtypeWrapped
 import mouse.all.*