Merge selected files payload into token count message

- number of selected files and token count should be sent to front-end in a single message, so that there are no desynchronized UI updates

└── apps
    └── editor
        └── src
            ├── context
            │   ├── context-initialization.ts
            │   ├── helpers
            │   │   └── get-selected-files.ts
            │   ├── providers
            │   │   └── workspace
            │   │       ├── modules
            │   │       │   └── token-calculator.ts
            │   │       └── workspace-provider.ts
            │   └── shared-context-state.ts
            └── views
                └── prompt
                    ├── backend
                    │   └── prompt-view-provider.ts
                    ├── frontend
                    │   ├── Main
                    │   │   ├── Main.tsx
                    │   │   └── MainView
                    │   │       ├── MainView.tsx
                    │   │       └── components
                    │   │           └── PromptAttachments
                    │   │               └── PromptAttachments.tsx
                    │   ├── Prompt.tsx
                    │   └── hooks
                    │       ├── use-editor-sync.ts
                    │       └── use-prompt
                    │           └── use-prompt.ts
                    └── types
                        └── messages.ts

Author: robertpiosik

apps/editor/src/views/prompt/backend/prompt-view-provider.ts

@@ -375,7 +375,7 @@ export class PromptViewProvider implements vscode.WebviewViewProvider {
       vscode.workspace.onDidChangeWorkspaceFolders(() => {
         handle_get_tasks(this)
         handle_get_workspace_state(this)
-        this.send_selected_files()
+        this.send_token_count()
       })
     )
 
@@ -428,7 +428,6 @@ export class PromptViewProvider implements vscode.WebviewViewProvider {
       this.current_selected_files_token_count = token_count
       if (this.webview_view) {
         this.send_token_count()
-        this.send_selected_files()
       }
     })
 
@@ -579,6 +578,11 @@ export class PromptViewProvider implements vscode.WebviewViewProvider {
       symbols_cache: this.symbols_cache
     })
 
+    const selected_files = get_selected_files({
+      workspace_provider: this.workspace_provider,
+      shared_context_state: this.shared_context_state
+    })
+
     this.send_message({
       command: 'TOKEN_COUNT_UPDATED',
       selected_files_token_count: this.current_selected_files_token_count,
@@ -597,19 +601,8 @@ export class PromptViewProvider implements vscode.WebviewViewProvider {
         ) +
           ask_instructions_result.skill_definitions.length) /
           4
-      )
-    })
-  }
-
-  send_selected_files() {
-    const file_paths = get_selected_files({
-      workspace_provider: this.workspace_provider,
-      shared_context_state: this.shared_context_state
-    })
-
-    this.send_message({
-      command: 'SELECTED_FILES',
-      file_paths
+      ),
+      selected_files
     })
   }
 
@@ -890,8 +883,6 @@ export class PromptViewProvider implements vscode.WebviewViewProvider {
             await handle_pick_tasks_workspace(this, message)
           } else if (message.command == 'GET_TOKEN_COUNT') {
             this.send_token_count()
-          } else if (message.command == 'GET_SELECTED_FILES') {
-            this.send_selected_files()
           } else if (message.command == 'PREVIEW_PROMPT') {
             await handle_preview_prompt({
               prompt_view_provider: this

apps/editor/src/views/prompt/frontend/hooks/use-editor-sync.ts

@@ -26,8 +26,8 @@ export const use_editor_sync = (vscode: any) => {
         set_currently_open_file_path(message.currently_open_file_path)
       } else if (message.command == 'EDITOR_SELECTION_CHANGED') {
         set_current_selection(message.current_selection)
-      } else if (message.command == 'SELECTED_FILES') {
-        set_selected_files(message.file_paths)
+      } else if (message.command == 'TOKEN_COUNT_UPDATED') {
+        set_selected_files(message.selected_files)
       } else if (message.command == 'CURRENTLY_OPEN_FILE_TEXT') {
         set_currently_open_file_text(message.text)
       } else if (message.command == 'WORKSPACE_STATE') {

apps/editor/src/views/prompt/frontend/hooks/use-prompt/use-prompt.ts

@@ -211,7 +211,6 @@ export const use_prompt = (vscode: any) => {
       { command: 'GET_VOICE_INPUT_PUSH_TO_TALK' },
       { command: 'GET_IS_MODERN_UI' },
       { command: 'GET_TOKEN_COUNT' },
-      { command: 'GET_SELECTED_FILES' },
       { command: 'GET_API_CONFIGURATIONS' }
     ]
     initial_messages.forEach((message) => post_message(vscode, message))

apps/editor/src/views/prompt/types/messages.ts

@@ -567,10 +567,6 @@ export interface GetTokenCountMessage extends BaseMessage {
   command: 'GET_TOKEN_COUNT'
 }
 
-export interface GetSelectedFilesMessage extends BaseMessage {
-  command: 'GET_SELECTED_FILES'
-}
-
 export interface PreviewPromptMessage extends BaseMessage {
   command: 'PREVIEW_PROMPT'
 }
@@ -711,7 +707,6 @@ export type FrontendMessage =
   | PickTasksWorkspaceMessage
   | GetIsModernUiMessage
   | GetTokenCountMessage
-  | GetSelectedFilesMessage
   | PreviewPromptMessage
   | InstallBrowserExtensionMessage
   | PreviewChangesSymbolMessage
@@ -816,6 +811,7 @@ export interface TokenCountMessage extends BaseMessage {
   selected_files_token_count: number
   edit_instructions_token_count: number
   ask_instructions_token_count: number
+  selected_files: string[]
 }
 
 export interface WebConfigurationUpdatedMessage extends BaseMessage {
@@ -871,11 +867,6 @@ export interface ResetApplyButtonTemporaryDisabledStateMessage extends BaseMessa
   command: 'RESET_APPLY_BUTTON_TEMPORARY_DISABLED_STATE'
 }
 
-export interface SelectedFilesMessage extends BaseMessage {
-  command: 'SELECTED_FILES'
-  file_paths: string[]
-}
-
 export interface SendWithShiftEnterMessage extends BaseMessage {
   command: 'SEND_WITH_SHIFT_ENTER'
   enabled: boolean
@@ -1022,7 +1013,6 @@ export type BackendMessage =
   | ApiPromptTypeMessage
   | CliPromptTypeMessage
   | VersionMessage
-  | SelectedFilesMessage
   | SendWithShiftEnterMessage
   | CanUndoChangedMessage
   | ResetApplyButtonTemporaryDisabledStateMessage