@yishiguji/tokenarena 0.10.0 → 0.11.0

package/dist/index.js

@@ -4713,37 +4713,185 @@ var GrokBuildParser = class {
 };
 registerParser(new GrokBuildParser());
 
+// src/parsers/atomcode.ts
+import { existsSync as existsSync24 } from "fs";
+import { homedir as homedir25 } from "os";
+import { basename as basename11, join as join26 } from "path";
+var TOOL_ID17 = "atomcode";
+var TOOL_NAME17 = "AtomCode";
+var DEFAULT_SESSIONS_DIR6 = join26(homedir25(), ".atomcode", "sessions");
+function getAtomCodeSessionsDirs(env = process.env) {
+  const dirs = [
+    env.TOKEN_ARENA_ATOMCODE_DIR,
+    env.ATOMCODE_HOME ? join26(env.ATOMCODE_HOME, "sessions") : void 0,
+    DEFAULT_SESSIONS_DIR6
+  ].filter((value) => Boolean(value));
+  return Array.from(new Set(dirs));
+}
+function toNonNegativeNumber3(value) {
+  const numberValue = Number(value);
+  return Number.isFinite(numberValue) && numberValue >= 0 ? numberValue : 0;
+}
+function parseTimestamp3(value) {
+  if (typeof value === "number" && Number.isFinite(value)) {
+    const timestamp = new Date(value);
+    return Number.isNaN(timestamp.getTime()) ? null : timestamp;
+  }
+  if (typeof value === "string" && value.trim()) {
+    const asNumber = Number(value);
+    if (Number.isFinite(asNumber)) {
+      const timestamp2 = new Date(asNumber);
+      if (!Number.isNaN(timestamp2.getTime())) {
+        return timestamp2;
+      }
+    }
+    const timestamp = new Date(value);
+    return Number.isNaN(timestamp.getTime()) ? null : timestamp;
+  }
+  return null;
+}
+function parseMeta(content) {
+  if (!content) return null;
+  try {
+    const parsed = JSON.parse(content);
+    return parsed && typeof parsed === "object" ? parsed : null;
+  } catch {
+    return null;
+  }
+}
+function getMetaModel(meta) {
+  for (const turn of meta?.turn_stats ?? []) {
+    for (const usage of turn.model_usage ?? []) {
+      if (typeof usage.model_id === "string" && usage.model_id) {
+        return usage.model_id;
+      }
+    }
+  }
+  return "unknown";
+}
+function getMetaProject(meta) {
+  if (typeof meta?.working_dir === "string" && meta.working_dir) {
+    return basename11(meta.working_dir) || "unknown";
+  }
+  return "unknown";
+}
+var AtomCodeParser = class {
+  tool;
+  sessionsDirs;
+  constructor(sessionsDir) {
+    this.sessionsDirs = sessionsDir ? [sessionsDir] : getAtomCodeSessionsDirs();
+    this.tool = {
+      id: TOOL_ID17,
+      name: TOOL_NAME17,
+      dataDir: this.sessionsDirs[0] ?? DEFAULT_SESSIONS_DIR6
+    };
+  }
+  async parse() {
+    const entries = [];
+    const sessionEvents = [];
+    const seenEntryKeys = /* @__PURE__ */ new Set();
+    for (const sessionsDir of this.sessionsDirs) {
+      for (const filePath of findJsonlFiles(sessionsDir)) {
+        const content = readFileSafe(filePath);
+        if (!content) continue;
+        const rows = parseJsonl(content);
+        if (rows.length === 0) continue;
+        const fallbackSessionId = extractSessionId(filePath);
+        const meta = parseMeta(
+          readFileSafe(filePath.replace(/\.jsonl$/, ".meta"))
+        );
+        const project = getMetaProject(meta);
+        const model = getMetaModel(meta);
+        for (const row of rows) {
+          const sessionId = typeof row.session_id === "string" && row.session_id ? row.session_id : fallbackSessionId;
+          const timestamp = parseTimestamp3(row.ts);
+          if (!timestamp) continue;
+          if (row.user !== void 0 || row.assistant !== void 0) {
+            sessionEvents.push({
+              sessionId,
+              source: TOOL_ID17,
+              project,
+              timestamp,
+              role: row.user !== void 0 ? "user" : "assistant"
+            });
+          }
+          const usage = row.usage;
+          if (!usage) continue;
+          const prompt = toNonNegativeNumber3(usage.prompt);
+          const completion = toNonNegativeNumber3(usage.completion);
+          const cached = toNonNegativeNumber3(usage.cached);
+          const inputTokens = Math.max(0, prompt - cached);
+          if (inputTokens + completion + cached === 0) {
+            continue;
+          }
+          const entryKey = [
+            sessionId,
+            timestamp.toISOString(),
+            model,
+            inputTokens,
+            completion,
+            cached
+          ].join("|");
+          if (seenEntryKeys.has(entryKey)) {
+            continue;
+          }
+          seenEntryKeys.add(entryKey);
+          entries.push({
+            sessionId,
+            source: TOOL_ID17,
+            model,
+            project,
+            timestamp,
+            inputTokens,
+            outputTokens: completion,
+            reasoningTokens: 0,
+            cachedTokens: cached
+          });
+        }
+      }
+    }
+    return {
+      buckets: aggregateToBuckets(entries),
+      sessions: extractSessions(sessionEvents, entries)
+    };
+  }
+  isInstalled() {
+    return this.sessionsDirs.some((dir) => existsSync24(dir));
+  }
+};
+registerParser(new AtomCodeParser());
+
 // src/cli.ts
 import { Command, Option } from "commander";
 
 // src/infrastructure/config/manager.ts
 import { randomUUID } from "crypto";
 import {
-  existsSync as existsSync24,
+  existsSync as existsSync25,
   mkdirSync,
   readFileSync as readFileSync9,
   unlinkSync,
   writeFileSync
 } from "fs";
-import { join as join27 } from "path";
+import { join as join28 } from "path";
 
 // src/infrastructure/xdg.ts
-import { homedir as homedir25 } from "os";
-import { join as join26 } from "path";
+import { homedir as homedir26 } from "os";
+import { join as join27 } from "path";
 function getConfigHome() {
-  return process.env.XDG_CONFIG_HOME || join26(homedir25(), ".config");
+  return process.env.XDG_CONFIG_HOME || join27(homedir26(), ".config");
 }
 function getStateHome() {
-  return process.env.XDG_STATE_HOME || join26(homedir25(), ".local", "state");
+  return process.env.XDG_STATE_HOME || join27(homedir26(), ".local", "state");
 }
 function getRuntimeDir() {
   return process.env.XDG_RUNTIME_DIR || getStateHome();
 }
 
 // src/infrastructure/config/manager.ts
-var CONFIG_DIR = join27(getConfigHome(), "tokenarena");
+var CONFIG_DIR = join28(getConfigHome(), "tokenarena");
 var isDev = process.env.TOKEN_ARENA_DEV === "1";
-var CONFIG_FILE = join27(CONFIG_DIR, isDev ? "config.dev.json" : "config.json");
+var CONFIG_FILE = join28(CONFIG_DIR, isDev ? "config.dev.json" : "config.json");
 var DEFAULT_API_URL = "https://token.guji.uno";
 var VALID_CONFIG_KEYS = [
   "apiKey",
@@ -4759,7 +4907,7 @@ function getConfigDir() {
   return CONFIG_DIR;
 }
 function loadConfig() {
-  if (!existsSync24(CONFIG_FILE)) return null;
+  if (!existsSync25(CONFIG_FILE)) return null;
   try {
     const raw = readFileSync9(CONFIG_FILE, "utf-8");
     const config = JSON.parse(raw);
@@ -4777,7 +4925,7 @@ function saveConfig(config) {
 `, "utf-8");
 }
 function deleteConfig() {
-  if (existsSync24(CONFIG_FILE)) {
+  if (existsSync25(CONFIG_FILE)) {
     unlinkSync(CONFIG_FILE);
   }
 }
@@ -5639,7 +5787,7 @@ var ApiClient = class {
 // src/infrastructure/runtime/lock.ts
 import {
   closeSync,
-  existsSync as existsSync25,
+  existsSync as existsSync26,
   openSync,
   readFileSync as readFileSync10,
   rmSync as rmSync3,
@@ -5648,22 +5796,22 @@ import {
 
 // src/infrastructure/runtime/paths.ts
 import { mkdirSync as mkdirSync2 } from "fs";
-import { join as join28 } from "path";
+import { join as join29 } from "path";
 var APP_NAME = "tokenarena";
 function getRuntimeDirPath() {
-  return join28(getRuntimeDir(), APP_NAME);
+  return join29(getRuntimeDir(), APP_NAME);
 }
 function getStateDir() {
-  return join28(getStateHome(), APP_NAME);
+  return join29(getStateHome(), APP_NAME);
 }
 function getSyncLockPath() {
-  return join28(getRuntimeDirPath(), "sync.lock");
+  return join29(getRuntimeDirPath(), "sync.lock");
 }
 function getSyncStatePath() {
-  return join28(getStateDir(), "status.json");
+  return join29(getStateDir(), "status.json");
 }
 function getUploadManifestPath() {
-  return join28(getStateDir(), "upload-manifest.json");
+  return join29(getStateDir(), "upload-manifest.json");
 }
 function ensureAppDirs() {
   mkdirSync2(getRuntimeDirPath(), { recursive: true });
@@ -5681,7 +5829,7 @@ function isProcessAlive(pid) {
   }
 }
 function readLockMetadata(lockPath) {
-  if (!existsSync25(lockPath)) {
+  if (!existsSync26(lockPath)) {
     return null;
   }
   try {
@@ -5755,13 +5903,13 @@ function describeExistingSyncLock() {
 }
 
 // src/infrastructure/runtime/state.ts
-import { existsSync as existsSync26, readFileSync as readFileSync11, writeFileSync as writeFileSync3 } from "fs";
+import { existsSync as existsSync27, readFileSync as readFileSync11, writeFileSync as writeFileSync3 } from "fs";
 function getDefaultState() {
   return { status: "idle" };
 }
 function loadSyncState() {
   const path = getSyncStatePath();
-  if (!existsSync26(path)) {
+  if (!existsSync27(path)) {
     return getDefaultState();
   }
   try {
@@ -5822,7 +5970,7 @@ function markSyncFailed(source, error, status) {
 }
 
 // src/infrastructure/runtime/upload-manifest.ts
-import { existsSync as existsSync27, readFileSync as readFileSync12, writeFileSync as writeFileSync4 } from "fs";
+import { existsSync as existsSync28, readFileSync as readFileSync12, writeFileSync as writeFileSync4 } from "fs";
 function isRecordOfStrings(value) {
   if (!value || typeof value !== "object" || Array.isArray(value)) {
     return false;
@@ -5838,7 +5986,7 @@ function isUploadManifest(value) {
 }
 function loadUploadManifest() {
   const path = getUploadManifestPath();
-  if (!existsSync27(path)) {
+  if (!existsSync28(path)) {
     return null;
   }
   try {
@@ -6378,18 +6526,18 @@ View your dashboard at: ${apiUrl}/usage`);
 
 // src/commands/init.ts
 import { execFileSync as execFileSync7, spawn } from "child_process";
-import { existsSync as existsSync30 } from "fs";
+import { existsSync as existsSync31 } from "fs";
 import { appendFile, mkdir, readFile } from "fs/promises";
-import { homedir as homedir28, platform as platform5 } from "os";
-import { dirname as dirname6, join as join29, posix as posix3, win32 } from "path";
+import { homedir as homedir29, platform as platform5 } from "os";
+import { dirname as dirname6, join as join30, posix as posix3, win32 } from "path";
 
 // src/infrastructure/service/index.ts
 import { platform as platform4 } from "os";
 
 // src/infrastructure/service/linux-systemd.ts
 import { execFileSync as execFileSync5 } from "child_process";
-import { existsSync as existsSync28, mkdirSync as mkdirSync3, rmSync as rmSync4, writeFileSync as writeFileSync5 } from "fs";
-import { homedir as homedir26, platform as platform2 } from "os";
+import { existsSync as existsSync29, mkdirSync as mkdirSync3, rmSync as rmSync4, writeFileSync as writeFileSync5 } from "fs";
+import { homedir as homedir27, platform as platform2 } from "os";
 import { posix } from "path";
 
 // src/utils/command.ts
@@ -6458,10 +6606,10 @@ function escapeXml(value) {
 
 // src/infrastructure/service/linux-systemd.ts
 var SYSTEMD_SERVICE_NAME = "tokenarena";
-function getLinuxSystemdServiceDir(homePath = homedir26()) {
+function getLinuxSystemdServiceDir(homePath = homedir27()) {
   return posix.join(homePath, ".config", "systemd", "user");
 }
-function getLinuxSystemdServiceFile(homePath = homedir26()) {
+function getLinuxSystemdServiceFile(homePath = homedir27()) {
   return posix.join(
     getLinuxSystemdServiceDir(homePath),
     `${SYSTEMD_SERVICE_NAME}.service`
@@ -6518,7 +6666,7 @@ function ensureSystemdAvailable() {
 }
 function createLinuxSystemdServiceBackend() {
   function isInstalled() {
-    return existsSync28(getLinuxSystemdServiceFile());
+    return existsSync29(getLinuxSystemdServiceFile());
   }
   async function setup(skipPrompt = false) {
     if (!ensureSystemdAvailable()) {
@@ -6643,7 +6791,7 @@ function createLinuxSystemdServiceBackend() {
   }
   async function uninstall(skipPrompt = false) {
     const serviceFile = getLinuxSystemdServiceFile();
-    if (!existsSync28(serviceFile)) {
+    if (!existsSync29(serviceFile)) {
       logger.info(formatBullet("\u670D\u52A1\u6587\u4EF6\u4E0D\u5B58\u5728\u3002", "warning"));
       return;
     }
@@ -6704,17 +6852,17 @@ function createLinuxSystemdServiceBackend() {
 
 // src/infrastructure/service/macos-launchd.ts
 import { execFileSync as execFileSync6 } from "child_process";
-import { existsSync as existsSync29, mkdirSync as mkdirSync4, rmSync as rmSync5, writeFileSync as writeFileSync6 } from "fs";
-import { homedir as homedir27, platform as platform3 } from "os";
+import { existsSync as existsSync30, mkdirSync as mkdirSync4, rmSync as rmSync5, writeFileSync as writeFileSync6 } from "fs";
+import { homedir as homedir28, platform as platform3 } from "os";
 import { posix as posix2 } from "path";
 var MACOS_LAUNCHD_LABEL = "com.guji.tokenarena";
 function getCurrentUid() {
   return typeof process.getuid === "function" ? process.getuid() : null;
 }
-function getMacosLaunchAgentDir(homePath = homedir27()) {
+function getMacosLaunchAgentDir(homePath = homedir28()) {
   return posix2.join(homePath, "Library", "LaunchAgents");
 }
-function getMacosLaunchAgentFile(homePath = homedir27()) {
+function getMacosLaunchAgentFile(homePath = homedir28()) {
   return posix2.join(
     getMacosLaunchAgentDir(homePath),
     `${MACOS_LAUNCHD_LABEL}.plist`
@@ -6841,7 +6989,7 @@ function writeLaunchAgentPlist() {
     label: MACOS_LAUNCHD_LABEL,
     programArguments: [command.execPath, ...command.args],
     environment: getManagedServiceEnvironment(),
-    workingDirectory: homedir27(),
+    workingDirectory: homedir28(),
     standardOutPath: stdoutPath,
     standardErrorPath: stderrPath
   });
@@ -6866,7 +7014,7 @@ function bootstrapLaunchAgent() {
 }
 function createMacosLaunchdServiceBackend() {
   function isInstalled() {
-    return existsSync29(getMacosLaunchAgentFile());
+    return existsSync30(getMacosLaunchAgentFile());
   }
   async function setup(skipPrompt = false) {
     if (!ensureLaunchctlAvailable()) {
@@ -7007,7 +7155,7 @@ function createMacosLaunchdServiceBackend() {
   }
   async function uninstall(skipPrompt = false) {
     const plistFile = getMacosLaunchAgentFile();
-    if (!existsSync29(plistFile)) {
+    if (!existsSync30(plistFile)) {
       logger.info(formatBullet("\u670D\u52A1\u6587\u4EF6\u4E0D\u5B58\u5728\u3002", "warning"));
       return;
     }
@@ -7118,7 +7266,7 @@ function resolvePowerShellProfilePath() {
   const systemRoot = process.env.SYSTEMROOT || "C:\\Windows";
   const candidates = [
     "pwsh.exe",
-    join29(systemRoot, "System32", "WindowsPowerShell", "v1.0", "powershell.exe")
+    join30(systemRoot, "System32", "WindowsPowerShell", "v1.0", "powershell.exe")
   ];
   for (const command of candidates) {
     try {
@@ -7147,8 +7295,8 @@ function resolvePowerShellProfilePath() {
 function resolveShellAliasSetup(options = {}) {
   const currentPlatform = options.currentPlatform ?? platform5();
   const env = options.env ?? process.env;
-  const homeDir = options.homeDir ?? homedir28();
-  const pathExists = options.exists ?? existsSync30;
+  const homeDir = options.homeDir ?? homedir29();
+  const pathExists = options.exists ?? existsSync31;
   const shellFromEnv = env.SHELL ? basenameLikeShell(env.SHELL).toLowerCase() : "";
   const shellName = shellFromEnv || (currentPlatform === "win32" ? "powershell" : "");
   const aliasName = "ta";
@@ -7345,7 +7493,7 @@ async function setupShellAlias() {
   try {
     await mkdir(dirname6(setup.configFile), { recursive: true });
     let existingContent = "";
-    if (existsSync30(setup.configFile)) {
+    if (existsSync31(setup.configFile)) {
       existingContent = await readFile(setup.configFile, "utf-8");
     }
     const normalizedContent = existingContent.toLowerCase();
@@ -7604,7 +7752,7 @@ function buildLocalUsageDashboardData(input2) {
 
 // src/infrastructure/runtime/cli-version.ts
 import { readFileSync as readFileSync13 } from "fs";
-import { dirname as dirname7, join as join30 } from "path";
+import { dirname as dirname7, join as join31 } from "path";
 import { fileURLToPath } from "url";
 var FALLBACK_VERSION = "0.0.0";
 var cachedVersion;
@@ -7612,7 +7760,7 @@ function getCliVersion(metaUrl = import.meta.url) {
   if (cachedVersion) {
     return cachedVersion;
   }
-  const packageJsonPath = join30(
+  const packageJsonPath = join31(
     dirname7(fileURLToPath(metaUrl)),
     "..",
     "package.json"
@@ -8023,8 +8171,8 @@ async function runSyncCommand(opts = {}) {
 }
 
 // src/commands/uninstall.ts
-import { existsSync as existsSync31, readFileSync as readFileSync14, rmSync as rmSync6, writeFileSync as writeFileSync7 } from "fs";
-import { homedir as homedir29, platform as platform6 } from "os";
+import { existsSync as existsSync32, readFileSync as readFileSync14, rmSync as rmSync6, writeFileSync as writeFileSync7 } from "fs";
+import { homedir as homedir30, platform as platform6 } from "os";
 function removeShellAlias() {
   const shell = process.env.SHELL;
   if (!shell) return;
@@ -8033,22 +8181,22 @@ function removeShellAlias() {
   let configFile;
   switch (shellName) {
     case "zsh":
-      configFile = `${homedir29()}/.zshrc`;
+      configFile = `${homedir30()}/.zshrc`;
       break;
     case "bash":
-      if (platform6() === "darwin" && existsSync31(`${homedir29()}/.bash_profile`)) {
-        configFile = `${homedir29()}/.bash_profile`;
+      if (platform6() === "darwin" && existsSync32(`${homedir30()}/.bash_profile`)) {
+        configFile = `${homedir30()}/.bash_profile`;
       } else {
-        configFile = `${homedir29()}/.bashrc`;
+        configFile = `${homedir30()}/.bashrc`;
       }
       break;
     case "fish":
-      configFile = `${homedir29()}/.config/fish/config.fish`;
+      configFile = `${homedir30()}/.config/fish/config.fish`;
       break;
     default:
       return;
   }
-  if (!existsSync31(configFile)) return;
+  if (!existsSync32(configFile)) return;
   try {
     let content = readFileSync14(configFile, "utf-8");
     const aliasPatterns = [
@@ -8087,7 +8235,7 @@ async function runUninstall() {
   const runtimeDir = getRuntimeDirPath();
   const serviceBackend = getServiceBackend();
   const hasInstalledService = serviceBackend?.isInstalled() ?? false;
-  const hasLocalArtifacts = existsSync31(configPath) || existsSync31(configDir) || existsSync31(stateDir) || existsSync31(runtimeDir) || hasInstalledService;
+  const hasLocalArtifacts = existsSync32(configPath) || existsSync32(configDir) || existsSync32(stateDir) || existsSync32(runtimeDir) || hasInstalledService;
   if (!hasLocalArtifacts) {
     logger.info(formatHeader("\u5378\u8F7D TokenArena"));
     logger.info(formatBullet("\u672A\u53D1\u73B0\u672C\u5730\u914D\u7F6E\uFF0C\u65E0\u9700\u5378\u8F7D\u3002"));
@@ -8131,22 +8279,22 @@ async function runUninstall() {
     }
   }
   logger.info(formatSection("\u6267\u884C\u7ED3\u679C"));
-  if (existsSync31(configPath)) {
+  if (existsSync32(configPath)) {
     deleteConfig();
     logger.info(formatBullet("\u5DF2\u5220\u9664\u914D\u7F6E\u6587\u4EF6\u3002", "success"));
   }
-  if (existsSync31(configDir)) {
+  if (existsSync32(configDir)) {
     try {
       rmSync6(configDir, { recursive: false, force: true });
       logger.info(formatBullet("\u5DF2\u5220\u9664\u914D\u7F6E\u76EE\u5F55\u3002", "success"));
     } catch {
     }
   }
-  if (existsSync31(stateDir)) {
+  if (existsSync32(stateDir)) {
     rmSync6(stateDir, { recursive: true, force: true });
     logger.info(formatBullet("\u5DF2\u5220\u9664\u72B6\u6001\u6570\u636E\u3002", "success"));
   }
-  if (existsSync31(runtimeDir)) {
+  if (existsSync32(runtimeDir)) {
     rmSync6(runtimeDir, { recursive: true, force: true });
     logger.info(formatBullet("\u5DF2\u5220\u9664\u8FD0\u884C\u65F6\u6570\u636E\u3002", "success"));
   }
@@ -8377,7 +8525,7 @@ function createCli() {
 }
 
 // src/infrastructure/runtime/main-module.ts
-import { existsSync as existsSync32, realpathSync as realpathSync2 } from "fs";
+import { existsSync as existsSync33, realpathSync as realpathSync2 } from "fs";
 import { resolve as resolve3 } from "path";
 import { fileURLToPath as fileURLToPath2 } from "url";
 function isMainModule(argvEntry = process.argv[1], metaUrl = import.meta.url) {
@@ -8388,7 +8536,7 @@ function isMainModule(argvEntry = process.argv[1], metaUrl = import.meta.url) {
   try {
     return realpathSync2(argvEntry) === realpathSync2(currentModulePath);
   } catch {
-    if (!existsSync32(argvEntry)) {
+    if (!existsSync33(argvEntry)) {
       return false;
     }
     return resolve3(argvEntry) === resolve3(currentModulePath);