From 5e6e34cfd3c777d982cf24458bdc1cd21593bc6c Mon Sep 17 00:00:00 2001
From: Chenlei Hu <hcl@comfy.org>
Date: Mon, 17 Mar 2025 15:05:24 -0400
Subject: [PATCH] Fix regex handling of folderName

---
 .../sidebar/tabs/WorkflowsSidebarTab.vue      | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/components/sidebar/tabs/WorkflowsSidebarTab.vue b/src/components/sidebar/tabs/WorkflowsSidebarTab.vue
index 4b298b2bd..fdaceab6a 100644
--- a/src/components/sidebar/tabs/WorkflowsSidebarTab.vue
+++ b/src/components/sidebar/tabs/WorkflowsSidebarTab.vue
@@ -4,6 +4,13 @@
     class="workflows-sidebar-tab bg-[var(--p-tree-background)]"
   >
     <template #tool-buttons>
+      <Button
+        icon="pi pi-folder-plus"
+        @click="addNewFolder"
+        severity="secondary"
+        text
+        v-tooltip.bottom="$t('g.newFolder')"
+      />
       <Button
         icon="pi pi-refresh"
         @click="workflowStore.syncWorkflows()"
@@ -91,6 +98,7 @@
             :root="renderTreeNode(workflowsTree, WorkflowTreeType.Browse)"
             v-model:expandedKeys="expandedKeys"
             :selectionKeys="selectionKeys"
+            ref="persistedWorkflowsTreeExplorerRef"
             v-if="workflowStore.persistedWorkflows.length > 0"
           >
             <template #node="{ node }">
@@ -262,7 +270,9 @@ const renderTreeNode = (
     : {
         handleClick,
         async handleAddFolder(folderName: string) {
-          const parentPath = this.key.replace(/^root\//, '')
+          if (folderName === '') return
+
+          const parentPath = this.key.replace(/^root\/?/, '')
           const folderPath = parentPath + '/' + folderName
           await workflowStore.createFolder(folderPath)
         }
@@ -286,4 +296,11 @@ const workflowBookmarkStore = useWorkflowBookmarkStore()
 onMounted(async () => {
   await workflowBookmarkStore.loadBookmarks()
 })
+
+const persistedWorkflowsTreeExplorerRef = ref<InstanceType<
+  typeof TreeExplorer
+> | null>(null)
+const addNewFolder = () => {
+  persistedWorkflowsTreeExplorerRef.value?.addFolderCommand?.('root')
+}
 </script>