dsh-vscode-mode 0.1.28 → 0.1.29

package/src/search/content.ts

@@ -0,0 +1,319 @@
+/**
+ * dsh-vscode-mode host — 工作区内容搜索（rg --json 主路径 + 有界编排）。
+ * 命中列由 rg 行内字节偏移转 UTF-16（1-based），可直接供 Monaco 跳转/高亮。
+ * 无 fallback：provider 失败抛错，由 RPC 层转错误响应（避免整树读文件）。
+ * 作者 ddj 2026年08月26号
+ */
+import type { Ctx, Session } from '../store.js'
+import { pathText } from './query.js'
+import { EXCLUDES, FILE_EXCLUDES, ripgrepPath, searchRoot } from './ripgrep.js'
+import { SearchCache } from './orchestrator.js'
+import type { ContentMatch, ContentSearchInput, ContentSearchProvider, ContentSearchResult } from './types.js'
+
+const STDOUT_CAP = 16 << 20
+const STDERR_CAP = 64 << 10
+const GRACE_MS = 20_000
+const DEFAULT_MAX_MATCHES = 500
+const DEFAULT_MAX_FILES = 100
+const CACHE_TTL = 60_000
+const CACHE_LIMIT = 100
+const PROVIDER_VERSION = 'content-ripgrep-v1'
+/** 单行 JSON 记录上限：巨型单行文件（源映射/打包产物）的命中行对搜索 UI 无意义，直接跳过。 */
+const MAX_RECORD_LINE = 1 << 20
+
+/** rg --json 单条 match 记录的宽松形状（未知字段忽略）。 */
+interface RgJsonItem {
+  type?: string
+  data?: {
+    path?: { text?: string }
+    lines?: { text?: string }
+    line_number?: number
+    submatches?: Array<{ start?: number; end?: number }>
+  }
+}
+
+/** 内容搜索请求（RPC 参数映射）。 */
+export interface ContentSearchRequest {
+  session: Session
+  cwd: string
+  query: string
+  matchCase?: boolean
+  wholeWord?: boolean
+  regex?: boolean
+  maxResults?: number
+  /** 正选 glob（仅在这些文件内搜索）。 */
+  include?: string[]
+  /** 排除 glob（这些文件不参与搜索）。 */
+  exclude?: string[]
+}
+
+type ContentSearchResponse = { matches: ContentMatch[]; truncated: boolean }
+
+/**
+ * 把行内字节偏移转成 UTF-16 列（1-based；越界钳到行尾）。
+ * @author ddj 2026年08月26号
+ * @param text 行文本
+ * @param byteOffset 行内字节偏移（rg submatches.start/end，实测为行相对）
+ * @returns 1-based UTF-16 列
+ */
+export function byteToUtf16Col(text: string, byteOffset: number): number {
+  let bytes = 0
+  let col = 1
+  for (const char of text) {
+    if (bytes >= byteOffset) break
+    bytes += Buffer.byteLength(char)
+    col++
+  }
+  return col
+}
+
+/**
+ * 根内绝对路径 → 工作区相对显示路径（供 edrv.read 与面板展示）。
+ * @author ddj 2026年08月26号
+ * @param value rg 输出路径
+ * @param root 搜索根（subprocess 执行世界）
+ * @returns 相对路径（根外路径原样返回）
+ */
+export function displayPathOf(value: string | undefined, root: string): string {
+  const path = pathText(String(value ?? ''))
+  if (!path) return path
+  const base = pathText(root).replace(/\/$/, '')
+  const lowerPath = path.toLocaleLowerCase('en-US')
+  const lowerBase = base.toLocaleLowerCase('en-US')
+  if (lowerPath === lowerBase) return '.'
+  if (lowerPath.startsWith(lowerBase + '/')) return path.slice(base.length + 1)
+  return path
+}
+
+/**
+ * 解析 rg --json 输出为扁平命中列表（多子匹配展开，列转 UTF-16）。
+ * @author ddj 2026年08月26号
+ * @param text stdout 文本
+ * @param root 搜索根
+ * @returns 命中列表
+ */
+export function parseRgJsonLines(text: string, root: string): ContentMatch[] {
+  const matches: ContentMatch[] = []
+  for (const line of text.split(/\r?\n/)) {
+    const trimmed = line.trim()
+    if (!trimmed) continue
+    let item: RgJsonItem
+    try {
+      item = JSON.parse(trimmed) as RgJsonItem
+    } catch {
+      continue
+    }
+    if (item?.type !== 'match' || !item.data) continue
+    const data = item.data
+    const path = displayPathOf(data.path?.text, root)
+    const lineText = (data.lines?.text ?? '').replace(/\r?\n$/, '').replace(/\r$/, '')
+    if (lineText.length > MAX_RECORD_LINE) continue
+    const lineNumber = data.line_number ?? 0
+    for (const sub of data.submatches ?? []) {
+      matches.push({
+        path,
+        line: lineNumber,
+        startColumn: byteToUtf16Col(lineText, sub.start ?? 0),
+        endColumn: byteToUtf16Col(lineText, sub.end ?? 0),
+        text: lineText,
+      })
+    }
+  }
+  return matches
+}
+
+/**
+ * 命中截断：匹配数/文件数双上限，超限标记 truncated。
+ * @author ddj 2026年08月26号
+ * @param matches 全部命中
+ * @param maxMatches 匹配数上限
+ * @param maxFiles 文件数上限
+ * @returns 截断后的命中与标志
+ */
+export function applyCaps(matches: ContentMatch[], maxMatches: number, maxFiles: number): { matches: ContentMatch[]; truncated: boolean } {
+  const seen = new Set<string>()
+  const out: ContentMatch[] = []
+  let truncated = false
+  for (const match of matches) {
+    if (out.length >= maxMatches) { truncated = true; break }
+    if (!seen.has(match.path)) {
+      if (seen.size >= maxFiles) { truncated = true; break }
+      seen.add(match.path)
+    }
+    out.push(match)
+  }
+  return { matches: out, truncated }
+}
+
+/**
+ * 构造 rg 内容搜索 argv（pattern 放 `--` 后首位置，支持前导 `-` 的查询）。
+ * include/exclude 为用户 glob（gitignore 式，原样透传不转义），
+ * 分别转正选 `--glob` 与排除 `--glob !`。
+ * @author ddj 2026年08月26号
+ * @param binary rg 路径
+ * @param root 搜索根
+ * @param input 搜索输入
+ * @returns argv 数组
+ */
+export function contentArgv(binary: string, root: string, input: ContentSearchInput): string[] {
+  const argv = [binary, '--no-config', '--json', '--line-number', '--no-heading', '--color', 'never', '--hidden', '--no-ignore']
+  if (input.matchCase === true) argv.push('--case-sensitive')
+  if (input.wholeWord === true) argv.push('--word-regexp')
+  if (input.regex !== true) argv.push('--fixed-strings')
+  for (const excluded of EXCLUDES) argv.push('--glob', '!**/' + excluded + '/**')
+  for (const excluded of FILE_EXCLUDES) argv.push('--glob', '!' + excluded)
+  for (const pattern of input.include ?? []) {
+    const glob = String(pattern).trim()
+    if (glob) argv.push('--glob', glob)
+  }
+  for (const pattern of input.exclude ?? []) {
+    const glob = String(pattern).trim()
+    if (glob) argv.push('--glob', '!' + glob)
+  }
+  argv.push('--', input.query, root)
+  return argv
+}
+
+/**
+ * 使用打包 ripgrep 搜索文件内容。
+ * @author ddj 2026年08月26号
+ * @param input provider 输入
+ * @returns 有界内容搜索结果
+ */
+export async function searchRipgrepContent(input: ContentSearchInput): Promise<ContentSearchResult> {
+  const sub = input.ctx.get('subprocess') as { spawn(spec: unknown): { done: Promise<{ exitCode: number | null; code?: number | null }>; collected?: { stdout?: { readFrom(offset: number): { text: string; lossy?: boolean } } } } } | undefined
+  if (!sub) throw new Error('缺少 subprocess')
+  const binary = ripgrepPath()
+  if (!binary) throw new Error('ripgrep 不可用')
+  const root = input.root ?? await searchRoot(input.ctx, input.session)
+  const handle = sub.spawn({
+    argv: contentArgv(binary, root, input),
+    cwd: root,
+    stdio: { stdin: 'ignore', stdout: { maxBytes: STDOUT_CAP }, stderr: { maxBytes: STDERR_CAP } },
+    graceMs: GRACE_MS,
+    signal: input.signal,
+  })
+  let outcome: { exitCode: number | null; code?: number | null }
+  try {
+    outcome = await handle.done
+  } catch (error) {
+    throw new Error('ripgrep 启动失败：' + String(error))
+  }
+  const code = outcome.exitCode ?? outcome.code
+  if (code !== 0 && code !== 1) throw new Error('ripgrep 退出码：' + String(code))
+  const reader = handle.collected?.stdout
+  if (!reader) throw new Error('ripgrep stdout 不可用')
+  const output = reader.readFrom(0)
+  const all = parseRgJsonLines(output.text, root)
+  const maxMatches = Math.max(1, Math.min(input.maxResults ?? DEFAULT_MAX_MATCHES, DEFAULT_MAX_MATCHES))
+  const capped = applyCaps(all, maxMatches, DEFAULT_MAX_FILES)
+  const truncated = Boolean(output.lossy) || capped.truncated
+  return { ...capped, complete: !truncated, source: 'ripgrep' }
+}
+
+/**
+ * 内容搜索编排器：短期缓存 + 同根在途取消 + 会话清理。
+ * @author ddj 2026年08月26号
+ */
+export class ContentSearcher {
+  private readonly cache = new SearchCache<ContentSearchResult>((r) => ({ ...r, matches: [...r.matches] }))
+  private readonly inflight = new Map<string, AbortController>()
+  private readonly rootsByCwd = new Map<string, Set<string>>()
+
+  /**
+   * 创建内容搜索编排器。
+   * @author ddj 2026年08月26号
+   * @param ctx DSH 上下文
+   * @param provider 内容搜索 provider（测试可替换）
+   */
+  constructor(private readonly ctx: Ctx, private readonly provider: ContentSearchProvider = { search: searchRipgrepContent }) {}
+
+  /**
+   * 执行一次工作区内容搜索。
+   * @author ddj 2026年08月26号
+   * @param request 搜索请求
+   * @returns RPC 响应字段（provider 失败抛错，abort 返回空）
+   */
+  async search(request: ContentSearchRequest): Promise<ContentSearchResponse> {
+    const query = String(request.query ?? '').trim()
+    if (query.length < 2) return { matches: [], truncated: false }
+    let root: string
+    try {
+      root = await searchRoot(this.ctx, request.session)
+    } catch {
+      return { matches: [], truncated: false }
+    }
+    const rootKey = pathText(root)
+    const roots = this.rootsByCwd.get(request.cwd) ?? new Set<string>()
+    roots.add(rootKey)
+    this.rootsByCwd.set(request.cwd, roots)
+    const includeKey = (request.include ?? []).join(',')
+    const excludeKey = (request.exclude ?? []).join(',')
+    const key = [rootKey, query, request.matchCase ? 'mc' : '', request.wholeWord ? 'ww' : '', request.regex ? 'rx' : '', includeKey, excludeKey, PROVIDER_VERSION].join('|')
+    const cached = this.cache.get(key)
+    if (cached) return { matches: cached.matches, truncated: cached.truncated }
+    this.inflight.get(root)?.abort()
+    const controller = new AbortController()
+    this.inflight.set(root, controller)
+    try {
+      const result = await this.provider.search({
+        ctx: this.ctx,
+        session: request.session,
+        cwd: request.cwd,
+        query,
+        matchCase: request.matchCase,
+        wholeWord: request.wholeWord,
+        regex: request.regex,
+        maxResults: request.maxResults,
+        include: request.include,
+        exclude: request.exclude,
+        signal: controller.signal,
+        root,
+      })
+      if (controller.signal.aborted) return { matches: [], truncated: false }
+      this.cache.set(key, result)
+      return { matches: result.matches, truncated: result.truncated }
+    } catch (error) {
+      if (controller.signal.aborted) return { matches: [], truncated: false }
+      throw error
+    } finally {
+      if (this.inflight.get(root) === controller) this.inflight.delete(root)
+    }
+  }
+
+  /**
+   * 清理会话对应根目录的缓存和在途搜索。
+   * @author ddj 2026年08月26号
+   * @param cwd 会话工作区
+   */
+  dispose(cwd: string): void {
+    const roots = this.rootsByCwd.get(cwd) ?? new Set<string>([cwd])
+    for (const root of roots) {
+      this.inflight.get(root)?.abort()
+      this.inflight.delete(root)
+      this.cache.clearRoot(root)
+    }
+    this.rootsByCwd.delete(cwd)
+  }
+
+  /**
+   * 清理全部状态。
+   * @author ddj 2026年08月26号
+   */
+  disposeAll(): void {
+    for (const controller of this.inflight.values()) controller.abort()
+    this.inflight.clear()
+    this.rootsByCwd.clear()
+    this.cache.clear()
+  }
+}
+
+/**
+ * 创建默认内容搜索编排器。
+ * @author ddj 2026年08月26号
+ * @param ctx DSH 上下文
+ * @returns 内容搜索编排器
+ */
+export function newContentSearcher(ctx: Ctx): ContentSearcher {
+  return new ContentSearcher(ctx)
+}

package/src/search/fallback.ts

@@ -17,7 +17,7 @@ interface SearchFs {
 }
 
 const RESULT_CAP = 500
-const EXCLUDED = new Set(['node_modules', '.git', '.tmp', '.cache', 'dist', 'build', 'vendor', 'coverage', '__pycache__'])
+const EXCLUDED = new Set(['node_modules', '.git', '.tmp', '.cache', 'dist', 'build', 'vendor', 'coverage', '__pycache__', '.pnpm-store', '.npm-cache', '.codegraph', '.workbuddy', '.dsh-edit-review.json', '.dsh-edit-review-archive.json', '.dsh-edit-review-debug.log'])
 
 /**
  * 在一个目录树中递归查找匹配文件。

package/src/search/orchestrator.ts

@@ -18,17 +18,24 @@ const RESULT_LIMIT = 50
 const PROVIDER_VERSION = 'ripgrep-v1'
 const POLICY_VERSION = 'search-policy-v1'
 
-type CacheEntry = { at: number; result: WorkspaceSearchResult }
+type CacheEntry<T> = { at: number; result: T }
 type SearchRequest = { session: Session; cwd: string; query: string; activePaths: string[] }
 type SearchResponse = { files: string[]; truncated: boolean }
 
 /**
- * 短期有界搜索缓存。
+ * 短期有界搜索缓存（泛型：文件/内容搜索共用；clone 由调用方保证写隔离）。
  * @author ddj 2026年08月24号
- * @returns 缓存对象
+ * @param clone 结果写隔离克隆（缺省恒等）
  */
-export class SearchCache {
-  private readonly entries = new Map<string, CacheEntry>()
+export class SearchCache<T> {
+  private readonly entries = new Map<string, CacheEntry<T>>()
+
+  /**
+   * 创建缓存。
+   * @author ddj 2026年08月24号
+   * @param clone 结果克隆函数（缺省直接引用）
+   */
+  constructor(private readonly clone: (value: T) => T = (value) => value) {}
 
   /**
    * 读取未过期条目。
@@ -37,13 +44,13 @@ export class SearchCache {
    * @param now 当前时间
    * @returns 缓存结果或 undefined
    */
-  get(key: string, now = Date.now()): WorkspaceSearchResult | undefined {
+  get(key: string, now = Date.now()): T | undefined {
     const entry = this.entries.get(key)
     if (!entry) return undefined
     if (now - entry.at >= CACHE_TTL) { this.entries.delete(key); return undefined }
     this.entries.delete(key)
     this.entries.set(key, entry)
-    return { ...entry.result, files: [...entry.result.files] }
+    return this.clone(entry.result)
   }
 
   /**
@@ -53,9 +60,9 @@ export class SearchCache {
    * @param result provider 结果
    * @param now 当前时间
    */
-  set(key: string, result: WorkspaceSearchResult, now = Date.now()): void {
+  set(key: string, result: T, now = Date.now()): void {
     this.entries.delete(key)
-    this.entries.set(key, { at: now, result: { ...result, files: [...result.files] } })
+    this.entries.set(key, { at: now, result: this.clone(result) })
     while (this.entries.size > CACHE_LIMIT) this.entries.delete(this.entries.keys().next().value as string)
   }
 
@@ -145,7 +152,7 @@ function mergeCandidates(result: WorkspaceSearchResult, activePaths: string[], r
  * @author ddj 2026年08月24号
  */
 export class SearchOrchestrator {
-  private readonly cache = new SearchCache()
+  private readonly cache = new SearchCache<WorkspaceSearchResult>((r) => ({ ...r, files: [...r.files] }))
   private readonly inflight = new Map<string, AbortController>()
   private readonly rootsByCwd = new Map<string, Set<string>>()
   private readonly provider: WorkspaceSearchProvider

package/src/search/ripgrep.ts

@@ -14,9 +14,23 @@ import type { WorkspaceSearchInput, WorkspaceSearchProvider, WorkspaceSearchResu
 const STDOUT_CAP = 4 << 20
 const STDERR_CAP = 64 << 10
 const GRACE_MS = 20_000
-const EXCLUDES = [
+
+/** 默认排除目录（文件/内容搜索共用，导出供 provider 复用）。 */
+export const EXCLUDES = [
   'node_modules', '.git', '.tmp', '.cache', 'dist', 'build', 'vendor',
-  'coverage', '__pycache__',
+  'coverage', '__pycache__', '.pnpm-store', '.npm-cache', '.codegraph', '.workbuddy',
+]
+
+/**
+ * 默认排除文件（glob 模式，文件/内容搜索共用）。
+ * 插件自身 sidecar（内容搜索会命中其历史编辑文本，且单行巨型 JSON 会撑爆输出上限）；
+ * *.map 源映射（纯构建产物，内容搜索无意义且常为单行巨文件）。
+ */
+export const FILE_EXCLUDES = [
+  '**/.dsh-edit-review.json',
+  '**/.dsh-edit-review-archive.json',
+  '**/.dsh-edit-review-debug.log',
+  '**/*.map',
 ]
 
 interface SearchHandle {
@@ -113,6 +127,7 @@ export async function searchRipgrep(input: WorkspaceSearchInput): Promise<Worksp
   const query = prepareQuery(input.query)
   const argv = [binary, '--no-config', '--files', '--hidden', '--no-ignore', '--glob-case-insensitive', '--glob', queryGlob(query.text)]
   for (const excluded of EXCLUDES) argv.push('--glob', '!**/' + excluded + '/**')
+  for (const excluded of FILE_EXCLUDES) argv.push('--glob', '!' + excluded)
   argv.push('--', root)
   const handle = sub.spawn({
     argv,

package/src/search/types.ts

@@ -45,3 +45,43 @@ export interface SearchCandidate {
 export interface CandidateRanker {
   rank(candidates: SearchCandidate[], query: PreparedQuery): SearchCandidate[]
 }
+
+/** 内容搜索单条命中（列 = 1-based UTF-16，直接供 Monaco 跳转/高亮）。 */
+export interface ContentMatch {
+  path: string
+  line: number
+  startColumn: number
+  endColumn: number
+  text: string
+}
+
+/** 内容搜索输入（options 缺省 = 大小写不敏感、字面匹配、非全词；include/exclude 为 rg glob）。 */
+export interface ContentSearchInput {
+  ctx: Ctx
+  session: Session
+  cwd: string
+  query: string
+  matchCase?: boolean
+  wholeWord?: boolean
+  regex?: boolean
+  maxResults?: number
+  /** 正选 glob（仅在这些文件内搜索；逗号已在客户端拆分）。 */
+  include?: string[]
+  /** 排除 glob（这些文件不参与搜索）。 */
+  exclude?: string[]
+  signal?: AbortSignal
+  root?: string
+}
+
+/** 内容搜索输出：扁平命中列表 + 截断标志。 */
+export interface ContentSearchResult {
+  matches: ContentMatch[]
+  truncated: boolean
+  complete: boolean
+  source: SearchSource
+}
+
+/** 内容搜索 provider（rg JSON 主路径；无 fallback，失败抛错由调用方处理）。 */
+export interface ContentSearchProvider {
+  search(input: ContentSearchInput): Promise<ContentSearchResult>
+}

package/src/shared/keybindings.ts

@@ -0,0 +1,39 @@
+/**
+ * dsh-vscode-mode shared — 快捷键命令目录与默认键位（双面契约）。
+ * 纯数据模块：host（settings schema 默认值）与 client（执行匹配/设置页）共用。
+ * 键位格式：修饰符 + 主键，`+` 连接，如 `Ctrl+Shift+F`；空串 = 未绑定。
+ * 作者 ddj 2026年08月26号
+ */
+
+/** 命令 id → 默认键位。命令目录以此为准，新增命令只需加一项。 */
+export const KEYBINDING_DEFAULTS: Record<string, string> = {
+  'edrv.save': 'Ctrl+S',
+  'edrv.quickOpen': 'Ctrl+P',
+  'edrv.toggleSidebar': 'Ctrl+B',
+  'edrv.searchInFiles': 'Ctrl+Shift+F',
+}
+
+/**
+ * 返回默认键位的独立副本（调用方修改不污染目录）。
+ * @author ddj 2026年08月26号
+ * @returns 默认键位映射
+ */
+export function defaultKeybindings(): Record<string, string> {
+  return { ...KEYBINDING_DEFAULTS }
+}
+
+/**
+ * 规整外部键位数据：只保留已知命令 id 且为字符串的值，未知 id 丢弃。
+ * 防止用户文档/旧版本写入的未知命令污染执行匹配。
+ * @author ddj 2026年08月26号
+ * @param raw 原始设置值
+ * @returns 规整后的键位映射（缺省项不补默认值，由调用方按需合并）
+ */
+export function normalizeKeybindings(raw: unknown): Record<string, string> {
+  const out: Record<string, string> = {}
+  if (!raw || typeof raw !== 'object') return out
+  for (const [id, chord] of Object.entries(raw as Record<string, unknown>)) {
+    if (id in KEYBINDING_DEFAULTS && typeof chord === 'string') out[id] = chord
+  }
+  return out
+}

package/src/shared/rpc.ts

@@ -29,6 +29,15 @@ export interface TreeEntry {
   size?: number
 }
 
+/** 内容搜索单条命中（edrv.searchContent 返回；path 相对工作区根，列为 1-based UTF-16）。 */
+export interface SearchContentMatch {
+  path: string
+  line: number
+  startColumn: number
+  endColumn: number
+  text: string
+}
+
 /** 一条批量决策项（decideBatch 的 items 元素，与 accept/reject 单条参数同构）。 */
 export interface DecideItem {
   callId: string
@@ -59,6 +68,7 @@ export interface RpcRequestMap {
   'edrv.rollback': { sessionId?: string; path: string; batch?: number }
   'edrv.debug': { sessionId?: string; text: string }
   'edrv.searchFiles': { sessionId?: string; query: string }
+  'edrv.searchContent': { sessionId?: string; query: string; matchCase?: boolean; wholeWord?: boolean; regex?: boolean; maxResults?: number; include?: string[]; exclude?: string[] }
   'edrv.listDir': { sessionId?: string; path: string }
   'edrv.revealInExplorer': { sessionId?: string; path: string }
   'mcp.list': {}
@@ -94,6 +104,7 @@ export interface RpcOkMap {
   'edrv.rollback': { path: string; batch: number | null }
   'edrv.debug': object
   'edrv.searchFiles': { files: string[]; truncated: boolean }
+  'edrv.searchContent': { matches: SearchContentMatch[]; truncated: boolean }
   'edrv.listDir': { root: string; path: string; entries: TreeEntry[] }
   'edrv.revealInExplorer': { revealed: string }
   'mcp.list': { servers: MpcServer[] }