Update open enum type rendering

kubukoz · Oct 10, 2023 · c309fed · c309fed
1 parent 2fb3cb0
commit c309fed
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/modules/core/src/test/smithy/demo.smithy b/modules/core/src/test/smithy/demo.smithy
@@ -143,15 +143,13 @@ map PowerMap {
     value: Hero
 }
 
-@openEnum
 enum Power {
     ICE = "Ice"
     FIRE = "Fire"
     LIGHTNING = "Lightning"
     WIND = "Wind"
 }
 
-@openEnum
 intEnum PrivacyTier {
     PUBLIC = 0
     PRIVATE = 1
@@ -279,3 +277,15 @@ map SampleSparseMap {
     key: String
     value: Integer
 }
+
+@openEnum
+enum OpenStringEnum {
+    ICE
+    FIRE
+}
+
+@openEnum
+intEnum OpenIntEnum {
+    ICE = 1
+    FIRE = 2
+}
diff --git a/modules/language-support/src/main/scala/playground/language/CompletionVisitor.scala b/modules/language-support/src/main/scala/playground/language/CompletionVisitor.scala
@@ -262,8 +262,8 @@ object CompletionItem {
       case e @ EnumerationSchema(_, _, _, _, _) =>
         e.tag match {
           case ClosedIntEnum     => now(s"intEnum ${e.shapeId.name}")
-          case OpenIntEnum(_)    => now(s"intEnum(open) ${e.shapeId.name}")
-          case OpenStringEnum(_) => now(s"enum(open) ${e.shapeId.name}")
+          case OpenIntEnum(_)    => now(s"@openEnum intEnum ${e.shapeId.name}")
+          case OpenStringEnum(_) => now(s"@openEnum enum ${e.shapeId.name}")
           case ClosedStringEnum  => now(s"enum ${e.shapeId.name}")
         }