ai-otel-setup 1.0.4 → 1.0.5

package/cli.js

@@ -280,10 +280,32 @@ function stripLegacyCodexHook(text) {
 
 function stripLegacyCodexHooksFlag(text) {
   // Codex 把 [features].codex_hooks 重命名为 [features].hooks，旧 key 启动时触发 deprecation 警告
-  // 删 = true 这行，由 managed 块统一写 hooks = true；= false 是显式 opt-out，保留
+  // 删 = 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，因为
@@ -293,11 +315,9 @@ function buildCodexManagedBlock(endpoint, hookDest, launcherDest) {
   //   - 只写 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]",
-    "hooks = true",
-    "",
     "[otel]",
     'environment = "prod"',
     "log_user_prompt = false",
@@ -336,11 +356,12 @@ function installCodex(home, endpoint) {
   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);
 
   // hook 同目录的 endpoint.json：hook 脚本运行时读它拿 logs endpoint，避免依赖
   // shell 前缀注入 env（cmd.exe 不认那种语法，跨平台必须改成走文件）。

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-otel-setup",
-  "version": "1.0.4",
+  "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",