@ashx-j/lunr 0.2.17 → 0.2.19

package/dist/builtin-extensions/pi-lsp-extension/src/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/builtin-extensions/pi-lsp-extension/src/index.ts"],"names":[],"mappings":"AACA;;;;;;;;;;;;;;;;;;;;;GAqBG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AA8EpE,MAAM,CAAC,OAAO,UAAU,YAAY,CAAC,EAAE,EAAE,YAAY,QAyfpD","sourcesContent":["// @ts-nocheck\n/**\n * pi-lsp-extension — Pi coding agent extension for LSP integration.\n *\n * Exposes Language Server Protocol capabilities as tools the LLM can call:\n * - lsp_diagnostics: compilation errors and warnings\n * - lsp_hover: type info and docs at a position\n * - lsp_definition: go to definition\n * - lsp_references: find all references\n * - lsp_symbols: file/workspace symbol search\n * - lsp_rename: preview rename refactoring\n * - lsp_completions: code completion suggestions at a position\n * - lsp_code_actions: quick fixes, refactorings, and source actions\n *\n * Position-based tools (hover, definition, references, rename, code_actions)\n * accept an optional `query` parameter as an alternative to line/character,\n * resolving a symbol name to its position automatically.\n *\n * Usage:\n *   1. npm install in this directory\n *   2. Add to pi via settings.json extensions, or: pi -e ./src/index.ts\n *   3. LSP servers start lazily when you first use a tool on a file\n */\n\nimport type { ExtensionAPI } from \"@ashx-j/lunr\";\nimport {\n  isReadToolResult,\n  isWriteToolResult,\n  isEditToolResult,\n} from \"@ashx-j/lunr\";\nimport { DiagnosticSeverity, type Diagnostic } from \"vscode-languageserver-protocol\";\n\nimport { LspManager, type ServerConfig, type LspManagerCallbacks } from \"./lsp-manager.js\";\nimport { FileSync } from \"./file-sync.js\";\nimport { TreeSitterManager } from \"./tree-sitter/parser-manager.js\";\nimport { WorkspaceIndex } from \"./tree-sitter/workspace-index.js\";\nimport { type WorkspaceProvider, DefaultWorkspaceProvider } from \"./workspace-provider.js\";\nimport { createDiagnosticsTool } from \"./tools/diagnostics.js\";\nimport { createHoverTool } from \"./tools/hover.js\";\nimport { createDefinitionTool } from \"./tools/definition.js\";\nimport { createReferencesTool } from \"./tools/references.js\";\nimport { createSymbolsTool } from \"./tools/symbols.js\";\nimport { createRenameTool } from \"./tools/rename.js\";\nimport { createCodeOverviewTool } from \"./tools/code-overview.js\";\nimport { createCompletionsTool } from \"./tools/completions.js\";\nimport { createCodeSearchTool } from \"./tools/code-search.js\";\nimport { createCodeRewriteTool } from \"./tools/code-rewrite.js\";\nimport { createCodeActionsTool } from \"./tools/code-actions.js\";\nimport { syntheticDotLocks } from \"./tools/completions.js\";\nimport { relative } from \"node:path\";\nimport { existsSync, readFileSync } from \"node:fs\";\nimport { join } from \"node:path\";\nimport { DIAGNOSTIC_SETTLE_DELAY_MS } from \"./shared/timing.js\";\n\n/**\n * Project-level LSP config — loaded from `.pi-lsp.json` in the workspace root.\n *\n * Example:\n * ```json\n * {\n *   \"autoStart\": [\"java\", \"typescript\"],\n *   \"lombokJar\": \"env/Lombok-1.18.x/runtime/lib/lombok-1.18.42.jar\",\n *   \"servers\": {\n *     \"python\": { \"command\": \"pylsp\", \"args\": [] }\n *   }\n * }\n * ```\n */\ninterface ProjectLspConfig {\n  /** Languages to start eagerly on session_start (e.g. [\"java\", \"typescript\"]) */\n  autoStart?: string[];\n  /** Path to Lombok jar (absolute or relative to project root). \"auto\" to auto-detect. */\n  lombokJar?: string;\n  /** Custom server configs keyed by language ID */\n  servers?: Record<string, { command: string; args?: string[]; env?: Record<string, string>; initializationOptions?: Record<string, unknown>; settings?: Record<string, unknown> }>;\n  /**\n   * Auto-inject LSP error diagnostics into write/edit tool results.\n   * Set to false to disable, or provide an array of language IDs to enable selectively.\n   * Default: true (all languages).\n   *\n   * Examples:\n   *   true              — inject for all languages\n   *   false             — never inject\n   *   [\"typescript\"]    — only inject for TypeScript files\n   */\n  autoInjectDiagnostics?: boolean | string[];\n}\n\n/** Load .pi-lsp.json from a directory. Returns null if not found or invalid. */\nfunction loadProjectConfig(dir: string): ProjectLspConfig | null {\n  const configPath = join(dir, \".pi-lsp.json\");\n  try {\n    if (!existsSync(configPath)) return null;\n    const raw = readFileSync(configPath, \"utf-8\");\n    const parsed = JSON.parse(raw);\n    if (typeof parsed !== \"object\" || parsed === null) return null;\n    return parsed as ProjectLspConfig;\n  } catch {\n    return null;\n  }\n}\n\nexport default function lspExtension(pi: ExtensionAPI) {\n  // Prevent EPIPE errors from LSP server exits from crashing the host process.\n  // When an LSP server exits unexpectedly, in-flight writes to its stdin pipe\n  // can produce EPIPE errors that escape all connection-level error handlers.\n  const origListeners = process.listeners(\"uncaughtException\");\n  process.on(\"uncaughtException\", (err: any) => {\n    if (err?.code === \"EPIPE\") return; // swallow — LSP server exited, harmless\n    // Re-throw for other handlers\n    for (const listener of origListeners) (listener as any)(err);\n    if (origListeners.length === 0) {\n      console.error(\"[LSP] Uncaught exception:\", err);\n    }\n  });\n\n  let manager: LspManager | null = null;\n  let fileSync: FileSync | null = null;\n  let treeSitter: TreeSitterManager | null = null;\n  let workspaceIndex: WorkspaceIndex | null = null;\n  let pendingProvider: WorkspaceProvider | null = null;\n  // Store latest ctx for lifecycle callbacks (updated on each event)\n  let latestCtx: any = null;\n  // Project config — loaded on session_start, used by auto-injection guard\n  let projectConfig: ProjectLspConfig | null = null;\n\n  /**\n   * Run `fn` with the currently captured ctx, swallowing stale-ctx errors.\n   *\n   * Lifecycle callbacks from LspManager (server ready, workspace setup, etc.)\n   * and `pi.events` listeners can fire AFTER the session is replaced or reloaded,\n   * at which point the captured ctx has been invalidated and any property access\n   * throws via ExtensionRunner.assertActive(). We can't use optional chaining to\n   * dodge it because the `ui` / `theme` getters themselves run assertActive().\n   */\n  const withLatestCtx = (fn: (ctx: any) => void) => {\n    const ctx = latestCtx;\n    if (!ctx) return;\n    try {\n      fn(ctx);\n    } catch (err: any) {\n      if (typeof err?.message === \"string\" && err.message.includes(\"stale after session\")) {\n        // Session was replaced/reloaded — drop the stale ctx and move on.\n        latestCtx = null;\n        return;\n      }\n      throw err;\n    }\n  };\n\n  // Listen for external workspace providers (e.g. bemol extension).\n  // Also check if one was already registered before we loaded (load order varies).\n  // Store as pendingProvider so it's available when the manager is created later.\n  const applyProvider = (data: unknown) => {\n    const provider = data as WorkspaceProvider;\n    pendingProvider = provider;\n    if (manager) {\n      manager.setWorkspaceProvider(provider);\n    }\n    // Update status if we have a UI context\n    const statusText = provider.getStatusText();\n    if (!statusText) return;\n    withLatestCtx((ctx) => {\n      if (!ctx.ui?.theme) return;\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"accent\", `LSP: ${statusText}`));\n    });\n  };\n\n  pi.events.on(\"lsp:register-workspace-provider\", applyProvider);\n\n  // Check for provider registered before our listener existed\n  const existing = (pi.events as any)[\"lsp:workspace-provider\"];\n  if (existing) applyProvider(existing);\n\n  /** Build lifecycle callbacks that update UI status */\n  const setLspStatus = (color: string, text: string) => {\n    withLatestCtx((ctx) => {\n      if (!ctx.ui?.theme) return;\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(color, text));\n    });\n  };\n\n  const makeCallbacks = (): LspManagerCallbacks => ({\n    onWorkspaceSetupStart: () => {\n      setLspStatus(\"warning\", \"LSP: workspace setup...\");\n    },\n    onWorkspaceSetupEnd: (success: boolean, duration: number) => {\n      const secs = (duration / 1000).toFixed(1);\n      if (success) {\n        setLspStatus(\"accent\", `LSP: workspace ready (${secs}s)`);\n      } else {\n        setLspStatus(\"warning\", `LSP: workspace setup failed (${secs}s)`);\n      }\n    },\n    onServerStart: (languageId: string, command: string) => {\n      setLspStatus(\"warning\", `LSP: starting ${languageId} (${command})...`);\n    },\n    onServerReady: (languageId: string) => {\n      setLspStatus(\"accent\", `LSP: ${languageId} ready`);\n    },\n    onServerError: (languageId: string, _error: string) => {\n      setLspStatus(\"error\", `LSP: ${languageId} failed`);\n    },\n    onServerCrash: (languageId: string, restarting: boolean, attempt: number) => {\n      if (restarting) {\n        setLspStatus(\"warning\", `LSP: restarting ${languageId}... (attempt ${attempt}/3)`);\n      } else {\n        setLspStatus(\"error\", `LSP: ${languageId} crashed — auto-restart exhausted`);\n      }\n    },\n  });\n\n  // Create manager eagerly so tools can reference it, but servers start lazily\n  const getManager = (): LspManager => {\n    if (!manager) {\n      manager = new LspManager(process.cwd(), undefined, makeCallbacks(), undefined, pendingProvider ?? undefined);\n      fileSync = new FileSync(manager);\n      fileSync.setSyntheticDotChecker((uri) => syntheticDotLocks.has(uri));\n      treeSitter = new TreeSitterManager();\n      workspaceIndex = new WorkspaceIndex(process.cwd(), treeSitter);\n      fileSync.setTreeSitter(treeSitter, workspaceIndex);\n    }\n    return manager;\n  };\n\n  const getFileSync = (): FileSync => {\n    if (!fileSync) {\n      getManager(); // ensures fileSync is created\n    }\n    return fileSync!;\n  };\n\n  const getTreeSitter = (): TreeSitterManager => {\n    if (!treeSitter) {\n      getManager(); // ensures treeSitter is created\n    }\n    return treeSitter!;\n  };\n\n  const getWorkspaceIndex = (): WorkspaceIndex => {\n    if (!workspaceIndex) {\n      getManager(); // ensures workspaceIndex is created\n    }\n    return workspaceIndex!;\n  };\n\n  // Initialize manager (uses cwd at session start time)\n  pi.on(\"session_start\", async (_event, ctx) => {\n    latestCtx = ctx;\n\n    // If manager was already created eagerly (e.g. by a tool before session_start),\n    // shut it down so we can re-create with the correct ctx.cwd.\n    if (manager) {\n      await manager.shutdownAll().catch(() => {});\n      if (treeSitter) treeSitter.shutdown();\n    }\n\n    manager = new LspManager(ctx.cwd, undefined, makeCallbacks(), undefined, pendingProvider ?? undefined);\n    fileSync = new FileSync(manager);\n    fileSync.setSyntheticDotChecker((uri) => syntheticDotLocks.has(uri));\n    treeSitter = new TreeSitterManager();\n    workspaceIndex = new WorkspaceIndex(ctx.cwd, treeSitter);\n    fileSync.setTreeSitter(treeSitter, workspaceIndex);\n\n    // Initialize tree-sitter in the background (don't block session start)\n    treeSitter.init().catch((err) => {\n      console.error(`[pi-lsp-extension] tree-sitter WASM init failed: ${err?.message ?? err}`);\n    });\n\n    // Show workspace provider status\n    const wsProvider = manager.workspace;\n    const statusText = wsProvider.getStatusText();\n    if (statusText) {\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"accent\", `LSP: ${statusText}`));\n    } else {\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"dim\", \"LSP: idle\"));\n    }\n\n    // Load project config and apply settings\n    projectConfig = loadProjectConfig(ctx.cwd);\n    if (projectConfig) {\n      // Apply custom server configs\n      if (projectConfig.servers) {\n        for (const [lang, serverConf] of Object.entries(projectConfig.servers)) {\n          manager.setServerConfig(lang, {\n            command: serverConf.command,\n            args: serverConf.args ?? [],\n            env: serverConf.env,\n            initializationOptions: serverConf.initializationOptions,\n            settings: serverConf.settings,\n          });\n        }\n      }\n\n      // Set Lombok jar path (explicit path or \"auto\" for auto-detection)\n      if (projectConfig.lombokJar) {\n        if (projectConfig.lombokJar !== \"auto\") {\n          manager.setLombokJar(projectConfig.lombokJar);\n        }\n        // \"auto\" is the default behavior — findLombokJar() already auto-detects.\n        // Setting it explicitly just confirms the user wants Lombok support.\n      }\n\n      // Auto-start configured languages in the background\n      if (projectConfig.autoStart && projectConfig.autoStart.length > 0) {\n        const langs = projectConfig.autoStart;\n        const lombokNote = langs.includes(\"java\") && manager.getLombokJar()\n          ? ` (lombok: ${manager.getLombokJar()?.split(\"/\").pop()})` : \"\";\n        setLspStatus(\"warning\", `LSP: auto-starting ${langs.join(\", \")}${lombokNote}...`);\n        manager.startEagerly(langs);\n      }\n    }\n  });\n\n  // Register all LSP tools\n  // Tools call getManager() lazily so they work even if session_start hasn't fired\n  const managerProxy = new Proxy({} as LspManager, {\n    get(_target, prop) {\n      return (getManager() as any)[prop];\n    },\n  });\n\n  const treeSitterProxy = new Proxy({} as TreeSitterManager, {\n    get(_target, prop) {\n      return (getTreeSitter() as any)[prop];\n    },\n  });\n\n  const workspaceIndexProxy = new Proxy({} as WorkspaceIndex, {\n    get(_target, prop) {\n      return (getWorkspaceIndex() as any)[prop];\n    },\n  });\n\n  pi.registerTool(createDiagnosticsTool(managerProxy, treeSitterProxy));\n  pi.registerTool(createHoverTool(managerProxy, treeSitterProxy));\n  pi.registerTool(createDefinitionTool(managerProxy, treeSitterProxy, workspaceIndexProxy));\n  pi.registerTool(createReferencesTool(managerProxy, treeSitterProxy));\n  pi.registerTool(createSymbolsTool(managerProxy, treeSitterProxy, workspaceIndexProxy));\n  pi.registerTool(createRenameTool(managerProxy, treeSitterProxy));\n  pi.registerTool(createCodeActionsTool(managerProxy, treeSitterProxy));\n  pi.registerTool(createCompletionsTool(managerProxy, {\n    getTrackedVersion: (uri) => getFileSync().getTrackedVersion(uri),\n    setTrackedVersion: (uri, v) => getFileSync().setTrackedVersion(uri, v),\n    isSyntheticDotActive: (uri) => syntheticDotLocks.has(uri),\n  }, treeSitterProxy));\n  const getRootDir = () => manager?.resolvePath(\".\") ?? process.cwd();\n  pi.registerTool(createCodeOverviewTool(getRootDir, treeSitterProxy, workspaceIndexProxy));\n  pi.registerTool(createCodeSearchTool(getRootDir, treeSitterProxy));\n  pi.registerTool(createCodeRewriteTool(getRootDir, treeSitterProxy, {\n    onFileModified: (filePath: string) => {\n      getFileSync().handleFileWrite(filePath).catch(() => {});\n    },\n  }));\n  pi.registerTool(createCodeActionsTool(managerProxy, treeSitterProxy));\n\n  // File sync: track file reads/writes/edits\n  // After writes/edits, append file-scoped error diagnostics to the tool result\n  pi.on(\"tool_result\", async (event) => {\n    const sync = getFileSync();\n\n    try {\n      if (isReadToolResult(event) && !event.isError) {\n        const path = (event.input as any)?.path;\n        if (path) await sync.handleFileRead(path);\n      }\n\n      if (isWriteToolResult(event) && !event.isError) {\n        const path = (event.input as any)?.path;\n        if (path) await sync.handleFileWrite(path);\n      }\n\n      if (isEditToolResult(event) && !event.isError) {\n        const path = (event.input as any)?.path;\n        if (path) await sync.handleFileWrite(path);\n      }\n    } catch {\n      // File sync errors are non-fatal\n    }\n\n    // Auto-append diagnostics for the changed file (write/edit only)\n    if ((isWriteToolResult(event) || isEditToolResult(event)) && !event.isError && manager) {\n      const path = (event.input as any)?.path;\n      if (!path) return;\n\n      const languageId = manager.getLanguageId(path);\n      if (!languageId) return;\n\n      // Check autoInjectDiagnostics config\n      const inject = projectConfig?.autoInjectDiagnostics;\n      if (inject === false) return;\n      if (Array.isArray(inject) && !inject.includes(languageId)) return;\n\n      const client = manager.getRunningClient(languageId);\n      if (!client) return;\n\n      // Wait briefly for the LSP to publish updated diagnostics\n      await new Promise((r) => setTimeout(r, DIAGNOSTIC_SETTLE_DELAY_MS));\n\n      const uri = manager.getFileUri(path);\n      const diagnostics = client.getDiagnostics(uri);\n      const errors = diagnostics.filter((d) => d.severity === DiagnosticSeverity.Error);\n\n      if (errors.length === 0) return;\n\n      // Build a compact summary — just errors, max 10 lines\n      const relPath = relative(manager.resolvePath(\".\"), manager.resolvePath(path));\n      const lines = errors.slice(0, 10).map((d) => {\n        const line = d.range.start.line + 1;\n        const col = d.range.start.character + 1;\n        const source = d.source ? ` [${d.source}]` : \"\";\n        return `${relPath}:${line}:${col} error: ${d.message}${source}`;\n      });\n      if (errors.length > 10) {\n        lines.push(`... and ${errors.length - 10} more error(s)`);\n      }\n\n      const summary = `\\n\\n⚠ LSP: ${errors.length} error(s) in ${relPath}:\\n${lines.join(\"\\n\")}`;\n\n      return {\n        content: [\n          ...event.content,\n          { type: \"text\" as const, text: summary },\n        ],\n      };\n    }\n  });\n\n  // Update status after tool execution ends\n  pi.on(\"tool_execution_end\", async (_event, ctx) => {\n    latestCtx = ctx;\n    if (!manager) return;\n    const statuses = manager.getStatus();\n    const running = statuses.filter((s) => s.running);\n    if (running.length === 0) {\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"dim\", \"LSP: idle\"));\n    } else {\n      const totalDiags = running.reduce((n, s) => n + s.diagnosticsCount, 0);\n      const langs = running.map((s) => s.languageId).join(\", \");\n      let status = `LSP: ${langs}`;\n      if (totalDiags > 0) {\n        status += ` (${totalDiags} diagnostic${totalDiags !== 1 ? \"s\" : \"\"})`;\n      }\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"accent\", status));\n    }\n  });\n\n  // /lsp command — show server status\n  pi.registerCommand(\"lsp\", {\n    description: \"Show LSP server status\",\n    handler: async (_args, ctx) => {\n      if (!manager) {\n        ctx.ui.notify(\"LSP manager not initialized\", \"warning\");\n        return;\n      }\n\n      const statuses = manager.getStatus();\n      if (statuses.length === 0) {\n        ctx.ui.notify(\"No LSP servers configured\", \"info\");\n        return;\n      }\n\n      const lines = statuses.map((s) => {\n        const icon = s.running ? \"🟢\" : \"⚪\";\n        const diags =\n          s.diagnosticsCount > 0 ? ` (${s.diagnosticsCount} diagnostics)` : \"\";\n        const shared = s.shared ? \" [shared]\" : \"\";\n        return `${icon} ${s.languageId}: ${s.command}${diags}${shared}`;\n      });\n\n      ctx.ui.notify(lines.join(\"\\n\"), \"info\");\n    },\n  });\n\n  // /lsp-restart command — restart a specific language server\n  pi.registerCommand(\"lsp-restart\", {\n    description: \"Restart an LSP server: /lsp-restart <language> (e.g. java, typescript)\",\n    handler: async (args, ctx) => {\n      if (!manager) {\n        ctx.ui.notify(\"LSP manager not initialized\", \"warning\");\n        return;\n      }\n\n      const languageId = args?.trim().toLowerCase();\n      if (!languageId) {\n        // Show running servers and usage\n        const statuses = manager.getStatus().filter((s) => s.running);\n        if (statuses.length === 0) {\n          ctx.ui.notify(\"No LSP servers are running.\\n\\nUsage: /lsp-restart <language>\", \"info\");\n        } else {\n          const langs = statuses.map((s) => s.languageId).join(\", \");\n          ctx.ui.notify(\n            `Running servers: ${langs}\\n\\nUsage: /lsp-restart <language>\\nExample: /lsp-restart java`,\n            \"info\"\n          );\n        }\n        return;\n      }\n\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"warning\", `LSP: restarting ${languageId}...`));\n      ctx.ui.notify(`Restarting ${languageId} server (kills daemon if shared)...`, \"info\");\n\n      try {\n        await manager.restartServer(languageId);\n        const lombokJar = languageId === \"java\" ? manager.getLombokJar() : null;\n        const lombokNote = lombokJar ? `\\nLombok: ${lombokJar}` : \"\";\n        ctx.ui.notify(`${languageId} server restarted successfully.${lombokNote}`, \"info\");\n        ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"accent\", `LSP: ${languageId} ready`));\n      } catch (err: any) {\n        ctx.ui.notify(`Failed to restart ${languageId}: ${err.message}`, \"error\");\n        ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"error\", `LSP: ${languageId} restart failed`));\n      }\n    },\n  });\n\n  // /lsp-config command — add or override server configuration\n  pi.registerCommand(\"lsp-config\", {\n    description:\n      \"Configure an LSP server: /lsp-config <language> <command> [args...]\",\n    handler: async (args, ctx) => {\n      if (!args?.trim()) {\n        ctx.ui.notify(\n          \"Usage: /lsp-config <language> <command> [args...]\\nExample: /lsp-config python pylsp\",\n          \"info\"\n        );\n        return;\n      }\n\n      const parts = args.trim().split(/\\s+/);\n      if (parts.length < 2) {\n        ctx.ui.notify(\n          \"Usage: /lsp-config <language> <command> [args...]\",\n          \"warning\"\n        );\n        return;\n      }\n\n      const [languageId, command, ...serverArgs] = parts;\n      const config: ServerConfig = { command, args: serverArgs };\n\n      getManager().setServerConfig(languageId, config);\n      ctx.ui.notify(\n        `Configured LSP for ${languageId}: ${command} ${serverArgs.join(\" \")}`,\n        \"info\"\n      );\n    },\n  });\n\n  // /lsp-lombok command — set Lombok jar path for Java\n  pi.registerCommand(\"lsp-lombok\", {\n    description:\n      \"Set Lombok jar path for Java: /lsp-lombok <path-to-lombok.jar>\",\n    handler: async (args, ctx) => {\n      const mgr = getManager();\n\n      if (!args?.trim()) {\n        const current = mgr.getLombokJar();\n        if (current) {\n          ctx.ui.notify(`Lombok jar: ${current}`, \"info\");\n        } else {\n          ctx.ui.notify(\n            \"No Lombok jar configured or detected.\\n\\n\" +\n            \"Usage: /lsp-lombok <path-to-lombok.jar>\\n\" +\n            \"Or set LOMBOK_JAR environment variable.\\n\\n\" +\n            \"Download from: https://projectlombok.org/download\",\n            \"info\"\n          );\n        }\n        return;\n      }\n\n      const jarPath = args.trim();\n      const { existsSync } = await import(\"node:fs\");\n      const { resolve } = await import(\"node:path\");\n      const resolved = resolve(ctx.cwd, jarPath);\n\n      if (!existsSync(resolved)) {\n        ctx.ui.notify(`File not found: ${resolved}`, \"error\");\n        return;\n      }\n\n      if (!resolved.endsWith(\".jar\")) {\n        ctx.ui.notify(`Warning: ${resolved} doesn't end in .jar — setting anyway`, \"warning\");\n      }\n\n      mgr.setLombokJar(resolved);\n      ctx.ui.notify(`Lombok jar set: ${resolved}`, \"info\");\n    },\n  });\n\n  // Clean shutdown (includes workspace provider, all LSP servers, and tree-sitter)\n  pi.on(\"session_shutdown\", async () => {\n    // Drop the captured ctx immediately — once shutdown fires, any late\n    // LSP manager callback that reaches setLspStatus/applyProvider would\n    // otherwise hit an invalidated ctx and throw an uncaught exception.\n    latestCtx = null;\n    if (manager) {\n      await manager.shutdownAll();\n      manager = null;\n      fileSync = null;\n    }\n    if (treeSitter) {\n      treeSitter.shutdown();\n      treeSitter = null;\n    }\n    workspaceIndex = null;\n  });\n}\n"]}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/builtin-extensions/pi-lsp-extension/src/index.ts"],"names":[],"mappings":"AACA;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAkB,MAAM,iCAAiC,CAAC;AAmHpF,MAAM,CAAC,OAAO,UAAU,YAAY,CAAC,EAAE,EAAE,YAAY,QAgepD;AAED,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC","sourcesContent":["// @ts-nocheck\n/**\n * pi-lsp-extension — Pi coding agent extension for LSP integration.\n *\n * Exposes Language Server Protocol capabilities as tools the LLM can call:\n * - lsp_diagnostics: compilation errors and warnings\n * - lsp_hover: type info and docs at a position\n * - lsp_definition: go to definition\n * - lsp_references: find all references\n * - lsp_symbols: file/workspace symbol search\n * - lsp_rename: preview rename refactoring\n * - lsp_completions: code completion suggestions at a position\n * - lsp_code_actions: quick fixes, refactorings, and source actions\n *\n * Position-based tools (hover, definition, references, rename, code_actions)\n * accept an optional `query` parameter as an alternative to line/character,\n * resolving a symbol name to its position automatically.\n *\n * Tool schemas, commands, and hooks register immediately. Heavy managers\n * (LspManager, FileSync, Tree-sitter/WASM, WorkspaceIndex) load on first use.\n *\n * Usage:\n *   1. npm install in this directory\n *   2. Add to pi via settings.json extensions, or: pi -e ./src/index.ts\n *   3. LSP servers start lazily when you first use a tool on a file\n */\n\nimport type { ExtensionAPI, ToolDefinition } from \"@ashx-j/lunr\";\nimport type { TSchema } from \"typebox\";\nimport { awaitWithAbort } from \"../../../utils/await-with-abort.ts\";\nimport {\n  isReadToolResult,\n  isWriteToolResult,\n  isEditToolResult,\n} from \"../../../core/extensions/types.js\";\nimport { DiagnosticSeverity } from \"vscode-languageserver-protocol\";\n\nimport type { WorkspaceProvider } from \"./workspace-provider.js\";\nimport { LspRuntimeHost, type LspRuntimeCallbacks } from \"./runtime.js\";\nimport { createDiagnosticsTool } from \"./tools/diagnostics.js\";\nimport { createHoverTool } from \"./tools/hover.js\";\nimport { createDefinitionTool } from \"./tools/definition.js\";\nimport { createReferencesTool } from \"./tools/references.js\";\nimport { createSymbolsTool } from \"./tools/symbols.js\";\nimport { createRenameTool } from \"./tools/rename.js\";\nimport { createCodeOverviewTool } from \"./tools/code-overview.js\";\nimport { createCompletionsTool } from \"./tools/completions.js\";\nimport { createCodeSearchTool } from \"./tools/code-search.js\";\nimport { createCodeRewriteTool } from \"./tools/code-rewrite.js\";\nimport { createCodeActionsTool } from \"./tools/code-actions.js\";\nimport { syntheticDotLocks } from \"./tools/completions.js\";\nimport { relative } from \"node:path\";\nimport { existsSync, readFileSync } from \"node:fs\";\nimport { join, resolve } from \"node:path\";\nimport { DIAGNOSTIC_SETTLE_DELAY_MS } from \"./shared/timing.js\";\n\n/**\n * Project-level LSP config — loaded from `.pi-lsp.json` in the workspace root.\n *\n * Example:\n * ```json\n * {\n *   \"autoStart\": [\"java\", \"typescript\"],\n *   \"lombokJar\": \"env/Lombok-1.18.x/runtime/lib/lombok-1.18.42.jar\",\n *   \"servers\": {\n *     \"python\": { \"command\": \"pylsp\", \"args\": [] }\n *   }\n * }\n * ```\n */\ninterface ProjectLspConfig {\n  /** Languages to start eagerly on session_start (e.g. [\"java\", \"typescript\"]) */\n  autoStart?: string[];\n  /** Path to Lombok jar (absolute or relative to project root). \"auto\" to auto-detect. */\n  lombokJar?: string;\n  /** Custom server configs keyed by language ID */\n  servers?: Record<string, { command: string; args?: string[]; env?: Record<string, string>; initializationOptions?: Record<string, unknown>; settings?: Record<string, unknown> }>;\n  /**\n   * Auto-inject LSP error diagnostics into write/edit tool results.\n   * Set to false to disable, or provide an array of language IDs to enable selectively.\n   * Default: true (all languages).\n   *\n   * Examples:\n   *   true              — inject for all languages\n   *   false             — never inject\n   *   [\"typescript\"]    — only inject for TypeScript files\n   */\n  autoInjectDiagnostics?: boolean | string[];\n}\n\ninterface ServerConfig {\n  command: string;\n  args: string[];\n  env?: Record<string, string>;\n  initializationOptions?: Record<string, unknown>;\n  settings?: Record<string, unknown>;\n}\n\n/** Load .pi-lsp.json from a directory. Returns null if not found or invalid. */\nfunction loadProjectConfig(dir: string): ProjectLspConfig | null {\n  const configPath = join(dir, \".pi-lsp.json\");\n  try {\n    if (!existsSync(configPath)) return null;\n    const raw = readFileSync(configPath, \"utf-8\");\n    const parsed = JSON.parse(raw);\n    if (typeof parsed !== \"object\" || parsed === null) return null;\n    return parsed as ProjectLspConfig;\n  } catch {\n    return null;\n  }\n}\n\nfunction createServiceProxy<T extends object>(getTarget: () => T | null, label: string): T {\n  return new Proxy({} as T, {\n    get(_target, prop, receiver) {\n      const service = getTarget();\n      if (!service) {\n        throw new Error(`${label} is not initialized`);\n      }\n      const value = Reflect.get(service, prop, service);\n      if (typeof value === \"function\") {\n        return value.bind(service);\n      }\n      return value;\n    },\n  });\n}\n\nfunction registerToolWithRuntime<T extends TSchema, D>(pi: ExtensionAPI, tool: ToolDefinition<T, D>, ensure: () => Promise<unknown>, host: LspRuntimeHost) {\n  pi.registerTool({\n    ...tool,\n    async execute(id, params, signal, onUpdate, ctx) {\n      const generation = host.getGeneration();\n      signal?.throwIfAborted();\n      await awaitWithAbort(ensure(), signal);\n      signal?.throwIfAborted();\n      if (host.getGeneration() !== generation) throw new Error(\"LSP session changed before execution\");\n      return tool.execute(id, params, signal, onUpdate, ctx);\n    },\n  });\n}\n\nexport default function lspExtension(pi: ExtensionAPI) {\n  // Prevent EPIPE errors from LSP server exits from crashing the host process.\n  // When an LSP server exits unexpectedly, in-flight writes to its stdin pipe\n  // can produce EPIPE errors that escape all connection-level error handlers.\n  const origListeners = process.listeners(\"uncaughtException\");\n  process.on(\"uncaughtException\", (err: any) => {\n    if (err?.code === \"EPIPE\") return; // swallow — LSP server exited, harmless\n    // Re-throw for other handlers\n    for (const listener of origListeners) (listener as any)(err);\n    if (origListeners.length === 0) {\n      console.error(\"[LSP] Uncaught exception:\", err);\n    }\n  });\n\n  const host = new LspRuntimeHost();\n  let pendingProvider: WorkspaceProvider | null = null;\n  let latestCtx: any = null;\n  let projectConfig: ProjectLspConfig | null = null;\n  let sessionCwd = process.cwd();\n\n  /**\n   * Run `fn` with the currently captured ctx, swallowing stale-ctx errors.\n   *\n   * Lifecycle callbacks from LspManager (server ready, workspace setup, etc.)\n   * and `pi.events` listeners can fire AFTER the session is replaced or reloaded,\n   * at which point the captured ctx has been invalidated and any property access\n   * throws via ExtensionRunner.assertActive(). We can't use optional chaining to\n   * dodge it because the `ui` / `theme` getters themselves run assertActive().\n   */\n  const withLatestCtx = (fn: (ctx: any) => void) => {\n    const ctx = latestCtx;\n    if (!ctx) return;\n    try {\n      fn(ctx);\n    } catch (err: any) {\n      if (typeof err?.message === \"string\" && err.message.includes(\"stale after session\")) {\n        // Session was replaced/reloaded — drop the stale ctx and move on.\n        latestCtx = null;\n        return;\n      }\n      throw err;\n    }\n  };\n\n  const applyProvider = (data: unknown) => {\n    const provider = data as WorkspaceProvider;\n    pendingProvider = provider;\n    host.setPendingProvider(provider);\n    const statusText = provider.getStatusText();\n    if (!statusText) return;\n    withLatestCtx((ctx) => {\n      if (!ctx.ui?.theme) return;\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"accent\", `LSP: ${statusText}`));\n    });\n  };\n\n  pi.events.on(\"lsp:register-workspace-provider\", applyProvider);\n\n  // Check for provider registered before our listener existed\n  const existing = (pi.events as any)[\"lsp:workspace-provider\"];\n  if (existing) applyProvider(existing);\n\n  const setLspStatus = (color: string, text: string) => {\n    withLatestCtx((ctx) => {\n      if (!ctx.ui?.theme) return;\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(color, text));\n    });\n  };\n\n  const makeCallbacks = (): LspRuntimeCallbacks => ({\n    onWorkspaceSetupStart: () => {\n      setLspStatus(\"warning\", \"LSP: workspace setup...\");\n    },\n    onWorkspaceSetupEnd: (success: boolean, duration: number) => {\n      const secs = (duration / 1000).toFixed(1);\n      if (success) {\n        setLspStatus(\"accent\", `LSP: workspace ready (${secs}s)`);\n      } else {\n        setLspStatus(\"warning\", `LSP: workspace setup failed (${secs}s)`);\n      }\n    },\n    onServerStart: (languageId: string, command: string) => {\n      setLspStatus(\"warning\", `LSP: starting ${languageId} (${command})...`);\n    },\n    onServerReady: (languageId: string) => {\n      setLspStatus(\"accent\", `LSP: ${languageId} ready`);\n    },\n    onServerError: (languageId: string, _error: string) => {\n      setLspStatus(\"error\", `LSP: ${languageId} failed`);\n    },\n    onServerCrash: (languageId: string, restarting: boolean, attempt: number) => {\n      if (restarting) {\n        setLspStatus(\"warning\", `LSP: restarting ${languageId}... (attempt ${attempt}/3)`);\n      } else {\n        setLspStatus(\"error\", `LSP: ${languageId} crashed — auto-restart exhausted`);\n      }\n    },\n  });\n\n  const applyProjectConfigToManager = (manager: { setServerConfig: Function; setLombokJar: Function }, config: ProjectLspConfig | null) => {\n    if (!config) return;\n    if (config.servers) {\n      for (const [lang, serverConf] of Object.entries(config.servers)) {\n        manager.setServerConfig(lang, {\n          command: serverConf.command,\n          args: serverConf.args ?? [],\n          env: serverConf.env,\n          initializationOptions: serverConf.initializationOptions,\n          settings: serverConf.settings,\n        });\n      }\n    }\n    if (config.lombokJar && config.lombokJar !== \"auto\") {\n      manager.setLombokJar(config.lombokJar);\n    }\n  };\n\n  const buildBindOptions = (cwd: string) => ({\n    cwd,\n    callbacks: makeCallbacks(),\n    pendingProvider,\n    syntheticDotChecker: (uri: string) => syntheticDotLocks.has(uri),\n    configureManager: (manager) => {\n      applyProjectConfigToManager(manager, projectConfig);\n    },\n  });\n\n  const ensureRuntime = async () => {\n    return host.ensureServices(\n      buildBindOptions(sessionCwd || process.cwd()),\n    );\n  };\n\n  // session_start binds cwd/config only. Heavy managers load on first use,\n  // unless autoStart is configured (then ensure + startEagerly).\n  pi.on(\"session_start\", async (_event, ctx) => {\n    latestCtx = ctx;\n    sessionCwd = ctx.cwd;\n\n    projectConfig = loadProjectConfig(ctx.cwd);\n    await host.bindSession(buildBindOptions(ctx.cwd));\n\n    const statusText = pendingProvider?.getStatusText?.() ?? \"\";\n    if (statusText) {\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"accent\", `LSP: ${statusText}`));\n    } else {\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"dim\", \"LSP: idle\"));\n    }\n\n    if (projectConfig?.autoStart && projectConfig.autoStart.length > 0) {\n      const langs = projectConfig.autoStart;\n      const generation = host.getGeneration();\n      void ensureRuntime().then((services) => {\n        if (host.getGeneration() !== generation) return;\n        const lombokJar = services.manager.getLombokJar?.() ?? null;\n        const lombokNote = langs.includes(\"java\") && lombokJar\n          ? ` (lombok: ${String(lombokJar).split(/[/\\\\]/).pop()})`\n          : \"\";\n        setLspStatus(\"warning\", `LSP: auto-starting ${langs.join(\", \")}${lombokNote}...`);\n        services.manager.startEagerly(langs);\n      }).catch((error) => {\n        if (host.getGeneration() === generation) setLspStatus(\"error\", `LSP startup failed: ${String(error)}`);\n      });\n    }\n  });\n\n  const managerProxy = createServiceProxy(\n    () => host.getServicesIfReady()?.manager ?? null,\n    \"LSP manager\",\n  );\n  const treeSitterProxy = createServiceProxy(\n    () => host.getServicesIfReady()?.treeSitter ?? null,\n    \"Tree-sitter manager\",\n  );\n  const workspaceIndexProxy = createServiceProxy(\n    () => host.getServicesIfReady()?.workspaceIndex ?? null,\n    \"Workspace index\",\n  );\n\n  registerToolWithRuntime(pi, createDiagnosticsTool(managerProxy, treeSitterProxy), ensureRuntime, host);\n  registerToolWithRuntime(pi, createHoverTool(managerProxy, treeSitterProxy), ensureRuntime, host);\n  registerToolWithRuntime(pi, createDefinitionTool(managerProxy, treeSitterProxy, workspaceIndexProxy), ensureRuntime, host);\n  registerToolWithRuntime(pi, createReferencesTool(managerProxy, treeSitterProxy), ensureRuntime, host);\n  registerToolWithRuntime(pi, createSymbolsTool(managerProxy, treeSitterProxy, workspaceIndexProxy), ensureRuntime, host);\n  registerToolWithRuntime(pi, createRenameTool(managerProxy, treeSitterProxy), ensureRuntime, host);\n  registerToolWithRuntime(pi, createCodeActionsTool(managerProxy, treeSitterProxy), ensureRuntime, host);\n  registerToolWithRuntime(pi, createCompletionsTool(\n      managerProxy,\n      {\n        getTrackedVersion: (uri) => {\n          const services = host.getServicesIfReady();\n          if (!services) return null;\n          return services.fileSync.getTrackedVersion(uri);\n        },\n        setTrackedVersion: (uri, v) => {\n          host.getServicesIfReady()?.fileSync.setTrackedVersion(uri, v);\n        },\n        isSyntheticDotActive: (uri) => syntheticDotLocks.has(uri),\n      },\n      treeSitterProxy,\n    ), ensureRuntime, host);\n  const getRootDir = () => {\n    const manager = host.getServicesIfReady()?.manager;\n    if (manager) return manager.resolvePath(\".\");\n    return sessionCwd || process.cwd();\n  };\n  registerToolWithRuntime(pi, createCodeOverviewTool(getRootDir, treeSitterProxy, workspaceIndexProxy), ensureRuntime, host);\n  registerToolWithRuntime(pi, createCodeSearchTool(getRootDir, treeSitterProxy), ensureRuntime, host);\n  registerToolWithRuntime(pi, createCodeRewriteTool(getRootDir, treeSitterProxy, {\n      onFileModified: (filePath: string) => {\n        const services = host.getServicesIfReady();\n        if (!services) return;\n        services.fileSync.handleFileWrite(filePath).catch(() => {});\n      },\n    }), ensureRuntime, host);\n\n  // File sync: track file reads/writes/edits without loading runtime on plain reads\n  // when LSP was never used. Writes may start a server (existing behavior) so they\n  // ensure runtime. After writes/edits, append file-scoped error diagnostics.\n  pi.on(\"tool_result\", async (event) => {\n    const generation = host.getGeneration();\n    try {\n      if (isReadToolResult(event) && !event.isError) {\n        const services = host.getServicesIfReady();\n        if (services) {\n          const path = (event.input as any)?.path;\n          if (path) await services.fileSync.handleFileRead(path);\n          if (host.getGeneration() !== generation) return;\n        }\n      }\n\n      if ((isWriteToolResult(event) || isEditToolResult(event)) && !event.isError) {\n        const path = (event.input as any)?.path;\n        if (path) {\n          const services = await ensureRuntime();\n          if (host.getGeneration() !== generation) return;\n          await services.fileSync.handleFileWrite(path);\n        }\n      }\n    } catch {\n      // File sync errors are non-fatal\n    }\n\n    if (host.getGeneration() !== generation) return;\n    // Auto-append diagnostics for the changed file (write/edit only)\n    const services = host.getServicesIfReady();\n    if ((isWriteToolResult(event) || isEditToolResult(event)) && !event.isError && services) {\n      const path = (event.input as any)?.path;\n      if (!path) return;\n\n      const manager = services.manager;\n      const languageId = manager.getLanguageId(path);\n      if (!languageId) return;\n\n      // Check autoInjectDiagnostics config\n      const inject = projectConfig?.autoInjectDiagnostics;\n      if (inject === false) return;\n      if (Array.isArray(inject) && !inject.includes(languageId)) return;\n\n      const client = manager.getRunningClient(languageId);\n      if (!client) return;\n\n      // Wait briefly for the LSP to publish updated diagnostics\n      await new Promise((r) => setTimeout(r, DIAGNOSTIC_SETTLE_DELAY_MS));\n      if (host.getGeneration() !== generation) return;\n\n      const uri = manager.getFileUri(path);\n      const diagnostics = client.getDiagnostics(uri);\n      const errors = diagnostics.filter((d) => d.severity === DiagnosticSeverity.Error);\n\n      if (errors.length === 0) return;\n\n      // Build a compact summary — just errors, max 10 lines\n      const relPath = relative(manager.resolvePath(\".\"), manager.resolvePath(path));\n      const lines = errors.slice(0, 10).map((d) => {\n        const line = d.range.start.line + 1;\n        const col = d.range.start.character + 1;\n        const source = d.source ? ` [${d.source}]` : \"\";\n        return `${relPath}:${line}:${col} error: ${d.message}${source}`;\n      });\n      if (errors.length > 10) {\n        lines.push(`... and ${errors.length - 10} more error(s)`);\n      }\n\n      const summary = `\\n\\n⚠ LSP: ${errors.length} error(s) in ${relPath}:\\n${lines.join(\"\\n\")}`;\n\n      return {\n        content: [\n          ...event.content,\n          { type: \"text\" as const, text: summary },\n        ],\n      };\n    }\n  });\n\n  // Update status after tool execution ends\n  pi.on(\"tool_execution_end\", async (_event, ctx) => {\n    latestCtx = ctx;\n    const services = host.getServicesIfReady();\n    if (!services) return;\n    const statuses = services.manager.getStatus();\n    const running = statuses.filter((s) => s.running);\n    if (running.length === 0) {\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"dim\", \"LSP: idle\"));\n    } else {\n      const totalDiags = running.reduce((n, s) => n + s.diagnosticsCount, 0);\n      const langs = running.map((s) => s.languageId).join(\", \");\n      let status = `LSP: ${langs}`;\n      if (totalDiags > 0) {\n        status += ` (${totalDiags} diagnostic${totalDiags !== 1 ? \"s\" : \"\"})`;\n      }\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"accent\", status));\n    }\n  });\n\n  // /lsp command — show server status\n  pi.registerCommand(\"lsp\", {\n    description: \"Show LSP server status\",\n    handler: async (_args, ctx) => {\n      const services = host.getServicesIfReady();\n      if (!services) {\n        ctx.ui.notify(\n          host.isInitializing() ? \"LSP runtime is starting\" : \"LSP runtime is idle (not started)\",\n          \"info\",\n        );\n        return;\n      }\n\n      const statuses = services.manager.getStatus();\n      if (statuses.length === 0) {\n        ctx.ui.notify(\"No LSP servers configured\", \"info\");\n        return;\n      }\n\n      const lines = statuses.map((s) => {\n        const icon = s.running ? \"🟢\" : \"⚪\";\n        const diags =\n          s.diagnosticsCount > 0 ? ` (${s.diagnosticsCount} diagnostics)` : \"\";\n        const shared = s.shared ? \" [shared]\" : \"\";\n        return `${icon} ${s.languageId}: ${s.command}${diags}${shared}`;\n      });\n\n      ctx.ui.notify(lines.join(\"\\n\"), \"info\");\n    },\n  });\n\n  // /lsp-restart command — restart a specific language server\n  pi.registerCommand(\"lsp-restart\", {\n    description: \"Restart an LSP server: /lsp-restart <language> (e.g. java, typescript)\",\n    handler: async (args, ctx) => {\n      const services = host.getServicesIfReady();\n      if (!services) {\n        if (args?.trim()) {\n          ctx.ui.notify(\n            host.isInitializing()\n              ? \"LSP runtime is starting\"\n              : \"LSP runtime is idle. Run an LSP tool before restarting a server.\",\n            \"info\",\n          );\n        } else {\n          ctx.ui.notify(\"No LSP servers are running.\\n\\nUsage: /lsp-restart <language>\", \"info\");\n        }\n        return;\n      }\n\n      const languageId = args?.trim().toLowerCase();\n      if (!languageId) {\n        // Show running servers and usage\n        const statuses = services.manager.getStatus().filter((s) => s.running);\n        if (statuses.length === 0) {\n          ctx.ui.notify(\"No LSP servers are running.\\n\\nUsage: /lsp-restart <language>\", \"info\");\n        } else {\n          const langs = statuses.map((s) => s.languageId).join(\", \");\n          ctx.ui.notify(\n            `Running servers: ${langs}\\n\\nUsage: /lsp-restart <language>\\nExample: /lsp-restart java`,\n            \"info\"\n          );\n        }\n        return;\n      }\n\n      ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"warning\", `LSP: restarting ${languageId}...`));\n      ctx.ui.notify(`Restarting ${languageId} server (kills daemon if shared)...`, \"info\");\n\n      try {\n        await services.manager.restartServer(languageId);\n        const lombokJar = languageId === \"java\" ? services.manager.getLombokJar() : null;\n        const lombokNote = lombokJar ? `\\nLombok: ${lombokJar}` : \"\";\n        ctx.ui.notify(`${languageId} server restarted successfully.${lombokNote}`, \"info\");\n        ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"accent\", `LSP: ${languageId} ready`));\n      } catch (err: any) {\n        ctx.ui.notify(`Failed to restart ${languageId}: ${err.message}`, \"error\");\n        ctx.ui.setStatus(\"lsp\", ctx.ui.theme.fg(\"error\", `LSP: ${languageId} restart failed`));\n      }\n    },\n  });\n\n  // /lsp-config command — add or override server configuration\n  pi.registerCommand(\"lsp-config\", {\n    description:\n      \"Configure an LSP server: /lsp-config <language> <command> [args...]\",\n    handler: async (args, ctx) => {\n      if (!args?.trim()) {\n        ctx.ui.notify(\n          \"Usage: /lsp-config <language> <command> [args...]\\nExample: /lsp-config python pylsp\",\n          \"info\"\n        );\n        return;\n      }\n\n      const parts = args.trim().split(/\\s+/);\n      if (parts.length < 2) {\n        ctx.ui.notify(\n          \"Usage: /lsp-config <language> <command> [args...]\",\n          \"warning\"\n        );\n        return;\n      }\n\n      const [languageId, command, ...serverArgs] = parts;\n      const config: ServerConfig = { command, args: serverArgs };\n\n      const services = await ensureRuntime();\n      services.manager.setServerConfig(languageId, config);\n      ctx.ui.notify(\n        `Configured LSP for ${languageId}: ${command} ${serverArgs.join(\" \")}`,\n        \"info\"\n      );\n    },\n  });\n\n  // /lsp-lombok command — set Lombok jar path for Java\n  pi.registerCommand(\"lsp-lombok\", {\n    description:\n      \"Set Lombok jar path for Java: /lsp-lombok <path-to-lombok.jar>\",\n    handler: async (args, ctx) => {\n      const services = await ensureRuntime();\n      const mgr = services.manager;\n\n      if (!args?.trim()) {\n        const current = mgr.getLombokJar();\n        if (current) {\n          ctx.ui.notify(`Lombok jar: ${current}`, \"info\");\n        } else {\n          ctx.ui.notify(\n            \"No Lombok jar configured or detected.\\n\\n\" +\n            \"Usage: /lsp-lombok <path-to-lombok.jar>\\n\" +\n            \"Or set LOMBOK_JAR environment variable.\\n\\n\" +\n            \"Download from: https://projectlombok.org/download\",\n            \"info\"\n          );\n        }\n        return;\n      }\n\n      const jarPath = args.trim();\n      const resolved = resolve(ctx.cwd, jarPath);\n\n      if (!existsSync(resolved)) {\n        ctx.ui.notify(`File not found: ${resolved}`, \"error\");\n        return;\n      }\n\n      if (!resolved.endsWith(\".jar\")) {\n        ctx.ui.notify(`Warning: ${resolved} doesn't end in .jar — setting anyway`, \"warning\");\n      }\n\n      mgr.setLombokJar(resolved);\n      ctx.ui.notify(`Lombok jar set: ${resolved}`, \"info\");\n    },\n  });\n\n  // Clean shutdown — never loads heavy modules if they were unused\n  pi.on(\"session_shutdown\", async () => {\n    // Drop the captured ctx immediately — once shutdown fires, any late\n    // LSP manager callback that reaches setLspStatus/applyProvider would\n    // otherwise hit an invalidated ctx and throw an uncaught exception.\n    latestCtx = null;\n    await host.shutdown();\n  });\n}\n\nexport { LspRuntimeHost } from \"./runtime.js\";\n"]}

package/dist/builtin-extensions/pi-lsp-extension/src/index.js

@@ -16,18 +16,18 @@
  * accept an optional `query` parameter as an alternative to line/character,
  * resolving a symbol name to its position automatically.
  *
+ * Tool schemas, commands, and hooks register immediately. Heavy managers
+ * (LspManager, FileSync, Tree-sitter/WASM, WorkspaceIndex) load on first use.
+ *
  * Usage:
  *   1. npm install in this directory
  *   2. Add to pi via settings.json extensions, or: pi -e ./src/index.ts
  *   3. LSP servers start lazily when you first use a tool on a file
  */
-import { isReadToolResult, isWriteToolResult, isEditToolResult, } from "@ashx-j/lunr";
+import { awaitWithAbort } from "../../../utils/await-with-abort.js";
+import { isReadToolResult, isWriteToolResult, isEditToolResult, } from "../../../core/extensions/types.js";
 import { DiagnosticSeverity } from "vscode-languageserver-protocol";
-import { LspManager } from "./lsp-manager.js";
-import { FileSync } from "./file-sync.js";
-import { TreeSitterManager } from "./tree-sitter/parser-manager.js";
-import { WorkspaceIndex } from "./tree-sitter/workspace-index.js";
-import "./workspace-provider.js";
+import { LspRuntimeHost } from "./runtime.js";
 import { createDiagnosticsTool } from "./tools/diagnostics.js";
 import { createHoverTool } from "./tools/hover.js";
 import { createDefinitionTool } from "./tools/definition.js";
@@ -42,7 +42,7 @@ import { createCodeActionsTool } from "./tools/code-actions.js";
 import { syntheticDotLocks } from "./tools/completions.js";
 import { relative } from "node:path";
 import { existsSync, readFileSync } from "node:fs";
-import { join } from "node:path";
+import { join, resolve } from "node:path";
 import { DIAGNOSTIC_SETTLE_DELAY_MS } from "./shared/timing.js";
 /** Load .pi-lsp.json from a directory. Returns null if not found or invalid. */
 function loadProjectConfig(dir) {
@@ -60,6 +60,35 @@ function loadProjectConfig(dir) {
         return null;
     }
 }
+function createServiceProxy(getTarget, label) {
+    return new Proxy({}, {
+        get(_target, prop, receiver) {
+            const service = getTarget();
+            if (!service) {
+                throw new Error(`${label} is not initialized`);
+            }
+            const value = Reflect.get(service, prop, service);
+            if (typeof value === "function") {
+                return value.bind(service);
+            }
+            return value;
+        },
+    });
+}
+function registerToolWithRuntime(pi, tool, ensure, host) {
+    pi.registerTool({
+        ...tool,
+        async execute(id, params, signal, onUpdate, ctx) {
+            const generation = host.getGeneration();
+            signal?.throwIfAborted();
+            await awaitWithAbort(ensure(), signal);
+            signal?.throwIfAborted();
+            if (host.getGeneration() !== generation)
+                throw new Error("LSP session changed before execution");
+            return tool.execute(id, params, signal, onUpdate, ctx);
+        },
+    });
+}
 export default function lspExtension(pi) {
     // Prevent EPIPE errors from LSP server exits from crashing the host process.
     // When an LSP server exits unexpectedly, in-flight writes to its stdin pipe
@@ -75,15 +104,11 @@ export default function lspExtension(pi) {
             console.error("[LSP] Uncaught exception:", err);
         }
     });
-    let manager = null;
-    let fileSync = null;
-    let treeSitter = null;
-    let workspaceIndex = null;
+    const host = new LspRuntimeHost();
     let pendingProvider = null;
-    // Store latest ctx for lifecycle callbacks (updated on each event)
     let latestCtx = null;
-    // Project config — loaded on session_start, used by auto-injection guard
     let projectConfig = null;
+    let sessionCwd = process.cwd();
     /**
      * Run `fn` with the currently captured ctx, swallowing stale-ctx errors.
      *
@@ -109,16 +134,10 @@ export default function lspExtension(pi) {
             throw err;
         }
     };
-    // Listen for external workspace providers (e.g. bemol extension).
-    // Also check if one was already registered before we loaded (load order varies).
-    // Store as pendingProvider so it's available when the manager is created later.
     const applyProvider = (data) => {
         const provider = data;
         pendingProvider = provider;
-        if (manager) {
-            manager.setWorkspaceProvider(provider);
-        }
-        // Update status if we have a UI context
+        host.setPendingProvider(provider);
         const statusText = provider.getStatusText();
         if (!statusText)
             return;
@@ -133,7 +152,6 @@ export default function lspExtension(pi) {
     const existing = pi.events["lsp:workspace-provider"];
     if (existing)
         applyProvider(existing);
-    /** Build lifecycle callbacks that update UI status */
     const setLspStatus = (color, text) => {
         withLatestCtx((ctx) => {
             if (!ctx.ui?.theme)
@@ -172,165 +190,144 @@ export default function lspExtension(pi) {
             }
         },
     });
-    // Create manager eagerly so tools can reference it, but servers start lazily
-    const getManager = () => {
-        if (!manager) {
-            manager = new LspManager(process.cwd(), undefined, makeCallbacks(), undefined, pendingProvider ?? undefined);
-            fileSync = new FileSync(manager);
-            fileSync.setSyntheticDotChecker((uri) => syntheticDotLocks.has(uri));
-            treeSitter = new TreeSitterManager();
-            workspaceIndex = new WorkspaceIndex(process.cwd(), treeSitter);
-            fileSync.setTreeSitter(treeSitter, workspaceIndex);
-        }
-        return manager;
-    };
-    const getFileSync = () => {
-        if (!fileSync) {
-            getManager(); // ensures fileSync is created
+    const applyProjectConfigToManager = (manager, config) => {
+        if (!config)
+            return;
+        if (config.servers) {
+            for (const [lang, serverConf] of Object.entries(config.servers)) {
+                manager.setServerConfig(lang, {
+                    command: serverConf.command,
+                    args: serverConf.args ?? [],
+                    env: serverConf.env,
+                    initializationOptions: serverConf.initializationOptions,
+                    settings: serverConf.settings,
+                });
+            }
         }
-        return fileSync;
-    };
-    const getTreeSitter = () => {
-        if (!treeSitter) {
-            getManager(); // ensures treeSitter is created
+        if (config.lombokJar && config.lombokJar !== "auto") {
+            manager.setLombokJar(config.lombokJar);
         }
-        return treeSitter;
     };
-    const getWorkspaceIndex = () => {
-        if (!workspaceIndex) {
-            getManager(); // ensures workspaceIndex is created
-        }
-        return workspaceIndex;
+    const buildBindOptions = (cwd) => ({
+        cwd,
+        callbacks: makeCallbacks(),
+        pendingProvider,
+        syntheticDotChecker: (uri) => syntheticDotLocks.has(uri),
+        configureManager: (manager) => {
+            applyProjectConfigToManager(manager, projectConfig);
+        },
+    });
+    const ensureRuntime = async () => {
+        return host.ensureServices(buildBindOptions(sessionCwd || process.cwd()));
     };
-    // Initialize manager (uses cwd at session start time)
+    // session_start binds cwd/config only. Heavy managers load on first use,
+    // unless autoStart is configured (then ensure + startEagerly).
     pi.on("session_start", async (_event, ctx) => {
         latestCtx = ctx;
-        // If manager was already created eagerly (e.g. by a tool before session_start),
-        // shut it down so we can re-create with the correct ctx.cwd.
-        if (manager) {
-            await manager.shutdownAll().catch(() => { });
-            if (treeSitter)
-                treeSitter.shutdown();
-        }
-        manager = new LspManager(ctx.cwd, undefined, makeCallbacks(), undefined, pendingProvider ?? undefined);
-        fileSync = new FileSync(manager);
-        fileSync.setSyntheticDotChecker((uri) => syntheticDotLocks.has(uri));
-        treeSitter = new TreeSitterManager();
-        workspaceIndex = new WorkspaceIndex(ctx.cwd, treeSitter);
-        fileSync.setTreeSitter(treeSitter, workspaceIndex);
-        // Initialize tree-sitter in the background (don't block session start)
-        treeSitter.init().catch((err) => {
-            console.error(`[pi-lsp-extension] tree-sitter WASM init failed: ${err?.message ?? err}`);
-        });
-        // Show workspace provider status
-        const wsProvider = manager.workspace;
-        const statusText = wsProvider.getStatusText();
+        sessionCwd = ctx.cwd;
+        projectConfig = loadProjectConfig(ctx.cwd);
+        await host.bindSession(buildBindOptions(ctx.cwd));
+        const statusText = pendingProvider?.getStatusText?.() ?? "";
         if (statusText) {
             ctx.ui.setStatus("lsp", ctx.ui.theme.fg("accent", `LSP: ${statusText}`));
         }
         else {
             ctx.ui.setStatus("lsp", ctx.ui.theme.fg("dim", "LSP: idle"));
         }
-        // Load project config and apply settings
-        projectConfig = loadProjectConfig(ctx.cwd);
-        if (projectConfig) {
-            // Apply custom server configs
-            if (projectConfig.servers) {
-                for (const [lang, serverConf] of Object.entries(projectConfig.servers)) {
-                    manager.setServerConfig(lang, {
-                        command: serverConf.command,
-                        args: serverConf.args ?? [],
-                        env: serverConf.env,
-                        initializationOptions: serverConf.initializationOptions,
-                        settings: serverConf.settings,
-                    });
-                }
-            }
-            // Set Lombok jar path (explicit path or "auto" for auto-detection)
-            if (projectConfig.lombokJar) {
-                if (projectConfig.lombokJar !== "auto") {
-                    manager.setLombokJar(projectConfig.lombokJar);
-                }
-                // "auto" is the default behavior — findLombokJar() already auto-detects.
-                // Setting it explicitly just confirms the user wants Lombok support.
-            }
-            // Auto-start configured languages in the background
-            if (projectConfig.autoStart && projectConfig.autoStart.length > 0) {
-                const langs = projectConfig.autoStart;
-                const lombokNote = langs.includes("java") && manager.getLombokJar()
-                    ? ` (lombok: ${manager.getLombokJar()?.split("/").pop()})` : "";
+        if (projectConfig?.autoStart && projectConfig.autoStart.length > 0) {
+            const langs = projectConfig.autoStart;
+            const generation = host.getGeneration();
+            void ensureRuntime().then((services) => {
+                if (host.getGeneration() !== generation)
+                    return;
+                const lombokJar = services.manager.getLombokJar?.() ?? null;
+                const lombokNote = langs.includes("java") && lombokJar
+                    ? ` (lombok: ${String(lombokJar).split(/[/\\]/).pop()})`
+                    : "";
                 setLspStatus("warning", `LSP: auto-starting ${langs.join(", ")}${lombokNote}...`);
-                manager.startEagerly(langs);
-            }
+                services.manager.startEagerly(langs);
+            }).catch((error) => {
+                if (host.getGeneration() === generation)
+                    setLspStatus("error", `LSP startup failed: ${String(error)}`);
+            });
         }
     });
-    // Register all LSP tools
-    // Tools call getManager() lazily so they work even if session_start hasn't fired
-    const managerProxy = new Proxy({}, {
-        get(_target, prop) {
-            return getManager()[prop];
-        },
-    });
-    const treeSitterProxy = new Proxy({}, {
-        get(_target, prop) {
-            return getTreeSitter()[prop];
+    const managerProxy = createServiceProxy(() => host.getServicesIfReady()?.manager ?? null, "LSP manager");
+    const treeSitterProxy = createServiceProxy(() => host.getServicesIfReady()?.treeSitter ?? null, "Tree-sitter manager");
+    const workspaceIndexProxy = createServiceProxy(() => host.getServicesIfReady()?.workspaceIndex ?? null, "Workspace index");
+    registerToolWithRuntime(pi, createDiagnosticsTool(managerProxy, treeSitterProxy), ensureRuntime, host);
+    registerToolWithRuntime(pi, createHoverTool(managerProxy, treeSitterProxy), ensureRuntime, host);
+    registerToolWithRuntime(pi, createDefinitionTool(managerProxy, treeSitterProxy, workspaceIndexProxy), ensureRuntime, host);
+    registerToolWithRuntime(pi, createReferencesTool(managerProxy, treeSitterProxy), ensureRuntime, host);
+    registerToolWithRuntime(pi, createSymbolsTool(managerProxy, treeSitterProxy, workspaceIndexProxy), ensureRuntime, host);
+    registerToolWithRuntime(pi, createRenameTool(managerProxy, treeSitterProxy), ensureRuntime, host);
+    registerToolWithRuntime(pi, createCodeActionsTool(managerProxy, treeSitterProxy), ensureRuntime, host);
+    registerToolWithRuntime(pi, createCompletionsTool(managerProxy, {
+        getTrackedVersion: (uri) => {
+            const services = host.getServicesIfReady();
+            if (!services)
+                return null;
+            return services.fileSync.getTrackedVersion(uri);
         },
-    });
-    const workspaceIndexProxy = new Proxy({}, {
-        get(_target, prop) {
-            return getWorkspaceIndex()[prop];
+        setTrackedVersion: (uri, v) => {
+            host.getServicesIfReady()?.fileSync.setTrackedVersion(uri, v);
         },
-    });
-    pi.registerTool(createDiagnosticsTool(managerProxy, treeSitterProxy));
-    pi.registerTool(createHoverTool(managerProxy, treeSitterProxy));
-    pi.registerTool(createDefinitionTool(managerProxy, treeSitterProxy, workspaceIndexProxy));
-    pi.registerTool(createReferencesTool(managerProxy, treeSitterProxy));
-    pi.registerTool(createSymbolsTool(managerProxy, treeSitterProxy, workspaceIndexProxy));
-    pi.registerTool(createRenameTool(managerProxy, treeSitterProxy));
-    pi.registerTool(createCodeActionsTool(managerProxy, treeSitterProxy));
-    pi.registerTool(createCompletionsTool(managerProxy, {
-        getTrackedVersion: (uri) => getFileSync().getTrackedVersion(uri),
-        setTrackedVersion: (uri, v) => getFileSync().setTrackedVersion(uri, v),
         isSyntheticDotActive: (uri) => syntheticDotLocks.has(uri),
-    }, treeSitterProxy));
-    const getRootDir = () => manager?.resolvePath(".") ?? process.cwd();
-    pi.registerTool(createCodeOverviewTool(getRootDir, treeSitterProxy, workspaceIndexProxy));
-    pi.registerTool(createCodeSearchTool(getRootDir, treeSitterProxy));
-    pi.registerTool(createCodeRewriteTool(getRootDir, treeSitterProxy, {
+    }, treeSitterProxy), ensureRuntime, host);
+    const getRootDir = () => {
+        const manager = host.getServicesIfReady()?.manager;
+        if (manager)
+            return manager.resolvePath(".");
+        return sessionCwd || process.cwd();
+    };
+    registerToolWithRuntime(pi, createCodeOverviewTool(getRootDir, treeSitterProxy, workspaceIndexProxy), ensureRuntime, host);
+    registerToolWithRuntime(pi, createCodeSearchTool(getRootDir, treeSitterProxy), ensureRuntime, host);
+    registerToolWithRuntime(pi, createCodeRewriteTool(getRootDir, treeSitterProxy, {
         onFileModified: (filePath) => {
-            getFileSync().handleFileWrite(filePath).catch(() => { });
+            const services = host.getServicesIfReady();
+            if (!services)
+                return;
+            services.fileSync.handleFileWrite(filePath).catch(() => { });
         },
-    }));
-    pi.registerTool(createCodeActionsTool(managerProxy, treeSitterProxy));
-    // File sync: track file reads/writes/edits
-    // After writes/edits, append file-scoped error diagnostics to the tool result
+    }), ensureRuntime, host);
+    // File sync: track file reads/writes/edits without loading runtime on plain reads
+    // when LSP was never used. Writes may start a server (existing behavior) so they
+    // ensure runtime. After writes/edits, append file-scoped error diagnostics.
     pi.on("tool_result", async (event) => {
-        const sync = getFileSync();
+        const generation = host.getGeneration();
         try {
             if (isReadToolResult(event) && !event.isError) {
-                const path = event.input?.path;
-                if (path)
-                    await sync.handleFileRead(path);
-            }
-            if (isWriteToolResult(event) && !event.isError) {
-                const path = event.input?.path;
-                if (path)
-                    await sync.handleFileWrite(path);
+                const services = host.getServicesIfReady();
+                if (services) {
+                    const path = event.input?.path;
+                    if (path)
+                        await services.fileSync.handleFileRead(path);
+                    if (host.getGeneration() !== generation)
+                        return;
+                }
             }
-            if (isEditToolResult(event) && !event.isError) {
+            if ((isWriteToolResult(event) || isEditToolResult(event)) && !event.isError) {
                 const path = event.input?.path;
-                if (path)
-                    await sync.handleFileWrite(path);
+                if (path) {
+                    const services = await ensureRuntime();
+                    if (host.getGeneration() !== generation)
+                        return;
+                    await services.fileSync.handleFileWrite(path);
+                }
             }
         }
         catch {
             // File sync errors are non-fatal
         }
+        if (host.getGeneration() !== generation)
+            return;
         // Auto-append diagnostics for the changed file (write/edit only)
-        if ((isWriteToolResult(event) || isEditToolResult(event)) && !event.isError && manager) {
+        const services = host.getServicesIfReady();
+        if ((isWriteToolResult(event) || isEditToolResult(event)) && !event.isError && services) {
             const path = event.input?.path;
             if (!path)
                 return;
+            const manager = services.manager;
             const languageId = manager.getLanguageId(path);
             if (!languageId)
                 return;
@@ -345,6 +342,8 @@ export default function lspExtension(pi) {
                 return;
             // Wait briefly for the LSP to publish updated diagnostics
             await new Promise((r) => setTimeout(r, DIAGNOSTIC_SETTLE_DELAY_MS));
+            if (host.getGeneration() !== generation)
+                return;
             const uri = manager.getFileUri(path);
             const diagnostics = client.getDiagnostics(uri);
             const errors = diagnostics.filter((d) => d.severity === DiagnosticSeverity.Error);
@@ -373,9 +372,10 @@ export default function lspExtension(pi) {
     // Update status after tool execution ends
     pi.on("tool_execution_end", async (_event, ctx) => {
         latestCtx = ctx;
-        if (!manager)
+        const services = host.getServicesIfReady();
+        if (!services)
             return;
-        const statuses = manager.getStatus();
+        const statuses = services.manager.getStatus();
         const running = statuses.filter((s) => s.running);
         if (running.length === 0) {
             ctx.ui.setStatus("lsp", ctx.ui.theme.fg("dim", "LSP: idle"));
@@ -394,11 +394,12 @@ export default function lspExtension(pi) {
     pi.registerCommand("lsp", {
         description: "Show LSP server status",
         handler: async (_args, ctx) => {
-            if (!manager) {
-                ctx.ui.notify("LSP manager not initialized", "warning");
+            const services = host.getServicesIfReady();
+            if (!services) {
+                ctx.ui.notify(host.isInitializing() ? "LSP runtime is starting" : "LSP runtime is idle (not started)", "info");
                 return;
             }
-            const statuses = manager.getStatus();
+            const statuses = services.manager.getStatus();
             if (statuses.length === 0) {
                 ctx.ui.notify("No LSP servers configured", "info");
                 return;
@@ -416,14 +417,22 @@ export default function lspExtension(pi) {
     pi.registerCommand("lsp-restart", {
         description: "Restart an LSP server: /lsp-restart <language> (e.g. java, typescript)",
         handler: async (args, ctx) => {
-            if (!manager) {
-                ctx.ui.notify("LSP manager not initialized", "warning");
+            const services = host.getServicesIfReady();
+            if (!services) {
+                if (args?.trim()) {
+                    ctx.ui.notify(host.isInitializing()
+                        ? "LSP runtime is starting"
+                        : "LSP runtime is idle. Run an LSP tool before restarting a server.", "info");
+                }
+                else {
+                    ctx.ui.notify("No LSP servers are running.\n\nUsage: /lsp-restart <language>", "info");
+                }
                 return;
             }
             const languageId = args?.trim().toLowerCase();
             if (!languageId) {
                 // Show running servers and usage
-                const statuses = manager.getStatus().filter((s) => s.running);
+                const statuses = services.manager.getStatus().filter((s) => s.running);
                 if (statuses.length === 0) {
                     ctx.ui.notify("No LSP servers are running.\n\nUsage: /lsp-restart <language>", "info");
                 }
@@ -436,8 +445,8 @@ export default function lspExtension(pi) {
             ctx.ui.setStatus("lsp", ctx.ui.theme.fg("warning", `LSP: restarting ${languageId}...`));
             ctx.ui.notify(`Restarting ${languageId} server (kills daemon if shared)...`, "info");
             try {
-                await manager.restartServer(languageId);
-                const lombokJar = languageId === "java" ? manager.getLombokJar() : null;
+                await services.manager.restartServer(languageId);
+                const lombokJar = languageId === "java" ? services.manager.getLombokJar() : null;
                 const lombokNote = lombokJar ? `\nLombok: ${lombokJar}` : "";
                 ctx.ui.notify(`${languageId} server restarted successfully.${lombokNote}`, "info");
                 ctx.ui.setStatus("lsp", ctx.ui.theme.fg("accent", `LSP: ${languageId} ready`));
@@ -463,7 +472,8 @@ export default function lspExtension(pi) {
             }
             const [languageId, command, ...serverArgs] = parts;
             const config = { command, args: serverArgs };
-            getManager().setServerConfig(languageId, config);
+            const services = await ensureRuntime();
+            services.manager.setServerConfig(languageId, config);
             ctx.ui.notify(`Configured LSP for ${languageId}: ${command} ${serverArgs.join(" ")}`, "info");
         },
     });
@@ -471,7 +481,8 @@ export default function lspExtension(pi) {
     pi.registerCommand("lsp-lombok", {
         description: "Set Lombok jar path for Java: /lsp-lombok <path-to-lombok.jar>",
         handler: async (args, ctx) => {
-            const mgr = getManager();
+            const services = await ensureRuntime();
+            const mgr = services.manager;
             if (!args?.trim()) {
                 const current = mgr.getLombokJar();
                 if (current) {
@@ -486,8 +497,6 @@ export default function lspExtension(pi) {
                 return;
             }
             const jarPath = args.trim();
-            const { existsSync } = await import("node:fs");
-            const { resolve } = await import("node:path");
             const resolved = resolve(ctx.cwd, jarPath);
             if (!existsSync(resolved)) {
                 ctx.ui.notify(`File not found: ${resolved}`, "error");
@@ -500,22 +509,14 @@ export default function lspExtension(pi) {
             ctx.ui.notify(`Lombok jar set: ${resolved}`, "info");
         },
     });
-    // Clean shutdown (includes workspace provider, all LSP servers, and tree-sitter)
+    // Clean shutdown — never loads heavy modules if they were unused
     pi.on("session_shutdown", async () => {
         // Drop the captured ctx immediately — once shutdown fires, any late
         // LSP manager callback that reaches setLspStatus/applyProvider would
         // otherwise hit an invalidated ctx and throw an uncaught exception.
         latestCtx = null;
-        if (manager) {
-            await manager.shutdownAll();
-            manager = null;
-            fileSync = null;
-        }
-        if (treeSitter) {
-            treeSitter.shutdown();
-            treeSitter = null;
-        }
-        workspaceIndex = null;
+        await host.shutdown();
     });
 }
+export { LspRuntimeHost } from "./runtime.js";
 //# sourceMappingURL=index.js.map