claude-code-wakatime 2.0.0 → 2.1.1

package/README.md

@@ -6,12 +6,19 @@
 
 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,11 @@ 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() {
+const logger = new logger_1.Logger();
+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,11 +41,57 @@ 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));
 }
-function sendHeartbeat(inp, logger) {
+function sendHeartbeat(inp) {
     const projectFolder = inp?.cwd;
     try {
         const args = [
@@ -58,7 +108,23 @@ function sendHeartbeat(inp, logger) {
             args.push('--project-folder');
             args.push(projectFolder);
         }
-        (0, child_process_1.execFileSync)(WAKATIME_CLI, args);
+        if (inp?.transcript_path) {
+            const lineChanges = calculateLineChanges(inp.transcript_path);
+            if (lineChanges) {
+                args.push('--ai-line-changes');
+                args.push(lineChanges.toString());
+            }
+        }
+        const options = utils_1.Utils.buildOptions();
+        (0, child_process_1.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}`);
+            }
+        });
     }
     catch (err) {
         logger.errorException(err);
@@ -68,7 +134,7 @@ function main() {
     const inp = parseInput();
     const options = new options_1.Options();
     const debug = options.getSetting('settings', 'debug');
-    const logger = new logger_1.Logger(debug === 'true' ? logger_1.LogLevel.DEBUG : logger_1.LogLevel.INFO);
+    logger.setLevel(debug === 'true' ? logger_1.LogLevel.DEBUG : logger_1.LogLevel.INFO);
     const deps = new dependencies_1.Dependencies(options, logger);
     if (inp) {
         try {
@@ -78,11 +144,11 @@ function main() {
             // ignore
         }
     }
-    if (inp?.hook_event_name === 'SessionStart') {
+    if (inp?.hook_event_name === 'SessionStart demo') {
         deps.checkAndInstallCli();
     }
-    if (shouldSendHeartbeat()) {
-        sendHeartbeat(inp, logger);
+    if (shouldSendHeartbeat(inp)) {
+        sendHeartbeat(inp);
         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/logger.js

@@ -7,7 +7,6 @@ exports.Logger = exports.LogLevel = void 0;
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const os_1 = __importDefault(require("os"));
-const utils_1 = require("./utils");
 var LogLevel;
 (function (LogLevel) {
     LogLevel[LogLevel["DEBUG"] = 0] = "DEBUG";
@@ -19,7 +18,7 @@ const LOG_FILE = path_1.default.join(os_1.default.homedir(), '.wakatime', 'claud
 class Logger {
     constructor(level) {
         this.level = LogLevel.INFO;
-        if (level)
+        if (level !== undefined)
             this.setLevel(level);
     }
     getLevel() {
@@ -30,7 +29,7 @@ class Logger {
     }
     log(level, msg) {
         if (level >= this.level) {
-            msg = `[${utils_1.Utils.timestamp()}][${LogLevel[level]}] ${msg}\n`;
+            msg = `[${new Date().toISOString()}][${LogLevel[level]}] ${msg}\n`;
             fs_1.default.mkdirSync(path_1.default.dirname(LOG_FILE), { recursive: true });
             fs_1.default.appendFileSync(LOG_FILE, msg);
         }

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.1.0';

package/package.json

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

package/src/index.ts

@@ -3,7 +3,7 @@
 import fs from 'fs';
 import path from 'path';
 import os from 'os';
-import { execFileSync } from 'child_process';
+import { execFile } from 'child_process';
 import { Options } from './options';
 import { VERSION } from './version';
 import { Dependencies } from './dependencies';
@@ -12,6 +12,7 @@ import { Logger, LogLevel } from './logger';
 
 const STATE_FILE = path.join(os.homedir(), '.wakatime', 'claude-code.json');
 const WAKATIME_CLI = path.join(os.homedir(), '.wakatime', 'wakatime-cli');
+const logger = new Logger();
 
 type State = {
   lastHeartbeatAt?: number;
@@ -24,7 +25,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;
@@ -100,7 +105,7 @@ function updateState() {
   fs.writeFileSync(STATE_FILE, JSON.stringify({ lastHeartbeatAt: Utils.timestamp() } as State, null, 2));
 }
 
-function sendHeartbeat(inp: Input | undefined, logger: Logger) {
+function sendHeartbeat(inp: Input | undefined) {
   const projectFolder = inp?.cwd;
   try {
     const args: string[] = [
@@ -120,13 +125,22 @@ function sendHeartbeat(inp: Input | undefined, logger: Logger) {
 
     if (inp?.transcript_path) {
       const lineChanges = calculateLineChanges(inp.transcript_path);
-      if (lineChanges !== 0) {
+      if (lineChanges) {
         args.push('--ai-line-changes');
         args.push(lineChanges.toString());
       }
     }
 
-    execFileSync(WAKATIME_CLI, 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}`);
+      }
+    });
   } catch (err: any) {
     logger.errorException(err);
   }
@@ -137,7 +151,7 @@ function main() {
 
   const options = new Options();
   const debug = options.getSetting('settings', 'debug');
-  const logger = new Logger(debug === 'true' ? LogLevel.DEBUG : LogLevel.INFO);
+  logger.setLevel(debug === 'true' ? LogLevel.DEBUG : LogLevel.INFO);
   const deps = new Dependencies(options, logger);
 
   if (inp) {
@@ -148,12 +162,12 @@ function main() {
     }
   }
 
-  if (inp?.hook_event_name === 'SessionStart') {
+  if (inp?.hook_event_name === 'SessionStart demo') {
     deps.checkAndInstallCli();
   }
 
-  if (shouldSendHeartbeat()) {
-    sendHeartbeat(inp, logger);
+  if (shouldSendHeartbeat(inp)) {
+    sendHeartbeat(inp);
     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;

package/src/logger.ts

@@ -16,7 +16,7 @@ export class Logger {
   private level: LogLevel = LogLevel.INFO;
 
   constructor(level?: LogLevel) {
-    if (level) this.setLevel(level);
+    if (level !== undefined) this.setLevel(level);
   }
 
   public getLevel(): LogLevel {
@@ -29,7 +29,7 @@ export class Logger {
 
   public log(level: LogLevel, msg: string): void {
     if (level >= this.level) {
-      msg = `[${Utils.timestamp()}][${LogLevel[level]}] ${msg}\n`;
+      msg = `[${new Date().toISOString()}][${LogLevel[level]}] ${msg}\n`;
       fs.mkdirSync(path.dirname(LOG_FILE), { recursive: true });
       fs.appendFileSync(LOG_FILE, msg);
     }