dsh-vscode-mode 0.1.30 → 0.1.35

package/src/lsp/providers.ts

@@ -0,0 +1,235 @@
+/**
+ * dsh-vscode-mode host — 语言服务器 provider 解析。
+ * 优先级：手动配置 > extmgr 扩展源(Phase 2 挂载) > 自动发现(~/.vscode/extensions、
+ * ~/.dsh/dsh-vscode-mode/extensions、PATH)。产物 LspProviderSpec：{ ready, argv, cwd, reason }，
+ * 供 manager 启动服务器。
+ * 作者 ddj 2026-08-27
+ */
+import { existsSync, readdirSync, type Dirent } from 'node:fs'
+import { homedir } from 'node:os'
+import { join, sep } from 'node:path'
+import { isAbsolute } from 'node:path'
+import { extensionsRoot } from './extmgr.js'
+import { dshHome } from '../paths.js'
+
+/** provider 来源标记（与 shared/lsp.ts 的 status.source 一致）。 */
+export type LspProviderKind = 'extension' | 'discover' | 'manual' | 'none'
+
+export interface LspProviderSpec {
+  languageId: string
+  kind: LspProviderKind
+  argv: string[]
+  cwd?: string
+  ready: boolean
+  reason?: string
+}
+
+/** 语言 → 扩展名（用于判断哪些文件触发该语言服务器）。 */
+export const LSP_LANG_EXT: Record<string, string[]> = {
+  lua: ['lua'],
+  csharp: ['cs', 'csx'],
+}
+
+/** 平台目录名（LuaLS/OmniSharp 的 server/bin/<平台>）。 */
+export function platformServerDir(platform = process.platform): string {
+  if (platform === 'win32') return 'Windows'
+  if (platform === 'darwin') return 'macOS'
+  return 'Linux'
+}
+
+/** exe 后缀（Windows 为 .exe）。 */
+export function exeSuffix(platform = process.platform): string {
+  return platform === 'win32' ? '.exe' : ''
+}
+
+/** 列出目录下匹配前缀的子目录（如 ~/.vscode/extensions/sumneko.lua-*）。 */
+export function listMatchingDirs(dir: string, prefix: string): string[] {
+  if (!existsSync(dir)) return []
+  let entries: Dirent[]
+  try {
+    entries = readdirSync(dir, { withFileTypes: true })
+  } catch (error) {
+    return []
+  }
+  return entries.filter((e) => e.isDirectory() && e.name.startsWith(prefix)).map((e) => join(dir, e.name))
+}
+
+/** VSCode 扩展目录（~/.vscode/extensions、~/.vscode-server/extensions + extmgr 目录；无则空数组）。 */
+export function vscodeExtensionsDirs(home = dshHome()): string[] {
+  const userHome = homedir()
+  return [
+    join(userHome, '.vscode', 'extensions'),
+    join(userHome, '.vscode-server', 'extensions'),
+    extensionsRoot(home),
+  ].filter((dir) => existsSync(dir))
+}
+
+// --region 扩展源 provider 挂载点（extmgr 安装扩展后注册；kind=extension）
+/** 扩展源 provider 注册表：languageId → 启动参数。 */
+const extProviders = new Map<string, { argv: string[]; cwd?: string; version?: string }>()
+
+/**
+ * 注册扩展源 provider（extmgr 安装的 VSIX 扩展调用；覆盖/清除传空 argv 可注销）。
+ * @author ddj 2026年08月27号
+ * @param languageId 语言 id
+ * @param argv 启动参数（argv[0] 为可执行）
+ * @param cwd 工作目录
+ * @param version 扩展版本（诊断展示）
+ */
+export function registerExtensionProvider(languageId: string, argv: string[], cwd?: string, version?: string): void {
+  if (!argv.length) {
+    extProviders.delete(languageId)
+    return
+  }
+  extProviders.set(languageId, { argv, cwd, version })
+}
+
+/** 取扩展源 provider spec（未注册返回 null）。 */
+export function extensionProviderFor(languageId: string): LspProviderSpec | null {
+  const p = extProviders.get(languageId)
+  if (!p || !p.argv.length) return null
+  return { languageId, kind: 'extension', argv: p.argv, cwd: p.cwd, ready: true }
+}
+
+/** 清空全部扩展源注册（插件卸载/扩展目录重建时）。 */
+export function clearExtensionProviders(): void {
+  extProviders.clear()
+}
+// --endregion
+
+/** PATH 中查找可执行文件（不含扩展名校验，交由 spawn 报错）。 */
+export function findInPath(name: string): string | null {
+  const pathEnv = process.env.PATH ?? ''
+  const exts = process.platform === 'win32' ? ['', '.exe', '.cmd', '.bat'] : ['']
+  for (const dir of pathEnv.split(sep === '\\' ? ';' : ':')) {
+    if (!dir) continue
+    for (const ext of exts) {
+      const candidate = join(dir, name + ext)
+      if (existsSync(candidate)) return candidate
+    }
+  }
+  return null
+}
+
+/** 查找 dotnet（PATH + 常见安装路径）。 */
+export function findDotnet(): string | null {
+  const inPath = findInPath('dotnet')
+  if (inPath) return inPath
+  const candidates = [
+    'C:/Program Files/dotnet/dotnet.exe',
+    'C:/Program Files (x86)/dotnet/dotnet.exe',
+  ]
+  for (const c of candidates) if (existsSync(c)) return c
+  return null
+}
+
+/** LuaLS 自动发现候选（VSCode 扩展目录 + extmgr 目录下 sumneko.lua-* 的 server/bin 可执行）。 */
+export function candidateLuaServers(home = dshHome()): string[] {
+  const plat = platformServerDir()
+  const suffix = exeSuffix()
+  const out: string[] = []
+  for (const dir of vscodeExtensionsDirs(home)) {
+    for (const extDir of listMatchingDirs(dir, 'sumneko.lua-')) {
+      const binDir = join(extDir, 'server', 'bin')
+      const binName = 'lua-language-server' + suffix
+      // 新布局（3.x 平台变体 vsix）：server/bin/xxx.exe；旧布局：server/bin/<平台>/xxx.exe
+      const bin = existsSync(join(binDir, binName)) ? join(binDir, binName) : join(binDir, plat, binName)
+      if (existsSync(bin) && !out.includes(bin)) out.push(bin)
+    }
+  }
+  return out
+}
+
+/** C# 自动发现候选：Roslyn dll（需 dotnet）或 OmniSharp 可执行。 */
+export function candidateCSharpServers(home = dshHome()): { roslynDll?: string; omnisharpExe?: string; dotnet?: string } {
+  const out: { roslynDll?: string; omnisharpExe?: string; dotnet?: string } = {}
+  for (const dir of vscodeExtensionsDirs(home)) {
+    for (const extDir of listMatchingDirs(dir, 'ms-dotnettools.csharp-')) {
+      const roslyn = join(extDir, '.roslyn', 'Microsoft.CodeAnalysis.LanguageServer.dll')
+      if (!out.roslynDll && existsSync(roslyn)) out.roslynDll = roslyn
+      const omni = join(extDir, '.omnisharp', 'OmniSharp' + exeSuffix())
+      if (!out.omnisharpExe && existsSync(omni)) out.omnisharpExe = omni
+    }
+  }
+  const dotnet = findDotnet()
+  if (dotnet) out.dotnet = dotnet
+  return out
+}
+
+/**
+ * 手动配置归一化：path 指向可执行文件/目录，command 为可执行名（PATH 解析）。
+ * 未配置时返回空 reason（由各 resolve 兜底输出语言相关安装指引文案）。
+ * @author ddj 2026年08月27号
+ */
+export function manualSpec(manual: { command?: string; path?: string } | undefined): { argv: string[]; cwd?: string; ready: boolean; reason?: string } {
+  if (!manual) return { argv: [], ready: false }
+  const pathValue = typeof manual.path === 'string' && manual.path.trim() ? manual.path.trim() : ''
+  const commandValue = typeof manual.command === 'string' && manual.command.trim() ? manual.command.trim() : ''
+  if (!pathValue && !commandValue) return { argv: [], ready: false }
+  if (pathValue) {
+    if (!existsSync(pathValue)) return { argv: [], ready: false, reason: '路径不存在：' + pathValue }
+    return { argv: [pathValue], ready: true }
+  }
+  const found = findInPath(commandValue)
+  if (!found) return { argv: [], ready: false, reason: 'PATH 中找不到命令：' + commandValue }
+  return { argv: [found], ready: true }
+}
+
+/**
+ * 解析 Lua provider（优先级：手动 > 扩展源 > 自动发现）。
+ * @author ddj 2026年08月27号
+ * @param manual 手动配置
+ * @param home DSH home（缺省真实；测试可注入临时目录）
+ * @returns provider 规格
+ */
+export function resolveLuaProvider(manual?: { command?: string; path?: string }, home = dshHome()): LspProviderSpec {
+  const base = { languageId: 'lua' }
+  const manualResolved = manualSpec(manual)
+  if (manualResolved.ready) return { ...base, kind: 'manual', argv: manualResolved.argv, cwd: manualResolved.cwd, ready: true }
+  const ext = extensionProviderFor('lua')
+  if (ext) return ext
+  const candidates = candidateLuaServers(home)
+  if (candidates.length) return { ...base, kind: 'discover', argv: [candidates[0]], cwd: undefined, ready: true }
+  const inPath = findInPath('lua-language-server' + exeSuffix())
+  if (inPath) return { ...base, kind: 'discover', argv: [inPath], ready: true }
+  return { ...base, kind: 'none', argv: [], ready: false, reason: manualResolved.reason ?? '未发现 lua-language-server（可安装 LuaLS 或手动指定路径）' }
+}
+
+/**
+ * 解析 C# provider（优先级：手动 > 扩展源 > 自动发现；需 dotnet + Roslyn dll；或 OmniSharp）。
+ * @author ddj 2026年08月27号
+ * @param manual 手动配置
+ * @param home DSH home（缺省真实；测试可注入临时目录）
+ * @returns provider 规格
+ */
+export function resolveCSharpProvider(manual?: { command?: string; path?: string }, home = dshHome()): LspProviderSpec {
+  const base = { languageId: 'csharp' }
+  const manualResolved = manualSpec(manual)
+  if (manualResolved.ready) return { ...base, kind: 'manual', argv: manualResolved.argv, cwd: manualResolved.cwd, ready: true }
+  const ext = extensionProviderFor('csharp')
+  if (ext) return ext
+  const discovered = candidateCSharpServers(home)
+  if (discovered.roslynDll && discovered.dotnet) {
+    return { ...base, kind: 'discover', argv: [discovered.dotnet, discovered.roslynDll, '--stdio'], ready: true }
+  }
+  if (discovered.omnisharpExe) {
+    return { ...base, kind: 'discover', argv: [discovered.omnisharpExe, '--stdio'], ready: true }
+  }
+  const reasons: string[] = []
+  if (!discovered.dotnet) reasons.push('未找到 dotnet')
+  if (!discovered.roslynDll && !discovered.omnisharpExe) reasons.push('未发现 ms-dotnettools.csharp 扩展')
+  return { ...base, kind: 'none', argv: [], ready: false, reason: reasons.join('；') + '（可手动指定 Roslyn/OmniSharp 入口）' }
+}
+
+/** 语言 → provider 解析器（可扩展；Phase 2 由 extmgr 注册扩展源 provider）。 */
+export type ProviderResolver = (manual?: { command?: string; path?: string }) => LspProviderSpec
+
+export const PROVIDER_RESOLVERS: Record<string, ProviderResolver> = {
+  lua: resolveLuaProvider,
+  csharp: resolveCSharpProvider,
+}
+
+/** 手动配置是否为绝对路径可执行（供设置页判断输入类型）。 */
+export function looksAbsolute(value: string): boolean {
+  return isAbsolute(value)
+}

package/src/lsp/rpc.ts

@@ -0,0 +1,334 @@
+/**
+ * dsh-vscode-mode host — edrv.lsp.* RPC handlers。
+ * 文档跟踪（openDocs 集合 + root|lang 引用计数）→ 懒启动服务器、用完释放。
+ * 作者 ddj 2026-08-27
+ */
+import type { Ctx } from '../store.js'
+import { searchRoot } from '../search/ripgrep.js'
+import { sessionOf, cwdOf } from '../registry.js'
+import { resolveProviderSpec, langOfPath, configFromPlugin, configFromSettings, LSP_SETTINGS_NS, type LspConfig } from './config.js'
+import type { LspManager } from './manager.js'
+import type { RpcHandlerMap } from '../shared/rpc.js'
+import type { LspLocation } from '../shared/lsp.js'
+import { installFromMarket, installVsixFile, listInstalled, marketGet, marketSearch, uninstall, updateExtension, type ExtInfo } from './extmgr.js'
+
+export interface LspRpcDeps {
+  ctx: Ctx
+  pluginConfig: unknown
+  manager: LspManager
+}
+
+/**
+ * 打开文档跟踪：openDocs(docKey→sessionId) + rootLangRefs 计数 + sessionAcquired 归属。
+ * 保证 manager 的 acquire/release 一一配对（多会话共享同一 root|lang 也平衡）。
+ */
+function createDocTracker() {
+  const openDocs = new Map<string, Set<string>>()
+  const rootLangRefs = new Map<string, number>()
+  const sessionDocs = new Map<string, Set<string>>()
+  const keyOf = (root: string, lang: string, path: string): string => root + '|' + lang + '|' + path
+  const rootLangOf = (docKey: string): string => docKey.slice(0, docKey.lastIndexOf('|'))
+  const rememberDoc = (sessionId: string, docKey: string): void => {
+    const docs = sessionDocs.get(sessionId) ?? new Set<string>()
+    docs.add(docKey)
+    sessionDocs.set(sessionId, docs)
+  }
+  const forgetDoc = (sessionId: string, docKey: string): void => {
+    const docs = sessionDocs.get(sessionId)
+    docs?.delete(docKey)
+    if (docs?.size === 0) sessionDocs.delete(sessionId)
+  }
+  const releaseRef = (rl: string): boolean => {
+    const next = (rootLangRefs.get(rl) ?? 1) - 1
+    if (next <= 0) {
+      rootLangRefs.delete(rl)
+      return true
+    }
+    rootLangRefs.set(rl, next)
+    return false
+  }
+  return {
+    /** 记录文档打开；同一文档可由多个会话打开，但每个会话只计一次。 */
+    open(root: string, lang: string, path: string, sessionId: string): boolean {
+      const docKey = keyOf(root, lang, path)
+      const owners = openDocs.get(docKey) ?? new Set<string>()
+      if (owners.has(sessionId)) return false
+      owners.add(sessionId)
+      openDocs.set(docKey, owners)
+      rememberDoc(sessionId, docKey)
+      const rl = root + '|' + lang
+      const count = rootLangRefs.get(rl) ?? 0
+      rootLangRefs.set(rl, count + 1)
+      return count === 0
+    },
+    /** 记录文档关闭；返回 true 表示 root|lang 已无文档（应 release）。 */
+    close(root: string, lang: string, path: string, sessionId: string): boolean {
+      const docKey = keyOf(root, lang, path)
+      const owners = openDocs.get(docKey)
+      if (!owners?.has(sessionId)) return false
+      owners.delete(sessionId)
+      if (owners.size === 0) openDocs.delete(docKey)
+      forgetDoc(sessionId, docKey)
+      return releaseRef(root + '|' + lang)
+    },
+    /** 关闭某会话全部文档；返回全局引用归零、需要 release 的 root|lang。 */
+    closeSession(sessionId: string): string[] {
+      const docs = sessionDocs.get(sessionId)
+      if (!docs) return []
+      const toRelease = new Set<string>()
+      for (const docKey of docs) {
+        const owners = openDocs.get(docKey)
+        owners?.delete(sessionId)
+        if (owners?.size === 0) openDocs.delete(docKey)
+        if (releaseRef(rootLangOf(docKey))) toRelease.add(rootLangOf(docKey))
+      }
+      sessionDocs.delete(sessionId)
+      return [...toRelease]
+    },
+  }
+}
+
+/**
+ * 构造 edrv.lsp.* handlers（一次性创建，tracker 状态跨请求保留）。
+ * @author ddj 2026年08月27号
+ * @param deps 依赖（ctx/pluginConfig/manager）
+ * @returns { handlers, disposeSession } handlers 为局部 map，disposeSession 供 session/disposed 清理
+ */
+export function createLspRpc(deps: LspRpcDeps): { handlers: Partial<RpcHandlerMap>; disposeSession: (sessionId: string) => void } {
+  const { ctx, pluginConfig, manager } = deps
+  const tracker = createDocTracker()
+
+  /** 解析会话与工作区根（失败 → null + error）。 */
+  const rootOf = async (sessionId: string | undefined): Promise<{ root: string; sessionId: string } | { err: string }> => {
+    const session = sessionOf(ctx, sessionId)
+    if (!session) return { err: '会话不存在' }
+    const cwd = cwdOf(session)
+    if (!cwd) return { err: '会话无工作区' }
+    try {
+      const root = await searchRoot(ctx, session)
+      const sid = typeof session.id === 'string' && session.id ? session.id : sessionId ?? ''
+      return { root, sessionId: sid }
+    } catch (error) {
+      return { err: String(error) }
+    }
+  }
+
+  const acquireServer = (root: string, lang: string) =>
+    manager.acquire(root, lang, (languageId: string) => resolveProviderSpec(ctx, pluginConfig, languageId))
+
+  const serverOf = (root: string, lang: string) =>
+    manager.peek(root, lang)
+
+  const rootLocations = (locations: LspLocation[], root: string): LspLocation[] =>
+    locations.map((location) => ({ ...location, root }))
+
+  const disposeSession = (sessionId: string): void => {
+    for (const rl of tracker.closeSession(sessionId)) {
+      const sep = rl.lastIndexOf('|')
+      manager.release(rl.slice(0, sep), rl.slice(sep + 1))
+    }
+  }
+
+  const handlers: Partial<RpcHandlerMap> = {
+    'edrv.lsp.status': async (args) => {
+      if (!args.sessionId) return { ok: true, servers: manager.statusAll() }
+      const sc = await rootOf(args.sessionId)
+      if ('err' in sc) return { ok: false, error: sc.err }
+      return { ok: true, servers: manager.statusAll().filter((status) => status.root === sc.root) }
+    },
+
+    'edrv.lsp.configGet': async () => {
+      const settings = configFromSettings(ctx)
+      const plugin = configFromPlugin(pluginConfig)
+      const merged: LspConfig = { ...plugin }
+      for (const lang of Object.keys(settings)) merged[lang] = { ...plugin[lang], ...settings[lang] }
+      return { ok: true, config: merged as unknown as Record<string, unknown> }
+    },
+
+    'edrv.lsp.configUpdate': async (args) => {
+      const lang = args.languageId
+      if (!lang || !resolveProviderSpec(ctx, pluginConfig, lang).languageId) {
+        return { ok: false, error: '不支持的语言：' + lang }
+      }
+      const current = configFromSettings(ctx)
+      const next: LspConfig = { ...current, [lang]: { ...(current[lang] ?? {}) } }
+      const target = next[lang]!
+      if (args.enabled !== undefined) target.enabled = args.enabled
+      if (args.command !== undefined) target.command = args.command || undefined
+      if (args.path !== undefined) target.path = args.path || undefined
+      try {
+        const settings = ctx.get('settings')
+        if (!settings?.update) return { ok: false, error: '设置服务不可用' }
+        await settings.update(LSP_SETTINGS_NS, { [lang]: target })
+        const descriptor = settings.describe?.({ redactSecrets: true })?.find((item: { ns?: string }) => item.ns === LSP_SETTINGS_NS)
+        const stored = descriptor?.value as Record<string, unknown> | undefined
+        const merged: LspConfig = { ...configFromPlugin(pluginConfig), ...(stored as LspConfig) }
+        return { ok: true, config: merged as unknown as Record<string, unknown> }
+      } catch (error) {
+        return { ok: false, error: '配置保存失败：' + String(error) }
+      }
+    },
+
+    'edrv.lsp.sync': async (args) => {
+      const lang = langOfPath(args.path)
+      if (!lang) return { ok: true }
+      const sc = await rootOf(args.sessionId)
+      if ('err' in sc) return { ok: false, error: sc.err }
+      const first = tracker.open(sc.root, lang, args.path, sc.sessionId)
+      const server = first ? acquireServer(sc.root, lang) : serverOf(sc.root, lang)
+      if (!server) return { ok: false, error: '语言服务器未注册' }
+      server.sync(args.path, args.text, args.version)
+      return { ok: true }
+    },
+
+    'edrv.lsp.close': async (args) => {
+      const lang = langOfPath(args.path)
+      if (!lang) return { ok: true }
+      const sc = await rootOf(args.sessionId)
+      if ('err' in sc) return { ok: true }
+      if (tracker.close(sc.root, lang, args.path, sc.sessionId)) {
+        manager.release(sc.root, lang)
+      }
+      return { ok: true }
+    },
+
+    'edrv.lsp.definition': async (args) => {
+      const lang = langOfPath(args.path)
+      if (!lang) return { ok: true, locations: [] }
+      const sc = await rootOf(args.sessionId)
+      if ('err' in sc) return { ok: false, error: sc.err }
+      try {
+        const server = serverOf(sc.root, lang)
+        if (!server) return { ok: true, locations: [] }
+        const locations = await server.definition(args.path, args.position.line, args.position.character)
+        const rooted = rootLocations(locations, sc.root)
+        const truncated = rooted.length > 500
+        return { ok: true, locations: truncated ? rooted.slice(0, 500) : rooted, truncated: truncated || undefined }
+      } catch (error) {
+        return { ok: false, error: 'LSP 定义查询失败：' + String(error) }
+      }
+    },
+
+    'edrv.lsp.references': async (args) => {
+      const lang = langOfPath(args.path)
+      if (!lang) return { ok: true, locations: [] }
+      const sc = await rootOf(args.sessionId)
+      if ('err' in sc) return { ok: false, error: sc.err }
+      try {
+        const server = serverOf(sc.root, lang)
+        if (!server) return { ok: true, locations: [] }
+        const locations = await server.references(args.path, args.position.line, args.position.character, args.includeDeclaration !== false)
+        const rooted = rootLocations(locations, sc.root)
+        const truncated = rooted.length > 500
+        return { ok: true, locations: truncated ? rooted.slice(0, 500) : rooted, truncated: truncated || undefined }
+      } catch (error) {
+        return { ok: false, error: 'LSP 引用查询失败：' + String(error) }
+      }
+    },
+
+    'edrv.lsp.documentSymbol': async (args) => {
+      const lang = langOfPath(args.path)
+      if (!lang) return { ok: true, symbols: [] }
+      const sc = await rootOf(args.sessionId)
+      if ('err' in sc) return { ok: false, error: sc.err }
+      try {
+        const server = serverOf(sc.root, lang)
+        if (!server) return { ok: true, symbols: [] }
+        const symbols = await server.documentSymbol(args.path)
+        return { ok: true, symbols }
+      } catch (error) {
+        return { ok: false, error: 'LSP 大纲查询失败：' + String(error) }
+      }
+    },
+
+    'edrv.lsp.workspaceSymbol': async (args) => {
+      const sc = await rootOf(args.sessionId)
+      if ('err' in sc) return { ok: false, error: sc.err }
+      const out: { symbols: never[]; truncated?: boolean } = { symbols: [] }
+      for (const lang of ['lua', 'csharp']) {
+        const existing = serverOf(sc.root, lang)
+        if (!existing) continue
+        try {
+          const symbols = await existing.workspaceSymbol(args.query)
+          if (symbols.length) out.symbols.push(...(symbols as never[]))
+        } catch (error) {
+          /* 单语言失败忽略 */
+        }
+      }
+      const truncated = out.symbols.length > 500
+      return { ok: true, symbols: out.symbols.slice(0, 500), truncated: truncated || undefined }
+    },
+
+    'edrv.lsp.hover': async (args) => {
+      const lang = langOfPath(args.path)
+      if (!lang) return { ok: true, hover: undefined }
+      const sc = await rootOf(args.sessionId)
+      if ('err' in sc) return { ok: false, error: sc.err }
+      try {
+        const server = serverOf(sc.root, lang)
+        if (!server) return { ok: true, hover: undefined }
+        const hover = await server.hover(args.path, args.position.line, args.position.character)
+        return { ok: true, hover: hover ?? undefined }
+      } catch (error) {
+        return { ok: false, error: 'LSP hover 查询失败：' + String(error) }
+      }
+    },
+
+    // --region 扩展管理（edrv.lsp.ext.*：已装/市场/安装/卸载/更新）
+    'edrv.lsp.ext.list': async () => ({ ok: true, extensions: listInstalled() }),
+
+    'edrv.lsp.ext.market': async (args) => {
+      try {
+        const items = await marketSearch(args.query, args.size ?? 12)
+        return { ok: true, extensions: items }
+      } catch (error) {
+        return { ok: false, error: '市场搜索失败：' + String(error) }
+      }
+    },
+
+    'edrv.lsp.ext.install': async (args) => {
+      try {
+        let info: ExtInfo
+        if (args.vsixPath) info = await installVsixFile(args.vsixPath)
+        else if (args.namespace && args.name) info = await installFromMarket(args.namespace, args.name, args.version)
+        else return { ok: false, error: '缺少安装源（vsixPath 或 namespace+name）' }
+        return { ok: true, extension: info }
+      } catch (error) {
+        return { ok: false, error: '安装失败：' + String(error) }
+      }
+    },
+
+    'edrv.lsp.ext.uninstall': async (args) => {
+      return uninstall(args.id) ? { ok: true } : { ok: false, error: '扩展未安装：' + args.id }
+    },
+
+    'edrv.lsp.ext.update': async (args) => {
+      try {
+        const updated = await updateExtension(args.id)
+        if (!updated) {
+          const current = listInstalled().find((e) => e.id === args.id)
+          return current ? { ok: true, extension: current, updated: false } : { ok: false, error: '扩展未安装：' + args.id }
+        }
+        return { ok: true, extension: updated, updated: true }
+      } catch (error) {
+        return { ok: false, error: '更新失败：' + String(error) }
+      }
+    },
+
+    'edrv.lsp.ext.updates': async () => {
+      try {
+        const updates = []
+        for (const ext of listInstalled()) {
+          const meta = await marketGet(ext.namespace, ext.name)
+          const latest = String(meta?.version ?? '')
+          if (latest && latest !== ext.version) updates.push({ id: ext.id, current: ext.version, latest, displayName: ext.displayName })
+        }
+        return { ok: true, updates }
+      } catch (error) {
+        return { ok: false, error: '检查更新失败：' + String(error) }
+      }
+    },
+    // --endregion
+  }
+  return { handlers, disposeSession }
+}