Implement interactive search prompt for relevant files during response application

- add internal command to 'search files commands' that allows providing results set (array of file objects (path, checked state)) in param. Use it in 'apply response command', in place of the current flow that uses prompt view.
- when using 'codeWebChat.internal.searchFilesWithResults', don't send response to the panel (remove any leftover logic of not using panel for file search anymore)
- it's been reported that 'search in results' doesn't work when invoked with [Fragment]

└── apps
    └── editor
        └── src
            ├── commands
            │   ├── apply-response-command
            │   │   ├── apply-response-command.ts
            │   │   ├── handlers
            │   │   │   ├── diff-handler.ts
            │   │   │   ├── fast-replace-handler.ts
            │   │   │   ├── restore-preview-handler.ts
            │   │   │   ├── search-replace-handler.ts
            │   │   │   └── truncated-handler.ts
            │   │   ├── response-processor.ts
            │   │   ├── types
            │   │   │   └── original-file-state.ts
            │   │   └── utils
            │   │       ├── file-checks.ts
            │   │       ├── file-operations.ts
            │   │       ├── preview
            │   │       │   ├── diff-utils.ts
            │   │       │   ├── file-preparer.ts
            │   │       │   ├── index.ts
            │   │       │   ├── preview.ts
            │   │       │   ├── temp-file-manager.ts
            │   │       │   ├── types.ts
            │   │       │   ├── virtual-document-provider.ts
            │   │       │   ├── vscode-ui.ts
            │   │       │   └── workspace-listener.ts
            │   │       ├── preview-handler.ts
            │   │       └── state-manager.ts
            │   ├── generate-commit-message-commands
            │   │   └── generate-commit-message-commands.ts
            │   └── search-files-commands.ts
            └── features
                └── search-files
                    ├── index.ts
                    ├── search-files.ts
                    ├── search-modes
                    │   ├── perform-intelligent-search-mode.ts
                    │   ├── perform-keywords-search-mode.ts
                    │   ├── perform-phrase-search-mode.ts
                    │   └── perform-semantic-search-mode.ts
                    └── utils
                        ├── analyze-files.ts
                        ├── check-matches-condition.ts
                        ├── create-search-regex.ts
                        ├── prompt-for-config.ts
                        ├── prompt-for-intelligent-search-results.ts
                        ├── prompt-for-keywords-match-mode.ts
                        ├── prompt-for-keywords-target.ts
                        ├── prompt-for-provided-results.ts
                        ├── prompt-for-search-mode.ts
                        ├── prompt-for-search-results.ts
                        ├── prompt-for-search-term.ts
                        ├── prompt-for-shrink-mode.ts
                        ├── search-files-by-intelligent.ts
                        ├── search-files-by-keywords.ts
                        ├── search-files-by-phrase.ts
                        └── search-files-by-semantic.ts

Author: robertpiosik

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

@@ -115,7 +115,11 @@ export const apply_response_command = (params: {
         workspace_files
       })
 
-      if (resolve_fn) {
+      const is_relevant_files = response_items.some(
+        (i) => i.type == 'relevant-files'
+      )
+
+      if (resolve_fn && !is_relevant_files) {
         const history = params.prompt_view_provider.response_history
 
         let created_at_for_switch: number
@@ -184,9 +188,6 @@ export const apply_response_command = (params: {
       let saved_tab_groups: SavedTabGroups | undefined
 
       try {
-        const is_relevant_files = response_items.some(
-          (i) => i.type == 'relevant-files'
-        )
         if (!is_relevant_files) {
           // Save current tab groups before entering preview
           saved_tab_groups = {

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

@@ -28,12 +28,7 @@ import { handle_truncated_edit } from './handlers/truncated-handler'
 import { WorkspaceProvider } from '@/context/providers/workspace/workspace-provider'
 import { natural_sort } from '@/utils/natural-sort'
 import { t } from '@/i18n'
-import {
-  RelevantFileInPreview,
-  ItemInPreview
-} from '@shared/types/file-in-preview'
-import { PreviewDecision } from './utils/preview/types'
-import { set_response_preview_promise_resolve } from './utils/preview/preview'
+import { RelevantFileInPreview } from '@shared/types/file-in-preview'
 import { is_truncation_line } from '@/utils/changes-integration/truncations-processor/utils/is-truncation-line'
 
 export type PreviewData = {
@@ -119,7 +114,6 @@ export const process_response = async (params: {
     const workspace_roots = params.workspace_provider.get_workspace_roots()
 
     const files_for_preview: RelevantFileInPreview[] = []
-    const items_for_preview: ItemInPreview[] = []
 
     for (const item of params.response_items) {
       if (item.type == 'relevant-files') {
@@ -180,71 +174,24 @@ export const process_response = async (params: {
 
         local_files.sort((a, b) => natural_sort(a.file_path, b.file_path))
         files_for_preview.push(...local_files)
-        items_for_preview.push(...local_files)
-      } else if (item.type == 'text') {
-        items_for_preview.push({
-          type: 'text',
-          content: item.content
-        })
-      } else if (item.type == 'inline-file') {
-        items_for_preview.push({
-          type: 'inline-file',
-          content: item.content,
-          language: item.language
-        })
       }
     }
 
-    const created_at_for_preview = params.args?.created_at ?? Date.now()
-
-    const history = params.prompt_view_provider.response_history
-    if (!params.args?.created_at) {
-      const new_item = {
-        response: params.response,
-        raw_instructions: params.args?.raw_instructions,
-        created_at: created_at_for_preview,
-        relevant_files: files_for_preview,
-        url: params.args?.url,
-        recent_api_configuration: params.args?.recent_api_configuration
-      }
-      history.push(new_item)
-    } else {
-      const existing = history.find(
-        (i) => i.created_at === params.args!.created_at
-      )
-      if (existing) {
-        existing.relevant_files = files_for_preview
-      }
-    }
-    params.prompt_view_provider.send_message({
-      command: 'RESPONSE_HISTORY',
-      history
-    })
-
     params.prompt_view_provider.send_message({
       command: 'HIDE_PROGRESS'
     })
 
-    params.prompt_view_provider.send_message({
-      command: 'RESPONSE_PREVIEW_STARTED',
-      items: items_for_preview,
-      raw_instructions: params.args?.raw_instructions,
-      created_at: created_at_for_preview,
-      url: params.args?.url,
-      recent_api_configuration: params.args?.recent_api_configuration
-    })
-
-    const decision = await new Promise<PreviewDecision>((resolve) => {
-      set_response_preview_promise_resolve(resolve)
-    }).finally(() => {
-      set_response_preview_promise_resolve(undefined)
-    })
-
-    params.prompt_view_provider.send_message({
-      command: 'RESPONSE_PREVIEW_FINISHED'
-    })
+    const decision = await vscode.commands.executeCommand<
+      { selected_paths: string[]; matched_paths: string[] } | undefined
+    >(
+      'codeWebChat.internal.searchFilesWithResults',
+      files_for_preview.map((f) => ({
+        path: f.absolute_path,
+        checked: f.is_checked
+      }))
+    )
 
-    if ('accepted_files' in decision && decision.accepted_files.length > 0) {
+    if (decision) {
       const shared_context_state =
         params.prompt_view_provider.shared_context_state
       const was_frf = params.workspace_provider.is_frf_mode
@@ -253,9 +200,8 @@ export const process_response = async (params: {
         shared_context_state.switch_context_state(false)
       }
 
-      const accepted_files = decision.accepted_files as RelevantFileInPreview[]
-      const presented_files = files_for_preview.map((f) => f.absolute_path!)
-      const selected_files = accepted_files.map((f) => f.absolute_path!)
+      const selected_files = decision.selected_paths
+      const presented_files = decision.matched_paths
 
       const filtered_current_files = current_checked_files.filter(
         (f) => !presented_files.includes(f)
@@ -277,16 +223,6 @@ export const process_response = async (params: {
       })
 
       await params.prompt_view_provider.switch_to_edit_files()
-    } else if ('created_at' in decision && decision.created_at) {
-      const history = params.prompt_view_provider.response_history
-      const new_history = history.filter(
-        (item) => item.created_at !== created_at_for_preview
-      )
-      params.prompt_view_provider.response_history = new_history
-      params.prompt_view_provider.send_message({
-        command: 'RESPONSE_HISTORY',
-        history: new_history
-      })
     }
 
     return null

apps/editor/src/commands/search-files-commands.ts

@@ -6,6 +6,7 @@ import { Logger } from '@shared/utils/logger'
 import { dictionary } from '@shared/constants/dictionary'
 import { t } from '@/i18n'
 import { search_files } from '@/features/search-files'
+import { prompt_for_provided_results } from '@/features/search-files/utils/prompt-for-provided-results'
 
 export const get_target_folder_path = async (
   item?: any
@@ -129,6 +130,37 @@ export const search_files_commands = (
     vscode.commands.registerCommand(
       'codeWebChat.searchFilesFromFile',
       (item: any) => search_handler(item)
+    ),
+    vscode.commands.registerCommand(
+      'codeWebChat.internal.searchFilesWithResults',
+      async (files: { path: string; checked: boolean }[]) => {
+        while (true) {
+          const result = await prompt_for_provided_results({
+            files,
+            workspace_provider
+          })
+
+          if (!result) return undefined
+
+          if ('action' in result) {
+            const sub_search_result = await search_files({
+              get_files: async () => result.matched_paths,
+              workspace_provider,
+              extension_context,
+              show_back_button: true,
+              is_sub_search: true
+            })
+
+            if (sub_search_result === 'back') {
+              continue
+            }
+
+            return sub_search_result
+          }
+
+          return result
+        }
+      }
     )
   ]
 }

apps/editor/src/extension.ts

@@ -33,13 +33,13 @@ import {
   generate_commit_message_commands,
   set_ranges_command,
   search_files_commands,
-  select_referencing_files_commands,
   select_definition_file_command,
   select_unstaged_files_command,
   select_commit_files_command,
   select_clipboard_paths_command,
   select_changed_files_command,
-  select_parent_folder_command
+  select_parent_folder_command,
+  select_referencing_files_commands
 } from './commands'
 import { setup_git_discard_file_watcher } from './services/git-discard-file-watcher'
 import { select_imported_files_command } from './commands/select-imported-files-command'
@@ -178,7 +178,7 @@ export const activate = async (extension_context: vscode.ExtensionContext) => {
       workspace_provider,
       extension_context
     ),
-    ...select_referencing_files_commands(workspace_provider, extension_context),
+    // ...select_referencing_files_commands(workspace_provider, extension_context),
     ...search_files_commands(workspace_provider, extension_context),
     select_definition_file_command(workspace_provider),
     open_url_command({