ai-otel-setup 1.0.4 → 1.0.6

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 不认那种语法，跨平台必须改成走文件）。
@@ -454,7 +475,7 @@ function main() {
   };
 
   // UserPromptSubmit 兜底 hook：复用同一脚本，由 stdin.hook_event_name 在脚本内部
-  // 分流。客户端做 5 分钟节流，服务端见 entry 已存在则仅补空。用于救 SessionStart
+  // 分流。客户端做 2 分钟节流，服务端见 entry 已存在则仅补空。用于救 SessionStart
   // 因网络/超时丢失的场景（线上观测约 60% 事件因此空 git/hostname）。
   const promptHookEntry = {
     matcher: "*",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-otel-setup",
-  "version": "1.0.4",
+  "version": "1.0.6",
   "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/on-session-start.js

@@ -20,8 +20,8 @@
  *
  * 节流（仅对 UserPromptSubmit）：
  *   - 在 ~/.claude/cc-otel-state/sent-<sid>.flag 写 marker
- *   - 5 分钟内同 sid 跳过 OTLP 上报，避免高频敲键狂发
- *   - 5 分钟后过期允许重试，给丢包/瞬时故障留救命窗口
+ *   - 2 分钟内同 sid 跳过 OTLP 上报，避免高频敲键狂发
+ *   - 2 分钟后过期允许重试，给丢包/瞬时故障留救命窗口
  */
 
 "use strict";
@@ -34,8 +34,8 @@ const http = require("http");
 const https = require("https");
 const { URL } = require("url");
 
-// UserPromptSubmit 节流窗口：5 分钟
-const PROMPT_THROTTLE_MS = 5 * 60 * 1000;
+// UserPromptSubmit 节流窗口：2 分钟
+const PROMPT_THROTTLE_MS = 2 * 60 * 1000;
 
 // -------- 环境变量读取 ----------
 
@@ -120,7 +120,7 @@ function safeGit(args) {
     // CC 在 stdin 里告诉脚本是哪个 hook 触发的；UserPromptSubmit 走"兜底"分支
     const isPromptFallback = input.hook_event_name === "UserPromptSubmit";
 
-    // 兜底路径节流：sid 维度 5 分钟最多一次（marker 文件 mtime 判断）。
+    // 兜底路径节流：sid 维度 2 分钟最多一次（marker 文件 mtime 判断）。
     // 失败重试窗口同时由此控制：marker 过期后允许下次 prompt 再发一次。
     const stateDir = path.join(os.homedir(), ".claude", "cc-otel-state");
     const markerPath = sessionId ? path.join(stateDir, `sent-${sessionId}.flag`) : null;
@@ -217,8 +217,8 @@ function safeGit(args) {
     req.on("timeout", () => { req.destroy(); done(); });
 
     // 在真正发包前 touch marker 文件——把"已尝试上报"持久化下来，
-    // 让后续 5 分钟内的 UserPromptSubmit 跳过重复 POST。失败也照写，
-    // 因为 5 分钟后 marker 会过期允许重试，不会永久卡住。
+    // 让后续 2 分钟内的 UserPromptSubmit 跳过重复 POST。失败也照写，
+    // 因为 2 分钟后 marker 会过期允许重试，不会永久卡住。
     if (markerPath) {
       try {
         fs.mkdirSync(stateDir, { recursive: true });

package/templates/settings.template.json

@@ -5,8 +5,8 @@
     "OTEL_LOGS_EXPORTER": "otlp",
     "OTEL_EXPORTER_OTLP_PROTOCOL": "grpc",
     "OTEL_EXPORTER_OTLP_ENDPOINT": "PLACEHOLDER_ENDPOINT",
-    "OTEL_LOGS_EXPORT_INTERVAL": "300000",
-    "OTEL_METRIC_EXPORT_INTERVAL": "600000",
+    "OTEL_LOGS_EXPORT_INTERVAL": "120000",
+    "OTEL_METRIC_EXPORT_INTERVAL": "300000",
     "OTEL_METRICS_INCLUDE_VERSION": "true",
     "OTEL_LOG_USER_PROMPTS": "0",
     "OTEL_LOG_TOOL_DETAILS": "1",