getaimeter 0.2.0 → 0.2.2

package/cli.js

@@ -5,6 +5,7 @@ const { getApiKey, saveApiKey, getWatchPaths, AIMETER_DIR } = require('./config'
 const { startWatching }   = require('./watcher');
 const { install, uninstall, isInstalled, startNow, stopNow } = require('./service');
 const { checkForUpdate, getCurrentVersion } = require('./update-check');
+const { startTray, stopTray } = require('./tray');
 
 const command = process.argv[2] || 'help';
 
@@ -187,8 +188,18 @@ function runWatch() {
 
   const cleanup = startWatching();
 
+  // Launch system tray icon (non-blocking, fails silently if systray2 not available)
+  const showTray = !process.argv.includes('--no-tray');
+  if (showTray) {
+    startTray(() => {
+      cleanup();
+      try { fs.unlinkSync(lockFile); } catch {}
+    }).catch(() => {}); // ignore tray errors
+  }
+
   const cleanupAll = () => {
     cleanup();
+    stopTray();
     try { fs.unlinkSync(lockFile); } catch {}
     process.exit(0);
   };
@@ -196,6 +207,7 @@ function runWatch() {
   process.on('SIGINT',  cleanupAll);
   process.on('SIGTERM', cleanupAll);
   process.on('exit', () => {
+    stopTray();
     try { fs.unlinkSync(lockFile); } catch {}
   });
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "getaimeter",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "description": "Track your Claude AI usage across CLI, VS Code, and Desktop App. One command to start.",
   "bin": {
     "aimeter": "cli.js"

package/watcher.js

@@ -31,19 +31,55 @@ function logError(...args) {
 // Source detection from file path
 // ---------------------------------------------------------------------------
 
+// Cache detected sources per file to avoid re-reading headers
+const _sourceCache = new Map();
+
 function detectSource(filePath) {
+  if (_sourceCache.has(filePath)) return _sourceCache.get(filePath);
+
   const normalized = filePath.replace(/\\/g, '/');
-  if (normalized.includes('local-agent-mode-sessions')) return 'desktop_agent_mode';
-  // VS Code uses lowercase 'c' in the sanitized project dir name on Windows
-  const projectsMatch = normalized.match(/\.claude\/projects\/([^/])/);
-  if (projectsMatch) {
-    const firstChar = projectsMatch[1];
-    // On Windows: CLI produces uppercase (C--Users), VS Code produces lowercase (c--Users)
-    if (firstChar === firstChar.toLowerCase() && firstChar !== firstChar.toUpperCase()) {
-      return 'claude_code_vscode';
-    }
+  if (normalized.includes('local-agent-mode-sessions')) {
+    _sourceCache.set(filePath, 'desktop_app');
+    return 'desktop_app';
   }
-  return 'claude_code_cli';
+
+  // For subagent files, inherit the parent session's source
+  if (normalized.includes('/subagents/')) {
+    const parentDir = normalized.replace(/\/[^/]+\/subagents\/.*$/, '');
+    // Find the parent JSONL (same dir, .jsonl file)
+    try {
+      const parentDirNative = parentDir.replace(/\//g, path.sep);
+      const entries = fs.readdirSync(parentDirNative);
+      for (const entry of entries) {
+        if (entry.endsWith('.jsonl')) {
+          const parentFile = path.join(parentDirNative, entry);
+          const parentSource = detectSource(parentFile); // recursive, will cache
+          _sourceCache.set(filePath, parentSource);
+          return parentSource;
+        }
+      }
+    } catch {}
+  }
+
+  // Read first 10KB of the file to find entrypoint or IDE markers
+  let source = 'cli'; // default
+  try {
+    const fd = fs.openSync(filePath, 'r');
+    const buf = Buffer.alloc(Math.min(10240, fs.fstatSync(fd).size));
+    fs.readSync(fd, buf, 0, buf.length, 0);
+    fs.closeSync(fd);
+    const header = buf.toString('utf8');
+
+    if (header.includes('"entrypoint":"claude-desktop"')) {
+      source = 'desktop_app';
+    } else if (header.includes('ide_opened_file') || header.includes('"entrypoint":"vscode"')) {
+      source = 'vscode';
+    }
+    // else remains 'cli'
+  } catch {}
+
+  _sourceCache.set(filePath, source);
+  return source;
 }
 
 // ---------------------------------------------------------------------------