claude-code-wakatime 2.1.3 → 3.0.0

package/hooks/hooks.json

@@ -0,0 +1,92 @@
+{
+  "hooks": {
+    "SessionStart": [
+      {
+        "matcher": "",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "node ${CLAUDE_PLUGIN_ROOT}/dist/index.js"
+          }
+        ]
+      }
+    ],
+    "SessionEnd": [
+      {
+        "matcher": "",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "node ${CLAUDE_PLUGIN_ROOT}/dist/index.js"
+          }
+        ]
+      }
+    ],
+    "PreToolUse": [
+      {
+        "matcher": "",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "node ${CLAUDE_PLUGIN_ROOT}/dist/index.js"
+          }
+        ]
+      }
+    ],
+    "PostToolUse": [
+      {
+        "matcher": "",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "node ${CLAUDE_PLUGIN_ROOT}/dist/index.js"
+          }
+        ]
+      }
+    ],
+    "UserPromptSubmit": [
+      {
+        "matcher": "",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "node ${CLAUDE_PLUGIN_ROOT}/dist/index.js"
+          }
+        ]
+      }
+    ],
+    "PreCompact": [
+      {
+        "matcher": "",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "node ${CLAUDE_PLUGIN_ROOT}/dist/index.js"
+          }
+        ]
+      }
+    ],
+    "SubagentStop": [
+      {
+        "matcher": "",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "node ${CLAUDE_PLUGIN_ROOT}/dist/index.js"
+          }
+        ]
+      }
+    ],
+    "Stop": [
+      {
+        "matcher": "",
+        "hooks": [
+          {
+            "type": "command",
+            "command": "node ${CLAUDE_PLUGIN_ROOT}/dist/index.js"
+          }
+        ]
+      }
+    ]
+  }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "claude-code-wakatime",
-  "version": "2.1.3",
+  "version": "3.0.0",
   "description": "WakaTime plugin for Claude Code",
   "bin": {
     "claude-code-wakatime": "dist/index.js"
@@ -8,7 +8,7 @@
   "scripts": {
     "postinstall": "node dist/install-hooks.js",
     "prebuild": "node scripts/generate-version.js",
-    "build": "tsc",
+    "build": "esbuild src/index.ts --bundle --platform=node --outfile=dist/index.js",
     "watch": "npm run prebuild && tsc --watch",
     "release:major": "npm version major && npm publish && git push && git push --tags",
     "release:minor": "npm version minor && npm publish && git push && git push --tags",
@@ -24,28 +24,27 @@
     "time-tracking",
     "ai"
   ],
-  "dependencies": {
-    "adm-zip": "0.5.16",
-    "request": "2.88.2",
-    "semver": "^7.7.2",
-    "which": "^5.0.0"
-  },
   "devDependencies": {
+    "@types/adm-zip": "^0.5.7",
+    "@types/azdata": "^1.46.6",
     "@types/eslint": "^8.44.7",
     "@types/node": "^20.9.2",
     "@types/pidusage": "^2.0.5",
     "@types/prettier": "^2.7.3",
-    "@types/adm-zip": "^0.5.7",
-    "@types/azdata": "^1.46.6",
     "@types/request": "^2.48.12",
     "@types/semver": "^7.7.0",
+    "@types/which": "^3.0.4",
     "@typescript-eslint/eslint-plugin": "6.11.0",
     "@typescript-eslint/parser": "6.11.0",
-    "@types/which": "^3.0.4",
-    "eslint": "^8.54.0",
+    "adm-zip": "0.5.16",
     "azdata": "^1.0.0",
+    "esbuild": "^0.25.12",
+    "eslint": "^8.54.0",
     "prettier": "^3.1.0",
-    "typescript": "^5.2.2"
+    "request": "2.88.2",
+    "semver": "^7.7.2",
+    "typescript": "^5.2.2",
+    "which": "^5.0.0"
   },
   "author": "WakaTime",
   "license": "BSD-3-Clause",

package/src/dependencies.ts

@@ -153,7 +153,7 @@ export class Dependencies {
       url: this.githubReleasesUrl,
       json: true,
       headers: {
-        'User-Agent': 'github.com/wakatime/vscode-wakatime',
+        'User-Agent': 'github.com/wakatime/claude-code-wakatime',
       },
     };
     this.logger.debug(`Fetching latest wakatime-cli version from GitHub API: ${options.url}`);
@@ -378,7 +378,7 @@ export class Dependencies {
   }
 
   private reportMissingPlatformSupport(osname: string, architecture: string): void {
-    const url = `https://api.wakatime.com/api/v1/cli-missing?osname=${osname}&architecture=${architecture}&plugin=vscode`;
+    const url = `https://api.wakatime.com/api/v1/cli-missing?osname=${osname}&architecture=${architecture}&plugin=claude-code`;
     const proxy = this.options.getSetting('settings', 'proxy');
     const noSSLVerify = this.options.getSetting('settings', 'no_ssl_verify');
     let options: request.CoreOptions & { url: string } = { url: url };

package/src/index.ts

@@ -60,6 +60,81 @@ function getLastHeartbeat() {
   }
 }
 
+function getModifiedFile(transcriptPath: string): string | undefined {
+  try {
+    if (!transcriptPath || !fs.existsSync(transcriptPath)) {
+      return undefined;
+    }
+
+    const content = fs.readFileSync(transcriptPath, 'utf-8');
+    const lines = content.split('\n');
+    const fileLineChanges = new Map<string, number>();
+
+    const lastHeartbeatAt = getLastHeartbeat();
+    for (const line of lines) {
+      if (line.trim()) {
+        try {
+          const logEntry = JSON.parse(line);
+          if (!logEntry.timestamp) continue;
+
+          const entryTimestamp = new Date(logEntry.timestamp).getTime() / 1000;
+          if (entryTimestamp >= lastHeartbeatAt) {
+            let filePath: string | undefined;
+
+            // Check for file paths in tool use results
+            if (logEntry.toolUse?.parameters?.file_path) {
+              filePath = logEntry.toolUse.parameters.file_path;
+            }
+
+            // Check for file paths in tool use results for multi-edit
+            if (logEntry.toolUse?.parameters?.edits) {
+              filePath = logEntry.toolUse.parameters.file_path;
+            }
+
+            // Check for file paths and line changes in structured patch
+            if (logEntry.toolUseResult?.structuredPatch) {
+              const patches = logEntry.toolUseResult.structuredPatch;
+              for (const patch of patches) {
+                if (patch.file) {
+                  filePath = patch.file as string;
+                  if (patch.newLines !== undefined && patch.oldLines !== undefined) {
+                    const lineChanges = Math.abs(patch.newLines - patch.oldLines);
+                    fileLineChanges.set(filePath, (fileLineChanges.get(filePath) || 0) + lineChanges);
+                  }
+                }
+              }
+            }
+
+            if (filePath && !fileLineChanges.has(filePath)) {
+              fileLineChanges.set(filePath, 0);
+            }
+          }
+        } catch {
+          // ignore
+        }
+      }
+    }
+
+    if (fileLineChanges.size === 0) {
+      return undefined;
+    }
+
+    // Find file with most line changes
+    let maxChanges = 0;
+    let mostChangedFile: string | undefined;
+    for (const [file, changes] of fileLineChanges.entries()) {
+      if (changes > maxChanges) {
+        maxChanges = changes;
+        mostChangedFile = file;
+      }
+    }
+
+    return mostChangedFile;
+  } catch {
+    return undefined;
+  }
+}
+
 function calculateLineChanges(transcriptPath: string): number {
   try {
     if (!transcriptPath || !fs.existsSync(transcriptPath)) {
@@ -108,11 +183,19 @@ function updateState() {
 function sendHeartbeat(inp: Input | undefined) {
   const projectFolder = inp?.cwd;
   try {
+    let entity = 'claude code';
+    if (inp?.transcript_path) {
+      const modifiedFile = getModifiedFile(inp.transcript_path);
+      if (modifiedFile) {
+        entity = modifiedFile;
+      }
+    }
+
     const args: string[] = [
       '--entity',
-      'claude code',
+      entity,
       '--entity-type',
-      'app',
+      entity === 'claude code' ? 'app' : 'file',
       '--category',
       'ai coding',
       '--plugin',
@@ -131,15 +214,13 @@ function sendHeartbeat(inp: Input | undefined) {
       }
     }
 
+    logger.debug(`Sending heartbeat: ${args}`);
+
     const options = Utils.buildOptions();
-    execFile(WAKATIME_CLI, args, options, (error, _stdout, stderr) => {
-      const output = _stdout.toString().trim() + stderr.toString().trim();
-      if (output) {
-        logger.error(output);
-      }
-      if (!(error != null)) {
-        logger.debug(`Sending heartbeat: ${args}`);
-      }
+    execFile(WAKATIME_CLI, args, options, (error, stdout, stderr) => {
+      const output = stdout.toString().trim() + stderr.toString().trim();
+      if (output) logger.error(output);
+      if (error) logger.error(error.toString());
     });
   } catch (err: any) {
     logger.errorException(err);

package/src/install-hooks.ts

@@ -3,7 +3,16 @@ import path from 'path';
 import os from 'os';
 
 const CLAUDE_SETTINGS = path.join(os.homedir(), '.claude', 'settings.json');
-const HOOK_EVENTS = ['PreToolUse', 'PostToolUse', 'UserPromptSubmit', 'SessionStart', 'Stop'];
+const HOOK_EVENTS = [
+  'PreToolUse',
+  'PostToolUse',
+  'SessionStart',
+  'SessionEnd',
+  'UserPromptSubmit',
+  'PreCompact',
+  'SubagentStop',
+  'Stop',
+];
 
 function loadSettings(): any {
   if (!fs.existsSync(CLAUDE_SETTINGS)) {