Code completion for enum entries

gradle · Nov 14, 2024 · ae3766b · ae3766b
1 parent 65b1eee
commit ae3766b
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/lsp/src/main/kotlin/org/gradle/declarative/lsp/DeclarativeTextDocumentService.kt b/lsp/src/main/kotlin/org/gradle/declarative/lsp/DeclarativeTextDocumentService.kt
@@ -40,13 +40,7 @@ import org.eclipse.lsp4j.SignatureInformation
 import org.eclipse.lsp4j.jsonrpc.messages.Either
 import org.eclipse.lsp4j.services.LanguageClient
 import org.eclipse.lsp4j.services.TextDocumentService
-import org.gradle.declarative.dsl.schema.AnalysisSchema
-import org.gradle.declarative.dsl.schema.DataClass
-import org.gradle.declarative.dsl.schema.DataParameter
-import org.gradle.declarative.dsl.schema.DataType
-import org.gradle.declarative.dsl.schema.DataTypeRef
-import org.gradle.declarative.dsl.schema.FunctionSemantics
-import org.gradle.declarative.dsl.schema.SchemaFunction
+import org.gradle.declarative.dsl.schema.*
 import org.gradle.declarative.lsp.build.model.DeclarativeResourcesModel
 import org.gradle.declarative.lsp.extension.indexBasedOverlayResultFromDocuments
 import org.gradle.declarative.lsp.extension.toLspRange
@@ -416,9 +410,9 @@ private fun computeTypedPlaceholder(
     type: DataTypeRef,
     analysisSchema: AnalysisSchema
 ): String {
-    val resolvedType = SchemaTypeRefContext(analysisSchema).resolveRef(type)
-    return when (resolvedType) {
+    return when (val resolvedType = SchemaTypeRefContext(analysisSchema).resolveRef(type)) {
         is DataType.BooleanDataType -> "\${$index|true,false|}"
+        is EnumClass -> "\${$index|${resolvedType.entryNames.joinToString(",")}|}"
         is DataType.IntDataType -> "\${$index:0}"
         is DataType.LongDataType -> "\${$index:0L}"
         is DataType.StringDataType -> "\"\${$index}\""