atabey 0.0.13 → 0.0.14

package/framework-mcp/dist/framework-mcp/src/utils/auth.js

@@ -0,0 +1,78 @@
+/**
+ * ─── MCP AUTHENTICATION ──────────────────────────────────────────
+ *
+ * Simple API key authentication for MCP server.
+ * When MCP_AUTH_TOKEN is set, all MCP and API requests require
+ * Authorization: Bearer <token> header.
+ *
+ * Environment Variables:
+ *   MCP_AUTH_TOKEN     - API key for authentication (optional)
+ *   MCP_AUTH_USERS     - Comma-separated list of user:token pairs (optional)
+ *                        Example: "alice:key1,bob:key2"
+ *
+ * If neither is set, authentication is disabled (open access).
+ */
+let authUsers = [];
+let masterToken = null;
+/**
+ * Initialize authentication from environment variables.
+ */
+export function initAuth() {
+    masterToken = process.env.MCP_AUTH_TOKEN || null;
+    const usersStr = process.env.MCP_AUTH_USERS || "";
+    if (usersStr) {
+        authUsers = usersStr.split(",").map(pair => {
+            const [name, token] = pair.split(":");
+            return { name: name?.trim() || "unknown", token: token?.trim() || "" };
+        }).filter(u => u.token);
+    }
+    if (masterToken) {
+        console.error(`[AUTH] Master token configured (${masterToken.length} chars)`);
+    }
+    if (authUsers.length > 0) {
+        console.error(`[AUTH] ${authUsers.length} user(s) configured: ${authUsers.map(u => u.name).join(", ")}`);
+    }
+    if (!masterToken && authUsers.length === 0) {
+        console.error("[AUTH] No authentication configured - OPEN ACCESS");
+        console.error("[AUTH] Set MCP_AUTH_TOKEN=<key> or MCP_AUTH_USERS=user1:key1,user2:key2 to enable");
+    }
+}
+/**
+ * Authenticate a request using the Authorization header.
+ * Returns the authenticated user name or null if unauthorized.
+ */
+export function authenticate(req) {
+    if (!masterToken && authUsers.length === 0) {
+        // Auth disabled - allow all
+        return { authenticated: true, user: "anonymous" };
+    }
+    const authHeader = req.headers["authorization"];
+    if (!authHeader) {
+        return { authenticated: false, user: "" };
+    }
+    // Support both "Bearer <key>" and "<key>" formats
+    const token = authHeader.startsWith("Bearer ") ? authHeader.slice(7) : authHeader;
+    // Check master token first
+    if (masterToken && token === masterToken) {
+        return { authenticated: true, user: "admin" };
+    }
+    // Check user tokens
+    for (const user of authUsers) {
+        if (token === user.token) {
+            return { authenticated: true, user: user.name };
+        }
+    }
+    return { authenticated: false, user: "" };
+}
+/**
+ * Generate authentication headers for MCP client configuration.
+ */
+export function getAuthHeader(user, token) {
+    return `Bearer ${token}`;
+}
+/**
+ * Check if authentication is enabled.
+ */
+export function isAuthEnabled() {
+    return !!masterToken || authUsers.length > 0;
+}

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

@@ -0,0 +1,265 @@
+/**
+ * ─── 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 path from "path";
+class SnapshotManager {
+    snapshots = new Map();
+    /**
+     * Capture a pre-write snapshot of a file.
+     */
+    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) */ }
+        this.snapshots.set(resolvedPath, {
+            filePath: resolvedPath,
+            originalContent,
+            timestamp: Date.now(),
+            traceId,
+            restored: false,
+        });
+    }
+    /**
+     * Get a snapshot for a file.
+     */
+    get(filePath) {
+        return this.snapshots.get(path.resolve(filePath));
+    }
+    /**
+     * Restore a file to its pre-write state.
+     * Returns true if restored, false if no snapshot exists.
+     */
+    restore(filePath) {
+        const resolvedPath = path.resolve(filePath);
+        const snapshot = this.snapshots.get(resolvedPath);
+        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(process.cwd(), resolvedPath)}\n`);
+            }
+            else {
+                // File was new – delete it
+                if (fs.existsSync(resolvedPath)) {
+                    fs.unlinkSync(resolvedPath);
+                    process.stderr.write(`[ROLLBACK] Deleted (new file): ${path.relative(process.cwd(), resolvedPath)}\n`);
+                }
+            }
+            snapshot.restored = true;
+            return true;
+        }
+        catch (error) {
+            process.stderr.write(`[ROLLBACK] Failed to restore ${resolvedPath}: ${error}\n`);
+            return false;
+        }
+    }
+    /**
+     * Clean up old snapshots.
+     */
+    cleanup(maxAgeMs = 300000) {
+        const cutoff = Date.now() - maxAgeMs;
+        for (const [filePath, snapshot] of this.snapshots) {
+            if (snapshot.timestamp < cutoff) {
+                this.snapshots.delete(filePath);
+            }
+        }
+    }
+    /**
+     * 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(process.cwd(), filePath)} due to ${violations.length} violation(s)\n`);
+        }
+        return violations;
+    }
+    /**
+     * 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 };

package/framework-mcp/dist/framework-mcp/src/utils/compliance.js

@@ -53,7 +53,14 @@ export function verifyCorporateCompliance(content, filePath) {
                 }
                 // 5. Secure CLI Policy (No direct child_process)
                 if (moduleSpecifier.text === "child_process") {
-                    errors.push(`[ERROR] Corporate Compliance Breach: Direct 'child_process' usage is forbidden at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}. Use secure framework APIs.`);
+                    const isExempt = filePath.includes("src/cli/commands/") ||
+                        filePath.includes("shared/fs.ts") ||
+                        filePath.includes("shared/lock.ts") ||
+                        filePath.includes("quality-gate.ts") ||
+                        filePath.includes("providers/shared.ts");
+                    if (!isExempt) {
+                        errors.push(`[ERROR] Corporate Compliance Breach: Direct 'child_process' usage is forbidden at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}. Use secure framework APIs.`);
+                    }
                 }
             }
         }
@@ -75,10 +82,11 @@ export function verifyCorporateCompliance(content, filePath) {
             const prop = node.expression;
             if (ts.isIdentifier(prop.expression) && prop.expression.text === "fs") {
                 if (["writeFileSync", "appendFileSync"].includes(prop.name.text)) {
-                    // Lock files are exempt
+                    // Lock files, core atomic fs wrapper, and mcp configs are exempt
                     const args = node.arguments;
                     const firstArgText = args.length > 0 ? args[0].getText(sourceFile).toLowerCase() : "";
-                    if (!firstArgText.includes("lock")) {
+                    const isExemptFile = filePath.includes("shared/fs.ts") || filePath.includes("src/cli/commands/mcp.ts");
+                    if (!firstArgText.includes("lock") && !isExemptFile) {
                         errors.push(`[ERROR] Corporate Compliance Breach: Raw 'fs.${prop.name.text}' mutation detected at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}. Use atomic utilities instead.`);
                     }
                 }
@@ -118,7 +126,7 @@ export function scanProjectCompliance(targetDir = "src") {
     const files = getAllFiles(targetDir);
     for (const file of files) {
         // Skip exempt files
-        if (file.includes("compliance") || file.includes("definitions") || file.includes("logger") || file.includes("cli/index.ts"))
+        if (file.includes("compliance") || file.includes("definitions") || file.includes("logger") || file.includes("cli/index.ts") || file.includes("shared/fs.ts") || file.includes("shared/lock.ts"))
             continue;
         const content = fs.readFileSync(file, "utf8");
         try {

package/framework-mcp/dist/framework-mcp/src/utils/context-optimizer.js

@@ -0,0 +1,156 @@
+/**
+ * ─── CONTEXT OPTIMIZER ─────────────────────────────────────────────
+ *
+ * Prevents context poisoning and runaway token consumption by AI CLIs.
+ * Acts as a Circuit Breaker at the MCP middleware layer.
+ *
+ * Features:
+ * - Token Budget Enforcer: Max tokens per tool call response
+ * - File Read Limiter: Auto-truncate large file reads
+ * - Token-Based Rate Limiter: Max token spend per window
+ * - Response Size Governor: Truncates oversized responses
+ *
+ * [KVKK/GDPR] Also prevents PII leakage via oversized responses.
+ */
+// ─── Configuration ────────────────────────────────────────────────
+const CONFIG = {
+    /** Max tokens allowed in a single tool response */
+    MAX_TOKENS_PER_CALL: parseInt(process.env.MCP_MAX_TOKENS_PER_CALL || "4000", 10),
+    /** Max tokens per minute across all calls for an agent */
+    MAX_TOKENS_PER_MINUTE: parseInt(process.env.MCP_MAX_TOKENS_PER_MINUTE || "20000", 10),
+    /** Max tokens per hour across all calls for an agent */
+    MAX_TOKENS_PER_HOUR: parseInt(process.env.MCP_MAX_TOKENS_PER_HOUR || "100000", 10),
+    /** Max file size in bytes before auto-truncation (100KB) */
+    MAX_FILE_READ_SIZE: parseInt(process.env.MCP_MAX_FILE_READ_SIZE || "102400", 10),
+    /** Max lines to return for truncated files */
+    MAX_FILE_LINES: parseInt(process.env.MCP_MAX_FILE_LINES || "200", 10),
+    /** Token estimation ratio (chars per token) */
+    CHARS_PER_TOKEN: 4,
+};
+const agentBudgets = new Map();
+function getOrCreateBudget(agent) {
+    if (!agentBudgets.has(agent)) {
+        agentBudgets.set(agent, {
+            records: [],
+            totalTokens: 0,
+            lastWarning: 0,
+            blockedUntil: null,
+        });
+    }
+    return agentBudgets.get(agent);
+}
+function pruneOldRecords(agent) {
+    const oneHourAgo = Date.now() - 3600000;
+    agent.records = agent.records.filter(r => r.timestamp > oneHourAgo);
+}
+function sumTokensInWindow(agent, windowMs) {
+    const cutoff = Date.now() - windowMs;
+    return agent.records
+        .filter(r => r.timestamp > cutoff)
+        .reduce((sum, r) => sum + r.tokens, 0);
+}
+/**
+ * Estimate token count from text.
+ * Uses 4 chars per token as a conservative estimate.
+ */
+export function estimateTokens(text) {
+    if (!text)
+        return 0;
+    return Math.ceil(text.length / CONFIG.CHARS_PER_TOKEN);
+}
+/**
+ * Check if a tool response is within token budget.
+ * Returns null if allowed, or an error message if blocked.
+ */
+export function checkTokenBudget(agent, toolName, responseText) {
+    const budget = getOrCreateBudget(agent);
+    // Check if agent is currently blocked
+    if (budget.blockedUntil && Date.now() < budget.blockedUntil) {
+        const remaining = Math.ceil((budget.blockedUntil - Date.now()) / 1000);
+        return `[TOKEN BUDGET] Agent ${agent} is blocked for ${remaining}s. Token limit exceeded.`;
+    }
+    const estimatedTokens = estimateTokens(responseText);
+    // 1. Check per-call limit
+    if (estimatedTokens > CONFIG.MAX_TOKENS_PER_CALL) {
+        return `[TOKEN BUDGET] Response too large: ~${estimatedTokens} tokens (max: ${CONFIG.MAX_TOKENS_PER_CALL}). Truncate your request.`;
+    }
+    // Record the token usage
+    budget.records.push({
+        tokens: estimatedTokens,
+        timestamp: Date.now(),
+        toolName,
+    });
+    budget.totalTokens += estimatedTokens;
+    // Prune old records
+    pruneOldRecords(budget);
+    // 2. Check per-minute limit
+    const minuteTokens = sumTokensInWindow(budget, 60000);
+    if (minuteTokens > CONFIG.MAX_TOKENS_PER_MINUTE) {
+        budget.blockedUntil = Date.now() + 30000; // 30s cooldown
+        return `[TOKEN BUDGET] Agent ${agent} exceeded ${CONFIG.MAX_TOKENS_PER_MINUTE} tokens/min. Cooldown 30s.`;
+    }
+    // 3. Check per-hour limit
+    const hourTokens = sumTokensInWindow(budget, 3600000);
+    if (hourTokens > CONFIG.MAX_TOKENS_PER_HOUR) {
+        budget.blockedUntil = Date.now() + 120000; // 2min cooldown
+        return `[TOKEN BUDGET] Agent ${agent} exceeded ${CONFIG.MAX_TOKENS_PER_HOUR} tokens/hour. Cooldown 2min.`;
+    }
+    // 4. Warning at 80% of per-minute limit
+    if (minuteTokens > CONFIG.MAX_TOKENS_PER_MINUTE * 0.8) {
+        const now = Date.now();
+        if (now - budget.lastWarning > 10000) { // Throttle warnings to once per 10s
+            budget.lastWarning = now;
+            process.stderr.write(`[TOKEN BUDGET WARNING] Agent ${agent} at ${Math.round(minuteTokens / CONFIG.MAX_TOKENS_PER_MINUTE * 100)}% of minute token budget.\n`);
+        }
+    }
+    return null;
+}
+/**
+ * Truncate file content to fit within token budget.
+ * Returns truncated content with a header note.
+ */
+export function truncateFileContent(content, filePath, maxTokens = CONFIG.MAX_TOKENS_PER_CALL) {
+    const estimatedTokens = estimateTokens(content);
+    if (estimatedTokens <= maxTokens)
+        return content;
+    const lines = content.split("\n");
+    const maxLines = Math.min(CONFIG.MAX_FILE_LINES, Math.floor(maxTokens * CONFIG.CHARS_PER_TOKEN / 80));
+    const truncated = lines.slice(0, maxLines).join("\n");
+    const remaining = lines.length - maxLines;
+    return `[TRUNCATED] File "${filePath}" is too large (~${estimatedTokens} tokens). Showing first ${maxLines}/${lines.length} lines.\n\n${truncated}\n\n[... ${remaining} more lines truncated. Use grep_search for targeted queries.]`;
+}
+/**
+ * Get token budget stats for an agent.
+ */
+export function getTokenBudgetStats(agent) {
+    const budget = agentBudgets.get(agent);
+    if (!budget) {
+        return { totalTokens: 0, recentMinute: 0, recentHour: 0, blocked: false, blockedRemaining: 0, callCount: 0 };
+    }
+    pruneOldRecords(budget);
+    return {
+        totalTokens: budget.totalTokens,
+        recentMinute: sumTokensInWindow(budget, 60000),
+        recentHour: sumTokensInWindow(budget, 3600000),
+        blocked: budget.blockedUntil !== null && Date.now() < budget.blockedUntil,
+        blockedRemaining: budget.blockedUntil ? Math.max(0, budget.blockedUntil - Date.now()) : 0,
+        callCount: budget.records.length,
+    };
+}
+/**
+ * Get all agents' token budget stats.
+ */
+export function getAllTokenBudgetStats() {
+    const stats = {};
+    for (const [agent] of agentBudgets) {
+        stats[agent] = getTokenBudgetStats(agent);
+    }
+    return stats;
+}
+/**
+ * Reset all token budgets.
+ */
+export function resetTokenBudgets() {
+    agentBudgets.clear();
+}
+export { CONFIG as TokenBudgetConfig };