Add EnumSection to allow decorators to modify enum member attributes

.changelog/1740703869.md

@@ -0,0 +1,8 @@
+---
+applies_to: [both]
+authors: [Dorenavant]
+references: []
+breaking: false
+new_feature: true
+bug_fix: false
+---

codegen-core/src/main/kotlin/software/amazon/smithy/rust/codegen/core/smithy/generators/EnumGenerator.kt

@@ -29,6 +29,9 @@ import software.amazon.smithy.rust.codegen.core.smithy.MaybeRenamed
 import software.amazon.smithy.rust.codegen.core.smithy.RuntimeType
 import software.amazon.smithy.rust.codegen.core.smithy.RuntimeType.Companion.preludeScope
 import software.amazon.smithy.rust.codegen.core.smithy.RustSymbolProvider
+import software.amazon.smithy.rust.codegen.core.smithy.customize.NamedCustomization
+import software.amazon.smithy.rust.codegen.core.smithy.customize.Section
+import software.amazon.smithy.rust.codegen.core.smithy.customize.writeCustomizations
 import software.amazon.smithy.rust.codegen.core.smithy.expectRustMetadata
 import software.amazon.smithy.rust.codegen.core.smithy.renamedFrom
 import software.amazon.smithy.rust.codegen.core.util.REDACTION
@@ -39,6 +42,17 @@ import software.amazon.smithy.rust.codegen.core.util.orNull
 import software.amazon.smithy.rust.codegen.core.util.shouldRedact
 import software.amazon.smithy.rust.codegen.core.util.toPascalCase
 
+/** EnumGenerator customization sections */
+sealed class EnumSection(name: String) : Section(name) {
+    abstract val definition: EnumDefinition
+
+    /** Hook to add additional attributes to an enum member */
+    data class AdditionalMemberAttributes(override val definition: EnumDefinition) : EnumSection("AdditionalMemberAttributes")
+}
+
+/** Customizations for EnumGenerator */
+abstract class EnumCustomization : NamedCustomization<EnumSection>()
+
 data class EnumGeneratorContext(
     val enumName: String,
     val enumMeta: RustMetadata,
@@ -86,6 +100,7 @@ class EnumMemberModel(
     private val parentShape: Shape,
     private val definition: EnumDefinition,
     private val symbolProvider: RustSymbolProvider,
+    private val customizations: List<EnumCustomization>,
 ) {
     companion object {
         /**
@@ -140,6 +155,7 @@ class EnumMemberModel(
     fun render(writer: RustWriter) {
         renderDocumentation(writer)
         renderDeprecated(writer)
+        writer.writeCustomizations(customizations, EnumSection.AdditionalMemberAttributes(definition))
         writer.write("${derivedName()},")
     }
 }
@@ -167,6 +183,7 @@ open class EnumGenerator(
     private val symbolProvider: RustSymbolProvider,
     private val shape: StringShape,
     private val enumType: EnumType,
+    private val customizations: List<EnumCustomization>,
 ) {
     companion object {
         /** Name of the function on the enum impl to get a vec of value names */
@@ -180,7 +197,7 @@ open class EnumGenerator(
             enumName = symbol.name,
             enumMeta = symbol.expectRustMetadata(),
             enumTrait = enumTrait,
-            sortedMembers = enumTrait.values.sortedBy { it.value }.map { EnumMemberModel(shape, it, symbolProvider) },
+            sortedMembers = enumTrait.values.sortedBy { it.value }.map { EnumMemberModel(shape, it, symbolProvider, customizations) },
         )
 
     fun render(writer: RustWriter) {

codegen-core/src/test/kotlin/software/amazon/smithy/rust/codegen/core/smithy/generators/EnumGeneratorTest.kt

@@ -65,6 +65,7 @@ class EnumGeneratorTest {
                 testModel.lookup("test#EnumWithUnknown"),
                 enumTrait.values.first { it.name.orNull() == name },
                 symbolProvider,
+                emptyList(),
             )
 
         @Test
@@ -112,7 +113,7 @@ class EnumGeneratorTest {
             shape: StringShape,
             enumType: EnumType = TestEnumType,
         ) {
-            EnumGenerator(model, provider, shape, enumType).render(this)
+            EnumGenerator(model, provider, shape, enumType, emptyList()).render(this)
         }
 
         @Test

codegen-core/src/test/kotlin/software/amazon/smithy/rust/codegen/core/smithy/protocols/parse/JsonParserGeneratorTest.kt

@@ -227,7 +227,7 @@ class JsonParserGeneratorTest {
             project.moduleFor(top) {
                 UnionGenerator(model, symbolProvider, this, model.lookup("test#Choice")).render()
                 val enum = model.lookup<StringShape>("test#FooEnum")
-                EnumGenerator(model, symbolProvider, enum, TestEnumType).render(this)
+                EnumGenerator(model, symbolProvider, enum, TestEnumType, emptyList()).render(this)
             }
         }
         model.lookup<OperationShape>("test#Op").outputShape(model).also { output ->

codegen-core/src/test/kotlin/software/amazon/smithy/rust/codegen/core/smithy/protocols/parse/XmlBindingTraitParserGeneratorTest.kt

@@ -208,7 +208,7 @@ internal class XmlBindingTraitParserGeneratorTest {
             project.moduleFor(top) {
                 UnionGenerator(model, symbolProvider, this, choiceShape).render()
                 model.lookup<StringShape>("test#FooEnum").also { enum ->
-                    EnumGenerator(model, symbolProvider, enum, TestEnumType).render(this)
+                    EnumGenerator(model, symbolProvider, enum, TestEnumType, emptyList()).render(this)
                 }
             }
         }

codegen-core/src/test/kotlin/software/amazon/smithy/rust/codegen/core/smithy/protocols/serialize/AwsQuerySerializerGeneratorTest.kt

@@ -147,7 +147,7 @@ class AwsQuerySerializerGeneratorTest {
                     renderUnknownVariant = generateUnknownVariant,
                 ).render()
                 val enum = model.lookup<StringShape>("test#FooEnum")
-                EnumGenerator(model, symbolProvider, enum, TestEnumType).render(this)
+                EnumGenerator(model, symbolProvider, enum, TestEnumType, emptyList()).render(this)
             }
         }
 
@@ -316,7 +316,7 @@ class AwsQuerySerializerGeneratorTest {
                     renderUnknownVariant = generateUnknownVariant,
                 ).render()
                 val enum = model.lookup<StringShape>("test#FooEnum")
-                EnumGenerator(model, symbolProvider, enum, TestEnumType).render(this)
+                EnumGenerator(model, symbolProvider, enum, TestEnumType, emptyList()).render(this)
             }
         }
 

codegen-core/src/test/kotlin/software/amazon/smithy/rust/codegen/core/smithy/protocols/serialize/Ec2QuerySerializerGeneratorTest.kt

@@ -143,7 +143,7 @@ class Ec2QuerySerializerGeneratorTest {
             project.moduleFor(top) {
                 UnionGenerator(model, symbolProvider, this, model.lookup("test#Choice")).render()
                 val enum = model.lookup<StringShape>("test#FooEnum")
-                EnumGenerator(model, symbolProvider, enum, TestEnumType).render(this)
+                EnumGenerator(model, symbolProvider, enum, TestEnumType, emptyList()).render(this)
             }
         }
 
@@ -298,7 +298,7 @@ class Ec2QuerySerializerGeneratorTest {
             project.moduleFor(top) {
                 UnionGenerator(model, symbolProvider, this, model.lookup("test#Choice")).render()
                 val enum = model.lookup<StringShape>("test#FooEnum")
-                EnumGenerator(model, symbolProvider, enum, TestEnumType).render(this)
+                EnumGenerator(model, symbolProvider, enum, TestEnumType, emptyList()).render(this)
             }
         }
 

codegen-core/src/test/kotlin/software/amazon/smithy/rust/codegen/core/smithy/protocols/serialize/JsonSerializerGeneratorTest.kt

@@ -159,7 +159,7 @@ class JsonSerializerGeneratorTest {
             project.moduleFor(top) {
                 UnionGenerator(model, symbolProvider, this, model.lookup("test#Choice")).render()
                 val enum = model.lookup<StringShape>("test#FooEnum")
-                EnumGenerator(model, symbolProvider, enum, TestEnumType).render(this)
+                EnumGenerator(model, symbolProvider, enum, TestEnumType, emptyList()).render(this)
             }
         }
 
@@ -333,7 +333,7 @@ class JsonSerializerGeneratorTest {
             project.moduleFor(top) {
                 UnionGenerator(model, symbolProvider, this, model.lookup("test#Choice")).render()
                 val enum = model.lookup<StringShape>("test#FooEnum")
-                EnumGenerator(model, symbolProvider, enum, TestEnumType).render(this)
+                EnumGenerator(model, symbolProvider, enum, TestEnumType, emptyList()).render(this)
             }
         }
 

codegen-core/src/test/kotlin/software/amazon/smithy/rust/codegen/core/smithy/protocols/serialize/XmlBindingTraitSerializerGeneratorTest.kt

@@ -164,7 +164,7 @@ internal class XmlBindingTraitSerializerGeneratorTest {
             project.moduleFor(top) {
                 UnionGenerator(model, symbolProvider, this, model.lookup("test#Choice")).render()
                 val enum = model.lookup<StringShape>("test#FooEnum")
-                EnumGenerator(model, symbolProvider, enum, TestEnumType).render(this)
+                EnumGenerator(model, symbolProvider, enum, TestEnumType, emptyList()).render(this)
             }
         }
         model.lookup<OperationShape>("test#Op").inputShape(model).also { input ->
@@ -334,7 +334,7 @@ internal class XmlBindingTraitSerializerGeneratorTest {
             project.moduleFor(top) {
                 UnionGenerator(model, symbolProvider, this, model.lookup("test#Choice")).render()
                 val enum = model.lookup<StringShape>("test#FooEnum")
-                EnumGenerator(model, symbolProvider, enum, TestEnumType).render(this)
+                EnumGenerator(model, symbolProvider, enum, TestEnumType, emptyList()).render(this)
             }
         }
         model.lookup<OperationShape>("test#Op").inputShape(model).also { input ->