codegen: Fix enum value generated name to not be lower cased

Fixes the generated enum value names to not be lower cased when capitalized. Related to aws/aws-sdk-go-v2#1013
jasdel · Jan 6, 2021 · 0ebb7db · 0ebb7db
1 parent 12a5ddb
commit 0ebb7db
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/codegen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/EnumGenerator.java b/codegen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/EnumGenerator.java
@@ -59,10 +59,15 @@ public void run() {
                 for (EnumDefinition definition : enumTrait.getValues()) {
                     StringBuilder labelBuilder = new StringBuilder(symbol.getName());
                     String name = definition.getName().get();
-                    for (String part : name.split("(?U)[\\W_]")) {
-                        labelBuilder.append(StringUtils.capitalize(part.toLowerCase(Locale.US)));
+                    String label = StringUtils.capitalize(name);
+
+                    // Only split, capital case, and join the elements if there are non-word characters.
+                    if (name.matches("(?U)[\\W_]")) {
+                        for (String part : name.split("(?U)[\\W_]")) {
+                            labelBuilder.append(StringUtils.capitalize(part.toLowerCase(Locale.US)));
+                        }
+                        label = labelBuilder.toString();
                     }
-                    String label = labelBuilder.toString();
 
                     // If camel-casing would cause a conflict, don't camel-case this enum value.
                     if (constants.contains(label)) {