@trim21/personal-pi-extensions 0.1.467 → 0.1.470

package/README.md

@@ -365,10 +365,11 @@ read/edit/write 工具内置 LSP 诊断（写文件后等待并报告 ERROR 级
       "bin": "gopls",
       "args": [],
       "cwd": "{root}", // 支持 {root} / {cwd} 模板
+      "env": { "VIRTUAL_ENV": "{root}/.venv", "GITHUB_TOKEN": { "sh": ["gh", "auth", "token"] } }, // 追加到子进程环境变量；string 值支持 {root} / {cwd} 模板与 ${VAR} 引用，{sh} 启动时执行命令取 stdout（失败则服务器启动失败）
       "languageIdByExtension": { ".go": "go" },
       "startupTimeoutMs": 45000,
       "diagnosticsWaitMs": 1500,
-      "initializationOptions": {}, // → initialize 请求
+      "initializationOptions": { "pythonPath": "${VIRTUAL_ENV:-/opt/venv}/bin/python" }, // → initialize 请求；字符串值支持 ${VAR} / ${VAR:-default} 插值
       "settings": {}, // → didChangeConfiguration / workspace/configuration 请求
     },
   },
@@ -389,7 +390,31 @@ read/edit/write 工具内置 LSP 诊断（写文件后等待并报告 ERROR 级
 - `bin`：可执行文件——绝对路径、相对调用 cwd 的路径，或名字（先在项目内 `node_modules/.bin`、`.venv/bin`、`venv/bin` 找，再走 PATH）
 - `languageIdByExtension`：扩展名 → LSP languageId（didOpen 用）；缺省回退内置映射表
 - `startupTimeoutMs` / `diagnosticsWaitMs`：per-server 超时，覆盖全局配置与默认值
-- `initializationOptions` 与 `settings` 按 LSP 语义分离：前者进 initialize 请求，后者进 didChangeConfiguration / workspace/configuration 请求
+- `env`：追加到 LSP 子进程的环境变量（在 `process.env` 之上合并）。string 值支持 `{root}` / `{cwd}` 模板与 `${VAR}` 环境变量引用；`{ "sh": [...] }` 在服务器启动时执行命令（argv 直接执行、不经 shell，需要 shell 特性时自行包 `["bash", "-c", "..."]`），stdout trim 后作为值，命令失败（非零退出或输出为空）时该服务器启动失败并报错
+- `initializationOptions` 与 `settings` 按 LSP 语义分离：前者进 initialize 请求，后者进 didChangeConfiguration / workspace/configuration 请求；`initializationOptions` 的字符串值（含嵌套对象/数组）在启动时做 `${VAR}` 插值，`${VAR:-default}` 在变量未定义或为空时用 default，未定义且无 default 替换为空字符串；插值时可引用 `env` 里配置的变量
+
+`env` 的 `{sh}` 命令与 `initializationOptions` 插值可以组合使用，例如用 `gh auth token` 给服务器的 initialize 请求提供 session token：
+
+```jsonc
+{
+  "servers": {
+    "github-lsp": {
+      "bin": "github-lsp",
+      "args": ["--stdio"],
+      "env": {
+        // 启动时执行 gh auth token，stdout（trim 后）成为环境变量 GITHUB_TOKEN
+        "GITHUB_TOKEN": { "sh": ["gh", "auth", "token"] },
+      },
+      "initializationOptions": {
+        // 插值引用上面命令的输出，随 initialize 请求发给服务器
+        "sessionToken": "${GITHUB_TOKEN}",
+      },
+    },
+  },
+}
+```
+
+命令失败（`gh` 未登录 / 不在 PATH）时该服务器启动失败并报错。
 
 没有内置默认服务器：`servers` 的 key 就是服务器 id，全部来自你的配置，未定义 `servers` 时不启动任何语言服务器。executable 的发现逻辑（如 tsserver 路径、venv 里的 python）不内置，需要时用 `bin` / `args` / `settings` 自行表达。
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trim21/personal-pi-extensions",
-  "version": "0.1.467",
+  "version": "0.1.470",
   "type": "module",
   "description": "Custom pi coding-agent extensions: bwrap sandbox, workspace guard, opencode edit, and more",
   "keywords": [

package/src/lib/lsp/server-config.ts

@@ -12,8 +12,10 @@
  * 特殊探测逻辑（tsserver 路径、venv python 等）。
  */
 
+import { execFile as nodeExecFile } from "node:child_process";
 import { existsSync } from "node:fs";
 import { isAbsolute, join, relative, sep } from "node:path";
+import { promisify } from "node:util";
 
 import { minimatch } from "minimatch";
 import { type Static, Type } from "typebox";
@@ -39,13 +41,24 @@ export const serverConfigSchema = Type.Object({
   args: Type.Optional(Type.Array(Type.String())),
   /** 启动工作目录，支持 {root} / {cwd} 模板；缺省 {root}。 */
   cwd: Type.Optional(Type.String()),
+  /**
+   * 追加到子进程的环境变量。string 值支持 {root} / {cwd} 模板与 ${VAR} 引用；
+   * {sh: [...]} 在启动时执行命令（argv 直接执行、不经 shell），stdout trim 后作为值，
+   * 失败（非零退出 / 空输出）时服务器启动失败并报错。
+   */
+  env: Type.Optional(
+    Type.Record(
+      Type.String(),
+      Type.Union([Type.String(), Type.Object({ sh: Type.Array(Type.String()) })]),
+    ),
+  ),
   /** 文件扩展名（含点）→ LSP languageId，didOpen 用；缺省回退内置映射表。 */
   languageIdByExtension: Type.Optional(Type.Record(Type.String(), Type.String())),
   /** initialize 握手超时（ms）；缺省用全局配置 / client 默认。 */
   startupTimeoutMs: Type.Optional(Type.Number({ minimum: 1 })),
   /** 写文件后等待诊断的时长（ms）；缺省用全局配置 / client 默认。 */
   diagnosticsWaitMs: Type.Optional(Type.Number({ minimum: 1 })),
-  /** initialize 请求的 initializationOptions。 */
+  /** initialize 请求的 initializationOptions；字符串值支持 ${VAR} / ${VAR:-default} 插值。 */
   initializationOptions: Type.Optional(Type.Record(Type.String(), Type.Unknown())),
   /** didChangeConfiguration / workspace/configuration 请求的 settings。 */
   settings: Type.Optional(Type.Record(Type.String(), Type.Unknown())),
@@ -77,6 +90,74 @@ function resolveTemplate(template: string, root: string, cwd: string): string {
   return template.split("{root}").join(root).split("{cwd}").join(cwd);
 }
 
+/** ${VAR} / ${VAR:-default} 插值；shell 语义：未定义或空时用 default（缺省空字符串）。 */
+const envVarPattern = /\$\{([A-Za-z_][A-Za-z0-9_]*)(?::-([^}]*))?\}/g;
+
+function interpolateEnvVars(value: string, env: NodeJS.ProcessEnv): string {
+  return value.replaceAll(envVarPattern, (_match, name: string, fallback?: string) => {
+    const resolved = env[name];
+    return resolved !== undefined && resolved !== "" ? resolved : (fallback ?? "");
+  });
+}
+
+/** 深遍历配置值，对所有字符串做环境变量插值。 */
+function interpolateEnvDeep(value: unknown, env: NodeJS.ProcessEnv): unknown {
+  if (typeof value === "string") return interpolateEnvVars(value, env);
+  if (Array.isArray(value)) return value.map((item) => interpolateEnvDeep(item, env));
+  if (value !== null && typeof value === "object") {
+    return Object.fromEntries(
+      Object.entries(value).map(([key, item]) => [key, interpolateEnvDeep(item, env)]),
+    );
+  }
+  return value;
+}
+
+/** 解析 per-server env：string 值做 {root} / {cwd} 模板 + 环境变量引用插值，{sh} 执行命令取 stdout。 */
+async function resolveEnv(
+  config: Record<string, string | { sh: string[] }> | undefined,
+  root: string,
+  cwd: string,
+): Promise<NodeJS.ProcessEnv | undefined> {
+  const entries = Object.entries(config ?? {});
+  if (entries.length === 0) return undefined;
+  const resolved = await Promise.all(
+    entries.map(async ([key, value]): Promise<[string, string]> => {
+      if (typeof value === "string") {
+        return [key, interpolateEnvVars(resolveTemplate(value, root, cwd), process.env)];
+      }
+      return [key, await runEnvCommand(value.sh, cwd)];
+    }),
+  );
+  return Object.fromEntries(resolved);
+}
+
+const execFile = promisify(nodeExecFile);
+
+/**
+ * argv 直接执行（不经 shell，避免注入面；需要 shell 特性时配置里自行包
+ * ["bash", "-c", "..."]），stdout trim 后作为 env 值。
+ * 失败（spawn 错误 / 非零退出 / 空输出）时抛错，由 lsp.ts 捕获后 notify 给用户。
+ */
+async function runEnvCommand(argv: string[], cwd: string): Promise<string> {
+  let stdout: string;
+  try {
+    ({ stdout } = await execFile(argv[0], argv.slice(1), { cwd }));
+  } catch (error) {
+    const err = error as NodeJS.ErrnoException & { stderr?: string };
+    const reason = typeof err.code === "number" ? `exit code ${err.code}` : err.message;
+    const stderr = err.stderr?.trim();
+    throw new Error(
+      `env command failed (${argv.join(" ")}): ${reason}${stderr ? `: ${stderr}` : ""}`,
+      { cause: error },
+    );
+  }
+  const output = stdout.trim();
+  if (!output) {
+    throw new Error(`env command (${argv.join(" ")}) produced empty output`);
+  }
+  return output;
+}
+
 /** 解析可执行文件：绝对/相对路径直接用；名字走项目工作区（node_modules/.bin 等）→ PATH。 */
 async function resolveBinary(bin: string, root: string, cwd: string): Promise<string | undefined> {
   if (isAbsolute(bin)) return existsSync(bin) ? bin : undefined;
@@ -144,11 +225,17 @@ export class ConfigAdapter implements LspServerAdapter {
     if (!bin) return undefined;
     const resolved = await resolveBinary(bin, root, cwd);
     if (!resolved) return undefined;
+    const spawnCwd = resolveTemplate(this.config.cwd ?? "{root}", root, cwd);
+    const env = await resolveEnv(this.config.env, root, cwd);
     return {
       process: spawnProcess(resolved, this.config.args ?? [], {
-        cwd: resolveTemplate(this.config.cwd ?? "{root}", root, cwd),
+        cwd: spawnCwd,
+        env,
       }),
-      initialization: this.config.initializationOptions,
+      initialization: interpolateEnvDeep(this.config.initializationOptions, {
+        ...process.env,
+        ...env,
+      }) as Record<string, unknown> | undefined,
       settings: this.config.settings,
       languageIds: this.config.languageIdByExtension,
     };