mcp-agent-foundry 1.0.0

package/dist/worktrees/resource-manager.js

@@ -0,0 +1,228 @@
+/**
+ * Resource Manager
+ *
+ * Manages worktree resource limits including count, disk usage,
+ * stale detection, and automatic cleanup.
+ */
+import { execSync } from "child_process";
+import { createLogger } from "../observability/logger.js";
+const DEFAULT_LIMITS = {
+    maxWorktrees: 10,
+    maxDiskMB: 10 * 1024, // 10 GB
+    cleanupThresholdMB: 8 * 1024, // 8 GB
+    maxIdleTimeMs: 24 * 60 * 60 * 1000, // 24 hours
+};
+export class ResourceManager {
+    worktreeManager;
+    limits;
+    logger;
+    constructor(worktreeManager, limits) {
+        this.worktreeManager = worktreeManager;
+        this.limits = { ...DEFAULT_LIMITS, ...limits };
+        this.logger = createLogger({ name: "resource-manager" });
+    }
+    /**
+     * Get current resource usage.
+     */
+    async getCurrentUsage() {
+        const worktrees = this.worktreeManager.list();
+        const now = Date.now();
+        let totalDiskMB = 0;
+        let oldestAge = 0;
+        for (const wt of worktrees) {
+            const diskMB = await this.calculateDiskUsage(wt.path);
+            totalDiskMB += diskMB;
+            const age = now - wt.createdAt;
+            if (age > oldestAge)
+                oldestAge = age;
+        }
+        return {
+            currentWorktrees: worktrees.length,
+            availableSlots: Math.max(0, this.limits.maxWorktrees - worktrees.length),
+            diskUsageMB: totalDiskMB,
+            oldestWorktreeAge: oldestAge,
+        };
+    }
+    /**
+     * Check if a new worktree can be allocated.
+     */
+    async canAllocate() {
+        const usage = await this.getCurrentUsage();
+        if (usage.currentWorktrees >= this.limits.maxWorktrees) {
+            return {
+                allowed: false,
+                reason: `Maximum worktrees (${this.limits.maxWorktrees}) reached`,
+            };
+        }
+        if (usage.diskUsageMB >= this.limits.maxDiskMB) {
+            return {
+                allowed: false,
+                reason: `Disk limit (${this.limits.maxDiskMB}MB) reached`,
+            };
+        }
+        return { allowed: true };
+    }
+    /**
+     * Get detailed metrics for all worktrees.
+     */
+    async getAllWorktreeMetrics() {
+        const worktrees = this.worktreeManager.list();
+        const metrics = [];
+        for (const wt of worktrees) {
+            metrics.push({
+                id: wt.id,
+                path: wt.path,
+                branch: wt.branch,
+                status: wt.status,
+                taskId: wt.taskId,
+                createdAt: wt.createdAt,
+                lastActivityAt: wt.lastActivityAt,
+                diskUsageMB: await this.calculateDiskUsage(wt.path),
+            });
+        }
+        return metrics;
+    }
+    /**
+     * Get metrics for a specific worktree.
+     */
+    async getWorktreeMetrics(worktreeId) {
+        const wt = this.worktreeManager.get(worktreeId);
+        if (!wt)
+            return null;
+        return {
+            id: wt.id,
+            path: wt.path,
+            branch: wt.branch,
+            status: wt.status,
+            taskId: wt.taskId,
+            createdAt: wt.createdAt,
+            lastActivityAt: wt.lastActivityAt,
+            diskUsageMB: await this.calculateDiskUsage(wt.path),
+        };
+    }
+    /**
+     * Find stale worktrees (idle beyond threshold).
+     */
+    findStaleWorktrees(maxIdleTimeMs) {
+        const threshold = maxIdleTimeMs ?? this.limits.maxIdleTimeMs;
+        const cutoff = Date.now() - threshold;
+        return this.worktreeManager.list().filter((wt) => (wt.status === "available" ||
+            wt.status === "stale" ||
+            wt.status === "completing") &&
+            wt.lastActivityAt < cutoff);
+    }
+    /**
+     * Check if cleanup should be triggered.
+     */
+    async shouldTriggerCleanup() {
+        const usage = await this.getCurrentUsage();
+        return usage.diskUsageMB >= this.limits.cleanupThresholdMB;
+    }
+    /**
+     * Cleanup stale worktrees.
+     */
+    async cleanupStale(maxIdleTimeMs) {
+        const stale = this.findStaleWorktrees(maxIdleTimeMs);
+        const cleaned = [];
+        for (const wt of stale) {
+            try {
+                await this.worktreeManager.remove(wt.id, true);
+                cleaned.push(wt.id);
+                this.logger.info("Cleaned up stale worktree", {
+                    id: wt.id,
+                    path: wt.path,
+                });
+            }
+            catch (error) {
+                this.logger.warn("Failed to cleanup worktree", {
+                    id: wt.id,
+                    error: error instanceof Error ? error.message : String(error),
+                });
+            }
+        }
+        return cleaned;
+    }
+    /**
+     * Cleanup worktrees by status.
+     */
+    async cleanupByStatus(status) {
+        const worktrees = this.worktreeManager.listByStatus(status);
+        const cleaned = [];
+        for (const wt of worktrees) {
+            try {
+                await this.worktreeManager.remove(wt.id, true);
+                cleaned.push(wt.id);
+            }
+            catch (error) {
+                this.logger.warn("Failed to cleanup worktree", { id: wt.id, error });
+            }
+        }
+        return cleaned;
+    }
+    /**
+     * Force cleanup specific worktrees.
+     */
+    async forceCleanup(worktreeIds) {
+        const cleaned = [];
+        const failed = [];
+        for (const id of worktreeIds) {
+            try {
+                await this.worktreeManager.remove(id, true);
+                cleaned.push(id);
+            }
+            catch (error) {
+                this.logger.error("Force cleanup failed", {
+                    id,
+                    error: error instanceof Error ? error : undefined,
+                });
+                failed.push(id);
+            }
+        }
+        return { cleaned, failed };
+    }
+    /**
+     * Get full resource report.
+     */
+    async getReport() {
+        const usage = await this.getCurrentUsage();
+        const worktrees = await this.getAllWorktreeMetrics();
+        const stale = this.findStaleWorktrees();
+        return {
+            usage,
+            limits: this.limits,
+            worktrees,
+            staleCount: stale.length,
+            shouldCleanup: await this.shouldTriggerCleanup(),
+        };
+    }
+    /**
+     * Calculate disk usage for a path in MB.
+     *
+     * Note: Uses execSync with shell for the du command. The dirPath parameter
+     * is system-controlled (comes from WorktreeManager internal state) and is
+     * quoted to prevent any injection. This is safe as worktree paths are
+     * generated internally and not from user input.
+     */
+    async calculateDiskUsage(dirPath) {
+        try {
+            // Use du command for accurate disk usage
+            // dirPath is system-controlled from WorktreeManager, quoted for safety
+            const result = execSync(`du -sm "${dirPath}" 2>/dev/null || echo "0"`, {
+                encoding: "utf-8",
+                timeout: 5000,
+            });
+            const match = result.match(/^(\d+)/);
+            return match ? parseInt(match[1], 10) : 0;
+        }
+        catch {
+            return 0;
+        }
+    }
+    /**
+     * Get the configured limits.
+     */
+    getLimits() {
+        return { ...this.limits };
+    }
+}
+//# sourceMappingURL=resource-manager.js.map

package/dist/worktrees/resource-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resource-manager.js","sourceRoot":"","sources":["../../src/worktrees/resource-manager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAW1D,MAAM,cAAc,GAAmB;IACrC,YAAY,EAAE,EAAE;IAChB,SAAS,EAAE,EAAE,GAAG,IAAI,EAAE,QAAQ;IAC9B,kBAAkB,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO;IACrC,aAAa,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,WAAW;CAChD,CAAC;AAEF,MAAM,OAAO,eAAe;IACT,eAAe,CAAkB;IACjC,MAAM,CAAiB;IACvB,MAAM,CAAC;IAExB,YACE,eAAgC,EAChC,MAAgC;QAEhC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;QAC/C,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe;QACnB,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;QAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;YACtD,WAAW,IAAI,MAAM,CAAC;YAEtB,MAAM,GAAG,GAAG,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC;YAC/B,IAAI,GAAG,GAAG,SAAS;gBAAE,SAAS,GAAG,GAAG,CAAC;QACvC,CAAC;QAED,OAAO;YACL,gBAAgB,EAAE,SAAS,CAAC,MAAM;YAClC,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,GAAG,SAAS,CAAC,MAAM,CAAC;YACxE,WAAW,EAAE,WAAW;YACxB,iBAAiB,EAAE,SAAS;SAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW;QACf,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAE3C,IAAI,KAAK,CAAC,gBAAgB,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YACvD,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,sBAAsB,IAAI,CAAC,MAAM,CAAC,YAAY,WAAW;aAClE,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC/C,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,eAAe,IAAI,CAAC,MAAM,CAAC,SAAS,aAAa;aAC1D,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,qBAAqB;QACzB,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAsB,EAAE,CAAC;QAEtC,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC;gBACX,EAAE,EAAE,EAAE,CAAC,EAAE;gBACT,IAAI,EAAE,EAAE,CAAC,IAAI;gBACb,MAAM,EAAE,EAAE,CAAC,MAAM;gBACjB,MAAM,EAAE,EAAE,CAAC,MAAM;gBACjB,MAAM,EAAE,EAAE,CAAC,MAAM;gBACjB,SAAS,EAAE,EAAE,CAAC,SAAS;gBACvB,cAAc,EAAE,EAAE,CAAC,cAAc;gBACjC,WAAW,EAAE,MAAM,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC,IAAI,CAAC;aACpD,CAAC,CAAC;QACL,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CAAC,UAAkB;QACzC,MAAM,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAChD,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC;QAErB,OAAO;YACL,EAAE,EAAE,EAAE,CAAC,EAAE;YACT,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,MAAM,EAAE,EAAE,CAAC,MAAM;YACjB,MAAM,EAAE,EAAE,CAAC,MAAM;YACjB,MAAM,EAAE,EAAE,CAAC,MAAM;YACjB,SAAS,EAAE,EAAE,CAAC,SAAS;YACvB,cAAc,EAAE,EAAE,CAAC,cAAc;YACjC,WAAW,EAAE,MAAM,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC,IAAI,CAAC;SACpD,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,aAAsB;QACvC,MAAM,SAAS,GAAG,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;QAC7D,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAEtC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,MAAM,CACvC,CAAC,EAAE,EAAE,EAAE,CACL,CAAC,EAAE,CAAC,MAAM,KAAK,WAAW;YACxB,EAAE,CAAC,MAAM,KAAK,OAAO;YACrB,EAAE,CAAC,MAAM,KAAK,YAAY,CAAC;YAC7B,EAAE,CAAC,cAAc,GAAG,MAAM,CAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB;QACxB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAC3C,OAAO,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC;IAC7D,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,aAAsB;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC,CAAC;QACrD,MAAM,OAAO,GAAa,EAAE,CAAC;QAE7B,KAAK,MAAM,EAAE,IAAI,KAAK,EAAE,CAAC;YACvB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;gBAC/C,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE;oBAC5C,EAAE,EAAE,EAAE,CAAC,EAAE;oBACT,IAAI,EAAE,EAAE,CAAC,IAAI;iBACd,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE;oBAC7C,EAAE,EAAE,EAAE,CAAC,EAAE;oBACT,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,MAAsB;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAa,EAAE,CAAC;QAE7B,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;YAC3B,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;gBAC/C,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACtB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACvE,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAChB,WAAqB;QAErB,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAa,EAAE,CAAC;QAE5B,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;gBAC5C,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACnB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE;oBACxC,EAAE;oBACF,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;iBAClD,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAClB,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS;QACb,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAC3C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,qBAAqB,EAAE,CAAC;QACrD,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAExC,OAAO;YACL,KAAK;YACL,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS;YACT,UAAU,EAAE,KAAK,CAAC,MAAM;YACxB,aAAa,EAAE,MAAM,IAAI,CAAC,oBAAoB,EAAE;SACjD,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACK,KAAK,CAAC,kBAAkB,CAAC,OAAe;QAC9C,IAAI,CAAC;YACH,yCAAyC;YACzC,uEAAuE;YACvE,MAAM,MAAM,GAAG,QAAQ,CAAC,WAAW,OAAO,2BAA2B,EAAE;gBACrE,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACrC,OAAO,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;CACF"}

package/package.json

@@ -0,0 +1,88 @@
+{
+  "name": "mcp-agent-foundry",
+  "version": "1.0.0",
+  "description": "Multi-Agent AI Orchestration with Git Worktree Isolation for Claude Code",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "bin": {
+    "agent-foundry": "./dist/cli.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsx watch src/index.ts",
+    "start": "node dist/index.js",
+    "test": "vitest",
+    "test:run": "vitest run",
+    "test:coverage": "vitest run --coverage",
+    "lint": "eslint src --ext .ts",
+    "lint:fix": "eslint src --ext .ts --fix",
+    "typecheck": "tsc --noEmit",
+    "clean": "rm -rf dist",
+    "prepublishOnly": "npm run build"
+  },
+  "engines": {
+    "node": ">=20.0.0"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/sashabogi/agent-foundry.git"
+  },
+  "homepage": "https://github.com/sashabogi/agent-foundry#readme",
+  "bugs": {
+    "url": "https://github.com/sashabogi/agent-foundry/issues"
+  },
+  "keywords": [
+    "mcp",
+    "model-context-protocol",
+    "claude",
+    "claude-code",
+    "multi-agent",
+    "orchestration",
+    "worktree",
+    "git",
+    "ai",
+    "llm",
+    "openai",
+    "anthropic",
+    "gemini",
+    "deepseek",
+    "ollama",
+    "parallel",
+    "isolation",
+    "pipeline"
+  ],
+  "author": "Sasha Bogojevic",
+  "license": "MIT",
+  "dependencies": {
+    "@anthropic-ai/claude-agent-sdk": "^0.2.25",
+    "@clack/prompts": "^0.11.0",
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "ajv": "^8.17.1",
+    "picocolors": "^1.1.1",
+    "pino": "^9.6.0",
+    "simple-git": "^3.22.0",
+    "yaml": "^2.7.0",
+    "zod": "^3.24.1"
+  },
+  "devDependencies": {
+    "@eslint/js": "^9.39.2",
+    "@types/node": "^22.10.7",
+    "@typescript-eslint/eslint-plugin": "^8.21.0",
+    "@typescript-eslint/parser": "^8.21.0",
+    "@vitest/coverage-v8": "^3.0.4",
+    "eslint": "^9.39.2",
+    "eslint-plugin-import": "^2.32.0",
+    "prettier": "^3.8.1",
+    "tsx": "^4.19.2",
+    "typescript": "^5.7.3",
+    "typescript-eslint": "^8.54.0",
+    "vitest": "^3.0.4"
+  },
+  "files": [
+    "dist",
+    "config",
+    "LICENSE",
+    "README.md"
+  ]
+}