claude-code-wakatime 2.0.0 → 2.1.0

package/README.md

@@ -6,12 +6,17 @@
 
 1. `npm install -g claude-code-wakatime`
 
-2. Install WakaTime for [VS Code](https://wakatime.com/vs-code) to setup your `~/.wakatime/` folder and api key.
+2. Add your [api key](https://wakatime.com/settings/api-key) to `~/.wakatime.cfg`:
+
+    [settings]
+    api_key = waka_123
 
 4. Use Claude Code and your AI coding activity will be displayed on your [WakaTime dashboard](https://wakatime.com)
 
 ## Usage
 
+New: See the lines of code generated by AI on your dashboard!
+
 Visit [https://wakatime.com](https://wakatime.com) to see your coding activity.
 
 ![Project Overview](https://wakatime.com/static/img/ScreenShots/Screen-Shot-2016-03-21.png)

package/dist/index.js

@@ -15,7 +15,10 @@ const utils_1 = require("./utils");
 const logger_1 = require("./logger");
 const STATE_FILE = path_1.default.join(os_1.default.homedir(), '.wakatime', 'claude-code.json');
 const WAKATIME_CLI = path_1.default.join(os_1.default.homedir(), '.wakatime', 'wakatime-cli');
-function shouldSendHeartbeat() {
+function shouldSendHeartbeat(inp) {
+    if (inp?.hook_event_name === 'Stop') {
+        return true;
+    }
     try {
         const last = JSON.parse(fs_1.default.readFileSync(STATE_FILE, 'utf-8')).lastHeartbeatAt ?? utils_1.Utils.timestamp();
         return utils_1.Utils.timestamp() - last >= 60;
@@ -37,6 +40,52 @@ function parseInput() {
     }
     return undefined;
 }
+function getLastHeartbeat() {
+    try {
+        const stateData = JSON.parse(fs_1.default.readFileSync(STATE_FILE, 'utf-8'));
+        return stateData.lastHeartbeatAt ?? 0;
+    }
+    catch {
+        return 0;
+    }
+}
+function calculateLineChanges(transcriptPath) {
+    try {
+        if (!transcriptPath || !fs_1.default.existsSync(transcriptPath)) {
+            return 0;
+        }
+        const content = fs_1.default.readFileSync(transcriptPath, 'utf-8');
+        const lines = content.split('\n');
+        let totalLineChanges = 0;
+        const lastHeartbeatAt = getLastHeartbeat();
+        for (const line of lines) {
+            if (line.trim()) {
+                try {
+                    const logEntry = JSON.parse(line);
+                    // Only count changes since last heartbeat
+                    if (logEntry.timestamp && logEntry.toolUseResult?.structuredPatch) {
+                        const entryTimestamp = new Date(logEntry.timestamp).getTime() / 1000;
+                        if (entryTimestamp >= lastHeartbeatAt) {
+                            const patches = logEntry.toolUseResult.structuredPatch;
+                            for (const patch of patches) {
+                                if (patch.newLines !== undefined && patch.oldLines !== undefined) {
+                                    totalLineChanges += patch.newLines - patch.oldLines;
+                                }
+                            }
+                        }
+                    }
+                }
+                catch {
+                    // ignore
+                }
+            }
+        }
+        return totalLineChanges;
+    }
+    catch {
+        return 0;
+    }
+}
 function updateState() {
     fs_1.default.mkdirSync(path_1.default.dirname(STATE_FILE), { recursive: true });
     fs_1.default.writeFileSync(STATE_FILE, JSON.stringify({ lastHeartbeatAt: utils_1.Utils.timestamp() }, null, 2));
@@ -58,6 +107,13 @@ function sendHeartbeat(inp, logger) {
             args.push('--project-folder');
             args.push(projectFolder);
         }
+        if (inp?.transcript_path) {
+            const lineChanges = calculateLineChanges(inp.transcript_path);
+            if (lineChanges !== 0) {
+                args.push('--ai-line-changes');
+                args.push(lineChanges.toString());
+            }
+        }
         (0, child_process_1.execFileSync)(WAKATIME_CLI, args);
     }
     catch (err) {
@@ -81,7 +137,7 @@ function main() {
     if (inp?.hook_event_name === 'SessionStart') {
         deps.checkAndInstallCli();
     }
-    if (shouldSendHeartbeat()) {
+    if (shouldSendHeartbeat(inp)) {
         sendHeartbeat(inp, logger);
         updateState();
     }

package/dist/install-hooks.js

@@ -7,7 +7,7 @@ const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const os_1 = __importDefault(require("os"));
 const CLAUDE_SETTINGS = path_1.default.join(os_1.default.homedir(), '.claude', 'settings.json');
-const HOOK_EVENTS = ['PreToolUse', 'PostToolUse', 'UserPromptSubmit', 'SessionStart'];
+const HOOK_EVENTS = ['PreToolUse', 'PostToolUse', 'UserPromptSubmit', 'SessionStart', 'Stop'];
 function loadSettings() {
     if (!fs_1.default.existsSync(CLAUDE_SETTINGS)) {
         return {};

package/dist/version.js

@@ -2,4 +2,4 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.VERSION = void 0;
 // This file is auto-generated during build. Do not edit manually.
-exports.VERSION = '1.0.1';
+exports.VERSION = '2.0.0';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "claude-code-wakatime",
-  "version": "2.0.0",
+  "version": "2.1.0",
   "description": "WakaTime plugin for Claude Code",
   "bin": {
     "claude-code-wakatime": "dist/index.js"

package/src/index.ts

@@ -24,7 +24,11 @@ type Input = {
   hook_event_name: string;
 };
 
-function shouldSendHeartbeat(): boolean {
+function shouldSendHeartbeat(inp?: Input): boolean {
+  if (inp?.hook_event_name === 'Stop') {
+    return true;
+  }
+
   try {
     const last = (JSON.parse(fs.readFileSync(STATE_FILE, 'utf-8')) as State).lastHeartbeatAt ?? Utils.timestamp();
     return Utils.timestamp() - last >= 60;
@@ -152,7 +156,7 @@ function main() {
     deps.checkAndInstallCli();
   }
 
-  if (shouldSendHeartbeat()) {
+  if (shouldSendHeartbeat(inp)) {
     sendHeartbeat(inp, logger);
     updateState();
   }

package/src/install-hooks.ts

@@ -3,7 +3,7 @@ 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'];
+const HOOK_EVENTS = ['PreToolUse', 'PostToolUse', 'UserPromptSubmit', 'SessionStart', 'Stop'];
 
 function loadSettings(): any {
   if (!fs.existsSync(CLAUDE_SETTINGS)) {
@@ -33,19 +33,18 @@ function installHooks(): void {
   };
 
   let hookAlreadyExists = true;
-  
+
   for (const event of HOOK_EVENTS) {
     settings.hooks[event] = settings.hooks[event] || [];
-    
+
     // Check if a hook with the same command already exists
-    const existingHook = settings.hooks[event].find((existingHook: any) =>
-      existingHook.hooks && 
-      Array.isArray(existingHook.hooks) &&
-      existingHook.hooks.some((hookItem: any) => 
-        hookItem.command === 'claude-code-wakatime'
-      )
+    const existingHook = settings.hooks[event].find(
+      (existingHook: any) =>
+        existingHook.hooks &&
+        Array.isArray(existingHook.hooks) &&
+        existingHook.hooks.some((hookItem: any) => hookItem.command === 'claude-code-wakatime'),
     );
-    
+
     if (!existingHook) {
       settings.hooks[event].push(hook);
       hookAlreadyExists = false;