Add expand and collapse folder commands with state management for the Code Web Chat view

Author: robertpiosik

apps/editor/package.json

@@ -153,6 +153,12 @@
         "category": "Code Web Chat",
         "icon": "$(clear-all)"
       },
+      {
+        "command": "codeWebChat.expandFolders",
+        "title": "Expand Folders",
+        "category": "Code Web Chat",
+        "icon": "$(expand-all)"
+      },
       {
         "command": "codeWebChat.collapseFolders",
         "title": "Collapse Folders",
@@ -370,6 +376,10 @@
           "command": "codeWebChat.checkAllOpenEditors",
           "when": "false"
         },
+        {
+          "command": "codeWebChat.expandFolders",
+          "when": "false"
+        },
         {
           "command": "codeWebChat.collapseFolders",
           "when": "false"
@@ -462,6 +472,16 @@
           "when": "view == codeWebChatViewContext",
           "group": "navigation@2"
         },
+        {
+          "command": "codeWebChat.expandFolders",
+          "when": "view == codeWebChatViewContext",
+          "group": "navigation@3"
+        },
+        {
+          "command": "codeWebChat.collapseFolders",
+          "when": "view == codeWebChatViewContext",
+          "group": "navigation@4"
+        },
         {
           "command": "codeWebChat.checkAll",
           "when": "view == codeWebChatViewWorkspace",

apps/editor/src/context/context-initialization.ts

@@ -151,8 +151,6 @@ export const context_initialization = async (
   const shared_state = SharedFileState.get_instance()
   shared_state.set_providers(workspace_provider, open_editors_provider)
 
-  workspace_provider.load_checked_files_state()
-
   context.subscriptions.push({
     dispose: () => shared_state.dispose()
   })
@@ -177,7 +175,7 @@ export const context_initialization = async (
     manageCheckboxStateManually: true
   })
 
-  const context_view = vscode.window.createTreeView('codeWebChatViewContext', {
+  let context_view = vscode.window.createTreeView('codeWebChatViewContext', {
     treeDataProvider: context_provider,
     manageCheckboxStateManually: true
   })
@@ -299,8 +297,38 @@ export const context_initialization = async (
         )
       }
     ),
+    vscode.commands.registerCommand('codeWebChat.expandFolders', async () => {
+      workspace_provider.set_context_view_collapsible_state(
+        vscode.TreeItemCollapsibleState.Expanded
+      )
+      workspace_view.dispose()
+      context_view.dispose()
+      await new Promise((resolve) => setTimeout(resolve, 0))
+
+      workspace_view = vscode.window.createTreeView(
+        'codeWebChatViewWorkspace',
+        {
+          treeDataProvider: workspace_provider!,
+          manageCheckboxStateManually: true
+        }
+      )
+
+      context_view = vscode.window.createTreeView('codeWebChatViewContext', {
+        treeDataProvider: context_provider,
+        manageCheckboxStateManually: true
+      })
+
+      register_workspace_view_handlers(workspace_view)
+      register_workspace_view_handlers(context_view)
+      context.subscriptions.push(workspace_view)
+      context.subscriptions.push(context_view)
+    }),
     vscode.commands.registerCommand('codeWebChat.collapseFolders', async () => {
+      workspace_provider.set_context_view_collapsible_state(
+        vscode.TreeItemCollapsibleState.Collapsed
+      )
       workspace_view.dispose()
+      context_view.dispose()
       await new Promise((resolve) => setTimeout(resolve, 0))
 
       workspace_view = vscode.window.createTreeView(
@@ -311,8 +339,15 @@ export const context_initialization = async (
         }
       )
 
+      context_view = vscode.window.createTreeView('codeWebChatViewContext', {
+        treeDataProvider: context_provider,
+        manageCheckboxStateManually: true
+      })
+
       register_workspace_view_handlers(workspace_view)
+      register_workspace_view_handlers(context_view)
       context.subscriptions.push(workspace_view)
+      context.subscriptions.push(context_view)
     }),
     vscode.commands.registerCommand('codeWebChat.clearChecks', async () => {
       await workspace_provider!.clear_checks()

apps/editor/src/context/providers/workspace/workspace-provider.ts

@@ -66,6 +66,14 @@ export class WorkspaceProvider
   public gitignore_initialization: Promise<void>
   public ranges_initialization: Promise<void>
   public use_compact_token_count: boolean = false
+  private _context_view_collapsible_state: vscode.TreeItemCollapsibleState =
+    vscode.TreeItemCollapsibleState.Expanded
+
+  public set_context_view_collapsible_state(
+    state: vscode.TreeItemCollapsibleState
+  ) {
+    this._context_view_collapsible_state = state
+  }
 
   public set_use_compact_token_count(use_compact: boolean) {
     if (this.use_compact_token_count != use_compact) {
@@ -715,7 +723,7 @@ export class WorkspaceProvider
           name,
           uri,
           context_view
-            ? vscode.TreeItemCollapsibleState.Expanded
+            ? this._context_view_collapsible_state
             : vscode.TreeItemCollapsibleState.Collapsed,
           true,
           this._checked_items.get(root) ??
@@ -974,7 +982,7 @@ export class WorkspaceProvider
           uri,
           is_directory
             ? context_view
-              ? vscode.TreeItemCollapsibleState.Expanded
+              ? this._context_view_collapsible_state
               : vscode.TreeItemCollapsibleState.Collapsed
             : vscode.TreeItemCollapsibleState.None,
           is_directory,