Add more support on Kotlin script

Omico · Jun 8, 2023 · f6d2682 · f6d2682
1 parent 952b790
commit f6d2682
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/...erator/src/main/kotlin/me/omico/elucidator/function/basic/KtFileScopeExtensionFunction.kt b/...erator/src/main/kotlin/me/omico/elucidator/function/basic/KtFileScopeExtensionFunction.kt
@@ -21,6 +21,7 @@ internal val BasicExtensionFunctions_KtFileScope: List<BasicExtensionFunction> b
         add(BasicExtensionFunction_addFileComment)
         add(BasicExtensionFunction_addFunction)
         add(BasicExtensionFunction_addProperty)
+        add(BasicExtensionFunction_addStatement)
         add(BasicExtensionFunction_addType)
     }
 }
diff --git a/elucidator/src/main/kotlin/me/omico/elucidator/KtFile.kt b/elucidator/src/main/kotlin/me/omico/elucidator/KtFile.kt
@@ -48,4 +48,8 @@ public fun KtFileScope.addProperty(
 ): Unit =
     property(name = name, type = type, modifiers = modifiers, block = block).let(::addProperty)
 
+public fun KtFileScope.addNewline() {
+    builder.addStatement("")
+}
+
 public fun KtFileScope.writeTo(path: Path): Unit = build().writeTo(path)
diff --git a/elucidator/src/main/kotlin/me/omico/elucidator/KtsFile.kt b/elucidator/src/main/kotlin/me/omico/elucidator/KtsFile.kt
@@ -23,3 +23,10 @@ public fun ktsFile(
     block: KtFileScope.() -> Unit,
 ): FileSpec =
     FileSpec.scriptBuilder(fileName = fileName, packageName = packageName).applyDslBuilder(block).build()
+
+public fun KtFileScope.addCallExpression(format: String, vararg args: Any, block: KtFileScope.() -> Unit) {
+    require(builder.isScript) { "CallExpression can only be added to a script file." }
+    builder.beginControlFlow(controlFlow = format, args = args)
+    block()
+    builder.endControlFlow()
+}