Fix DSL builder when only functions or only properties are configured

ALikhachev · Sep 19, 2024 · 788fb2b · 788fb2b
1 parent baab444
commit 788fb2b
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/plugin/src/main/kotlin/org/jetbrains/hackathon2024/dsl/dslBuilder.kt b/plugin/src/main/kotlin/org/jetbrains/hackathon2024/dsl/dslBuilder.kt
@@ -23,19 +23,27 @@ internal class KapiFenceRootBuilder(project: Project) : KapiFenceRootDsl {
         }
     }
 
+    private fun String.maybeWrapIntoClass(className: String): String? {
+        return if (isEmpty()) {
+            null
+        } else {
+            "class $className {\n$this\n}"
+        }
+    }
+
     override fun deprecateMembers(
         name: String,
         body: KapiFenceClassDsl.() -> Unit,
     ) {
         val funBuilder = StringBuilder()
-        funBuilder.appendLine("class $name {")
         val propBuilder = StringBuilder()
-        propBuilder.appendLine("class $name {")
         KapiFenceClassBuilder(funBuilder, propBuilder).body()
-        funBuilder.appendLine("}")
-        propBuilder.appendLine("}")
-        records_.add(funBuilder.toString())
-        records_.add(propBuilder.toString())
+        funBuilder.toString().maybeWrapIntoClass(name)?.let {
+            records_.add(it)
+        }
+        propBuilder.toString().maybeWrapIntoClass(name)?.let {
+            records_.add(it)
+        }
     }
 }