Add back button navigation to all context selection flows in apply context command

Author: robertpiosik

README.md

@@ -14,7 +14,7 @@
 
 Code Web Chat is a free and open-source AI coding toolkit. You can use CWC in **VS Code** family of editors (Cursor, Antigravity, VSCodium etc.) for a **much faster and cost efficient\*** development experience.
 
-<sup>\* Compared to coding agents.</sub>
+<sup>\* Compared to coding agents, thanks to its a single request/response design.</sub>
 
 ✍️ **Construct coding prompts for...**
 
@@ -32,7 +32,7 @@ Code Web Chat is a free and open-source AI coding toolkit. You can use CWC in **
 
 ## Introduction
 
-With its focus on context engineering, CWC proposes a non-agentic workflow, where all the necessary for the task files, includnig useful examples, are picked with engineer's intent. Such guided model works well with simple prompts and as accurate as it can get. The tool is a perfect companion for all kinds of coding tasks; from mass renamings to complex features.
+With its focus on context engineering, CWC proposes **a non-agentic workflow**, where all the necessary for the task files, includnig useful examples, are **picked with engineer's intent**. Such guided model works well with simple prompts and is as accurate as it can get. The tool is a **perfect companion for all kinds of coding tasks**; from laborious refactors to complex features.
 
 > [!TIP]
 > Feeling lost in the codebase? Just pick a broad context, set _Prune context_, type what you're about to do and AI will help narrow it down to a reasonable size.

apps/editor/src/commands/apply-context-command/apply-context-command.ts

@@ -243,8 +243,13 @@ export const apply_context_command = (
             show_main_menu = true
           }
         } else if (main_selection.value == 'unstaged') {
-          await handle_unstaged_files_source(workspace_provider)
-          return
+          const result = await handle_unstaged_files_source(
+            workspace_provider,
+            extension_context
+          )
+          if (result == 'back') {
+            show_main_menu = true
+          }
         } else if (main_selection.value == 'commit_files') {
           const result = await handle_commit_files_source(
             workspace_provider,

apps/editor/src/commands/apply-context-command/helpers/applying/apply-saved-context.ts

@@ -10,7 +10,7 @@ export const apply_saved_context = async (
   workspace_root: string,
   workspace_provider: WorkspaceProvider,
   extension_context: vscode.ExtensionContext
-): Promise<void> => {
+): Promise<'back' | void> => {
   const resolved_paths = await resolve_context_paths(
     context,
     workspace_root,
@@ -55,6 +55,8 @@ export const apply_saved_context = async (
       const quick_pick = vscode.window.createQuickPick()
       quick_pick.items = quick_pick_options
       quick_pick.placeholder = `How would you like to apply "${context.name}"?`
+      quick_pick.buttons = [vscode.QuickInputButtons.Back]
+
       if (last_choice_label) {
         const active_item = quick_pick_options.find(
           (opt) => opt.label === last_choice_label
@@ -64,23 +66,33 @@ export const apply_saved_context = async (
         }
       }
 
-      const choice = await new Promise<vscode.QuickPickItem | undefined>(
-        (resolve) => {
-          let is_accepted = false
-          quick_pick.onDidAccept(() => {
-            is_accepted = true
-            resolve(quick_pick.selectedItems[0])
+      const choice = await new Promise<
+        vscode.QuickPickItem | 'back' | undefined
+      >((resolve) => {
+        let is_accepted = false
+        quick_pick.onDidTriggerButton((button) => {
+          if (button === vscode.QuickInputButtons.Back) {
+            resolve('back')
             quick_pick.hide()
-          })
-          quick_pick.onDidHide(() => {
-            if (!is_accepted) {
-              resolve(undefined)
-            }
-            quick_pick.dispose()
-          })
-          quick_pick.show()
-        }
-      )
+          }
+        })
+        quick_pick.onDidAccept(() => {
+          is_accepted = true
+          resolve(quick_pick.selectedItems[0])
+          quick_pick.hide()
+        })
+        quick_pick.onDidHide(() => {
+          if (!is_accepted) {
+            resolve('back')
+          }
+          quick_pick.dispose()
+        })
+        quick_pick.show()
+      })
+
+      if (choice === 'back') {
+        return 'back'
+      }
 
       if (!choice) {
         return