getaimeter 0.7.7 → 0.7.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "getaimeter",
-  "version": "0.7.7",
+  "version": "0.7.9",
   "description": "Track AI coding costs across Claude, Cursor, Codex, and Gemini. Optimization recommendations that cut costs by 30%.",
   "bin": {
     "aimeter": "cli.js"

package/tray.ps1

@@ -125,7 +125,7 @@ $checkUpdateItem.Add_Click({
                     # Update and auto-restart: stop watcher, update npm, restart watcher
                     $nodePath = (Get-Command node -ErrorAction SilentlyContinue).Source
                     $aimeter = (Get-Command aimeter -ErrorAction SilentlyContinue).Source
-                    Start-Process "cmd.exe" -ArgumentList "/k echo Updating AIMeter... && npm install -g getaimeter@latest && echo. && echo Restarting watcher... && aimeter stop && timeout /t 2 /nobreak >nul && aimeter start && echo. && echo Updated to latest version! This window can be closed. && echo."
+                    Start-Process "cmd.exe" -ArgumentList "/k echo Updating AIMeter... && npm install -g getaimeter@latest && echo. && echo Reinstalling shortcuts... && aimeter install && echo. && echo Updated to latest version! This window can be closed. && echo."
 
                     # Give the cmd window time to stop us, then exit this tray
                     Start-Sleep -Seconds 3

package/watcher.js

@@ -75,8 +75,24 @@ function detectSource(filePath) {
     return 'gemini_cli';
   }
   if (normalized.includes('.codex/') || normalized.includes('/codex/')) {
-    _sourceCache.set(filePath, 'codex_cli');
-    return 'codex_cli';
+    // Codex sessions may come from CLI or VS Code extension.
+    // Read session_meta to check originator/source field.
+    let codexSource = 'codex_cli';
+    try {
+      const fd = fs.openSync(filePath, 'r');
+      const buf = Buffer.alloc(Math.min(2048, fs.fstatSync(fd).size));
+      fs.readSync(fd, buf, 0, buf.length, 0);
+      fs.closeSync(fd);
+      const firstLine = buf.toString('utf8').split('\n')[0];
+      const meta = JSON.parse(firstLine);
+      if (meta.type === 'session_meta' && meta.payload) {
+        if (meta.payload.source === 'vscode' || meta.payload.originator === 'codex_vscode') {
+          codexSource = 'codex_vscode';
+        }
+      }
+    } catch {}
+    _sourceCache.set(filePath, codexSource);
+    return codexSource;
   }
   if (normalized.includes('/Cursor/') || normalized.includes('/cursor/') || normalized.includes('.cursor/')) {
     _sourceCache.set(filePath, 'cursor');