allow line numbers in autocomplete

Author: ewired

packages/opencode/src/cli/cmd/tui/component/prompt/autocomplete.tsx

@@ -11,6 +11,42 @@ import { useCommandDialog } from "@tui/component/dialog-command"
 import { Locale } from "@/util/locale"
 import type { PromptInfo } from "./history"
 
+function removeLineRange(input: string) {
+  const colonIndex = input.lastIndexOf(":")
+  return colonIndex !== -1 ? input.substring(0, colonIndex) : input
+}
+
+function extractLineRange(input: string) {
+  const colonIndex = input.lastIndexOf(":")
+  if (colonIndex === -1) {
+    return { baseQuery: input }
+  }
+
+  const baseName = input.substring(0, colonIndex)
+  const linePart = input.substring(colonIndex + 1)
+  const lineMatch = linePart.match(/^(\d+)(?:-(\d+))?$/)
+
+  if (!lineMatch) {
+    return { baseQuery: baseName }
+  }
+
+  const startLine = Number(lineMatch[1])
+  const endLine = lineMatch[2] ? Number(lineMatch[2]) : undefined
+
+  if (endLine !== undefined && startLine > endLine) {
+    return { baseQuery: baseName }
+  }
+
+  return {
+    lineRange: {
+      baseName,
+      startLine,
+      endLine,
+    },
+    baseQuery: baseName,
+  }
+}
+
 export type AutocompleteRef = {
   onInput: (value: string) => void
   onKeyDown: (e: KeyEvent) => void
@@ -115,10 +151,12 @@ export function Autocomplete(props: {
     async (query) => {
       if (!store.visible || store.visible === "/") return []
 
+      const { lineRange, baseQuery } = extractLineRange(query ?? "")
+
       // Get files from SDK
       const result = await sdk.client.find.files({
         query: {
-          query: query ?? "",
+          query: baseQuery,
         },
       })
 
@@ -128,15 +166,27 @@ export function Autocomplete(props: {
       if (!result.error && result.data) {
         const width = store.position.width - 4
         options.push(
-          ...result.data.map(
-            (item): AutocompleteOption => ({
-              display: Locale.truncateMiddle(item, width),
+          ...result.data.map((item): AutocompleteOption => {
+            let url = `file://${process.cwd()}/${item}`
+            let filename = item
+            if (lineRange && !item.endsWith("/")) {
+              filename = `${item}:${lineRange.startLine}${lineRange.endLine ? `-${lineRange.endLine}` : ""}`
+              const urlObj = new URL(url)
+              urlObj.searchParams.set("start", String(lineRange.startLine))
+              if (lineRange.endLine !== undefined) {
+                urlObj.searchParams.set("end", String(lineRange.endLine))
+              }
+              url = urlObj.toString()
+            }
+
+            return {
+              display: Locale.truncateMiddle(filename, width),
               onSelect: () => {
-                insertPart(item, {
+                insertPart(filename, {
                   type: "file",
                   mime: "text/plain",
-                  filename: item,
-                  url: `file://${process.cwd()}/${item}`,
+                  filename,
+                  url,
                   source: {
                     type: "file",
                     text: {
@@ -148,8 +198,8 @@ export function Autocomplete(props: {
                   },
                 })
               },
-            }),
-          ),
+            }
+          }),
         )
       }
 
@@ -331,8 +381,8 @@ export function Autocomplete(props: {
     ).filter((x) => x.disabled !== true)
     const currentFilter = filter()
     if (!currentFilter) return mixed.slice(0, 10)
-    const result = fuzzysort.go(currentFilter, mixed, {
-      keys: [(obj) => obj.display.trimEnd(), "description", (obj) => obj.aliases?.join(" ") ?? ""],
+    const result = fuzzysort.go(removeLineRange(currentFilter), mixed, {
+      keys: [(obj) => removeLineRange(obj.display.trimEnd()), "description", (obj) => obj.aliases?.join(" ") ?? ""],
       limit: 10,
     })
     return result.map((arr) => arr.obj)