claude-code-wakatime 1.0.0 → 2.0.0

package/src/options.ts

@@ -0,0 +1,198 @@
+import * as child_process from 'child_process';
+import * as fs from 'fs';
+import * as path from 'path';
+
+import { Utils } from './utils';
+
+export interface Setting {
+  key: string;
+  value: string;
+  error?: string;
+}
+
+export class Options {
+  private configFile: string;
+  private internalConfigFile: string;
+  private logFile: string;
+  public resourcesLocation: string;
+
+  constructor() {
+    const home = Utils.getHomeDirectory();
+    const wakaFolder = path.join(home, '.wakatime');
+    try {
+      if (!fs.existsSync(wakaFolder)) {
+        fs.mkdirSync(wakaFolder, { recursive: true });
+      }
+      this.resourcesLocation = wakaFolder;
+    } catch (e) {
+      console.error(e);
+      throw e;
+    }
+
+    this.configFile = path.join(home, '.wakatime.cfg');
+    this.internalConfigFile = path.join(this.resourcesLocation, 'wakatime-internal.cfg');
+    this.logFile = path.join(this.resourcesLocation, 'wakatime.log');
+  }
+
+  public getSetting(section: string, key: string, internal?: boolean): string | undefined {
+    const content = fs.readFileSync(this.getConfigFile(internal ?? false), 'utf-8');
+    if (content.trim()) {
+      let currentSection = '';
+      let lines = content.split('\n');
+      for (var i = 0; i < lines.length; i++) {
+        let line = lines[i];
+        if (this.startsWith(line.trim(), '[') && this.endsWith(line.trim(), ']')) {
+          currentSection = line
+            .trim()
+            .substring(1, line.trim().length - 1)
+            .toLowerCase();
+        } else if (currentSection === section) {
+          let parts = line.split('=');
+          let currentKey = parts[0].trim();
+          if (currentKey === key && parts.length > 1) {
+            return this.removeNulls(parts[1].trim());
+          }
+        }
+      }
+
+      return undefined;
+    }
+  }
+
+  public setSetting(section: string, key: string, val: string, internal: boolean): void {
+    const configFile = this.getConfigFile(internal);
+    fs.readFile(configFile, 'utf-8', (err: NodeJS.ErrnoException | null, content: string) => {
+      // ignore errors because config file might not exist yet
+      if (err) content = '';
+
+      let contents: string[] = [];
+      let currentSection = '';
+
+      let found = false;
+      let lines = content.split('\n');
+      for (var i = 0; i < lines.length; i++) {
+        let line = lines[i];
+        if (this.startsWith(line.trim(), '[') && this.endsWith(line.trim(), ']')) {
+          if (currentSection === section && !found) {
+            contents.push(this.removeNulls(key + ' = ' + val));
+            found = true;
+          }
+          currentSection = line
+            .trim()
+            .substring(1, line.trim().length - 1)
+            .toLowerCase();
+          contents.push(this.removeNulls(line));
+        } else if (currentSection === section) {
+          let parts = line.split('=');
+          let currentKey = parts[0].trim();
+          if (currentKey === key) {
+            if (!found) {
+              contents.push(this.removeNulls(key + ' = ' + val));
+              found = true;
+            }
+          } else {
+            contents.push(this.removeNulls(line));
+          }
+        } else {
+          contents.push(this.removeNulls(line));
+        }
+      }
+
+      if (!found) {
+        if (currentSection !== section) {
+          contents.push('[' + section + ']');
+        }
+        contents.push(this.removeNulls(key + ' = ' + val));
+      }
+
+      fs.writeFile(configFile as string, contents.join('\n'), (err) => {
+        if (err) throw err;
+      });
+    });
+  }
+
+  public setSettings(section: string, settings: Setting[], internal: boolean): void {
+    const configFile = this.getConfigFile(internal);
+    fs.readFile(configFile, 'utf-8', (err: NodeJS.ErrnoException | null, content: string) => {
+      // ignore errors because config file might not exist yet
+      if (err) content = '';
+
+      let contents: string[] = [];
+      let currentSection = '';
+
+      const found: Record<string, boolean> = {};
+      let lines = content.split('\n');
+      for (var i = 0; i < lines.length; i++) {
+        let line = lines[i];
+        if (this.startsWith(line.trim(), '[') && this.endsWith(line.trim(), ']')) {
+          if (currentSection === section) {
+            settings.forEach((setting) => {
+              if (!found[setting.key]) {
+                contents.push(this.removeNulls(setting.key + ' = ' + setting.value));
+                found[setting.key] = true;
+              }
+            });
+          }
+          currentSection = line
+            .trim()
+            .substring(1, line.trim().length - 1)
+            .toLowerCase();
+          contents.push(this.removeNulls(line));
+        } else if (currentSection === section) {
+          let parts = line.split('=');
+          let currentKey = parts[0].trim();
+          let keepLineUnchanged = true;
+          settings.forEach((setting) => {
+            if (currentKey === setting.key) {
+              keepLineUnchanged = false;
+              if (!found[setting.key]) {
+                contents.push(this.removeNulls(setting.key + ' = ' + setting.value));
+                found[setting.key] = true;
+              }
+            }
+          });
+          if (keepLineUnchanged) {
+            contents.push(this.removeNulls(line));
+          }
+        } else {
+          contents.push(this.removeNulls(line));
+        }
+      }
+
+      settings.forEach((setting) => {
+        if (!found[setting.key]) {
+          if (currentSection !== section) {
+            contents.push('[' + section + ']');
+            currentSection = section;
+          }
+          contents.push(this.removeNulls(setting.key + ' = ' + setting.value));
+          found[setting.key] = true;
+        }
+      });
+
+      fs.writeFile(configFile as string, contents.join('\n'), (err) => {
+        if (err) throw err;
+      });
+    });
+  }
+
+  public getConfigFile(internal: boolean): string {
+    return internal ? this.internalConfigFile : this.configFile;
+  }
+
+  public getLogFile(): string {
+    return this.logFile;
+  }
+
+  private startsWith(outer: string, inner: string): boolean {
+    return outer.slice(0, inner.length) === inner;
+  }
+
+  private endsWith(outer: string, inner: string): boolean {
+    return inner === '' || outer.slice(-inner.length) === inner;
+  }
+
+  private removeNulls(s: string): string {
+    return s.replace(/\0/g, '');
+  }
+}

package/src/utils.ts

@@ -0,0 +1,97 @@
+import * as fs from 'fs';
+import * as os from 'os';
+import * as child_process from 'child_process';
+import { StdioOptions } from 'child_process';
+
+export class Utils {
+  public static quote(str: string): string {
+    if (str.includes(' ')) return `"${str.replace('"', '\\"')}"`;
+    return str;
+  }
+
+  public static apiKeyInvalid(key?: string): string {
+    const err = 'Invalid api key... check https://wakatime.com/api-key for your key';
+    if (!key) return err;
+    const re = new RegExp('^(waka_)?[0-9A-F]{8}-[0-9A-F]{4}-4[0-9A-F]{3}-[89AB][0-9A-F]{3}-[0-9A-F]{12}$', 'i');
+    if (!re.test(key)) return err;
+    return '';
+  }
+
+  public static formatDate(date: Date): String {
+    let months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'];
+    let ampm = 'AM';
+    let hour = date.getHours();
+    if (hour > 11) {
+      ampm = 'PM';
+      hour = hour - 12;
+    }
+    if (hour == 0) {
+      hour = 12;
+    }
+    let minute = date.getMinutes();
+    return `${months[date.getMonth()]} ${date.getDate()}, ${date.getFullYear()} ${hour}:${minute < 10 ? `0${minute}` : minute} ${ampm}`;
+  }
+
+  public static obfuscateKey(key: string): string {
+    let newKey = '';
+    if (key) {
+      newKey = key;
+      if (key.length > 4) newKey = 'XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXX' + key.substring(key.length - 4);
+    }
+    return newKey;
+  }
+
+  public static wrapArg(arg: string): string {
+    if (arg.indexOf(' ') > -1) return '"' + arg.replace(/"/g, '\\"') + '"';
+    return arg;
+  }
+
+  public static formatArguments(binary: string, args: string[]): string {
+    let clone = args.slice(0);
+    clone.unshift(this.wrapArg(binary));
+    let newCmds: string[] = [];
+    let lastCmd = '';
+    for (let i = 0; i < clone.length; i++) {
+      if (lastCmd == '--key') newCmds.push(this.wrapArg(this.obfuscateKey(clone[i])));
+      else newCmds.push(this.wrapArg(clone[i]));
+      lastCmd = clone[i];
+    }
+    return newCmds.join(' ');
+  }
+
+  public static apiUrlToDashboardUrl(url: string): string {
+    url = url
+      .replace('://api.', '://')
+      .replace('/api/v1', '')
+      .replace(/^api\./, '')
+      .replace('/api', '');
+    return url;
+  }
+
+  public static isWindows(): boolean {
+    return os.platform() === 'win32';
+  }
+
+  public static getHomeDirectory(): string {
+    let home = process.env.WAKATIME_HOME;
+    if (home && home.trim() && fs.existsSync(home.trim())) return home.trim();
+    return process.env[this.isWindows() ? 'USERPROFILE' : 'HOME'] || process.cwd();
+  }
+
+  public static buildOptions(stdin?: boolean): Object {
+    const options: child_process.ExecFileOptions = {
+      windowsHide: true,
+    };
+    if (stdin) {
+      (options as any).stdio = ['pipe', 'pipe', 'pipe'] as StdioOptions;
+    }
+    if (!this.isWindows() && !process.env.WAKATIME_HOME && !process.env.HOME) {
+      options['env'] = { ...process.env, WAKATIME_HOME: this.getHomeDirectory() };
+    }
+    return options;
+  }
+
+  public static timestamp() {
+    return Date.now() / 1000;
+  }
+}