atabey 0.0.37 → 0.1.0

package/dist/atabey-mcp/src/mcp/utils/auto-rollback.js

@@ -0,0 +1,418 @@
+/**
+ * ─── AUTO-ROLLBACK + REGENERATE ───────────────────────────────────
+ *
+ * When the AI generates code that violates governance rules (e.g., `any` type,
+ * console.log, copyleft license), this module:
+ *   1. Captures the file content before the write
+ *   2. Validates the new content against governance rules
+ *   3. If violation found: blocks the write, restores original, and sends
+ *      a regenerate instruction back to the AI via MCP notification
+ *
+ * Flow:
+ *   [Pre-Write Snapshot] → [Validate Content] → [Violation?] ─yes→ [Rollback] → [Notify AI]
+ *                                                         └─no─→ [Allow Write]
+ *
+ * Key Innovation: Unlike simple blocking, this tells the AI WHY it was
+ * blocked and gives it specific instructions to self-correct.
+ */
+import fs from "fs";
+import os from "os";
+import path from "path";
+import { createHash } from "crypto";
+// Rollback directory for durable snapshots (survives MCP restarts)
+const ROLLBACK_DIR = ".atabey/rollbacks";
+// Base directory for snapshots — defaults to cwd (or temp dir in Vitest) but
+// should be set to the project root by the governance layer.
+let rollbackBaseDir = process.env.VITEST
+    ? path.join(os.tmpdir(), `atabey-vitest-worker-${process.pid}`)
+    : process.cwd();
+/** Set the project root used for durable rollback snapshots. */
+export function setAutoRollbackRoot(projectRoot) {
+    rollbackBaseDir = path.resolve(projectRoot);
+}
+class SnapshotManager {
+    snapshots = new Map();
+    getRollbackPath(resolvedPath, traceId) {
+        const safeName = resolvedPath.replace(/[^a-zA-Z0-9]/g, "_").slice(0, 100);
+        const hash = createHash("sha1").update(resolvedPath).digest("hex").slice(0, 8);
+        return path.join(rollbackBaseDir, ROLLBACK_DIR, `${safeName}__${hash}__${traceId}.bak`);
+    }
+    ensureRollbackDir() {
+        const dir = path.join(rollbackBaseDir, ROLLBACK_DIR);
+        if (!fs.existsSync(dir)) {
+            fs.mkdirSync(dir, { recursive: true });
+        }
+    }
+    saveDurableSnapshot(resolvedPath, snapshot) {
+        try {
+            this.ensureRollbackDir();
+            const backupPath = this.getRollbackPath(resolvedPath, snapshot.traceId);
+            const data = JSON.stringify({
+                filePath: resolvedPath,
+                originalContent: snapshot.originalContent,
+                timestamp: snapshot.timestamp,
+                traceId: snapshot.traceId,
+            });
+            fs.writeFileSync(backupPath, data, "utf8");
+        }
+        catch { /* best effort */ }
+    }
+    loadDurableSnapshot(filePath) {
+        try {
+            const resolved = path.resolve(filePath);
+            // We don't know traceId here, so scan recent backups for this file (simple approach)
+            const dir = path.join(rollbackBaseDir, ROLLBACK_DIR);
+            if (!fs.existsSync(dir))
+                return null;
+            const files = fs.readdirSync(dir)
+                .filter(f => f.endsWith(".bak"))
+                .sort()
+                .reverse(); // newest first
+            for (const f of files) {
+                try {
+                    const full = path.join(dir, f);
+                    const content = fs.readFileSync(full, "utf8");
+                    try {
+                        const data = JSON.parse(content);
+                        if (data.filePath === resolved) {
+                            return {
+                                filePath: data.filePath,
+                                originalContent: data.originalContent,
+                                timestamp: data.timestamp,
+                                traceId: data.traceId,
+                                restored: false,
+                            };
+                        }
+                    }
+                    catch { /* ignore */ }
+                }
+                catch { /* ignore */ }
+            }
+        }
+        catch { /* ignore */ }
+        return null;
+    }
+    /**
+     * Capture a pre-write snapshot of a file.
+     * Also persists to disk for restart durability.
+     */
+    capture(filePath, traceId) {
+        const resolvedPath = path.resolve(filePath);
+        let originalContent = null;
+        try {
+            if (fs.existsSync(resolvedPath)) {
+                originalContent = fs.readFileSync(resolvedPath, "utf8");
+            }
+        }
+        catch { /* file doesn't exist yet (new file) */ }
+        const snapshot = {
+            filePath: resolvedPath,
+            originalContent,
+            timestamp: Date.now(),
+            traceId,
+            restored: false,
+        };
+        this.snapshots.set(resolvedPath, snapshot);
+        this.saveDurableSnapshot(resolvedPath, snapshot);
+    }
+    /**
+     * Get a snapshot for a file.
+     * Falls back to durable backup if not in memory.
+     */
+    get(filePath) {
+        const resolved = path.resolve(filePath);
+        const mem = this.snapshots.get(resolved);
+        if (mem)
+            return mem;
+        const durable = this.loadDurableSnapshot(resolved);
+        if (durable) {
+            this.snapshots.set(resolved, durable);
+            return durable;
+        }
+        return undefined;
+    }
+    /**
+     * Restore a file to its pre-write state.
+     * Returns true if restored, false if no snapshot exists.
+     */
+    restore(filePath) {
+        const resolvedPath = path.resolve(filePath);
+        let snapshot = this.snapshots.get(resolvedPath);
+        // Try durable fallback
+        if (!snapshot) {
+            snapshot = this.loadDurableSnapshot(resolvedPath) || undefined;
+            if (snapshot)
+                this.snapshots.set(resolvedPath, snapshot);
+        }
+        if (!snapshot || snapshot.restored)
+            return false;
+        try {
+            if (snapshot.originalContent !== null) {
+                // File existed before – restore original
+                fs.writeFileSync(resolvedPath, snapshot.originalContent, "utf8");
+                process.stderr.write(`[ROLLBACK] Restored: ${path.relative(rollbackBaseDir, resolvedPath)}\n`);
+            }
+            else {
+                // File was new – delete it
+                if (fs.existsSync(resolvedPath)) {
+                    fs.unlinkSync(resolvedPath);
+                    process.stderr.write(`[ROLLBACK] Deleted (new file): ${path.relative(rollbackBaseDir, resolvedPath)}\n`);
+                }
+            }
+            snapshot.restored = true;
+            // Remove durable backup
+            try {
+                const backupPath = this.getRollbackPath(resolvedPath, snapshot.traceId);
+                if (fs.existsSync(backupPath))
+                    fs.unlinkSync(backupPath);
+            }
+            catch { /* ignore */ }
+            return true;
+        }
+        catch (error) {
+            process.stderr.write(`[ROLLBACK] Failed to restore ${resolvedPath}: ${error}\n`);
+            return false;
+        }
+    }
+    /**
+     * Clean up old snapshots (memory + durable files).
+     */
+    cleanup(maxAgeMs = 300000) {
+        const cutoff = Date.now() - maxAgeMs;
+        for (const [filePath, snapshot] of this.snapshots) {
+            if (snapshot.timestamp < cutoff) {
+                // Remove durable backup if exists
+                try {
+                    const backupPath = this.getRollbackPath(filePath, snapshot.traceId);
+                    if (fs.existsSync(backupPath))
+                        fs.unlinkSync(backupPath);
+                }
+                catch { /* ignore */ }
+                this.snapshots.delete(filePath);
+            }
+        }
+        // Also clean old durable files on disk
+        try {
+            const dir = path.join(rollbackBaseDir, ROLLBACK_DIR);
+            if (fs.existsSync(dir)) {
+                const files = fs.readdirSync(dir);
+                for (const f of files) {
+                    const full = path.join(dir, f);
+                    try {
+                        const stat = fs.statSync(full);
+                        if (stat.mtimeMs < cutoff) {
+                            fs.unlinkSync(full);
+                        }
+                    }
+                    catch { /* ignore */ }
+                }
+            }
+        }
+        catch { /* ignore */ }
+    }
+    /**
+     * Get all active snapshots.
+     */
+    getAll() {
+        return Array.from(this.snapshots.values());
+    }
+}
+// ─── Auto-Rollback Engine ─────────────────────────────────────────
+export class AutoRollback {
+    static snapshots = new SnapshotManager();
+    /**
+     * Initialize the auto-rollback system.
+     * Sets up file watcher cleanup.
+     */
+    static initialize() {
+        // Periodic cleanup of old snapshots (every 5 minutes)
+        setInterval(() => {
+            AutoRollback.snapshots.cleanup();
+        }, 300000);
+    }
+    /**
+     * Prepare for a file write by capturing the current state.
+     * Call this BEFORE the tool handler writes to the file.
+     */
+    static prepareWrite(filePath, traceId) {
+        AutoRollback.snapshots.capture(filePath, traceId);
+    }
+    /**
+     * Validate content and rollback if violations are found.
+     * This is called after the tool handler writes the file, but
+     * BEFORE the response is returned to the AI.
+     *
+     * Returns:
+     *   - null if content is clean
+     *   - ViolationRecord[] if violations were found and rolled back
+     */
+    static validateAndRollback(filePath, content, violations) {
+        if (violations.length === 0)
+            return null;
+        // Restore the original file
+        const restored = AutoRollback.snapshots.restore(filePath);
+        if (restored) {
+            process.stderr.write(`[ROLLBACK] Rolled back ${path.relative(rollbackBaseDir, filePath)} due to ${violations.length} violation(s)\n`);
+        }
+        return violations;
+    }
+    /**
+     * Restore a set of files to their pre-write snapshots (atomic multi-file
+     * rollback). Files without a snapshot are skipped.
+     */
+    static restoreFiles(filePaths) {
+        let restored = 0;
+        for (const p of filePaths) {
+            try {
+                if (AutoRollback.snapshots.restore(p))
+                    restored++;
+            }
+            catch {
+                /* best-effort per file */
+            }
+        }
+        return restored;
+    }
+    /**
+     * Restore every snapshot captured under a given trace id (task-level
+     * transaction rollback).
+     */
+    static restoreTrace(traceId) {
+        const targets = AutoRollback.snapshots
+            .getAll()
+            .filter((s) => s.traceId === traceId && !s.restored)
+            .map((s) => s.filePath);
+        return AutoRollback.restoreFiles(targets);
+    }
+    /**
+     * Build a regenerate instruction for the AI.
+     * This is what gets sent back to the AI to tell it to fix its output.
+     */
+    static buildRegenerateInstruction(violations, toolName) {
+        const criticalViolations = violations.filter(v => v.severity === "CRITICAL");
+        const lines = [
+            "## ⛔ AI Output Blocked – Governance Violation",
+            "",
+            `The code you generated for tool "${toolName}" has been BLOCKED and ROLLED BACK.`,
+            "The file has been restored to its previous state.",
+            "",
+        ];
+        if (criticalViolations.length > 0) {
+            lines.push(`### 🔴 Critical Violations (${criticalViolations.length})`);
+            lines.push("");
+            for (const v of criticalViolations) {
+                lines.push(`- **${v.rule}** (${v.filePath}:${v.line})`);
+                lines.push(`  - Detail: ${v.detail}`);
+                lines.push(`  - Fix: ${v.regenerateInstruction}`);
+                lines.push("");
+            }
+        }
+        const otherViolations = violations.filter(v => v.severity !== "CRITICAL");
+        if (otherViolations.length > 0) {
+            lines.push(`### 🟠 Other Violations (${otherViolations.length})`);
+            lines.push("");
+            for (const v of otherViolations) {
+                lines.push(`- **${v.rule}** (${v.filePath}:${v.line})`);
+                lines.push(`  - Detail: ${v.detail}`);
+                lines.push(`  - Fix: ${v.regenerateInstruction}`);
+                lines.push("");
+            }
+        }
+        lines.push("---");
+        lines.push("> **Instruction:** Please regenerate the code for this operation.");
+        lines.push("> Address ALL violations listed above.");
+        lines.push("> Do NOT reintroduce the same violations.");
+        return lines.join("\n");
+    }
+    /**
+     * Validate tool response content against governance rules.
+     * Used by the MCP middleware to check file write content before returning.
+     */
+    static checkWriteContent(filePath, content, traceId) {
+        const ext = path.extname(filePath).toLowerCase();
+        const violations = [];
+        // Only check source files
+        if (![".ts", ".tsx", ".js", ".jsx", ".py", ".go", ".java", ".rs", ".swift", ".kt"].includes(ext)) {
+            return { allowed: true, violations: [], instruction: null };
+        }
+        const lines = content.split("\n");
+        // 1. Check for `any` type (CRITICAL)
+        if ([".ts", ".tsx"].includes(ext)) {
+            for (let i = 0; i < lines.length; i++) {
+                const anyMatch = lines[i].match(/:\s*any\b/);
+                if (anyMatch) {
+                    violations.push({
+                        rule: "No `any` Type",
+                        severity: "CRITICAL",
+                        filePath,
+                        line: i + 1,
+                        detail: `Line ${i + 1}: ${anyMatch[0].substring(0, 60)}`,
+                        regenerateInstruction: "Replace `any` with `unknown` and add proper type guards. Or use a specific interface/type.",
+                    });
+                }
+            }
+        }
+        // 2. Check for console.log (CRITICAL)
+        for (let i = 0; i < lines.length; i++) {
+            const consoleMatch = lines[i].match(/console\.(log|error|warn|debug)\s*\(/);
+            if (consoleMatch) {
+                // Check if file is exempt
+                if (!filePath.includes("logger.ts") && !filePath.includes("cli.ts") && !filePath.includes("compliance.ts")) {
+                    violations.push({
+                        rule: "No Console Log",
+                        severity: "CRITICAL",
+                        filePath,
+                        line: i + 1,
+                        detail: `Line ${i + 1}: ${consoleMatch[0].substring(0, 60)}`,
+                        regenerateInstruction: "Replace with the project's logger (import from the appropriate logger module).",
+                    });
+                }
+            }
+        }
+        // 3. Check for hardcoded secrets (HIGH)
+        const secretPatterns = [
+            { pattern: /(['"])sk-[a-zA-Z0-9]{20,}['"]/, name: "OpenAI API Key" },
+            { pattern: /(['"])ghp_[a-zA-Z0-9]{36,}['"]/, name: "GitHub Token" },
+            { pattern: /(['"])AIza[0-9A-Za-z_-]{35}['"]/, name: "Google API Key" },
+        ];
+        for (let i = 0; i < lines.length; i++) {
+            for (const { pattern, name } of secretPatterns) {
+                if (pattern.test(lines[i])) {
+                    violations.push({
+                        rule: `Hardcoded Secret: ${name}`,
+                        severity: "HIGH",
+                        filePath,
+                        line: i + 1,
+                        detail: `Potential ${name} hardcoded at line ${i + 1}`,
+                        regenerateInstruction: `Use environment variables (process.env.*) instead of hardcoding ${name}.`,
+                    });
+                }
+            }
+        }
+        // If violations found, prepare snapshot and rollback
+        if (violations.length > 0) {
+            AutoRollback.prepareWrite(filePath, traceId);
+            const rolledBack = AutoRollback.validateAndRollback(filePath, content, violations);
+            const instruction = AutoRollback.buildRegenerateInstruction(violations, "write_file");
+            return {
+                allowed: false,
+                violations: rolledBack || violations,
+                instruction,
+            };
+        }
+        return { allowed: true, violations: [], instruction: null };
+    }
+    /**
+     * Get snapshot stats.
+     */
+    static getSnapshotStats() {
+        const all = AutoRollback.snapshots.getAll();
+        return {
+            total: all.length,
+            restored: all.filter(s => s.restored).length,
+            pending: all.filter(s => !s.restored).length,
+        };
+    }
+}
+export { AutoRollback as AutoRollbackEngine };
+//# sourceMappingURL=auto-rollback.js.map

package/dist/atabey-mcp/src/mcp/utils/auto-rollback.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-rollback.js","sourceRoot":"","sources":["../../../../../../atabey-mcp/src/mcp/utils/auto-rollback.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAEpC,mEAAmE;AACnE,MAAM,YAAY,GAAG,mBAAmB,CAAC;AAEzC,6EAA6E;AAC7E,6DAA6D;AAC7D,IAAI,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM;IACpC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,wBAAwB,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/D,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;AAEpB,gEAAgE;AAChE,MAAM,UAAU,mBAAmB,CAAC,WAAmB;IACnD,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;AAChD,CAAC;AAYD,MAAM,eAAe;IACT,SAAS,GAA8B,IAAI,GAAG,EAAE,CAAC;IAEjD,eAAe,CAAC,YAAoB,EAAE,OAAe;QACzD,MAAM,QAAQ,GAAG,YAAY,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC1E,MAAM,IAAI,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/E,OAAO,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,YAAY,EAAE,GAAG,QAAQ,KAAK,IAAI,KAAK,OAAO,MAAM,CAAC,CAAC;IAC5F,CAAC;IAEO,iBAAiB;QACrB,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;QACrD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACtB,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3C,CAAC;IACL,CAAC;IAEO,mBAAmB,CAAC,YAAoB,EAAE,QAAsB;QACpE,IAAI,CAAC;YACD,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YACxE,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC;gBACxB,QAAQ,EAAE,YAAY;gBACtB,eAAe,EAAE,QAAQ,CAAC,eAAe;gBACzC,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC5B,CAAC,CAAC;YACH,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC;IACjC,CAAC;IAEO,mBAAmB,CAAC,QAAgB;QACxC,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACxC,qFAAqF;YACrF,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;YACrD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,OAAO,IAAI,CAAC;YAErC,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC;iBAC5B,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;iBAC/B,IAAI,EAAE;iBACN,OAAO,EAAE,CAAC,CAAC,eAAe;YAE/B,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;gBACpB,IAAI,CAAC;oBACD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;oBAC/B,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;oBAC9C,IAAI,CAAC;wBACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBACjC,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;4BAC7B,OAAO;gCACH,QAAQ,EAAE,IAAI,CAAC,QAAQ;gCACvB,eAAe,EAAE,IAAI,CAAC,eAAe;gCACrC,SAAS,EAAE,IAAI,CAAC,SAAS;gCACzB,OAAO,EAAE,IAAI,CAAC,OAAO;gCACrB,QAAQ,EAAE,KAAK;6BAClB,CAAC;wBACN,CAAC;oBACL,CAAC;oBAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;gBAC5B,CAAC;gBAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;YAC5B,CAAC;QACL,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;QACxB,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;;OAGG;IACI,OAAO,CAAC,QAAgB,EAAE,OAAe;QAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,eAAe,GAAkB,IAAI,CAAC;QAE1C,IAAI,CAAC;YACD,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;gBAC9B,eAAe,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YAC5D,CAAC;QACL,CAAC;QAAC,MAAM,CAAC,CAAC,uCAAuC,CAAC,CAAC;QAEnD,MAAM,QAAQ,GAAiB;YAC3B,QAAQ,EAAE,YAAY;YACtB,eAAe;YACf,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,OAAO;YACP,QAAQ,EAAE,KAAK;SAClB,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC3C,IAAI,CAAC,mBAAmB,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACrD,CAAC;IAED;;;OAGG;IACI,GAAG,CAAC,QAAgB;QACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACzC,IAAI,GAAG;YAAE,OAAO,GAAG,CAAC;QAEpB,MAAM,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QACnD,IAAI,OAAO,EAAE,CAAC;YACV,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACtC,OAAO,OAAO,CAAC;QACnB,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAED;;;OAGG;IACI,OAAO,CAAC,QAAgB;QAC3B,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAEhD,uBAAuB;QACvB,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,YAAY,CAAC,IAAI,SAAS,CAAC;YAC/D,IAAI,QAAQ;gBAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC7D,CAAC;QAED,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,QAAQ;YAAE,OAAO,KAAK,CAAC;QAEjD,IAAI,CAAC;YACD,IAAI,QAAQ,CAAC,eAAe,KAAK,IAAI,EAAE,CAAC;gBACpC,yCAAyC;gBACzC,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;gBACjE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YACnG,CAAC;iBAAM,CAAC;gBACJ,2BAA2B;gBAC3B,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;oBAC9B,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;oBAC5B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,kCAAkC,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;gBAC7G,CAAC;YACL,CAAC;YAED,QAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC;YAEzB,wBAAwB;YACxB,IAAI,CAAC;gBACD,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;gBACxE,IAAI,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC;oBAAE,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YAC7D,CAAC;YAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;YAExB,OAAO,IAAI,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,gCAAgC,YAAY,KAAK,KAAK,IAAI,CAAC,CAAC;YACjF,OAAO,KAAK,CAAC;QACjB,CAAC;IACL,CAAC;IAED;;OAEG;IACI,OAAO,CAAC,WAAmB,MAAM;QACpC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC;QACrC,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAChD,IAAI,QAAQ,CAAC,SAAS,GAAG,MAAM,EAAE,CAAC;gBAC9B,kCAAkC;gBAClC,IAAI,CAAC;oBACD,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;oBACpE,IAAI,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC;wBAAE,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;gBAC7D,CAAC;gBAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;gBACxB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACpC,CAAC;QACL,CAAC;QAED,uCAAuC;QACvC,IAAI,CAAC;YACD,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;YACrD,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrB,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;gBAClC,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;oBACpB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;oBAC/B,IAAI,CAAC;wBACD,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;wBAC/B,IAAI,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,CAAC;4BACxB,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;wBACxB,CAAC;oBACL,CAAC;oBAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;gBAC5B,CAAC;YACL,CAAC;QACL,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACI,MAAM;QACT,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;IAC/C,CAAC;CACJ;AAaD,qEAAqE;AAErE,MAAM,OAAO,YAAY;IACb,MAAM,CAAC,SAAS,GAAG,IAAI,eAAe,EAAE,CAAC;IAEjD;;;OAGG;IACI,MAAM,CAAC,UAAU;QACpB,sDAAsD;QACtD,WAAW,CAAC,GAAG,EAAE;YACb,YAAY,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QACrC,CAAC,EAAE,MAAM,CAAC,CAAC;IACf,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,YAAY,CAAC,QAAgB,EAAE,OAAe;QACxD,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAED;;;;;;;;OAQG;IACI,MAAM,CAAC,mBAAmB,CAC7B,QAAgB,EAChB,OAAe,EACf,UAA6B;QAE7B,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAEzC,4BAA4B;QAC5B,MAAM,QAAQ,GAAG,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAE1D,IAAI,QAAQ,EAAE,CAAC;YACX,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,QAAQ,CAAC,WAAW,UAAU,CAAC,MAAM,iBAAiB,CAAC,CAAC;QAC1I,CAAC;QAED,OAAO,UAAU,CAAC;IACtB,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,YAAY,CAAC,SAAmB;QAC1C,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;YACxB,IAAI,CAAC;gBACD,IAAI,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;oBAAE,QAAQ,EAAE,CAAC;YACtD,CAAC;YAAC,MAAM,CAAC;gBACL,0BAA0B;YAC9B,CAAC;QACL,CAAC;QACD,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,YAAY,CAAC,OAAe;QACtC,MAAM,OAAO,GAAG,YAAY,CAAC,SAAS;aACjC,MAAM,EAAE;aACR,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,OAAO,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC;aACnD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC5B,OAAO,YAAY,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IAC9C,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,0BAA0B,CACpC,UAA6B,EAC7B,QAAgB;QAEhB,MAAM,kBAAkB,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,UAAU,CAAC,CAAC;QAE7E,MAAM,KAAK,GAAa;YACpB,+CAA+C;YAC/C,EAAE;YACF,oCAAoC,QAAQ,qCAAqC;YACjF,mDAAmD;YACnD,EAAE;SACL,CAAC;QAEF,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,KAAK,CAAC,IAAI,CAAC,+BAA+B,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC;YACxE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACf,KAAK,MAAM,CAAC,IAAI,kBAAkB,EAAE,CAAC;gBACjC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC;gBACxD,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;gBACtC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,qBAAqB,EAAE,CAAC,CAAC;gBAClD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACnB,CAAC;QACL,CAAC;QAED,MAAM,eAAe,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,UAAU,CAAC,CAAC;QAC1E,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,KAAK,CAAC,IAAI,CAAC,4BAA4B,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC;YAClE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACf,KAAK,MAAM,CAAC,IAAI,eAAe,EAAE,CAAC;gBAC9B,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC;gBACxD,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;gBACtC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,qBAAqB,EAAE,CAAC,CAAC;gBAClD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACnB,CAAC;QACL,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClB,KAAK,CAAC,IAAI,CAAC,mEAAmE,CAAC,CAAC;QAChF,KAAK,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;QACrD,KAAK,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;QAExD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,iBAAiB,CAC3B,QAAgB,EAChB,OAAe,EACf,OAAe;QAEf,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QACjD,MAAM,UAAU,GAAsB,EAAE,CAAC;QAEzC,0BAA0B;QAC1B,IAAI,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC/F,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QAChE,CAAC;QAED,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAElC,qCAAqC;QACrC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;gBAC7C,IAAI,QAAQ,EAAE,CAAC;oBACX,UAAU,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,eAAe;wBACrB,QAAQ,EAAE,UAAU;wBACpB,QAAQ;wBACR,IAAI,EAAE,CAAC,GAAG,CAAC;wBACX,MAAM,EAAE,QAAQ,CAAC,GAAG,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE;wBACxD,qBAAqB,EAAE,4FAA4F;qBACtH,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;QACL,CAAC;QAED,sCAAsC;QACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,MAAM,YAAY,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAC5E,IAAI,YAAY,EAAE,CAAC;gBACf,0BAA0B;gBAC1B,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;oBACzG,UAAU,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,gBAAgB;wBACtB,QAAQ,EAAE,UAAU;wBACpB,QAAQ;wBACR,IAAI,EAAE,CAAC,GAAG,CAAC;wBACX,MAAM,EAAE,QAAQ,CAAC,GAAG,CAAC,KAAK,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE;wBAC5D,qBAAqB,EAAE,gFAAgF;qBAC1G,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;QACL,CAAC;QAED,wCAAwC;QACxC,MAAM,cAAc,GAAG;YACnB,EAAE,OAAO,EAAE,+BAA+B,EAAE,IAAI,EAAE,gBAAgB,EAAE;YACpE,EAAE,OAAO,EAAE,gCAAgC,EAAE,IAAI,EAAE,cAAc,EAAE;YACnE,EAAE,OAAO,EAAE,iCAAiC,EAAE,IAAI,EAAE,gBAAgB,EAAE;SACzE,CAAC;QAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,cAAc,EAAE,CAAC;gBAC7C,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBACzB,UAAU,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,qBAAqB,IAAI,EAAE;wBACjC,QAAQ,EAAE,MAAM;wBAChB,QAAQ;wBACR,IAAI,EAAE,CAAC,GAAG,CAAC;wBACX,MAAM,EAAE,aAAa,IAAI,sBAAsB,CAAC,GAAG,CAAC,EAAE;wBACtD,qBAAqB,EAAE,mEAAmE,IAAI,GAAG;qBACpG,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;QACL,CAAC;QAED,qDAAqD;QACrD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,YAAY,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC7C,MAAM,UAAU,GAAG,YAAY,CAAC,mBAAmB,CAAC,QAAQ,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;YACnF,MAAM,WAAW,GAAG,YAAY,CAAC,0BAA0B,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YAEtF,OAAO;gBACH,OAAO,EAAE,KAAK;gBACd,UAAU,EAAE,UAAU,IAAI,UAAU;gBACpC,WAAW;aACd,CAAC;QACN,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;IAChE,CAAC;IAED;;OAEG;IACI,MAAM,CAAC,gBAAgB;QAK1B,MAAM,GAAG,GAAG,YAAY,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;QAC5C,OAAO;YACH,KAAK,EAAE,GAAG,CAAC,MAAM;YACjB,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,MAAM;YAC5C,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,MAAM;SAC/C,CAAC;IACN,CAAC;;AAGL,OAAO,EAAE,YAAY,IAAI,kBAAkB,EAAE,CAAC"}

package/dist/atabey-mcp/src/mcp/utils/context-optimizer.d.ts

@@ -0,0 +1,85 @@
+/**
+ * ─── CONTEXT OPTIMIZER v2 ─────────────────────────────────────────────
+ *
+ * Solves IDE/CLI limit problems:
+ * - Token limits (Claude 200K, GPT 128K, Cursor context)
+ * - Context window overflow
+ * - Rate limiting (tokens/min, calls/min)
+ * - File read size limits
+ * - Response size limits
+ * - Session context persistence (recovery after disconnect)
+ * - Smart prompt compression
+ *
+ * Multi-provider support: each IDE/CLI has different limits.
+ *
+ * ─── Provider Limits ──────────────────────────────────────────────
+ * | Platform    | Context | Max Call | Rate (min) | Rate (hour) |
+ * |-------------|---------|----------|------------|-------------|
+ * | Claude Code | 200K    | 8K       | 100K       | 500K        |
+ * | Cursor      | 100K    | 4K       | 50K        | 250K        |
+ * | Gemini CLI  | 1M      | 32K      | 500K       | 2M          |
+ * | VS Code     | 64K     | 4K       | 30K        | 150K        |
+ * | GitHub Copilot | 64K | 4K       | 30K        | 150K        |
+ */
+interface ProviderLimits {
+    maxContextTokens: number;
+    maxResponseTokens: number;
+    maxTokensPerMinute: number;
+    maxTokensPerHour: number;
+    maxFileReadSize: number;
+    maxFileLines: number;
+    charsPerToken: number;
+    maxToolCallsPerMinute: number;
+}
+/**
+ * Detect provider from environment or client name.
+ */
+export declare function detectProvider(clientName: string): string;
+/**
+ * Estimate token count from text.
+ *
+ * Delegates to the shared approximate cl100k tokenizer so that budget checks,
+ * smart truncation, and FinOps all report the SAME numbers (single source of
+ * truth). The `provider` argument is kept for signature compatibility.
+ */
+export declare function estimateTokens(text: string, _provider?: string): number;
+/**
+ * Smart file truncation — preserves imports, type definitions, and function signatures.
+ */
+export declare function smartTruncateFile(content: string, filePath: string, provider?: string): string;
+/**
+ * Enhanced token budget check with provider awareness.
+ */
+export declare function checkTokenBudget(agent: string, toolName: string, responseText: string, provider?: string): string | null;
+export declare function getTokenBudgetStats(agent: string): {
+    totalTokens: number;
+    recentMinute: number;
+    recentHour: number;
+    blocked: boolean;
+    blockedRemaining: number;
+    callCount: number;
+};
+export declare function getAllTokenBudgetStats(): Record<string, ReturnType<typeof getTokenBudgetStats>>;
+export declare function resetTokenBudgets(): void;
+/**
+ * Apply runtime token-budget overrides (from the dashboard API). Keys must be
+ * valid ProviderLimits fields; unknown keys are ignored. Env vars still win.
+ */
+export declare function updateTokenBudgetConfig(newConfig: Record<string, number>): void;
+export declare function getTokenBudgetConfig(): {
+    [x: string]: ProviderLimits;
+};
+/**
+ * Whether token/rate budget breaches should HARD-BLOCK tool calls.
+ * Opt-in via `ATABEY_TOKEN_BUDGET_ENFORCE=true` (default: advisory only).
+ */
+export declare function isTokenBudgetEnforced(): boolean;
+/**
+ * Pre-execution token/rate budget gate.
+ *
+ * Returns a block message when the agent is over its token/rate budget AND
+ * enforcement is enabled; `null` otherwise (advisory mode). Usage recording
+ * still happens post-execution via `checkTokenBudget`.
+ */
+export declare function enforceTokenBudgetPreCall(agent: string, _toolName: string): string | null;
+export {};