tokentracker-cli 0.21.0 → 0.21.1

package/src/lib/activation-check.js

@@ -1,341 +0,0 @@
-const os = require("node:os");
-const path = require("node:path");
-const fs = require("node:fs/promises");
-const { readJson } = require("./fs");
-
-/**
- * 跨 AI CLI 自动激活检测
- * 在 tokentracker 各种命令执行时顺带检测并完成配置
- */
-
-const AI_CLIS = [
-  {
-    name: "codex",
-    displayName: "Codex",
-    checkInstalled: checkCodexInstalled,
-    checkConfigured: checkCodexConfigured,
-    configure: configureCodex,
-  },
-  {
-    name: "claude-code",
-    displayName: "Claude Code", 
-    checkInstalled: checkClaudeCodeInstalled,
-    checkConfigured: checkClaudeCodeConfigured,
-    configure: configureClaudeCode,
-  },
-  {
-    name: "opencode",
-    displayName: "OpenCode",
-    checkInstalled: checkOpencodeInstalled,
-    checkConfigured: checkOpencodeConfigured,
-    configure: configureOpencode,
-  },
-  {
-    name: "every-code",
-    displayName: "Every Code",
-    checkInstalled: checkEveryCodeInstalled,
-    checkConfigured: checkEveryCodeConfigured,
-    configure: configureEveryCode,
-  },
-  {
-    name: "openclaw",
-    displayName: "OpenClaw",
-    checkInstalled: checkOpenclawInstalled,
-    checkConfigured: checkOpenclawConfigured,
-    configure: configureOpenclaw,
-  },
-];
-
-/**
- * 检测所有 AI CLI 并自动配置未完成的
- * @param {Object} options
- * @param {string} options.home - home目录
- * @param {boolean} options.silent - 是否静默模式
- * @param {boolean} options.autoConfigure - 是否自动配置（否则仅提示）
- */
-async function checkAndActivate({ home = os.homedir(), silent = true, autoConfigure = true } = {}) {
-  const results = [];
-  
-  for (const cli of AI_CLIS) {
-    try {
-      const isInstalled = await cli.checkInstalled({ home });
-      if (!isInstalled) continue;
-      
-      const isConfigured = await cli.checkConfigured({ home });
-      if (isConfigured) continue;
-      
-      // 发现已安装但未配置的 CLI
-      if (autoConfigure) {
-        const success = await cli.configure({ home, silent });
-        results.push({
-          name: cli.name,
-          displayName: cli.displayName,
-          action: success ? "configured" : "failed",
-        });
-        
-        if (!silent && success) {
-          console.log(`✅ 已自动配置 ${cli.displayName} 集成`);
-        }
-      } else {
-        results.push({
-          name: cli.name,
-          displayName: cli.displayName,
-          action: "pending",
-        });
-        
-        if (!silent) {
-          console.log(`⏳ 检测到 ${cli.displayName} 未配置，运行 'tokentracker init' 以配置`);
-        }
-      }
-    } catch (err) {
-      // 静默忽略错误，不影响主流程
-      if (!silent) {
-        console.error(`检查 ${cli.displayName} 失败:`, err.message);
-      }
-    }
-  }
-  
-  return results;
-}
-
-// ===== Codex 检测与配置 =====
-
-async function checkCodexInstalled({ home }) {
-  const configPath = path.join(home, ".codex", "config.toml");
-  try {
-    await fs.access(configPath);
-    return true;
-  } catch {
-    return false;
-  }
-}
-
-async function checkCodexConfigured({ home }) {
-  const configPath = path.join(home, ".codex", "config.toml");
-  try {
-    const content = await fs.readFile(configPath, "utf8");
-    // 检查是否已配置 notify
-    return content.includes("tokentracker") || content.includes("notify");
-  } catch {
-    return false;
-  }
-}
-
-async function configureCodex({ home, silent }) {
-  try {
-    // 使用现有的 codex-config 模块
-    const { upsertCodexNotify } = require("./codex-config");
-    const notifyCmd = path.join(home, ".tokentracker", "bin", "notify.cjs");
-    const codexConfigPath = path.join(home, ".codex", "config.toml");
-    const notifyOriginalPath = path.join(home, ".tokentracker", "backups", "codex-notify-original.json");
-    
-    await upsertCodexNotify({
-      codexConfigPath,
-      notifyCmd,
-      notifyOriginalPath,
-    });
-    return true;
-  } catch (err) {
-    if (!silent) console.error("配置 Codex 失败:", err.message);
-    return false;
-  }
-}
-
-// ===== Claude Code 检测与配置 =====
-
-async function checkClaudeCodeInstalled({ home }) {
-  const settingsPath = path.join(home, ".claude", "settings.json");
-  try {
-    await fs.access(settingsPath);
-    return true;
-  } catch {
-    return false;
-  }
-}
-
-async function checkClaudeCodeConfigured({ home }) {
-  const settingsPath = path.join(home, ".claude", "settings.json");
-  try {
-    const settings = await readJson(settingsPath);
-    // 检查是否已有 tokentracker 相关的 hook
-    const hooks = settings?.hooks?.SessionStart || [];
-    return hooks.some(h =>
-      h.hooks?.some(hook => hook.command?.includes("tokentracker"))
-    );
-  } catch {
-    return false;
-  }
-}
-
-async function configureClaudeCode({ home, silent }) {
-  try {
-    const settingsPath = path.join(home, ".claude", "settings.json");
-    const settings = (await readJson(settingsPath)) || {};
-    
-    // 添加 SessionStart hook
-    if (!settings.hooks) settings.hooks = {};
-    if (!settings.hooks.SessionStart) settings.hooks.SessionStart = [];
-    
-    // 检查是否已存在
-    const exists = settings.hooks.SessionStart.some(h => 
-      h.matcher === "startup" &&
-      h.hooks?.some(hook => hook.command?.includes("tokentracker activate-if-needed"))
-    );
-    
-    if (!exists) {
-      settings.hooks.SessionStart.push({
-        matcher: "startup",
-        hooks: [{
-          type: "command",
-          command: "tokentracker activate-if-needed --silent 2>/dev/null || true"
-        }]
-      });
-      
-      await fs.writeFile(settingsPath, JSON.stringify(settings, null, 2) + "\n", "utf8");
-    }
-    return true;
-  } catch (err) {
-    if (!silent) console.error("配置 Claude Code 失败:", err.message);
-    return false;
-  }
-}
-
-// ===== OpenCode 检测与配置 =====
-
-async function checkOpencodeInstalled({ home }) {
-  const configPath = path.join(home, ".config", "opencode", "opencode.json");
-  try {
-    await fs.access(configPath);
-    return true;
-  } catch {
-    return false;
-  }
-}
-
-async function checkOpencodeConfigured({ home }) {
-  const pluginDir = path.join(home, ".config", "opencode", "plugins");
-  try {
-    const files = await fs.readdir(pluginDir);
-    return files.some(f => f.includes("tokentracker"));
-  } catch {
-    return false;
-  }
-}
-
-async function configureOpencode({ home, silent }) {
-  try {
-    const pluginDir = path.join(home, ".config", "opencode", "plugins");
-    await fs.mkdir(pluginDir, { recursive: true });
-    
-    const pluginPath = path.join(pluginDir, "tokentracker-activation.js");
-    const pluginCode = `export const TokentrackerActivation = async ({ $ }) => {
-  return {
-    "session.created": async () => {
-      await $'tokentracker activate-if-needed --silent'.quiet().nothrow();
-    }
-  };
-};`;
-    
-    await fs.writeFile(pluginPath, pluginCode, "utf8");
-    return true;
-  } catch (err) {
-    if (!silent) console.error("配置 OpenCode 失败:", err.message);
-    return false;
-  }
-}
-
-// ===== Every Code 检测与配置 =====
-
-async function checkEveryCodeInstalled({ home }) {
-  const configPath = path.join(home, ".code", "config.toml");
-  try {
-    await fs.access(configPath);
-    return true;
-  } catch {
-    return false;
-  }
-}
-
-async function checkEveryCodeConfigured({ home }) {
-  const configPath = path.join(home, ".code", "config.toml");
-  try {
-    const content = await fs.readFile(configPath, "utf8");
-    return content.includes("tokentracker");
-  } catch {
-    return false;
-  }
-}
-
-async function configureEveryCode({ home, silent }) {
-  try {
-    // Every Code 配置类似 Codex
-    const configPath = path.join(home, ".code", "config.toml");
-    let content = "";
-    try {
-      content = await fs.readFile(configPath, "utf8");
-    } catch {
-      content = "";
-    }
-    
-    const notifyCmd = path.join(home, ".tokentracker", "bin", "notify.cjs");
-    const notifyLine = `notify = ["/usr/bin/env", "node", "${notifyCmd}"]`;
-
-    if (!content.includes("tokentracker")) {
-      content = content.trim() + "\n\n# tokentracker integration\n" + notifyLine + "\n";
-      await fs.writeFile(configPath, content, "utf8");
-    }
-    return true;
-  } catch (err) {
-    if (!silent) console.error("配置 Every Code 失败:", err.message);
-    return false;
-  }
-}
-
-module.exports = {
-  checkAndActivate,
-  AI_CLIS,
-};
-
-// ===== OpenClaw 检测与配置 =====
-
-async function checkOpenclawInstalled({ home }) {
-  const configPath = path.join(home, ".openclaw", "openclaw.json");
-  try {
-    await fs.access(configPath);
-    return true;
-  } catch {
-    return false;
-  }
-}
-
-async function checkOpenclawConfigured({ home }) {
-  const { probeOpenclawSessionPluginState } = require("./openclaw-session-plugin");
-  const { resolveTrackerPaths } = require("./tracker-paths");
-  try {
-    const { trackerDir } = await resolveTrackerPaths({ home });
-    const state = await probeOpenclawSessionPluginState({ home, trackerDir, env: process.env });
-    return state?.configured === true;
-  } catch {
-    return false;
-  }
-}
-
-async function configureOpenclaw({ home, silent }) {
-  try {
-    const { installOpenclawSessionPlugin } = require("./openclaw-session-plugin");
-    const { resolveTrackerPaths } = require("./tracker-paths");
-    const { trackerDir } = await resolveTrackerPaths({ home });
-    
-    const result = await installOpenclawSessionPlugin({
-      home,
-      trackerDir,
-      packageName: "tokentracker-cli",
-      env: process.env,
-    });
-    
-    return result?.configured === true;
-  } catch (err) {
-    if (!silent) console.error("配置 OpenClaw 失败:", err.message);
-    return false;
-  }
-}

package/src/lib/opencode-usage-audit.js

@@ -1,205 +0,0 @@
-const fs = require("node:fs/promises");
-const os = require("node:os");
-const path = require("node:path");
-
-const { listOpencodeMessageFiles, parseOpencodeIncremental } = require("./rollout");
-
-const BUCKET_SEPARATOR = "|";
-const DAY_RE = /^\d{4}-\d{2}-\d{2}$/;
-
-function formatHourKey(date) {
-  return `${date.getUTCFullYear()}-${String(date.getUTCMonth() + 1).padStart(2, "0")}-${String(
-    date.getUTCDate(),
-  ).padStart(2, "0")}T${String(date.getUTCHours()).padStart(2, "0")}:${String(
-    date.getUTCMinutes() >= 30 ? 30 : 0,
-  ).padStart(2, "0")}:00`;
-}
-
-function toBig(value) {
-  if (typeof value === "bigint") return value;
-  if (typeof value === "number") return BigInt(value);
-  if (typeof value === "string" && value.trim()) return BigInt(value);
-  return 0n;
-}
-
-function addTotals(target, delta) {
-  target.input_tokens += toBig(delta.input_tokens);
-  target.cached_input_tokens += toBig(delta.cached_input_tokens);
-  target.output_tokens += toBig(delta.output_tokens);
-  target.reasoning_output_tokens += toBig(delta.reasoning_output_tokens);
-  target.total_tokens += toBig(delta.total_tokens);
-}
-
-async function buildLocalHourlyTotals({ storageDir, source = "opencode" }) {
-  const messageFiles = await listOpencodeMessageFiles(storageDir);
-  const queuePath = path.join(
-    os.tmpdir(),
-    `tokentracker-opencode-audit-${process.pid}-${Date.now()}.jsonl`,
-  );
-  const cursors = { version: 1, files: {}, hourly: null, opencode: null };
-
-  await parseOpencodeIncremental({ messageFiles, cursors, queuePath, source });
-  await fs.rm(queuePath, { force: true }).catch(() => {});
-
-  const byHour = new Map();
-  let minDay = null;
-  let maxDay = null;
-
-  for (const [key, bucket] of Object.entries(cursors.hourly?.buckets || {})) {
-    const [bucketSource, , hourStart] = String(key).split(BUCKET_SEPARATOR);
-    if (bucketSource !== source || !hourStart) continue;
-    const dt = new Date(hourStart);
-    if (!Number.isFinite(dt.getTime())) continue;
-    const hourKey = formatHourKey(dt);
-    const dayKey = hourKey.slice(0, 10);
-
-    if (!minDay || dayKey < minDay) minDay = dayKey;
-    if (!maxDay || dayKey > maxDay) maxDay = dayKey;
-
-    const totals = byHour.get(hourKey) || {
-      input_tokens: 0n,
-      cached_input_tokens: 0n,
-      output_tokens: 0n,
-      reasoning_output_tokens: 0n,
-      total_tokens: 0n,
-    };
-    addTotals(totals, bucket.totals || {});
-    byHour.set(hourKey, totals);
-  }
-
-  return { byHour, minDay, maxDay };
-}
-
-function normalizeServerRows(rows) {
-  const map = new Map();
-  for (const row of rows || []) {
-    if (!row || !row.hour) continue;
-    map.set(row.hour, {
-      missing: Boolean(row.missing),
-      totals: {
-        input_tokens: toBig(row.input_tokens),
-        cached_input_tokens: toBig(row.cached_input_tokens),
-        output_tokens: toBig(row.output_tokens),
-        reasoning_output_tokens: toBig(row.reasoning_output_tokens),
-        total_tokens: toBig(row.total_tokens),
-      },
-    });
-  }
-  return map;
-}
-
-function diffTotals(local, server) {
-  return {
-    input_tokens: local.input_tokens - server.input_tokens,
-    cached_input_tokens: local.cached_input_tokens - server.cached_input_tokens,
-    output_tokens: local.output_tokens - server.output_tokens,
-    reasoning_output_tokens: local.reasoning_output_tokens - server.reasoning_output_tokens,
-    total_tokens: local.total_tokens - server.total_tokens,
-  };
-}
-
-function maxAbsDelta(delta) {
-  return [
-    delta.input_tokens,
-    delta.cached_input_tokens,
-    delta.output_tokens,
-    delta.reasoning_output_tokens,
-    delta.total_tokens,
-  ].reduce((acc, value) => {
-    const abs = value < 0n ? -value : value;
-    return abs > acc ? abs : acc;
-  }, 0n);
-}
-
-function isValidDay(value) {
-  if (!DAY_RE.test(value)) return false;
-  const dt = new Date(`${value}T00:00:00.000Z`);
-  return Number.isFinite(dt.getTime());
-}
-
-function listDays(from, to) {
-  if (!isValidDay(from) || !isValidDay(to)) return [];
-  if (from > to) return [];
-  const out = [];
-  const start = new Date(`${from}T00:00:00.000Z`);
-  const end = new Date(`${to}T00:00:00.000Z`);
-  for (let dt = start; dt <= end; dt = new Date(dt.getTime() + 24 * 60 * 60 * 1000)) {
-    const day = `${dt.getUTCFullYear()}-${String(dt.getUTCMonth() + 1).padStart(2, "0")}-${String(
-      dt.getUTCDate(),
-    ).padStart(2, "0")}`;
-    out.push(day);
-  }
-  return out;
-}
-
-async function auditOpencodeUsage({ storageDir, from, to, fetchHourly, includeMissing = false }) {
-  const local = await buildLocalHourlyTotals({ storageDir, source: "opencode" });
-  if (!local.minDay || !local.maxDay) {
-    throw new Error("No local opencode data found");
-  }
-
-  const fromDay = from || local.minDay;
-  const toDay = to || local.maxDay;
-  const days = listDays(fromDay, toDay);
-  if (days.length === 0) {
-    throw new Error("Invalid date range for audit");
-  }
-
-  const diffs = [];
-  let matched = 0;
-  let mismatched = 0;
-  let incomplete = 0;
-  let maxDelta = 0n;
-
-  for (const day of days) {
-    const server = await fetchHourly(day);
-    const serverByHour = normalizeServerRows(server?.data || []);
-    for (let h = 0; h < 24; h++) {
-      for (const m of [0, 30]) {
-        const hourKey = `${day}T${String(h).padStart(2, "0")}:${String(m).padStart(2, "0")}:00`;
-        const localTotals = local.byHour.get(hourKey) || {
-          input_tokens: 0n,
-          cached_input_tokens: 0n,
-          output_tokens: 0n,
-          reasoning_output_tokens: 0n,
-          total_tokens: 0n,
-        };
-        const serverEntry = serverByHour.get(hourKey) || null;
-        if (serverEntry?.missing && !includeMissing) {
-          incomplete += 1;
-          continue;
-        }
-        const serverTotals = serverEntry?.totals || {
-          input_tokens: 0n,
-          cached_input_tokens: 0n,
-          output_tokens: 0n,
-          reasoning_output_tokens: 0n,
-          total_tokens: 0n,
-        };
-        const delta = diffTotals(localTotals, serverTotals);
-        const deltaMax = maxAbsDelta(delta);
-        if (deltaMax === 0n) {
-          matched += 1;
-        } else {
-          mismatched += 1;
-          if (deltaMax > maxDelta) maxDelta = deltaMax;
-          diffs.push({ hour: hourKey, local: localTotals, server: serverTotals, delta });
-        }
-      }
-    }
-  }
-
-  return {
-    summary: {
-      days: days.length,
-      slots: days.length * 48,
-      matched,
-      mismatched,
-      incomplete,
-      maxDelta,
-    },
-    diffs,
-  };
-}
-
-module.exports = { auditOpencodeUsage, buildLocalHourlyTotals };