Add sync button to update saved contexts with currently selected files

Author: robertpiosik

packages/vscode/src/commands/apply-context-command/sources/json-file-source.ts

@@ -72,6 +72,11 @@ export const handle_json_file_source = async (
       await load_and_merge_file_contexts()
     const workspace_folders = vscode.workspace.workspaceFolders || []
 
+    const sync_button = {
+      iconPath: new vscode.ThemeIcon('sync'),
+      tooltip: 'Update with currently selected files'
+    }
+
     const edit_button = {
       iconPath: new vscode.ThemeIcon('edit'),
       tooltip: 'Rename'
@@ -81,7 +86,7 @@ export const handle_json_file_source = async (
       tooltip: 'Delete'
     }
     const open_file_button = {
-      iconPath: new vscode.ThemeIcon('file'),
+      iconPath: new vscode.ThemeIcon('go-to-file'),
       tooltip: 'Open contexts.json'
     }
 
@@ -123,7 +128,7 @@ export const handle_json_file_source = async (
               label: context.name,
               description,
               context: context,
-              buttons: [edit_button, delete_button]
+              buttons: [sync_button, edit_button, delete_button]
             })
           }
         }
@@ -224,6 +229,53 @@ export const handle_json_file_source = async (
         const item = selection
         const old_name = item.context.name
 
+        if (selection.triggeredButton === sync_button) {
+          const context_name = item.context.name
+          const checked_files = workspace_provider.get_checked_files()
+          const files_by_workspace = group_files_by_workspace(checked_files)
+
+          const current_roots = context_to_roots.get(context_name) || []
+          const all_roots = new Set([
+            ...current_roots,
+            ...files_by_workspace.keys()
+          ])
+
+          for (const root of all_roots) {
+            const files = files_by_workspace.get(root) || []
+            const p = get_contexts_file_path(root)
+
+            if (files.length === 0 && !fs.existsSync(p)) continue
+
+            if (files.length > 0) {
+              const dir = path.dirname(p)
+              if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true })
+            }
+
+            let contexts = load_contexts_from_file(p)
+            contexts = contexts.filter((c) => c.name !== context_name)
+
+            if (files.length > 0) {
+              const condensed_paths = condense_paths(
+                files,
+                root,
+                workspace_provider
+              )
+              const relative_paths = condensed_paths.map((p) =>
+                p.replace(/\\/g, '/')
+              )
+              contexts.unshift({
+                name: context_name,
+                paths: relative_paths
+              })
+            }
+            await save_contexts_to_file(contexts, p)
+          }
+          vscode.window.showInformationMessage(
+            dictionary.information_message.CONTEXT_SAVED_SUCCESSFULLY
+          )
+          return
+        }
+
         if (selection.triggeredButton === edit_button) {
           active_dialog_count++
           const input = vscode.window.createInputBox()

packages/vscode/src/commands/apply-context-command/sources/workspace-state-source.ts

@@ -35,6 +35,11 @@ export const handle_workspace_state_source = async (
       workspace_root
     )
 
+    const sync_button = {
+      iconPath: new vscode.ThemeIcon('sync'),
+      tooltip: 'Update with currently selected files'
+    }
+
     const edit_button = {
       iconPath: new vscode.ThemeIcon('edit'),
       tooltip: 'Rename'
@@ -60,7 +65,7 @@ export const handle_workspace_state_source = async (
         })
 
         contexts.forEach((context, index) => {
-          const buttons = [edit_button, delete_button]
+          const buttons = [sync_button, edit_button, delete_button]
 
           const description = `${context.paths.length} ${
             context.paths.length == 1 ? 'path' : 'paths'
@@ -205,6 +210,43 @@ export const handle_workspace_state_source = async (
             index: number
           }
 
+          if (event.button === sync_button) {
+            const context = item.context
+            const checked_files = workspace_provider.get_checked_files()
+            const files_by_workspace = group_files_by_workspace(checked_files)
+            let all_prefixed_paths: string[] = []
+
+            for (const [root, files] of files_by_workspace.entries()) {
+              if (files.length === 0) continue
+              const condensed_paths = condense_paths(
+                files,
+                root,
+                workspace_provider
+              )
+              const relative_paths = condensed_paths.map((p) =>
+                p.replace(/\\/g, '/')
+              )
+              const prefixed_paths = add_workspace_prefix(relative_paths, root)
+              all_prefixed_paths = [...all_prefixed_paths, ...prefixed_paths]
+            }
+            all_prefixed_paths.sort((a, b) => a.localeCompare(b))
+
+            const updated_contexts = internal_contexts.map((c) =>
+              c.name === context.name ? { ...c, paths: all_prefixed_paths } : c
+            )
+            save_contexts_for_workspace(
+              extension_context,
+              workspace_root,
+              updated_contexts
+            )
+            vscode.window.showInformationMessage(
+              dictionary.information_message.CONTEXT_SAVED_SUCCESSFULLY
+            )
+            quick_pick.hide()
+            resolve(undefined)
+            return
+          }
+
           if (event.button === edit_button) {
             active_dialog_count++
             const input_box = vscode.window.createInputBox()