getaimeter 0.2.0 → 0.2.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "getaimeter",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "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,37 @@ 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';
+
+  // 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;
 }
 
 // ---------------------------------------------------------------------------