Add line range selection functionality to files in the workspace context view for targeted code context usage

Author: robertpiosik

packages/vscode/package.json

@@ -280,6 +280,11 @@
         "title": "Generate Commit Message with CWC",
         "category": "Code Web Chat",
         "icon": "$(sparkle)"
+      },
+      {
+        "command": "codeWebChat.setRange",
+        "title": "Set Range",
+        "category": "Code Web Chat"
       }
     ],
     "menus": {
@@ -367,6 +372,10 @@
         {
           "command": "codeWebChat.generateCommitMessage",
           "when": "false"
+        },
+        {
+          "command": "codeWebChat.setRange",
+          "when": "false"
         }
       ],
       "view/title": [
@@ -494,6 +503,11 @@
           "when": "view == codeWebChatViewOpenEditors && viewItem == openEditor",
           "group": "1_modification@3"
         },
+        {
+          "command": "codeWebChat.setRange",
+          "when": "view == codeWebChatViewOpenEditors && viewItem == openEditor",
+          "group": "1_modification@4"
+        },
         {
           "command": "codeWebChat.newFile",
           "when": "view == codeWebChatViewWorkspace && viewItem == directory",
@@ -518,6 +532,11 @@
           "command": "codeWebChat.referenceInPrompt",
           "when": "view == codeWebChatViewWorkspace || view == codeWebChatViewContext",
           "group": "1_modification@3"
+        },
+        {
+          "command": "codeWebChat.setRange",
+          "when": "(view == codeWebChatViewWorkspace || view == codeWebChatViewContext) && viewItem == file",
+          "group": "1_modification@4"
         }
       ]
     },

packages/vscode/src/commands/index.ts

@@ -16,3 +16,4 @@ export * from './save-all-command'
 export * from './save-context-command'
 export * from './apply-context-command'
 export * from './find-paths-in-clipboard-command'
+export * from './set-range-command'

packages/vscode/src/commands/set-range-command.ts

@@ -0,0 +1,125 @@
+import * as vscode from 'vscode'
+import * as path from 'path'
+import * as fs from 'fs/promises'
+import { FileItem } from '../context/providers/workspace-provider'
+
+export const set_range_command = () => {
+  return vscode.commands.registerCommand(
+    'codeWebChat.setRange',
+    async (item: FileItem) => {
+      if (!item || !item.resourceUri) {
+        return
+      }
+      const file_path = item.resourceUri.fsPath
+      const workspace_folder = vscode.workspace.getWorkspaceFolder(
+        item.resourceUri
+      )
+
+      if (!workspace_folder) {
+        vscode.window.showErrorMessage('File is not in a workspace folder.')
+        return
+      }
+
+      const ranges_file_path = path.join(
+        workspace_folder.uri.fsPath,
+        '.vscode',
+        'ranges.json'
+      )
+      let ranges: Record<string, string> = {}
+      try {
+        const content = await fs.readFile(ranges_file_path, 'utf-8')
+        ranges = JSON.parse(content)
+      } catch (error) {
+        // File might not exist, which is fine
+      }
+
+      const relative_path = path.relative(
+        workspace_folder.uri.fsPath,
+        file_path
+      )
+      const current_range = ranges[relative_path]
+
+      const new_range = await vscode.window.showInputBox({
+        prompt: `Set range of lines for ${path.basename(file_path)}`,
+        title: 'Range',
+        placeHolder: 'e.g., 100-300 400- -500 or empty to clear',
+        value: current_range,
+        validateInput: (value) => {
+          if (value.trim() == '') return null
+
+          const parts = value.trim().split(/\s+/)
+          const parsed_ranges: {
+            start: number
+            end: number
+            original: string
+          }[] = []
+
+          for (const part of parts) {
+            const match = part.match(/^(\d+)?-(\d+)?$/)
+            if (!match) {
+              return `Invalid format in "${part}". Use formats like 100-300, 100-, -300.`
+            }
+
+            const [, start_str, end_str] = match
+
+            if (!start_str && !end_str) {
+              return 'Invalid range "-". Specify at least a start or an end line.'
+            }
+
+            const start = start_str ? parseInt(start_str, 10) : null
+            const end = end_str ? parseInt(end_str, 10) : null
+
+            if (start !== null && start < 1) {
+              return `Start line must be 1 or greater in "${part}".`
+            }
+
+            if (end !== null && end < 1) {
+              return `End line must be 1 or greater in "${part}".`
+            }
+
+            if (start !== null && end !== null) {
+              if (start > end) {
+                return `Start line cannot be greater than end line in "${part}".`
+              }
+              if (start == end) {
+                return `Start and end lines cannot be the same in "${part}".`
+              }
+            }
+            parsed_ranges.push({
+              start: start ?? 1,
+              end: end ?? Number.MAX_SAFE_INTEGER,
+              original: part
+            })
+          }
+          parsed_ranges.sort((a, b) => a.start - b.start)
+
+          for (let i = 0; i < parsed_ranges.length - 1; i++) {
+            if (parsed_ranges[i].end >= parsed_ranges[i + 1].start) {
+              return `Ranges cannot overlap: "${
+                parsed_ranges[i].original
+              }" and "${parsed_ranges[i + 1].original}".`
+            }
+          }
+          return null
+        }
+      })
+
+      if (new_range === undefined) {
+        return
+      }
+
+      if (new_range) {
+        ranges[relative_path] = new_range.trim().split(/\s+/).join(' ')
+      } else {
+        delete ranges[relative_path]
+      }
+
+      try {
+        await fs.mkdir(path.dirname(ranges_file_path), { recursive: true })
+        await fs.writeFile(ranges_file_path, JSON.stringify(ranges, null, 2))
+      } catch (error: any) {
+        vscode.window.showErrorMessage(`Failed to save range: ${error.message}`)
+      }
+    }
+  )
+}

packages/vscode/src/context/context-initialization.ts

@@ -268,7 +268,15 @@ export const context_initialization = async (
             const file_uri = vscode.Uri.file(file_path)
             const content_uint8_array =
               await vscode.workspace.fs.readFile(file_uri)
-            const content = new TextDecoder().decode(content_uint8_array)
+            let content = new TextDecoder().decode(content_uint8_array)
+
+            const range = workspace_provider.get_range(file_path)
+            if (range) {
+              content = workspace_provider.apply_range_to_content(
+                content,
+                range
+              )
+            }
 
             let display_path: string
             const workspace_folder =

packages/vscode/src/context/providers/open-editors-provider.ts

@@ -21,6 +21,7 @@ export class OpenEditorsProvider
   private _file_token_counts: Map<string, number> = new Map()
   private _tab_change_handler: vscode.Disposable
   private _file_change_watcher: vscode.Disposable
+  private _workspace_change_handler: vscode.Disposable
   private _initialized: boolean = false
   private _opened_from_workspace_view: Set<string> = new Set()
   private _non_preview_files: Set<string> = new Set()
@@ -39,6 +40,9 @@ export class OpenEditorsProvider
     this._shared_state = SharedFileState.get_instance()
     this.workspace_provider = workspace_provider
 
+    this._workspace_change_handler =
+      this.workspace_provider.onDidChangeTreeData(() => this.refresh())
+
     this._update_preview_tabs_state()
 
     this._tab_change_handler = vscode.window.tabGroups.onDidChangeTabs((e) => {
@@ -147,6 +151,7 @@ export class OpenEditorsProvider
   }
 
   dispose(): void {
+    this._workspace_change_handler.dispose()
     this._tab_change_handler.dispose()
     this._file_change_watcher.dispose()
     this._config_change_handler.dispose()
@@ -196,15 +201,28 @@ export class OpenEditorsProvider
 
     const token_count = element.tokenCount
 
+    let final_description = element.description || '' // relative path
+
+    const prefix_parts: string[] = []
     if (token_count !== undefined) {
-      const formatted_token_count = display_token_count(token_count)
-      if (element.description) {
-        element.description = `${formatted_token_count} · ${element.description}`
+      prefix_parts.push(display_token_count(token_count))
+    }
+    if (element.range) {
+      prefix_parts.push(element.range)
+    }
+
+    const prefix = prefix_parts.join(' · ')
+
+    if (prefix) {
+      if (final_description) {
+        final_description = `${prefix} · ${final_description}`
       } else {
-        element.description = formatted_token_count
+        final_description = prefix
       }
     }
 
+    element.description = final_description
+
     return element
   }
 
@@ -283,6 +301,7 @@ export class OpenEditorsProvider
         description = path.basename(workspace_root)
       }
 
+      const range = this.workspace_provider.get_range(file_path)
       const token_count =
         await this.workspace_provider.calculate_file_tokens(file_path)
 
@@ -296,7 +315,9 @@ export class OpenEditorsProvider
         true,
         token_count,
         undefined, // selectedTokenCount is undefined for open editor files
-        description
+        description,
+        false,
+        range
       )
 
       items.push(item)