tokentracker-cli 1.3.3 → 1.3.5

package/src/lib/notify-command.js

@@ -0,0 +1,38 @@
+"use strict";
+
+function powershellLiteral(value) {
+  return `'${String(value || "").replace(/'/g, "''")}'`;
+}
+
+function quotePosix(value) {
+  const text = String(value || "");
+  if (/^[A-Za-z0-9_\-./:@]+$/.test(text)) return text;
+  return `"${text.replace(/\\/g, "\\\\").replace(/"/g, '\\"').replace(/\$/g, '\\$').replace(/`/g, '\\`')}"`;
+}
+
+function buildNotifyCommand(notifyPath, source, { platform = process.platform, execPath = process.execPath } = {}) {
+  if (!/^[a-z0-9-]+$/i.test(source)) throw new Error("Invalid notify source");
+  if (platform === "win32") {
+    return `& ${powershellLiteral(execPath)} ${powershellLiteral(notifyPath)} --source=${source}`;
+  }
+  return `/usr/bin/env node ${quotePosix(notifyPath)} --source=${source}`;
+}
+
+// Recognize only our complete notify invocation, never a user command merely
+// containing notify.cjs or additional shell actions. Used to repair old hooks.
+function notifyIdentity(command) {
+  if (typeof command !== "string") return null;
+  const tokens = command.match(/"(?:\\.|[^"\\])*"|'(?:''|[^'])*'|[^\s]+/g) || [];
+  const decode = (token) => token.startsWith("'")
+    ? token.slice(1, -1).replace(/''/g, "'")
+    : token.startsWith('"') ? token.slice(1, -1).replace(/\\([\\"$`])/g, "$1") : token;
+  const args = tokens.map(decode);
+  if (args[0] === "&") args.shift();
+  if (args[0] === "/usr/bin/env" && args[1] === "node") args.shift();
+  if (args.length !== 3 || !/(?:^|[\\/])node(?:\.exe)?$/i.test(args[0])) return null;
+  if (!/(?:^|[\\/])notify\.cjs$/i.test(args[1]) || !/^--source=[a-z0-9-]+$/i.test(args[2])) return null;
+  const file = args[1].replace(/\\/g, "/");
+  return `${/^[a-z]:\//i.test(file) ? file.toLowerCase() : file}\u0000${args[2]}`;
+}
+
+module.exports = { buildNotifyCommand, notifyIdentity };

package/src/lib/openclaw-session-plugin.js

@@ -371,7 +371,7 @@ function buildSessionPluginIndex({ trackerDir, packageName = "tokentracker-cli",
     `      const agentId = normalize(ctx && ctx.agentId) || parseAgentId(sessionKey);\n` +
     `      if (!agentId) return;\n` +
     `\n` +
-    `      const sessionInfo = resolveSessionInfo(agentId, sessionKey);\n` +
+    `      const sessionInfo = await resolveSessionInfo(agentId, sessionKey);\n` +
     `      const sessionId = normalize(sessionInfo && sessionInfo.sessionId);\n` +
     `      if (!sessionId) return;\n` +
     `\n` +
@@ -440,25 +440,58 @@ function buildSessionPluginIndex({ trackerDir, packageName = "tokentracker-cli",
     `  return out;\n` +
     `}\n` +
     `\n` +
-    `function resolveSessionInfo(agentId, sessionKey) {\n` +
+    `async function resolveSessionInfo(agentId, sessionKey) {\n` +
     `  const key = normalize(sessionKey);\n` +
     `  if (!key) return null;\n` +
     `  const sessionsPath = path.join(openclawHome, 'agents', agentId, 'sessions', 'sessions.json');\n` +
-    `  try {\n` +
-    `    const raw = fs.readFileSync(sessionsPath, 'utf8');\n` +
-    `    const parsed = JSON.parse(raw);\n` +
-    `    if (!parsed || typeof parsed !== 'object') return null;\n` +
-    `    const entry = parsed[key];\n` +
-    `    if (!entry || typeof entry !== 'object') return null;\n` +
-    `    return {\n` +
-    `      sessionKey: key,\n` +
-    `      sessionId: normalize(entry.sessionId),\n` +
-    `      entry\n` +
-    `    };\n` +
-    `  } catch (_) {}\n` +
+    `  if (fs.existsSync(sessionsPath)) {\n` +
+    `    try {\n` +
+    `      const raw = fs.readFileSync(sessionsPath, 'utf8');\n` +
+    `      const parsed = JSON.parse(raw);\n` +
+    `      if (parsed && typeof parsed === 'object') {\n` +
+    `        const entry = parsed[key];\n` +
+    `        if (entry && typeof entry === 'object' && normalize(entry.sessionId)) {\n` +
+    `          return { sessionKey: key, sessionId: normalize(entry.sessionId), entry };\n` +
+    `        }\n` +
+    `      }\n` +
+    `    } catch (_) {}\n` +
+    `  }\n` +
+    `  // OpenClaw >= 2026.9.9 keeps session state in a per-agent SQLite\n` +
+    `  // database (~/.openclaw/agents/<id>/agent/openclaw-agent.sqlite);\n` +
+    `  // sessions.json is no longer written. Without this fallback the\n` +
+    `  // plugin would silently no-op on every agent_end for anyone on a\n` +
+    `  // current OpenClaw install (issue: openclaw migration to sqlite).\n` +
+    `  return resolveSessionInfoFromSqlite(agentId, key);\n` +
+    `}\n` +
+    `\n` +
+    `async function resolveSessionInfoFromSqlite(agentId, sessionKey) {\n` +
+    `  const sqliteModule = await importSqliteSync();\n` +
+    `  if (!sqliteModule) return null;\n` +
+    `  const { DatabaseSync } = sqliteModule;\n` +
+    `  const currentSessionQuery = "SELECT w.session_id FROM session_nodes n JOIN session_windows w ON w.session_id = n.current_session_id WHERE n.session_key = ? AND COALESCE(w.reason, '') != 'compaction'";\n` +
+    `  const sessionQuery = "SELECT session_id FROM session_windows WHERE session_key = ? AND COALESCE(reason, '') != 'compaction' ORDER BY updated_at DESC LIMIT 1";\n` +
+    `  const dbCandidates = [path.join(openclawHome, 'agents', agentId, 'agent', 'openclaw-agent.sqlite'), path.join(openclawHome, 'agents', agentId, 'openclaw-agent.sqlite')];\n` +
+    `  for (const dbPath of dbCandidates) {\n` +
+    `    if (!fs.existsSync(dbPath)) continue;\n` +
+    `    let db;\n` +
+    `    try { db = new DatabaseSync(dbPath, { readOnly: true }); } catch (_) { continue; }\n` +
+    `    try {\n` +
+    `      let row;\n` +
+    `      try { row = db.prepare(currentSessionQuery).get(sessionKey); } catch (_) {}\n` +
+    `      if (!row || !row.session_id) row = db.prepare(sessionQuery).get(sessionKey);\n` +
+    `      if (!row || !row.session_id) continue;\n` +
+    `      // The passive SQLite reader owns per-event accounting; never synthesize lifetime sums.\n` +
+    `      return { sessionKey, sessionId: row.session_id, entry: {} };\n` +
+    `    } catch (_) {}\n` +
+    `    finally { try { db.close(); } catch (_) {} }\n` +
+    `  }\n` +
     `  return null;\n` +
     `}\n` +
     `\n` +
+    `async function importSqliteSync() {\n` +
+    `  try { return await import('node:sqlite'); } catch (_) { return null; }\n` +
+    `}\n` +
+    `\n` +
     `function parseAgentId(sessionKey) {\n` +
     `  const s = normalize(sessionKey);\n` +
     `  if (!s || !s.startsWith('agent:')) return null;\n` +

package/src/lib/passive-mode.js

@@ -28,6 +28,7 @@
 
 const fs = require("node:fs");
 const path = require("node:path");
+const { expandHome } = require("./scan-roots");
 
 /**
  * @typedef {Object} PassiveProvider
@@ -171,6 +172,17 @@ function detectPassiveProviders({ home, hookStatus, env = process.env }) {
     settingsPath: path.join(home, ".workbuddy", "settings.json"),
   }));
 
+  // WorkBuddy AI — international build; data + hooks in ~/.workbuddy-ai
+  const workbuddyAiHome = expandHome(env.WORKBUDDY_AI_HOME, home) || path.join(home, ".workbuddy-ai");
+  out.push(buildEntry({
+    name: "workbuddy-ai",
+    hookExpected: true,
+    hookInstalled: Boolean(hookStatus?.["workbuddy-ai"]),
+    logsDir: workbuddyAiHome,
+    logsPredicate: (_full, name) => name === "projects" || name.endsWith(".jsonl"),
+    settingsPath: path.join(workbuddyAiHome, "settings.json"),
+  }));
+
   return out;
 }
 

package/src/lib/pricing/curated-overrides.json

@@ -3,9 +3,43 @@
     "note": "Curated price overrides. Always wins over LiteLLM. Two reasons to live here: (1) self-defined alias names that LiteLLM will never carry (kiro-*, hy3-*, composer-*, kimi-for-coding, free-tier OpenRouter routes); (2) prices we want to pin even if LiteLLM has the model (e.g. cache_write fields LiteLLM often omits). Units: USD per million tokens. Edit this file to override pricing without redeploying.",
     "units": "usd_per_million_tokens",
     "deepseek_v4_pricing_note": "Official DeepSeek V4 pricing verified on 2026-08-21. Peak hours are 01:00-04:00 and 06:00-10:00 UTC; all other hours are off-peak at half price. Exact entries store peak rates. computeRowCost applies the time-of-use discount from each row's UTC hour_start.",
-    "iflytek_maas_pricing_note": "iFlytek MaaS inference prices supplied for AStudio on 2026-08-31. AStudio homepage: https://agent.xfyun.cn/. Official pricing source: https://maas.xfyun.cn/modelSquare. Source values are RMB per million tokens, converted at 7.2 RMB/USD, then rounded to two decimal places for billing. Models without a dedicated cache-hit price use the normal input rate; cache writes also use the normal input rate. iFlytek MaaS prices are source-scoped and do not inherit DeepSeek time-of-use discounts."
+    "iflytek_maas_pricing_note": "iFlytek MaaS inference prices supplied for AStudio on 2026-08-31. AStudio homepage: https://agent.xfyun.cn/. Official pricing source: https://maas.xfyun.cn/modelSquare. Source values are RMB per million tokens, converted at 7.2 RMB/USD, then rounded to two decimal places for billing. Models without a dedicated cache-hit price use the normal input rate; cache writes also use the normal input rate. iFlytek MaaS prices are source-scoped and do not inherit DeepSeek time-of-use discounts. Spark-X2.5 entries (spark-x2.5, spark-x2.5-4b, spark-x2.5-1.7b) were read from the model square on 2026-10-08; spark-x2.5 is listed at a limited-time 30% rate (0.96 / 3.6 / 0.14 RMB input / output / cache-hit, list price 3.2 / 12 / 0.48), and the two small models are free."
   },
   "exact": {
+    "claude-opus-5-5": {
+      "input": 4,
+      "output": 20,
+      "cache_read": 0.2,
+      "cache_write": 5,
+      "note": "Official Anthropic pricing verified 2026-10-11: https://platform.claude.com/docs/en/about-claude/pricing"
+    },
+    "claude-opus-5.5": {
+      "input": 4,
+      "output": 20,
+      "cache_read": 0.2,
+      "cache_write": 5,
+      "note": "Official Anthropic pricing verified 2026-10-11: https://platform.claude.com/docs/en/about-claude/pricing"
+    },
+    "claude-sonnet-5-5": {
+      "input": 2,
+      "output": 10,
+      "cache_read": 0.1,
+      "cache_write": 2.5,
+      "note": "Official Anthropic pricing verified 2026-10-11: https://platform.claude.com/docs/en/about-claude/pricing"
+    },
+    "claude-sonnet-5.5": {
+      "input": 2,
+      "output": 10,
+      "cache_read": 0.1,
+      "cache_write": 2.5,
+      "note": "Official Anthropic pricing verified 2026-10-11: https://platform.claude.com/docs/en/about-claude/pricing"
+    },
+    "k3-256k": {
+      "input": 3,
+      "output": 15,
+      "cache_read": 0.3,
+      "note": "Bare alias emitted by Kimi Code (modelAlias \"kimi-code/k3\" -> \"k3\"). Same rates as kimi-k3."
+    },
     "cline-gateway-models": {
       "input": 0,
       "output": 0,
@@ -42,10 +76,10 @@
       "note": "Opus 4.8 not yet in LiteLLM. Pin to the standard Opus tier (same as 4.6/4.7); remove once LiteLLM carries it."
     },
     "claude-sonnet-5": {
-      "input": 3,
-      "output": 15,
-      "cache_read": 0.3,
-      "cache_write": 3.75,
+      "input": 2,
+      "output": 10,
+      "cache_read": 0.2,
+      "cache_write": 2.5,
       "note": "Sonnet 5. LiteLLM added this model on ~2026-06-30 with matching rates, but pin explicitly so pricing is correct immediately rather than depending on the 24h local pricing-cache refresh cycle."
     },
     "claude-haiku-4-5": {
@@ -482,6 +516,9 @@
   },
   "source_exact": {
     "acode": {
+      "spark-x2.5": { "input": 0.13, "output": 0.50, "cache_read": 0.02, "cache_write": 0.13 },
+      "spark-x2.5-4b": { "input": 0.00, "output": 0.00, "cache_read": 0.00, "cache_write": 0.00 },
+      "spark-x2.5-1.7b": { "input": 0.00, "output": 0.00, "cache_read": 0.00, "cache_write": 0.00 },
       "xopglm53": { "input": 1.11, "output": 3.89, "cache_read": 0.28, "cache_write": 1.11 },
       "xopdeepseekv4pro0813": { "input": 1.25, "output": 3.75, "cache_read": 0.04, "cache_write": 1.25 },
       "xopdeepseekv4flash0731": { "input": 0.14, "output": 0.28, "cache_read": 0.03, "cache_write": 0.14 },

package/src/lib/pricing/matcher.js

@@ -238,7 +238,13 @@ function lookupContainedExactCaseInsensitive(table, model) {
   const lower = model.toLowerCase();
   const keys = Object.keys(table).sort((a, b) => b.length - a.length);
   for (const key of keys) {
-    if (lower.includes(key.toLowerCase())) return table[key];
+    const start = lower.indexOf(key.toLowerCase());
+    if (start < 0) continue;
+    // A model generation must not match a later numeric version (Opus 5.5
+    // is a separate SKU from Opus 5). Dated and provider suffixes still work.
+    const suffix = lower.slice(start + key.length);
+    if (/^[.\d]/.test(suffix)) continue;
+    return table[key];
   }
   return null;
 }
@@ -256,6 +262,9 @@ function lookupPricing(model, { curated, litellm, source } = {}) {
   // 0. CURATED source exact. Source-specific prices apply only to their source,
   // preventing collisions with public prices for same-named models from other CLIs.
   const sourceKey = typeof source === "string" ? source.toLowerCase() : "";
+  if (sourceKey === "workbuddy-ai" && lower.trim() === "auto") {
+    return { hit: false, source: "miss", value: null };
+  }
   if (sourceKey === "cline" && lower.endsWith(":free")) {
     return {
       hit: true,
@@ -289,10 +298,6 @@ function lookupPricing(model, { curated, litellm, source } = {}) {
   if (curatedDotExact) {
     return { hit: true, source: "curated:exact-dot", value: curatedDotExact };
   }
-  const curatedDotContainedExact = lookupContainedExactCaseInsensitive(curated.exact, dotForm);
-  if (curatedDotContainedExact) {
-    return { hit: true, source: "curated:exact-dot", value: curatedDotContainedExact };
-  }
 
   // 2. LiteLLM exact
   if (litellm && litellm[lookupModel]) {
@@ -302,6 +307,10 @@ function lookupPricing(model, { curated, litellm, source } = {}) {
   if (litellmDotExact) {
     return { hit: true, source: "litellm:exact-dot", value: litellmDotExact };
   }
+  const curatedDotContainedExact = lookupContainedExactCaseInsensitive(curated.exact, dotForm);
+  if (curatedDotContainedExact) {
+    return { hit: true, source: "curated:exact-dot", value: curatedDotContainedExact };
+  }
 
   // 3. CURATED alias (literal mapping like "auto" -> "composer-1")
   if (curated.alias && curated.alias[lookupModel] && curated.exact[curated.alias[lookupModel]]) {