Update CodeBlock.kt

Co-authored-by: Zongle Wang <wangzongler@gmail.com>
ComposeGears · Jul 25, 2024 · 5471c3a · 5471c3a
1 parent 8bc7afc
commit 5471c3a
Showing 1 changed file with 5 additions and 12 deletions.
diff --git a/...nerator/common/src/main/kotlin/io/github/composegears/valkyrie/generator/ext/CodeBlock.kt b/...nerator/common/src/main/kotlin/io/github/composegears/valkyrie/generator/ext/CodeBlock.kt
@@ -5,24 +5,17 @@ import com.squareup.kotlinpoet.CodeBlock
 fun CodeBlock.Builder.argumentBlock(
     argumentFlow: String,
     vararg args: Any?,
+    isNested: Boolean = false,
     block: CodeBlock.Builder.() -> Unit,
 ) {
     add("$argumentFlow\n", *args)
     indention(block)
     newLine()
     add(")")
-}
-
-fun CodeBlock.Builder.nestedArgumentBlock(
-    argumentFlow: String,
-    vararg args: Any?,
-    block: CodeBlock.Builder.() -> Unit,
-) {
-    add("$argumentFlow\n", *args)
-    indention(block)
-    newLine()
-    add("),")
-    newLine()
+    if (isNested) {
+        add(",")
+        newLine()
+    }
 }
 
 fun CodeBlock.Builder.indention(block: CodeBlock.Builder.() -> Unit) {