Add Return Home navigation, enhance Find Relevant Files state management, and implement source code shrinking option

- in duplicate workspace, don't store or restore frf
- when switching to frf context, clear it
- once user accepts 'relevant files' and we switched to 'edit context' prompt type. Show auto-closing modal instead of [Fragment]
- [Fragment] should also change page to "Home". Also rename the method appropriately. The message sent to the frontend should be named "RETURN_HOME"
- when returning to home, use return_home_and_switch_to_edit_context
- Move checkbox with text [Fragment] above [Fragment]

Author: robertpiosik

apps/editor/package.json

@@ -2,7 +2,7 @@
   "name": "gemini-coder",
   "displayName": "Code Web Chat",
   "description": "Fast and safe way to code with AI (CWC)",
-  "version": "1.874.0",
+  "version": "1.875.0",
   "scripts": {
     "build": "npx vsce package --no-dependencies",
     "vscode:prepublish": "rimraf out && npm run compile",

apps/editor/src/commands/apply-chat-response-command/response-processor.ts

@@ -266,13 +266,14 @@ export const process_chat_response = async (
 
       if (was_frf) {
         shared_state.switch_context_state(true)
-        await workspace_provider.set_checked_files([])
-        await panel_provider.switch_to_edit_context()
+        await panel_provider.return_home_and_switch_to_edit_context()
       }
 
-      vscode.window.showInformationMessage(
-        t('command.apply-chat-response.relevant-files.success')
-      )
+      panel_provider.send_message({
+        command: 'SHOW_AUTO_CLOSING_MODAL',
+        title: t('command.apply-chat-response.relevant-files.success'),
+        type: 'success'
+      })
     }
 
     return null

apps/editor/src/commands/duplicate-workspace-command.ts

@@ -20,9 +20,6 @@ export const duplicate_workspace_command = (
       const checked_files = export_state.regular.checked_files
       const checked_files_timestamps = export_state.regular.checked_timestamps
 
-      const checked_files_frf = export_state.frf.checked_files
-      const checked_files_timestamps_frf = export_state.frf.checked_timestamps
-
       const workspace_root_folders =
         vscode.workspace.workspaceFolders?.map((folder) => folder.uri.fsPath) ??
         []
@@ -47,8 +44,6 @@ export const duplicate_workspace_command = (
       const context_to_save: DuplicateWorkspaceContext = {
         checked_files,
         checked_files_timestamps,
-        checked_files_frf,
-        checked_files_timestamps_frf,
         workspace_root_folders,
         timestamp: Date.now(),
         open_editors,

apps/editor/src/constants/state-keys.ts

@@ -89,8 +89,6 @@ export const LAST_SEARCH_FILES_FOR_CONTEXT_QUERY_STATE_KEY =
 export type DuplicateWorkspaceContext = {
   checked_files: string[]
   checked_files_timestamps: Record<string, number>
-  checked_files_frf: string[]
-  checked_files_timestamps_frf: Record<string, number>
   timestamp: number
   workspace_root_folders: string[]
   open_editors?: { path: string; view_column?: number }[]

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

@@ -65,14 +65,6 @@ const restore_duplicated_workspace_context = async (
           CONTEXT_CHECKED_TIMESTAMPS_STATE_KEY,
           duplicated_context.checked_files_timestamps
         )
-        await context.workspaceState.update(
-          CONTEXT_CHECKED_PATHS_FRF_STATE_KEY,
-          duplicated_context.checked_files_frf
-        )
-        await context.workspaceState.update(
-          CONTEXT_CHECKED_TIMESTAMPS_FRF_STATE_KEY,
-          duplicated_context.checked_files_timestamps_frf
-        )
 
         if (duplicated_context.ranges) {
           await context.workspaceState.update(

apps/editor/src/context/providers/open-editors/open-editors-provider.ts

@@ -55,6 +55,11 @@ export class OpenEditorsProvider
   public switch_context_state(is_frf: boolean) {
     if (this._is_frf_mode == is_frf) return
     this._is_frf_mode = is_frf
+
+    if (is_frf) {
+      this._checked_items.clear()
+    }
+
     this.refresh()
     this._dispatch_change_events()
   }

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

@@ -146,6 +146,13 @@ export class WorkspaceProvider
   public switch_context_state(is_frf: boolean) {
     if (this._is_frf_mode == is_frf) return
     this._is_frf_mode = is_frf
+
+    if (is_frf) {
+      this._checked_items.clear()
+      this._checked_timestamps.clear()
+      this._partially_checked_dirs.clear()
+    }
+
     this._token_calculator.clear_selected_counts()
     this.refresh()
     this._dispatch_change_events()

apps/editor/src/context/shared-file-state.ts

@@ -99,6 +99,14 @@ export class SharedFileState {
         this._open_editors_provider.switch_context_state(is_frf)
 
       this._is_frf_mode = is_frf
+
+      if (is_frf) {
+        this._unchecked_in_open_editors.clear()
+        this._unchecked_in_workspace.clear()
+        this._undo_stack = []
+        this._redo_stack = []
+      }
+
       this.update_checked_files_set()
       this._on_did_change_checked_files.fire()
     } finally {

apps/editor/src/i18n/translations/commands/apply-chat-response.ts

@@ -90,21 +90,21 @@ export const apply_chat_response = {
     bg: 'Потвърдете файловете за включване в контекста'
   },
   'command.apply-chat-response.relevant-files.success': {
-    en: 'Relevant files has been checked successfully.',
-    pl: 'Pomyślnie znaleziono odpowiednie pliki.',
-    'zh-cn': '成功找到相关文件。',
-    ja: '関連ファイルが正常に見つかりました。',
-    'zh-tw': '成功找到相關檔案。',
-    de: 'Relevante Dateien erfolgreich gefunden.',
-    es: 'Archivos relevantes encontrados exitosamente.',
-    fr: 'Fichiers pertinents trouvés avec succès.',
-    'pt-br': 'Arquivos relevantes encontrados com sucesso.',
-    ru: 'Соответствующие файлы успешно найдены.',
-    ko: '관련 파일을 성공적으로 찾았습니다.',
-    it: 'File rilevanti trovati con successo.',
-    tr: 'İlgili dosyalar başarıyla bulundu.',
-    cs: 'Relevantní soubory úspěšně nalezeny.',
-    hu: 'A releváns fájlok sikeresen megtalálva.',
-    bg: 'Съответните файлове са намерени успешно.'
+    en: 'Context has been updated',
+    pl: 'Kontekst został zaktualizowany',
+    'zh-cn': '上下文已更新',
+    ja: 'コンテキストが更新されました',
+    'zh-tw': '上下文已更新',
+    de: 'Kontext wurde aktualisiert',
+    es: 'El contexto ha sido actualizado',
+    fr: 'Le contexte a été mis à jour',
+    'pt-br': 'O contexto foi atualizado',
+    ru: 'Контекст был обновлен',
+    ko: '컨텍스트가 업데이트되었습니다',
+    it: 'Il contesto è stato aggiornato',
+    tr: 'Bağlam güncellendi',
+    cs: 'Kontext byl aktualizován',
+    hu: 'A kontextus frissítve lett',
+    bg: 'Контекстът е актуализиран'
   }
 } as const

apps/editor/src/views/panel/backend/panel-provider.ts

@@ -274,7 +274,7 @@ export class PanelProvider implements vscode.WebviewViewProvider {
     SharedFileState.get_instance().switch_context_state(is_find_relevant_files)
   }
 
-  public async switch_to_edit_context() {
+  public async return_home_and_switch_to_edit_context() {
     if (this.mode == MODE.WEB) {
       this.web_prompt_type = 'edit-context'
       await this.context.workspaceState.update(
@@ -298,6 +298,7 @@ export class PanelProvider implements vscode.WebviewViewProvider {
     }
     this.update_providers_shrink_mode()
     this.update_providers_context_state()
+    this.send_message({ command: 'RETURN_HOME' })
   }
 
   public async send_checkpoints() {
@@ -1005,6 +1006,8 @@ export class PanelProvider implements vscode.WebviewViewProvider {
             await handle_voice_input(this, message)
           } else if (message.command == 'GET_SETUP_PROGRESS') {
             await this.send_setup_progress()
+          } else if (message.command == 'RETURN_HOME') {
+            await this.return_home_and_switch_to_edit_context()
           } else if (
             message.command == 'GET_FIND_RELEVANT_FILES_SHRINK_SOURCE_CODE'
           ) {