Implement support for document symbols to handle long text pastes as referenced files and rename image handling commands for consistency

Author: robertpiosik

apps/editor/src/views/panel/backend/message-handlers/handle-code-completion.ts

@@ -27,6 +27,7 @@ import {
 import { replace_saved_context_symbol } from '@/views/panel/backend/utils/replace-saved-context-symbol'
 import { replace_skill_symbol } from '../utils/replace-skill-symbol'
 import { replace_image_symbol } from '../utils/replace-image-symbol'
+import { replace_document_symbol } from '../utils/replace-document-symbol'
 
 const get_code_completion_config = async (
   api_providers_manager: ModelProvidersManager,
@@ -411,6 +412,12 @@ export const handle_code_completion = async (
       })
     }
 
+    if (processed_completion_instructions.includes('#Document(')) {
+      processed_completion_instructions = await replace_document_symbol({
+        instruction: processed_completion_instructions
+      })
+    }
+
     const files_collector = new FilesCollector(
       panel_provider.workspace_provider,
       panel_provider.open_editors_provider

apps/editor/src/views/panel/backend/message-handlers/handle-copy-prompt.ts

@@ -10,6 +10,7 @@ import {
 import { replace_saved_context_symbol } from '@/views/panel/backend/utils/replace-saved-context-symbol'
 import { replace_skill_symbol } from '@/views/panel/backend/utils/replace-skill-symbol'
 import { replace_image_symbol } from '@/views/panel/backend/utils/replace-image-symbol'
+import { replace_document_symbol } from '../utils/replace-document-symbol'
 import {
   code_at_cursor_instructions_for_panel,
   prune_context_instructions_prefix,
@@ -148,6 +149,12 @@ export const handle_copy_prompt = async (params: {
       })
     }
 
+    if (processed_completion_instructions.includes('#Document(')) {
+      processed_completion_instructions = await replace_document_symbol({
+        instruction: processed_completion_instructions
+      })
+    }
+
     const missing_text_tag = processed_completion_instructions
       ? `<missing_text>${processed_completion_instructions}</missing_text>`
       : '<missing_text>'
@@ -220,6 +227,12 @@ export const handle_copy_prompt = async (params: {
       })
     }
 
+    if (processed_instructions.includes('#Document(')) {
+      processed_instructions = await replace_document_symbol({
+        instruction: processed_instructions
+      })
+    }
+
     let system_instructions_xml = ''
 
     if (params.panel_provider.web_prompt_type == 'edit-context') {

apps/editor/src/views/panel/backend/message-handlers/handle-edit-context.ts

@@ -25,6 +25,7 @@ import { replace_selection_symbol } from '@/views/panel/backend/utils/replace-se
 import { PanelProvider } from '@/views/panel/backend/panel-provider'
 import { replace_skill_symbol } from '@/views/panel/backend/utils/replace-skill-symbol'
 import { replace_image_symbol } from '@/views/panel/backend/utils/replace-image-symbol'
+import { replace_document_symbol } from '../utils/replace-document-symbol'
 import { apply_reasoning_effort } from '@/utils/apply-reasoning-effort'
 import { EditContextMessage } from '@/views/panel/types/messages'
 import { dictionary } from '@shared/constants/dictionary'
@@ -371,6 +372,12 @@ export const handle_edit_context = async (
     })
   }
 
+  if (processed_instructions.includes('#Document(')) {
+    processed_instructions = await replace_document_symbol({
+      instruction: processed_instructions
+    })
+  }
+
   const is_prune_context = panel_provider.api_prompt_type == 'prune-context'
   const collected_files = await files_collector.collect_files({
     compact: is_prune_context

apps/editor/src/views/panel/backend/message-handlers/handle-open-prompt-document.ts

@@ -0,0 +1,18 @@
+import * as vscode from 'vscode'
+import * as os from 'os'
+import * as path from 'path'
+
+export const handle_open_prompt_document = async (message: {
+  hash: string
+}) => {
+  try {
+    const hash = message.hash
+    const filename = `cwc-document-${hash}.txt`
+    const file_path = path.join(os.tmpdir(), filename)
+
+    const doc = await vscode.workspace.openTextDocument(file_path)
+    await vscode.window.showTextDocument(doc)
+  } catch (error) {
+    vscode.window.showErrorMessage(`Failed to open document: ${error}`)
+  }
+}

…ssage-handlers/handle-open-temp-image.ts …age-handlers/handle-open-prompt-image.tsapps/editor/src/views/panel/backend/message-handlers/handle-open-temp-image.ts renamed to apps/editor/src/views/panel/backend/message-handlers/handle-open-prompt-image.ts apps/editor/src/views/panel/backend/message-handlers/handle-open-temp-image.ts renamed to apps/editor/src/views/panel/backend/message-handlers/handle-open-prompt-image.ts

@@ -2,9 +2,11 @@ import * as vscode from 'vscode'
 import * as os from 'os'
 import * as path from 'path'
 import * as fs from 'fs'
-import { OpenTempImageMessage } from '../../types/messages'
+import { OpenPromptImageMessage } from '../../types/messages'
 
-export const handle_open_temp_image = async (message: OpenTempImageMessage) => {
+export const handle_open_prompt_image = async (
+  message: OpenPromptImageMessage
+) => {
   try {
     const hash = message.hash
     const txt_filename = `cwc-image-${hash}.txt`

apps/editor/src/views/panel/backend/message-handlers/handle-preview-preset.ts

@@ -12,6 +12,7 @@ import {
 import { Preset } from '@shared/types/preset'
 import { replace_skill_symbol } from '@/views/panel/backend/utils/replace-skill-symbol'
 import { replace_image_symbol } from '@/views/panel/backend/utils/replace-image-symbol'
+import { replace_document_symbol } from '../utils/replace-document-symbol'
 import { apply_preset_affixes_to_instruction } from '@/utils/apply-preset-affixes'
 import { dictionary } from '@shared/constants/dictionary'
 import {
@@ -128,6 +129,12 @@ export const handle_preview_preset = async (
       })
     }
 
+    if (processed_completion_instructions.includes('#Document(')) {
+      processed_completion_instructions = await replace_document_symbol({
+        instruction: processed_completion_instructions
+      })
+    }
+
     const missing_text_tag = processed_completion_instructions
       ? `<missing_text>${processed_completion_instructions}</missing_text>`
       : '<missing_text>'
@@ -209,6 +216,12 @@ export const handle_preview_preset = async (
       })
     }
 
+    if (processed_instructions.includes('#Document(')) {
+      processed_instructions = await replace_document_symbol({
+        instruction: processed_instructions
+      })
+    }
+
     let system_instructions_xml = ''
     if (panel_provider.web_prompt_type == 'edit-context') {
       const config = vscode.workspace.getConfiguration('codeWebChat')

apps/editor/src/views/panel/backend/message-handlers/handle-prune-context.ts

@@ -23,6 +23,7 @@ import { replace_selection_symbol } from '@/views/panel/backend/utils/replace-se
 import { PanelProvider } from '@/views/panel/backend/panel-provider'
 import { replace_skill_symbol } from '@/views/panel/backend/utils/replace-skill-symbol'
 import { replace_image_symbol } from '@/views/panel/backend/utils/replace-image-symbol'
+import { replace_document_symbol } from '@/views/panel/backend/utils/replace-document-symbol'
 import { apply_reasoning_effort } from '@/utils/apply-reasoning-effort'
 import { PruneContextMessage } from '@/views/panel/types/messages'
 import { dictionary } from '@shared/constants/dictionary'
@@ -352,6 +353,12 @@ export const handle_prune_context = async (
     })
   }
 
+  if (processed_instructions.includes('#Document(')) {
+    processed_instructions = await replace_document_symbol({
+      instruction: processed_instructions
+    })
+  }
+
   const collected_files = await files_collector.collect_files({
     compact: true
   })

apps/editor/src/views/panel/backend/message-handlers/handle-save-prompt-document.ts

@@ -0,0 +1,22 @@
+import * as vscode from 'vscode'
+import * as os from 'os'
+import * as path from 'path'
+import * as fs from 'fs'
+import * as crypto from 'crypto'
+import { SavePromptDocumentMessage } from '../../types/messages'
+import { PanelProvider } from '../panel-provider'
+
+export const handle_save_prompt_document = async (
+  panel_provider: PanelProvider,
+  message: SavePromptDocumentMessage
+) => {
+  try {
+    const hash = crypto.createHash('md5').update(message.text).digest('hex')
+    const filename = `cwc-document-${hash}.txt`
+    const file_path = path.join(os.tmpdir(), filename)
+    await fs.promises.writeFile(file_path, message.text, 'utf-8')
+    panel_provider.add_text_at_cursor_position(`#Document(${hash})`)
+  } catch (error) {
+    vscode.window.showErrorMessage(`Failed to save document: ${error}`)
+  }
+}

…ssage-handlers/handle-save-temp-image.ts …age-handlers/handle-save-prompt-image.tsapps/editor/src/views/panel/backend/message-handlers/handle-save-temp-image.ts renamed to apps/editor/src/views/panel/backend/message-handlers/handle-save-prompt-image.ts apps/editor/src/views/panel/backend/message-handlers/handle-save-temp-image.ts renamed to apps/editor/src/views/panel/backend/message-handlers/handle-save-prompt-image.ts

@@ -3,12 +3,12 @@ import * as os from 'os'
 import * as path from 'path'
 import * as fs from 'fs'
 import * as crypto from 'crypto'
-import { SaveTempImageMessage } from '../../types/messages'
+import { SavePromptImageMessage } from '../../types/messages'
 import { PanelProvider } from '../panel-provider'
 
-export const handle_save_temp_image = async (
+export const handle_save_prompt_image = async (
   panel_provider: PanelProvider,
-  message: SaveTempImageMessage
+  message: SavePromptImageMessage
 ) => {
   try {
     const buffer = Buffer.from(message.content_base64, 'base64')

apps/editor/src/views/panel/backend/message-handlers/handle-send-to-browser.ts

@@ -11,6 +11,7 @@ import {
 } from '@/views/panel/backend/utils/replace-git-symbols'
 import { replace_skill_symbol } from '@/views/panel/backend/utils/replace-skill-symbol'
 import { replace_image_symbol } from '@/views/panel/backend/utils/replace-image-symbol'
+import { replace_document_symbol } from '../utils/replace-document-symbol'
 import {
   code_at_cursor_instructions_for_panel,
   prune_context_instructions_prefix,
@@ -175,6 +176,12 @@ export const handle_send_to_browser = async (params: {
       })
     }
 
+    if (processed_completion_instructions.includes('#Document(')) {
+      processed_completion_instructions = await replace_document_symbol({
+        instruction: processed_completion_instructions
+      })
+    }
+
     const context_text = await files_collector.collect_files({
       exclude_path: active_path
     })
@@ -284,6 +291,12 @@ export const handle_send_to_browser = async (params: {
           })
         }
 
+        if (processed_instructions.includes('#Document(')) {
+          processed_instructions = await replace_document_symbol({
+            instruction: processed_instructions
+          })
+        }
+
         let system_instructions_xml = ''
         if (params.panel_provider.web_prompt_type == 'edit-context') {
           const config = vscode.workspace.getConfiguration('codeWebChat')