ai-otel-setup 1.0.3 → 1.0.5

package/cli.js

@@ -25,6 +25,28 @@ const fs = require("fs");
 const path = require("path");
 const os = require("os");
 
+const PKG_VERSION = require("./package.json").version;
+
+// 安装时这台机器的 node 绝对路径，给 hook 命令做兜底（见 buildHookCommand）。
+const NODE_BIN = process.execPath;
+
+// 跨平台 hook 命令：固定形式 `<NODE_BIN> <launcher> <hook>`，三段都是绝对路径，
+// 对 shell 完全透明（POSIX sh / cmd.exe / PowerShell 5.1+ / PowerShell 7+ 全 cover）。
+// "PATH 上 node 优先 → 否则用 baked 绝对路径" 的兜底逻辑放在 launch-hook.js 里做，
+// 不再依赖 shell `||` 操作符——PS 5.1 不支持 `||`，cc/gemini 在 Windows 上默认就
+// 是 PS，会被坑。
+function buildHookCommand(launcherPath, scriptPath) {
+  return `"${NODE_BIN}" "${launcherPath}" "${scriptPath}"`;
+}
+
+// 把 launcher 模板拷到 hook 同目录，返回 launcher 的绝对路径
+function installLauncher(installDir) {
+  const launcherDest = path.join(installDir, "launch-hook.js");
+  fs.copyFileSync(path.join(__dirname, "templates", "launch-hook.js"), launcherDest);
+  fs.chmodSync(launcherDest, 0o755);
+  return launcherDest;
+}
+
 const REQUIRED_KEYS = ["url"];
 const HOOK_ID = "team:session-start";
 // UserPromptSubmit 兜底 hook：复用同一脚本，靠 stdin.hook_event_name 分流；
@@ -256,7 +278,35 @@ function stripLegacyCodexHook(text) {
   );
 }
 
-function buildCodexManagedBlock(endpoint, hookDest, logsEndpoint) {
+function stripLegacyCodexHooksFlag(text) {
+  // Codex 把 [features].codex_hooks 重命名为 [features].hooks，旧 key 启动时触发 deprecation 警告
+  // 删 = true 这行，由 ensureFeaturesHooksTrue 统一写 hooks = true；= false 是显式 opt-out，保留
+  return text.replace(/^[ \t]*codex_hooks[ \t]*=[ \t]*true[ \t]*\r?\n/gm, "");
+}
+
+function ensureFeaturesHooksTrue(text) {
+  // 在用户已有的 [features] 块原地插入 hooks = true（如缺失）；没有 [features] 就新建。
+  // 不能写在 managed 块里——TOML 1.0 禁止同名 table 重复声明，会被严格解析器拒绝。
+  const lines = text.split(/\r?\n/);
+  let featuresIdx = -1;
+  let hooksKeyExists = false;
+  for (let i = 0; i < lines.length; i++) {
+    if (featuresIdx === -1) {
+      if (/^\s*\[features\]\s*$/.test(lines[i])) featuresIdx = i;
+      continue;
+    }
+    if (/^\s*\[/.test(lines[i])) break; // 下一个 section，结束 [features] 主块扫描
+    if (/^[ \t]*hooks[ \t]*=/.test(lines[i])) hooksKeyExists = true;
+  }
+  if (featuresIdx >= 0) {
+    if (hooksKeyExists) return text; // 任何 hooks = ... 都尊重，不覆盖用户显式选择
+    lines.splice(featuresIdx + 1, 0, "hooks = true");
+    return lines.join("\n");
+  }
+  return text.trimEnd() + "\n\n[features]\nhooks = true\n";
+}
+
+function buildCodexManagedBlock(endpoint, hookDest, launcherDest) {
   // exporter / trace_exporter / metrics_exporter 是 externally-tagged enum：
   //   - 写 scalar `exporter = "otlp-grpc"`：codex 解析为 unit variant，因为
   //     OtlpGrpc 是 struct variant（带 endpoint 等字段），报
@@ -265,11 +315,9 @@ function buildCodexManagedBlock(endpoint, hookDest, logsEndpoint) {
   //   - 只写 table `[otel.exporter."otlp-grpc"]`：✓ codex 把它解析为
   //     OtlpGrpc { endpoint }，tag 来自 key 名。
   // 官方 sample 之所以能 `exporter = "none"`，是因为 None 本身就是 unit variant。
+  // [features].hooks = true 由 ensureFeaturesHooksTrue 写到用户块里，避免重复声明 [features]
   return [
     CODEX_MANAGED_BEGIN,
-    "[features]",
-    "codex_hooks = true",
-    "",
     "[otel]",
     'environment = "prod"',
     "log_user_prompt = false",
@@ -288,7 +336,7 @@ function buildCodexManagedBlock(endpoint, hookDest, logsEndpoint) {
     "",
     "[[hooks.SessionStart.hooks]]",
     'type = "command"',
-    `command = ${JSON.stringify(`AI_OTEL_LOGS_ENDPOINT=${logsEndpoint} node "${hookDest}"`)}`,
+    `command = ${JSON.stringify(buildHookCommand(launcherDest, hookDest))}`,
     CODEX_MANAGED_END,
   ].join("\n");
 }
@@ -304,16 +352,24 @@ function installCodex(home, endpoint) {
   fs.mkdirSync(installDir, { recursive: true });
   fs.copyFileSync(path.join(__dirname, "templates", "codex", "on-session-start.js"), hookDest);
   fs.chmodSync(hookDest, 0o755);
+  const launcherDest = installLauncher(installDir);
   const bak = backup(configPath);
   let existing = fs.existsSync(configPath) ? fs.readFileSync(configPath, "utf8") : "";
 
-  // 三步去重：先剥离上一次的 managed 块，再清掉旧 schema 残留
+  // 先剥离上一次的 managed 块和旧 schema 残留，再保证用户块里有 hooks = true
   existing = stripCodexManagedBlock(existing);
   existing = stripLegacyCodexOtel(existing);
   existing = stripLegacyCodexHook(existing);
+  existing = stripLegacyCodexHooksFlag(existing);
+  existing = ensureFeaturesHooksTrue(existing);
 
-  const logsEndpoint = logsEndpointFromGrpc(endpoint);
-  const managed = buildCodexManagedBlock(endpoint, hookDest, logsEndpoint);
+  // hook 同目录的 endpoint.json：hook 脚本运行时读它拿 logs endpoint，避免依赖
+  // shell 前缀注入 env（cmd.exe 不认那种语法，跨平台必须改成走文件）。
+  writeJSONAtomic(path.join(installDir, "endpoint.json"), {
+    endpoint,
+    logsEndpoint: logsEndpointFromGrpc(endpoint),
+  });
+  const managed = buildCodexManagedBlock(endpoint, hookDest, launcherDest);
   const merged = (existing.trimEnd() + "\n\n" + managed + "\n").replace(/\n{3,}/g, "\n\n");
   fs.writeFileSync(configPath, merged, "utf8");
   return { tool: "codex", status: "installed", path: configPath, backup: bak };
@@ -330,6 +386,12 @@ function installGemini(home, endpoint) {
   fs.mkdirSync(installDir, { recursive: true });
   fs.copyFileSync(path.join(__dirname, "templates", "gemini", "on-session-start.js"), hookDest);
   fs.chmodSync(hookDest, 0o755);
+  const launcherDest = installLauncher(installDir);
+  // 同 Codex：endpoint.json 给 hook 脚本读，跨平台不依赖 env 前缀。
+  writeJSONAtomic(path.join(installDir, "endpoint.json"), {
+    endpoint,
+    logsEndpoint: logsEndpointFromGrpc(endpoint),
+  });
   const existing = readJSONSafe(settingsPath);
   const bak = backup(settingsPath);
   const merged = { ...existing };
@@ -350,7 +412,7 @@ function installGemini(home, endpoint) {
     : [];
   const hookEntry = {
     id: HOOK_ID,
-    command: `node "${hookDest}"`,
+    command: buildHookCommand(launcherDest, hookDest),
   };
   const idx = sessionStart.findIndex((h) => h && h.id === HOOK_ID);
   if (idx >= 0) sessionStart[idx] = hookEntry;
@@ -384,6 +446,7 @@ function main() {
   const installDir = path.join(claudeDir, "cc-otel");
   const settingsPath = path.join(claudeDir, "settings.json");
   const hookScriptDest = path.join(installDir, "on-session-start.js");
+  const launcherDest = path.join(installDir, "launch-hook.js");
 
   const templateDir = path.join(__dirname, "templates");
   const settingsTemplate = readJSONSafe(path.join(templateDir, "settings.template.json"));
@@ -402,7 +465,7 @@ function main() {
     hooks: [
       {
         type: "command",
-        command: `node "${hookScriptDest}"`,
+        command: buildHookCommand(launcherDest, hookScriptDest),
         timeout: 3,
       },
     ],
@@ -419,7 +482,7 @@ function main() {
     hooks: [
       {
         type: "command",
-        command: `node "${hookScriptDest}"`,
+        command: buildHookCommand(launcherDest, hookScriptDest),
         timeout: 3,
       },
     ],
@@ -431,6 +494,7 @@ function main() {
   fs.mkdirSync(installDir, { recursive: true });
   fs.copyFileSync(hookScriptSrc, hookScriptDest);
   fs.chmodSync(hookScriptDest, 0o755);
+  installLauncher(installDir);
 
   // v1.0.3：把 endpoint 写盘，给 hook 脚本的 resolveLogsEndpoint 当兜底。
   // 修的是 v1.0.2 的真实事故：settings.json 的 env 不一定能继承到 hook 子进程
@@ -469,6 +533,7 @@ function main() {
 
   console.log("[ai-otel-setup] 安装完成。");
   console.log("");
+  console.log(`  ${"version".padEnd(12)}: ${PKG_VERSION}`);
   console.log(`  ${"endpoint".padEnd(12)}: ${endpoint}`);
   for (const r of allResults) {
     console.log(`  ${r.tool.padEnd(12)}: ${r.status}${r.reason ? " (" + r.reason + ")" : ""}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-otel-setup",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "One-shot installer for AI CLI OpenTelemetry forwarding. Writes Claude Code, Codex CLI, and Gemini CLI telemetry config in a single npx command.",
   "bin": {
     "ai-otel-setup": "cli.js",

package/templates/codex/on-session-start.js

@@ -27,8 +27,17 @@ function safeGit(args) {
   }
 }
 
+// 解析 OTLP/HTTP logs endpoint。优先级：env 覆盖 → installer 写在 hook 同目录的
+// endpoint.json → localhost 兜底。原本走 shell 前缀 `AI_OTEL_LOGS_ENDPOINT=...` 注入
+// env，但那是 POSIX 独有语法、cmd.exe 把它当程序名就 G 了，所以 v1.0.4 起命令行
+// 不再带前缀，改让脚本自己读 endpoint.json，跨平台统一。env 留作 debug 覆盖口。
 function endpoint() {
-  return process.env.AI_OTEL_LOGS_ENDPOINT || "http://localhost:4318/v1/logs";
+  if (process.env.AI_OTEL_LOGS_ENDPOINT) return process.env.AI_OTEL_LOGS_ENDPOINT;
+  try {
+    const cfg = JSON.parse(fs.readFileSync(path.join(__dirname, "endpoint.json"), "utf8"));
+    if (cfg && cfg.logsEndpoint) return cfg.logsEndpoint;
+  } catch (_) { /* 文件不存在/解析失败：继续走 localhost */ }
+  return "http://localhost:4318/v1/logs";
 }
 
 (async () => {

package/templates/gemini/on-session-start.js

@@ -4,6 +4,7 @@
 const { execFileSync } = require("child_process");
 const os = require("os");
 const path = require("path");
+const fs = require("fs");
 const http = require("http");
 const https = require("https");
 const { URL } = require("url");
@@ -16,8 +17,19 @@ function safeGit(args) {
   }
 }
 
+// 解析 OTLP/HTTP logs endpoint。优先级：env 覆盖 → installer 写在 hook 同目录的
+// endpoint.json → localhost 兜底。v1.0.4 起 hook 自己读 endpoint.json，避免依赖
+// shell 前缀注入 env 那种 POSIX-only 写法（cmd.exe 不认）。
 function endpoint() {
-  const base = process.env.GEMINI_TELEMETRY_OTLP_ENDPOINT || "http://localhost:4317";
+  let base = process.env.GEMINI_TELEMETRY_OTLP_ENDPOINT;
+  if (!base) {
+    try {
+      const cfg = JSON.parse(fs.readFileSync(path.join(__dirname, "endpoint.json"), "utf8"));
+      if (cfg && cfg.logsEndpoint) return cfg.logsEndpoint;
+      if (cfg && cfg.endpoint) base = cfg.endpoint;
+    } catch (_) { /* 文件不存在/解析失败：继续走 localhost */ }
+  }
+  if (!base) base = "http://localhost:4317";
   const url = new URL(base);
   if (url.port === "4317") url.port = "4318";
   if (!url.pathname || url.pathname === "/") url.pathname = "/v1/logs";

package/templates/launch-hook.js

@@ -0,0 +1,34 @@
+#!/usr/bin/env node
+"use strict";
+
+// 跨平台 hook 启动器：优先 PATH 上的 node（用户升级 Node 时自动跟新版本），
+// 找不到再用安装时这台机器上 node 的绝对路径（即当前进程的 execPath，一定可用）。
+// 改用 JS 内部兜底替代 shell `||` 操作符链——PowerShell 5.1（Win10/11 默认 shell）
+// 不支持 `||`，cc/gemini 在 Windows 上默认走 PowerShell，会被坑。统一在 JS 里兜底
+// 后，命令字符串变成纯 `<node> <launcher> <hook>` 三段绝对路径调用，对 shell 透明，
+// POSIX sh / cmd.exe / PowerShell 5.1 / PowerShell 7 全 cover。
+//
+// stdio: "inherit"：stdin（Codex/CC 传 hook payload JSON）、stderr 都直通给 hook
+// 子进程，hook 那边的 readStdin / process.stderr 行为不受影响。退出码原样转发。
+
+const { spawnSync, execFileSync } = require("child_process");
+
+const scriptPath = process.argv[2];
+if (!scriptPath) process.exit(0);
+
+let nodeBin = process.execPath;
+try {
+  // -v 只打版本号立即退出，用来探 PATH 上是否有可执行 node。
+  // timeout 防 PATH 上的 "node" 是个会卡住的 wrapper（极少见但存在）。
+  execFileSync("node", ["-v"], { stdio: "ignore", timeout: 1500 });
+  nodeBin = "node";
+} catch (_) {
+  // PATH 上没 node 或探测失败 → 沿用当前进程 execPath（即 installer 焊死的那条
+  // 绝对路径）。注意此时如果用户连 baked 那个版本也卸了，那 launcher 自己根本就
+  // 启动不起来，根本走不到这里——也就是说 hook 真挂的时候用户会感知到，符合
+  // 不静默吞错的预期。
+}
+
+const r = spawnSync(nodeBin, [scriptPath], { stdio: "inherit" });
+// status 为 null 表示进程被信号杀掉（SIGTERM 等），按 1 处理
+process.exit(r.status === null ? 1 : r.status);