createFiles show file list ok

Author: doofin

src/main/scala/functorcoder/actions/Commands.scala

@@ -71,14 +71,12 @@ object Commands {
       case Some(value) =>
         // split the path
         val pathParts = value.split("/")
-        // generate the full path for parent and 1 to 5 levels up
+        // generate parent path for and 1 to 5 levels up
         val parentPaths =
           (1 to 5).map { i =>
             pathParts.take(pathParts.length - i).mkString("/")
           }
-        showMessageAndLog(
-          "parent paths: " + parentPaths.mkString(", ")
-        )
+
         quickPick.createQuickPick(
           title = "create files/folders",
           placeHolder = "select a parent folder",
@@ -90,26 +88,37 @@ object Commands {
                 // create the files and folders according to the tree
                 showMessageAndLog("creating files in: " + path)
                 quickPick.createInputBox(
-                  title = "Create files/folders description",
+                  title = "Create files/folders under " + path,
                   placeHolder = "describe your project",
                   onInput = { input =>
                     val respFuture = llm.sendPrompt(functorcoder.llm.llmPrompt.CreateFiles(input))
                   respFuture.foreach { response =>
                     // parse the response to a tree of files and folders
-                    val tree = treeParse.parse(response)
-                    showMessageAndLog("current directory: " + currDir)
+                    val treeOpt = treeParse.parse(response)
+                    val filesList = treeOpt.map(createFiles.tree2list).getOrElse(Seq()).mkString(", ")
 
                     quickPick.createQuickPick(
                       title = "Files and Folders",
                       placeHolder = "select to apply creating files and folders",
                       items = Seq(
                         (
-                          "files created!",
+                          s"create $filesList",
                           "",
                           { () =>
+                            treeOpt match {
+                              case scala.util.Success(tree) =>
+                                createFiles.createFilesAndFolders(
+                                  tree,
+                                  path
+                                )
+                              case scala.util.Failure(exception) =>
+                                showMessageAndLog(
+                                  s"Failed to parse tree: ${treeOpt.toString}, exception: ${exception.getMessage}"
+                                )
+                            }
                             // create the files and folders according to the tree
-                            tree.toString
-                            showMessageAndLog("files created!")
+
+                            // showMessageAndLog("files created!")
                           }
                         )
                       )

src/main/scala/functorcoder/actions/createFiles.scala

@@ -32,7 +32,7 @@ object createFiles {
 
     treeParse.parse(promptResponse) match {
       case scala.util.Success(tree) =>
-        createFilesAndFolders(tree)
+        createFilesAndFolders(tree, "")
       case scala.util.Failure(exception) =>
         showMessageAndLog(s"Trying again with $retry retries left")
         if (retry > 0) {
@@ -47,12 +47,26 @@ object createFiles {
     * @param tree
     *   the tree of files and folders
     */
-  def createFilesAndFolders(tree: TreeNode[String]): Unit = {
+  def createFilesAndFolders(tree: TreeNode[String], parentPath0: String): Unit = {
     // recursively create files and folders
     showMessageAndLog(s"Files and folders tree: $tree")
     val TreeNode(root, children) = tree
+    val parentPath: String = parentPath0 + "/" + root
+
     children.foreach { child =>
-      createFilesAndFolders(child)
+      val file = child.value
+      showMessageAndLog(s"Creating file in $parentPath, file: $file")
+      createFilesAndFolders(child, parentPath)
+    }
+  }
+
+  def tree2list(tree: TreeNode[String]): Seq[String] = {
+    val TreeNode(root, children) = tree
+    val childList = children.flatMap(tree2list)
+    if (childList.isEmpty) {
+      Seq(root)
+    } else {
+      Seq(root) ++ childList
     }
   }
 }

src/main/scala/functorcoder/llm/wk.worksheet.sc

@@ -2,6 +2,7 @@ import functorcoder.llm.llmPrompt
 import functorcoder.llm.llmPrompt.Prompt
 import scala.collection.mutable.ArrayBuffer
 import functorcoder.algo.treeParse
+import functorcoder.actions.createFiles.*
 
 val Modification = llmPrompt
   .Modification(code = "val x = 1", taskRequirement = "add documentation")
@@ -28,3 +29,5 @@ TreeNode("root", ArrayBuffer())
 
 val input = "(root [(folder1 [(file1 file2) folder2]) folder3])"
 val tree = treeParse.parse(input)
+
+tree2list(tree.get)

src/main/scala/vscextension/quickPick.scala

@@ -40,14 +40,13 @@ object quickPick {
     modifieF(quickPick)
     quickPick.buttons = js.Array(vscode.QuickInputButtons.Back)
 
-    quickPick.items = items.toJSArray.map { (itemStr, itemDesc, _) => //
+    quickPick.items = items.map { (itemStr, itemDesc, _) => //
       vscode
         .QuickPickItem(itemStr)
         .setAlwaysShow(true)
         .setButtons(js.Array(vscode.QuickInputButtons.Back))
-        .setDescription(itemStr)
         .setDetail(itemDesc)
-    }
+    }.toJSArray
 
     quickPick.onDidChangeSelection { selection =>
       println(s"selected: ${selection(0).label}")