token-guard 0.1.0

package/dist/monitor.js

@@ -0,0 +1,160 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TokenMonitor = void 0;
+const node_child_process_1 = require("node:child_process");
+const node_fs_1 = require("node:fs");
+const tokenizer_js_1 = require("./tokenizer.js");
+const pricing_js_1 = require("./pricing.js");
+const display_js_1 = require("./display.js");
+class TokenMonitor {
+    stats;
+    options;
+    process = null;
+    updateInterval = null;
+    lastStatusLength = 0;
+    constructor(command, options) {
+        const model = options.model || (0, pricing_js_1.detectModel)(command);
+        this.options = options;
+        this.stats = {
+            inputTokens: 0,
+            outputTokens: 0,
+            totalTokens: 0,
+            estimatedCost: 0,
+            model,
+            startTime: new Date(),
+            command,
+            budgetExceeded: false,
+            costExceeded: false,
+        };
+    }
+    async run(args) {
+        return new Promise((resolve) => {
+            const [cmd, ...cmdArgs] = args;
+            this.process = (0, node_child_process_1.spawn)(cmd, cmdArgs, {
+                stdio: ['inherit', 'pipe', 'pipe'],
+                shell: true,
+            });
+            // Track input (what we send to the process)
+            // For now, we count the command itself as input
+            this.addInputTokens(args.join(' '));
+            // Track stdout
+            this.process.stdout?.on('data', (data) => {
+                const text = data.toString();
+                this.addOutputTokens(text);
+                process.stdout.write(data);
+                this.checkLimits();
+            });
+            // Track stderr
+            this.process.stderr?.on('data', (data) => {
+                const text = data.toString();
+                this.addOutputTokens(text);
+                process.stderr.write(data);
+                this.checkLimits();
+            });
+            // Periodic status update
+            if (!this.options.quiet) {
+                this.updateInterval = setInterval(() => {
+                    this.showStatus();
+                }, 1000);
+            }
+            this.process.on('close', (code) => {
+                this.cleanup();
+                this.stats.endTime = new Date();
+                if (!this.options.quiet) {
+                    console.error((0, display_js_1.renderFinalReport)(this.stats, this.options));
+                }
+                if (this.options.output) {
+                    this.saveReport();
+                }
+                resolve(code ?? 0);
+            });
+            this.process.on('error', (err) => {
+                this.cleanup();
+                (0, display_js_1.error)(`Failed to start command: ${err.message}`);
+                resolve(1);
+            });
+        });
+    }
+    addInputTokens(text) {
+        const tokens = (0, tokenizer_js_1.countTokens)(text);
+        this.stats.inputTokens += tokens;
+        this.updateTotals();
+    }
+    addOutputTokens(text) {
+        const tokens = (0, tokenizer_js_1.countTokens)(text);
+        this.stats.outputTokens += tokens;
+        this.updateTotals();
+    }
+    updateTotals() {
+        this.stats.totalTokens = this.stats.inputTokens + this.stats.outputTokens;
+        this.stats.estimatedCost = (0, pricing_js_1.calculateCost)(this.stats.inputTokens, this.stats.outputTokens, this.stats.model);
+    }
+    checkLimits() {
+        const { budget, costLimit, warnPercent } = this.options;
+        // Check token budget
+        if (budget) {
+            const percent = (this.stats.totalTokens / budget) * 100;
+            if (percent >= 100) {
+                this.stats.budgetExceeded = true;
+                this.terminate('Token budget exceeded');
+                return;
+            }
+            if (percent >= warnPercent && percent < warnPercent + 5) {
+                (0, display_js_1.warn)(`${percent.toFixed(0)}% of token budget used`);
+            }
+        }
+        // Check cost limit
+        if (costLimit) {
+            const percent = (this.stats.estimatedCost / costLimit) * 100;
+            if (percent >= 100) {
+                this.stats.costExceeded = true;
+                this.terminate('Cost limit exceeded');
+                return;
+            }
+            if (percent >= warnPercent && percent < warnPercent + 5) {
+                (0, display_js_1.warn)(`${percent.toFixed(0)}% of cost limit used`);
+            }
+        }
+    }
+    terminate(reason) {
+        (0, display_js_1.error)(reason);
+        if (this.process) {
+            this.process.kill('SIGTERM');
+        }
+    }
+    showStatus() {
+        if (this.options.quiet)
+            return;
+        const status = (0, display_js_1.renderStatus)(this.stats, this.options, true);
+        (0, display_js_1.clearLine)();
+        process.stderr.write(status);
+    }
+    cleanup() {
+        if (this.updateInterval) {
+            clearInterval(this.updateInterval);
+            this.updateInterval = null;
+        }
+        (0, display_js_1.clearLine)();
+    }
+    saveReport() {
+        if (!this.options.output)
+            return;
+        const report = {
+            ...this.stats,
+            startTime: this.stats.startTime.toISOString(),
+            endTime: this.stats.endTime?.toISOString(),
+            options: this.options,
+        };
+        try {
+            (0, node_fs_1.writeFileSync)(this.options.output, JSON.stringify(report, null, 2));
+        }
+        catch (err) {
+            (0, display_js_1.error)(`Failed to save report: ${err}`);
+        }
+    }
+    getStats() {
+        return { ...this.stats };
+    }
+}
+exports.TokenMonitor = TokenMonitor;
+//# sourceMappingURL=monitor.js.map

package/dist/monitor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"monitor.js","sourceRoot":"","sources":["../src/monitor.ts"],"names":[],"mappings":";;;AAAA,2DAA8D;AAC9D,qCAAwC;AAExC,iDAA6C;AAC7C,6CAA0D;AAC1D,6CAAuF;AAEvF,MAAa,YAAY;IACf,KAAK,CAAa;IAClB,OAAO,CAAoB;IAC3B,OAAO,GAAwB,IAAI,CAAC;IACpC,cAAc,GAA0B,IAAI,CAAC;IAC7C,gBAAgB,GAAG,CAAC,CAAC;IAE7B,YAAY,OAAe,EAAE,OAA0B;QACrD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAA,wBAAW,EAAC,OAAO,CAAC,CAAC;QAEpD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,KAAK,GAAG;YACX,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,CAAC;YACd,aAAa,EAAE,CAAC;YAChB,KAAK;YACL,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,OAAO;YACP,cAAc,EAAE,KAAK;YACrB,YAAY,EAAE,KAAK;SACpB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,IAAc;QACtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,MAAM,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,IAAI,CAAC;YAE/B,IAAI,CAAC,OAAO,GAAG,IAAA,0BAAK,EAAC,GAAG,EAAE,OAAO,EAAE;gBACjC,KAAK,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC;gBAClC,KAAK,EAAE,IAAI;aACZ,CAAC,CAAC;YAEH,4CAA4C;YAC5C,gDAAgD;YAChD,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAEpC,eAAe;YACf,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;gBAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC7B,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;gBAC3B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC3B,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,CAAC,CAAC,CAAC;YAEH,eAAe;YACf,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;gBAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC7B,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;gBAC3B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC3B,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,CAAC,CAAC,CAAC;YAEH,yBAAyB;YACzB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBACxB,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;oBACrC,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpB,CAAC,EAAE,IAAI,CAAC,CAAC;YACX,CAAC;YAED,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;gBAChC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;gBAEhC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;oBACxB,OAAO,CAAC,KAAK,CAAC,IAAA,8BAAiB,EAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC7D,CAAC;gBAED,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;oBACxB,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpB,CAAC;gBAED,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;YACrB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC/B,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,IAAA,kBAAK,EAAC,4BAA4B,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;gBACjD,OAAO,CAAC,CAAC,CAAC,CAAC;YACb,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,IAAY;QACjC,MAAM,MAAM,GAAG,IAAA,0BAAW,EAAC,IAAI,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,MAAM,CAAC;QACjC,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAEO,eAAe,CAAC,IAAY;QAClC,MAAM,MAAM,GAAG,IAAA,0BAAW,EAAC,IAAI,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,MAAM,CAAC;QAClC,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAEO,YAAY;QAClB,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;QAC1E,IAAI,CAAC,KAAK,CAAC,aAAa,GAAG,IAAA,0BAAa,EACtC,IAAI,CAAC,KAAK,CAAC,WAAW,EACtB,IAAI,CAAC,KAAK,CAAC,YAAY,EACvB,IAAI,CAAC,KAAK,CAAC,KAAK,CACjB,CAAC;IACJ,CAAC;IAEO,WAAW;QACjB,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QAExD,qBAAqB;QACrB,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,GAAG,GAAG,CAAC;YAExD,IAAI,OAAO,IAAI,GAAG,EAAE,CAAC;gBACnB,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;gBACjC,IAAI,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;gBACxC,OAAO;YACT,CAAC;YAED,IAAI,OAAO,IAAI,WAAW,IAAI,OAAO,GAAG,WAAW,GAAG,CAAC,EAAE,CAAC;gBACxD,IAAA,iBAAI,EAAC,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;QAED,mBAAmB;QACnB,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,GAAG,SAAS,CAAC,GAAG,GAAG,CAAC;YAE7D,IAAI,OAAO,IAAI,GAAG,EAAE,CAAC;gBACnB,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;gBAC/B,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;gBACtC,OAAO;YACT,CAAC;YAED,IAAI,OAAO,IAAI,WAAW,IAAI,OAAO,GAAG,WAAW,GAAG,CAAC,EAAE,CAAC;gBACxD,IAAA,iBAAI,EAAC,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,MAAc;QAC9B,IAAA,kBAAK,EAAC,MAAM,CAAC,CAAC;QACd,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK;YAAE,OAAO;QAE/B,MAAM,MAAM,GAAG,IAAA,yBAAY,EAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC5D,IAAA,sBAAS,GAAE,CAAC;QACZ,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/B,CAAC;IAEO,OAAO;QACb,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAA,sBAAS,GAAE,CAAC;IACd,CAAC;IAEO,UAAU;QAChB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,OAAO;QAEjC,MAAM,MAAM,GAAG;YACb,GAAG,IAAI,CAAC,KAAK;YACb,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,WAAW,EAAE;YAC7C,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE;YAC1C,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;QAEF,IAAI,CAAC;YACH,IAAA,uBAAa,EAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACtE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAA,kBAAK,EAAC,0BAA0B,GAAG,EAAE,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,QAAQ;QACN,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;CACF;AArLD,oCAqLC"}

package/dist/pricing.d.ts

@@ -0,0 +1,6 @@
+import type { ModelPricing } from './types.js';
+export declare const MODEL_PRICING: Record<string, ModelPricing>;
+export declare function getPricing(model: string): ModelPricing;
+export declare function calculateCost(inputTokens: number, outputTokens: number, model: string): number;
+export declare function detectModel(command: string): string;
+//# sourceMappingURL=pricing.d.ts.map

package/dist/pricing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.d.ts","sourceRoot":"","sources":["../src/pricing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAG/C,eAAO,MAAM,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAmBtD,CAAC;AAEF,wBAAgB,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,YAAY,CAUtD;AAED,wBAAgB,aAAa,CAC3B,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,MAAM,GACZ,MAAM,CAKR;AAED,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAkBnD"}

package/dist/pricing.js

@@ -0,0 +1,60 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MODEL_PRICING = void 0;
+exports.getPricing = getPricing;
+exports.calculateCost = calculateCost;
+exports.detectModel = detectModel;
+// Pricing per 1M tokens (as of Jan 2026)
+exports.MODEL_PRICING = {
+    // Claude models
+    'claude-opus-4-5': { input: 15.00, output: 75.00 },
+    'claude-opus-4': { input: 15.00, output: 75.00 },
+    'claude-sonnet-4': { input: 3.00, output: 15.00 },
+    'claude-3-opus': { input: 15.00, output: 75.00 },
+    'claude-3-sonnet': { input: 3.00, output: 15.00 },
+    'claude-3-haiku': { input: 0.25, output: 1.25 },
+    // OpenAI models
+    'gpt-4o': { input: 2.50, output: 10.00 },
+    'gpt-4-turbo': { input: 10.00, output: 30.00 },
+    'gpt-4': { input: 30.00, output: 60.00 },
+    'gpt-3.5-turbo': { input: 0.50, output: 1.50 },
+    'o1': { input: 15.00, output: 60.00 },
+    'o1-mini': { input: 3.00, output: 12.00 },
+    // Default fallback (sonnet-level pricing)
+    'default': { input: 3.00, output: 15.00 },
+};
+function getPricing(model) {
+    const normalizedModel = model.toLowerCase().replace(/[_-]/g, '-');
+    for (const [key, pricing] of Object.entries(exports.MODEL_PRICING)) {
+        if (normalizedModel.includes(key.replace(/[_-]/g, '-'))) {
+            return pricing;
+        }
+    }
+    return exports.MODEL_PRICING['default'];
+}
+function calculateCost(inputTokens, outputTokens, model) {
+    const pricing = getPricing(model);
+    const inputCost = (inputTokens / 1_000_000) * pricing.input;
+    const outputCost = (outputTokens / 1_000_000) * pricing.output;
+    return inputCost + outputCost;
+}
+function detectModel(command) {
+    const lowerCommand = command.toLowerCase();
+    if (lowerCommand.includes('claude')) {
+        if (lowerCommand.includes('opus'))
+            return 'claude-opus-4';
+        if (lowerCommand.includes('haiku'))
+            return 'claude-3-haiku';
+        return 'claude-sonnet-4'; // default for claude commands
+    }
+    if (lowerCommand.includes('aider')) {
+        // Aider defaults vary, assume sonnet
+        return 'claude-sonnet-4';
+    }
+    if (lowerCommand.includes('gpt-4'))
+        return 'gpt-4o';
+    if (lowerCommand.includes('openai'))
+        return 'gpt-4o';
+    return 'default';
+}
+//# sourceMappingURL=pricing.js.map

package/dist/pricing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.js","sourceRoot":"","sources":["../src/pricing.ts"],"names":[],"mappings":";;;AAwBA,gCAUC;AAED,sCASC;AAED,kCAkBC;AA/DD,yCAAyC;AAC5B,QAAA,aAAa,GAAiC;IACzD,gBAAgB;IAChB,iBAAiB,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;IAClD,eAAe,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;IAChD,iBAAiB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE;IACjD,eAAe,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;IAChD,iBAAiB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE;IACjD,gBAAgB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAE/C,gBAAgB;IAChB,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE;IACxC,aAAa,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;IAC9C,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;IACxC,eAAe,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAC9C,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;IACrC,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE;IAEzC,0CAA0C;IAC1C,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE;CAC1C,CAAC;AAEF,SAAgB,UAAU,CAAC,KAAa;IACtC,MAAM,eAAe,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IAElE,KAAK,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,qBAAa,CAAC,EAAE,CAAC;QAC3D,IAAI,eAAe,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;YACxD,OAAO,OAAO,CAAC;QACjB,CAAC;IACH,CAAC;IAED,OAAO,qBAAa,CAAC,SAAS,CAAC,CAAC;AAClC,CAAC;AAED,SAAgB,aAAa,CAC3B,WAAmB,EACnB,YAAoB,EACpB,KAAa;IAEb,MAAM,OAAO,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IAClC,MAAM,SAAS,GAAG,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5D,MAAM,UAAU,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAC/D,OAAO,SAAS,GAAG,UAAU,CAAC;AAChC,CAAC;AAED,SAAgB,WAAW,CAAC,OAAe;IACzC,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAE3C,IAAI,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QACpC,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,eAAe,CAAC;QAC1D,IAAI,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,gBAAgB,CAAC;QAC5D,OAAO,iBAAiB,CAAC,CAAC,8BAA8B;IAC1D,CAAC;IAED,IAAI,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACnC,qCAAqC;QACrC,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,IAAI,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,QAAQ,CAAC;IACpD,IAAI,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,QAAQ,CAAC;IAErD,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/tokenizer.d.ts

@@ -0,0 +1,3 @@
+export declare function countTokens(text: string): number;
+export declare function formatTokens(count: number): string;
+//# sourceMappingURL=tokenizer.d.ts.map

package/dist/tokenizer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenizer.d.ts","sourceRoot":"","sources":["../src/tokenizer.ts"],"names":[],"mappings":"AAKA,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAyChD;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAQlD"}

package/dist/tokenizer.js

@@ -0,0 +1,54 @@
+"use strict";
+// Simple token estimation without external dependencies
+// Uses GPT-4 style tokenization rules (~4 chars per token)
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.countTokens = countTokens;
+exports.formatTokens = formatTokens;
+const CHARS_PER_TOKEN = 4;
+function countTokens(text) {
+    if (!text)
+        return 0;
+    // More accurate estimation:
+    // - Split on whitespace and punctuation
+    // - Count special tokens for code
+    // Count words
+    const words = text.split(/\s+/).filter(w => w.length > 0);
+    // Estimate tokens (rough approximation)
+    // - Short words (1-4 chars) = 1 token
+    // - Medium words (5-8 chars) = 1-2 tokens
+    // - Long words (9+ chars) = 2+ tokens
+    // - Code symbols often get their own tokens
+    let tokens = 0;
+    for (const word of words) {
+        if (word.length <= 4) {
+            tokens += 1;
+        }
+        else if (word.length <= 8) {
+            tokens += Math.ceil(word.length / 4);
+        }
+        else {
+            tokens += Math.ceil(word.length / 3);
+        }
+    }
+    // Add tokens for punctuation and special characters
+    const specialChars = text.match(/[{}()\[\]<>:;,."'`~!@#$%^&*+=|\\/?-]/g);
+    if (specialChars) {
+        tokens += Math.ceil(specialChars.length * 0.5);
+    }
+    // Add tokens for newlines (often separate tokens)
+    const newlines = text.match(/\n/g);
+    if (newlines) {
+        tokens += newlines.length;
+    }
+    return Math.max(1, Math.round(tokens));
+}
+function formatTokens(count) {
+    if (count >= 1_000_000) {
+        return `${(count / 1_000_000).toFixed(2)}M`;
+    }
+    if (count >= 1_000) {
+        return `${(count / 1_000).toFixed(1)}k`;
+    }
+    return count.toString();
+}
+//# sourceMappingURL=tokenizer.js.map

package/dist/tokenizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenizer.js","sourceRoot":"","sources":["../src/tokenizer.ts"],"names":[],"mappings":";AAAA,wDAAwD;AACxD,2DAA2D;;AAI3D,kCAyCC;AAED,oCAQC;AArDD,MAAM,eAAe,GAAG,CAAC,CAAC;AAE1B,SAAgB,WAAW,CAAC,IAAY;IACtC,IAAI,CAAC,IAAI;QAAE,OAAO,CAAC,CAAC;IAEpB,4BAA4B;IAC5B,wCAAwC;IACxC,kCAAkC;IAElC,cAAc;IACd,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAE1D,wCAAwC;IACxC,sCAAsC;IACtC,0CAA0C;IAC1C,sCAAsC;IACtC,4CAA4C;IAE5C,IAAI,MAAM,GAAG,CAAC,CAAC;IAEf,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,CAAC;QACd,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED,oDAAoD;IACpD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;IACzE,IAAI,YAAY,EAAE,CAAC;QACjB,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC;IACjD,CAAC;IAED,kDAAkD;IAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,IAAI,QAAQ,CAAC,MAAM,CAAC;IAC5B,CAAC;IAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;AACzC,CAAC;AAED,SAAgB,YAAY,CAAC,KAAa;IACxC,IAAI,KAAK,IAAI,SAAS,EAAE,CAAC;QACvB,OAAO,GAAG,CAAC,KAAK,GAAG,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC9C,CAAC;IACD,IAAI,KAAK,IAAI,KAAK,EAAE,CAAC;QACnB,OAAO,GAAG,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC1C,CAAC;IACD,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;AAC1B,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,31 @@
+export interface TokenGuardOptions {
+    budget?: number;
+    costLimit?: number;
+    model?: string;
+    warnPercent: number;
+    quiet: boolean;
+    output?: string;
+}
+export interface ModelPricing {
+    input: number;
+    output: number;
+}
+export interface UsageStats {
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+    estimatedCost: number;
+    model: string;
+    startTime: Date;
+    endTime?: Date;
+    command: string;
+    budgetExceeded: boolean;
+    costExceeded: boolean;
+}
+export interface Config {
+    defaultBudget?: number;
+    defaultCostLimit?: number;
+    warnPercent: number;
+    models: Record<string, ModelPricing>;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,iBAAiB;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,UAAU;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,IAAI,CAAC;IAChB,OAAO,CAAC,EAAE,IAAI,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc,EAAE,OAAO,CAAC;IACxB,YAAY,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,MAAM;IACrB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;CACtC"}

package/dist/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/package.json

@@ -0,0 +1,43 @@
+{
+  "name": "token-guard",
+  "version": "0.1.0",
+  "description": "CLI tool to monitor and limit token usage for AI coding assistants",
+  "main": "dist/index.js",
+  "bin": {
+    "token-guard": "dist/cli.js",
+    "tg": "dist/cli.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "start": "node dist/cli.js",
+    "test": "node --test",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "ai",
+    "llm",
+    "tokens",
+    "claude",
+    "openai",
+    "budget",
+    "cost",
+    "cli"
+  ],
+  "author": "Josh Duffy",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/joshduffy/token-guard.git"
+  },
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "typescript": "^5.3.0"
+  },
+  "dependencies": {
+    "tiktoken": "^1.0.0"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}

package/src/cli.ts

@@ -0,0 +1,166 @@
+#!/usr/bin/env node
+
+import { loadConfig } from './config.js';
+import { TokenMonitor } from './monitor.js';
+import type { TokenGuardOptions } from './types.js';
+
+const VERSION = '0.1.0';
+
+const HELP = `
+token-guard v${VERSION}
+Monitor and limit token usage for AI coding assistants
+
+USAGE:
+  token-guard [options] -- <command>
+
+OPTIONS:
+  -b, --budget <tokens>     Maximum tokens allowed (default: unlimited)
+  -c, --cost-limit <usd>    Maximum cost in USD (default: unlimited)
+  -m, --model <name>        Model for pricing (default: auto-detect)
+  -w, --warn <percent>      Warn at percentage of budget (default: 80)
+  -q, --quiet               Only show warnings and errors
+  -o, --output <file>       Save usage report to JSON file
+  -h, --help                Show this help message
+  -v, --version             Show version
+
+EXAMPLES:
+  token-guard -- claude "refactor the auth module"
+  token-guard -b 50000 -- aider --message "add tests"
+  token-guard -c 0.50 -- claude "explain this codebase"
+  token-guard -b 100000 -w 50 -o report.json -- claude "review PR"
+
+CONFIGURATION:
+  Create ~/.token-guard.json for persistent settings:
+  {
+    "defaultBudget": 50000,
+    "warnPercent": 80,
+    "models": {
+      "custom-model": { "input": 5.00, "output": 15.00 }
+    }
+  }
+
+`;
+
+function parseArgs(args: string[]): { options: TokenGuardOptions; command: string[] } {
+  const config = loadConfig();
+
+  const options: TokenGuardOptions = {
+    budget: config.defaultBudget,
+    costLimit: config.defaultCostLimit,
+    warnPercent: config.warnPercent,
+    quiet: false,
+  };
+
+  const command: string[] = [];
+  let i = 0;
+  let foundSeparator = false;
+
+  while (i < args.length) {
+    const arg = args[i];
+
+    if (foundSeparator) {
+      command.push(arg);
+      i++;
+      continue;
+    }
+
+    if (arg === '--') {
+      foundSeparator = true;
+      i++;
+      continue;
+    }
+
+    switch (arg) {
+      case '-h':
+      case '--help':
+        console.log(HELP);
+        process.exit(0);
+        break;
+
+      case '-v':
+      case '--version':
+        console.log(`token-guard v${VERSION}`);
+        process.exit(0);
+        break;
+
+      case '-b':
+      case '--budget':
+        options.budget = parseInt(args[++i], 10);
+        if (isNaN(options.budget)) {
+          console.error('Error: --budget requires a number');
+          process.exit(1);
+        }
+        break;
+
+      case '-c':
+      case '--cost-limit':
+        options.costLimit = parseFloat(args[++i]);
+        if (isNaN(options.costLimit)) {
+          console.error('Error: --cost-limit requires a number');
+          process.exit(1);
+        }
+        break;
+
+      case '-m':
+      case '--model':
+        options.model = args[++i];
+        break;
+
+      case '-w':
+      case '--warn':
+        options.warnPercent = parseInt(args[++i], 10);
+        if (isNaN(options.warnPercent)) {
+          console.error('Error: --warn requires a number');
+          process.exit(1);
+        }
+        break;
+
+      case '-q':
+      case '--quiet':
+        options.quiet = true;
+        break;
+
+      case '-o':
+      case '--output':
+        options.output = args[++i];
+        break;
+
+      default:
+        // If we hit an unknown arg before --, treat rest as command
+        command.push(arg);
+        foundSeparator = true;
+        break;
+    }
+
+    i++;
+  }
+
+  return { options, command };
+}
+
+async function main(): Promise<void> {
+  const args = process.argv.slice(2);
+
+  if (args.length === 0) {
+    console.log(HELP);
+    process.exit(0);
+  }
+
+  const { options, command } = parseArgs(args);
+
+  if (command.length === 0) {
+    console.error('Error: No command specified');
+    console.error('Usage: token-guard [options] -- <command>');
+    process.exit(1);
+  }
+
+  const monitor = new TokenMonitor(command.join(' '), options);
+  const exitCode = await monitor.run(command);
+
+  process.exit(exitCode);
+}
+
+main().catch((err) => {
+  console.error('Fatal error:', err);
+  process.exit(1);
+});

package/src/config.ts

@@ -0,0 +1,44 @@
+import { readFileSync, existsSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import type { Config } from './types.js';
+import { MODEL_PRICING } from './pricing.js';
+
+const CONFIG_FILENAME = '.token-guard.json';
+
+const DEFAULT_CONFIG: Config = {
+  warnPercent: 80,
+  models: MODEL_PRICING,
+};
+
+export function loadConfig(): Config {
+  const configPaths = [
+    join(process.cwd(), CONFIG_FILENAME),
+    join(homedir(), CONFIG_FILENAME),
+  ];
+
+  for (const configPath of configPaths) {
+    if (existsSync(configPath)) {
+      try {
+        const content = readFileSync(configPath, 'utf-8');
+        const userConfig = JSON.parse(content);
+        return {
+          ...DEFAULT_CONFIG,
+          ...userConfig,
+          models: {
+            ...DEFAULT_CONFIG.models,
+            ...userConfig.models,
+          },
+        };
+      } catch {
+        // Ignore invalid config files
+      }
+    }
+  }
+
+  return DEFAULT_CONFIG;
+}
+
+export function getConfigPath(): string {
+  return join(homedir(), CONFIG_FILENAME);
+}