@overlordai/worker 1.0.0

package/dist/worker-client.js

@@ -0,0 +1,303 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkerClient = void 0;
+const ws_1 = __importDefault(require("ws"));
+const protocol_1 = require("@overlordai/protocol");
+const hardware_js_1 = require("./hardware.js");
+const pino_1 = __importDefault(require("pino"));
+const log = (0, pino_1.default)({ name: 'worker-client' });
+const MAX_RECONNECT_DELAY_MS = 30_000;
+const INITIAL_RECONNECT_DELAY_MS = 1_000;
+const DEDUP_CLEANUP_INTERVAL_MS = 60_000;
+class WorkerClient {
+    ws = null;
+    options;
+    handlers;
+    heartbeatTimer = null;
+    reconnectTimer = null;
+    reconnectDelay = INITIAL_RECONNECT_DELAY_MS;
+    connected = false;
+    intentionalClose = false;
+    dedupSet = new Map();
+    dedupCleanupTimer = null;
+    constructor(options, handlers) {
+        this.options = options;
+        this.handlers = handlers;
+    }
+    isConnected() {
+        return this.connected;
+    }
+    connect() {
+        if (this.ws) {
+            try {
+                this.ws.removeAllListeners();
+                this.ws.close();
+            }
+            catch { }
+            this.ws = null;
+        }
+        this.intentionalClose = false;
+        const wsUrl = this.buildWsUrl();
+        log.info({ url: wsUrl }, 'Connecting to dispatcher');
+        try {
+            this.ws = new ws_1.default(wsUrl, {
+                headers: {
+                    'Authorization': `Bearer ${this.options.getToken()}`,
+                },
+            });
+        }
+        catch (err) {
+            log.error({ err }, 'Failed to create WebSocket');
+            this.scheduleReconnect();
+            return;
+        }
+        this.ws.on('open', () => {
+            log.info('WebSocket connected');
+            this.connected = true;
+            this.reconnectDelay = INITIAL_RECONNECT_DELAY_MS;
+            // Send auth frame as first message
+            this.sendFrame({
+                type: 'auth',
+                token: this.options.getToken(),
+                protocolVersion: protocol_1.PROTOCOL_VERSION,
+            });
+            // Start heartbeat
+            this.startHeartbeat();
+            // Start dedup cleanup
+            this.startDedupCleanup();
+            this.handlers.onConnected();
+        });
+        this.ws.on('message', (data) => {
+            this.handleMessage(data);
+        });
+        this.ws.on('close', (code, reason) => {
+            log.info({ code, reason: reason.toString() }, 'WebSocket closed');
+            this.cleanup();
+            this.handlers.onDisconnected();
+            if (!this.intentionalClose) {
+                this.scheduleReconnect();
+            }
+        });
+        this.ws.on('error', (err) => {
+            log.error({ err }, 'WebSocket error');
+            // The close event will fire after this, triggering reconnect
+        });
+    }
+    disconnect() {
+        this.intentionalClose = true;
+        this.cleanup();
+        if (this.ws) {
+            this.ws.close(1000, 'Worker shutting down');
+            this.ws = null;
+        }
+    }
+    sendHeartbeat() {
+        const metrics = (0, hardware_js_1.collectRuntimeMetrics)();
+        const frame = {
+            type: 'heartbeat',
+            machineId: this.options.machineId,
+            timestamp: new Date().toISOString(),
+            metrics: {
+                cpuUsage: metrics.cpuUsage,
+                memoryUsage: metrics.memoryUsage,
+                diskUsage: metrics.diskUsage,
+                activeSlots: this.options.getActiveSlots(),
+            },
+            capabilities: this.options.getCapabilities(),
+        };
+        this.sendFrame(frame);
+    }
+    sendAck(msgId, payload) {
+        const frame = {
+            type: 'ack',
+            msgId,
+            payload,
+        };
+        this.sendFrame(frame);
+    }
+    sendProgress(taskId, stage, status, message) {
+        const frame = {
+            type: 'progress',
+            taskId,
+            stage,
+            status,
+            message,
+        };
+        this.sendFrame(frame);
+    }
+    sendReconnect(tasks) {
+        const frame = {
+            type: 'reconnect',
+            machineId: this.options.machineId,
+            tasks,
+        };
+        this.sendFrame(frame);
+    }
+    sendRecoverySecretAck() {
+        const frame = {
+            type: 'recovery_secret_ack',
+        };
+        this.sendFrame(frame);
+    }
+    sendStageConfirmRequest(taskId, stageIndex, mode, timeout, prompt, choices) {
+        const frame = {
+            type: 'stage_confirm_request',
+            taskId,
+            stageIndex,
+            mode,
+            timeout,
+            ...(prompt !== undefined && { prompt }),
+            ...(choices !== undefined && { choices }),
+        };
+        this.sendFrame(frame);
+    }
+    sendFrame(frame) {
+        if (!this.ws || this.ws.readyState !== ws_1.default.OPEN) {
+            log.warn({ frameType: frame.type }, 'Cannot send frame, WebSocket not open');
+            return;
+        }
+        try {
+            this.ws.send(JSON.stringify(frame));
+        }
+        catch (err) {
+            log.error({ err, frameType: frame.type }, 'Failed to send frame');
+        }
+    }
+    handleMessage(data) {
+        let frame;
+        try {
+            frame = JSON.parse(data.toString());
+        }
+        catch (err) {
+            log.error({ err }, 'Failed to parse incoming frame');
+            return;
+        }
+        // Dedup messages with msgId
+        if ('msgId' in frame && frame.msgId) {
+            const msgId = frame.msgId;
+            if (this.dedupSet.has(msgId)) {
+                log.debug({ msgId, type: frame.type }, 'Duplicate message, ignoring');
+                return;
+            }
+            this.dedupSet.set(msgId, Date.now());
+        }
+        log.debug({ type: frame.type }, 'Received frame');
+        switch (frame.type) {
+            case 'heartbeat_ack':
+                this.handlers.onHeartbeatAck(frame);
+                break;
+            case 'execute':
+                this.handlers.onExecute(frame);
+                break;
+            case 'cancel':
+                this.handlers.onCancel(frame);
+                break;
+            case 'cleanup_workspace':
+                this.handlers.onCleanupWorkspace(frame);
+                break;
+            case 'resume':
+                this.handlers.onResume(frame);
+                break;
+            case 'server_shutting_down':
+                this.handlers.onServerShuttingDown(frame);
+                break;
+            case 'recovery_secret_bootstrap':
+                this.handlers.onRecoverySecretBootstrap(frame);
+                break;
+            case 'start_tunnel':
+                this.handlers.onStartTunnel(frame);
+                break;
+            case 'stop_tunnel':
+                this.handlers.onStopTunnel(frame);
+                break;
+            case 'stage_confirm_response':
+                this.handlers.onStageConfirmResponse(frame);
+                break;
+            default:
+                log.error({ frame }, 'Unknown frame type received from server');
+                log.warn({ frame }, 'Unknown frame type');
+        }
+    }
+    buildWsUrl() {
+        const host = this.options.overlordHost;
+        let wsBase;
+        if (host.startsWith('https://')) {
+            wsBase = 'wss://' + host.slice(8);
+        }
+        else if (host.startsWith('http://')) {
+            wsBase = 'ws://' + host.slice(7);
+        }
+        else {
+            wsBase = 'ws://' + host;
+        }
+        // Remove trailing slash
+        if (wsBase.endsWith('/')) {
+            wsBase = wsBase.slice(0, -1);
+        }
+        return `${wsBase}/api/workers/channel`;
+    }
+    startHeartbeat() {
+        this.stopHeartbeat();
+        this.heartbeatTimer = setInterval(() => {
+            this.sendHeartbeat();
+        }, protocol_1.HEARTBEAT_INTERVAL_MS);
+        if (this.heartbeatTimer.unref) {
+            this.heartbeatTimer.unref();
+        }
+        // Send first heartbeat immediately
+        this.sendHeartbeat();
+    }
+    stopHeartbeat() {
+        if (this.heartbeatTimer) {
+            clearInterval(this.heartbeatTimer);
+            this.heartbeatTimer = null;
+        }
+    }
+    startDedupCleanup() {
+        this.stopDedupCleanup();
+        this.dedupCleanupTimer = setInterval(() => {
+            const cutoff = Date.now() - protocol_1.MSGID_DEDUP_TTL_SEC * 1000;
+            for (const [msgId, timestamp] of this.dedupSet) {
+                if (timestamp < cutoff) {
+                    this.dedupSet.delete(msgId);
+                }
+            }
+        }, DEDUP_CLEANUP_INTERVAL_MS);
+        if (this.dedupCleanupTimer.unref) {
+            this.dedupCleanupTimer.unref();
+        }
+    }
+    stopDedupCleanup() {
+        if (this.dedupCleanupTimer) {
+            clearInterval(this.dedupCleanupTimer);
+            this.dedupCleanupTimer = null;
+        }
+    }
+    scheduleReconnect() {
+        if (this.reconnectTimer)
+            return;
+        const delay = this.reconnectDelay;
+        log.info({ delayMs: delay }, 'Scheduling reconnect');
+        this.reconnectTimer = setTimeout(() => {
+            this.reconnectTimer = null;
+            this.connect();
+        }, delay);
+        // Exponential backoff with jitter
+        const jitter = Math.random() * 0.3 * this.reconnectDelay;
+        this.reconnectDelay = Math.min(this.reconnectDelay * 2 + jitter, MAX_RECONNECT_DELAY_MS);
+    }
+    cleanup() {
+        this.connected = false;
+        this.stopHeartbeat();
+        this.stopDedupCleanup();
+        if (this.reconnectTimer) {
+            clearTimeout(this.reconnectTimer);
+            this.reconnectTimer = null;
+        }
+    }
+}
+exports.WorkerClient = WorkerClient;
+//# sourceMappingURL=worker-client.js.map

package/dist/worker-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-client.js","sourceRoot":"","sources":["../src/worker-client.ts"],"names":[],"mappings":";;;;;;AAAA,4CAA2B;AAkB3B,mDAI8B;AAC9B,+CAAsD;AACtD,gDAAwB;AAExB,MAAM,GAAG,GAAG,IAAA,cAAI,EAAC,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;AAE5C,MAAM,sBAAsB,GAAG,MAAM,CAAC;AACtC,MAAM,0BAA0B,GAAG,KAAK,CAAC;AACzC,MAAM,yBAAyB,GAAG,MAAM,CAAC;AAyBzC,MAAa,YAAY;IACf,EAAE,GAAqB,IAAI,CAAC;IAC5B,OAAO,CAAsB;IAC7B,QAAQ,CAAuB;IAC/B,cAAc,GAA0C,IAAI,CAAC;IAC7D,cAAc,GAAyC,IAAI,CAAC;IAC5D,cAAc,GAAG,0BAA0B,CAAC;IAC5C,SAAS,GAAG,KAAK,CAAC;IAClB,gBAAgB,GAAG,KAAK,CAAC;IACzB,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IACrC,iBAAiB,GAA0C,IAAI,CAAC;IAExE,YAAY,OAA4B,EAAE,QAA8B;QACtE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,OAAO;QACL,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC;gBAAC,IAAI,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC;gBAAC,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;YAC/D,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;QAED,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;QAE9B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAChC,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,0BAA0B,CAAC,CAAC;QAErD,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,GAAG,IAAI,YAAS,CAAC,KAAK,EAAE;gBAC7B,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE;iBACrD;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,4BAA4B,CAAC,CAAC;YACjD,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACtB,GAAG,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YAChC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,cAAc,GAAG,0BAA0B,CAAC;YAEjD,mCAAmC;YACnC,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE;gBAC9B,eAAe,EAAE,2BAAgB;aAClC,CAAC,CAAC;YAEH,kBAAkB;YAClB,IAAI,CAAC,cAAc,EAAE,CAAC;YAEtB,sBAAsB;YACtB,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAEzB,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAuB,EAAE,EAAE;YAChD,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YACnC,GAAG,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAClE,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;YAE/B,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBAC3B,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YAC1B,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,iBAAiB,CAAC,CAAC;YACtC,6DAA6D;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU;QACR,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;YAC5C,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IAED,aAAa;QACX,MAAM,OAAO,GAAG,IAAA,mCAAqB,GAAE,CAAC;QACxC,MAAM,KAAK,GAAkB;YAC3B,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS;YACjC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,OAAO,EAAE;gBACP,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,WAAW,EAAE,OAAO,CAAC,WAAW;gBAChC,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE;aAC3C;YACD,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE;SAC7C,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;IAED,OAAO,CAAC,KAAa,EAAE,OAAiC;QACtD,MAAM,KAAK,GAAkB;YAC3B,IAAI,EAAE,KAAK;YACX,KAAK;YACL,OAAO;SACR,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;IAED,YAAY,CAAC,MAAc,EAAE,KAAa,EAAE,MAAc,EAAE,OAAgB;QAC1E,MAAM,KAAK,GAAkB;YAC3B,IAAI,EAAE,UAAU;YAChB,MAAM;YACN,KAAK;YACL,MAAM;YACN,OAAO;SACR,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;IAED,aAAa,CAAC,KAOZ;QACA,MAAM,KAAK,GAAkB;YAC3B,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS;YACjC,KAAK;SACN,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;IAED,qBAAqB;QACnB,MAAM,KAAK,GAAkB;YAC3B,IAAI,EAAE,qBAAqB;SAC5B,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;IAED,uBAAuB,CACrB,MAAc,EACd,UAAkB,EAClB,IAAsB,EACtB,OAAe,EACf,MAAe,EACf,OAA6B;QAE7B,MAAM,KAAK,GAA6B;YACtC,IAAI,EAAE,uBAAuB;YAC7B,MAAM;YACN,UAAU;YACV,IAAI;YACJ,OAAO;YACP,GAAG,CAAC,MAAM,KAAK,SAAS,IAAI,EAAE,MAAM,EAAE,CAAC;YACvC,GAAG,CAAC,OAAO,KAAK,SAAS,IAAI,EAAE,OAAO,EAAE,CAAC;SAC1C,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;IAEO,SAAS,CAAC,KAAoB;QACpC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,YAAS,CAAC,IAAI,EAAE,CAAC;YACtD,GAAG,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE,uCAAuC,CAAC,CAAC;YAC7E,OAAO;QACT,CAAC;QACD,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE,sBAAsB,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,IAAuB;QAC3C,IAAI,KAA0B,CAAC;QAC/B,IAAI,CAAC;YACH,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAwB,CAAC;QAC7D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,gCAAgC,CAAC,CAAC;YACrD,OAAO;QACT,CAAC;QAED,4BAA4B;QAC5B,IAAI,OAAO,IAAI,KAAK,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;YAC1B,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC7B,GAAG,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE,6BAA6B,CAAC,CAAC;gBACtE,OAAO;YACT,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QACvC,CAAC;QAED,GAAG,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE,gBAAgB,CAAC,CAAC;QAElD,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,eAAe;gBAClB,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;gBACpC,MAAM;YACR,KAAK,SAAS;gBACZ,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;gBAC/B,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;gBAC9B,MAAM;YACR,KAAK,mBAAmB;gBACtB,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;gBACxC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;gBAC9B,MAAM;YACR,KAAK,sBAAsB;gBACzB,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBAC1C,MAAM;YACR,KAAK,2BAA2B;gBAC9B,IAAI,CAAC,QAAQ,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;gBAC/C,MAAM;YACR,KAAK,cAAc;gBACjB,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;gBACnC,MAAM;YACR,KAAK,aAAa;gBAChB,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBAClC,MAAM;YACR,KAAK,wBAAwB;gBAC3B,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;gBAC5C,MAAM;YACR;gBACE,GAAG,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,yCAAyC,CAAC,CAAC;gBAChE,GAAG,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,oBAAoB,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;QACvC,IAAI,MAAc,CAAC;QACnB,IAAI,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAChC,MAAM,GAAG,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACpC,CAAC;aAAM,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YACtC,MAAM,GAAG,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,OAAO,GAAG,IAAI,CAAC;QAC1B,CAAC;QACD,wBAAwB;QACxB,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/B,CAAC;QACD,OAAO,GAAG,MAAM,sBAAsB,CAAC;IACzC,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;YACrC,IAAI,CAAC,aAAa,EAAE,CAAC;QACvB,CAAC,EAAE,gCAAqB,CAAC,CAAC;QAC1B,IAAI,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC9B,CAAC;QACD,mCAAmC;QACnC,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAEO,aAAa;QACnB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;YACxC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,8BAAmB,GAAG,IAAI,CAAC;YACvD,KAAK,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC/C,IAAI,SAAS,GAAG,MAAM,EAAE,CAAC;oBACvB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC,EAAE,yBAAyB,CAAC,CAAC;QAC9B,IAAI,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;YACjC,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;QACjC,CAAC;IACH,CAAC;IAEO,gBAAgB;QACtB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACtC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO;QAEhC,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC;QAClC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,sBAAsB,CAAC,CAAC;QAErD,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,kCAAkC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,GAAG,IAAI,CAAC,cAAc,CAAC;QACzD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,GAAG,MAAM,EAAE,sBAAsB,CAAC,CAAC;IAC3F,CAAC;IAEO,OAAO;QACb,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAExB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;CACF;AAxUD,oCAwUC"}

package/dist/workspace-manager.d.ts

@@ -0,0 +1,51 @@
+import type { ConfigSnapshot } from '@overlordai/protocol';
+export interface WorkspaceInitParams {
+    taskId: number;
+    projectKey: string;
+    configSnapshot: ConfigSnapshot;
+    developer: {
+        gitName: string;
+        gitEmail: string;
+    };
+}
+export interface WorkspaceInfo {
+    path: string;
+    branch: string;
+    env: Record<string, string>;
+}
+export declare class WorkspaceManager {
+    private config;
+    private git;
+    private mutex;
+    private lastFetchTimes;
+    constructor(config: {
+        workspaceRoot: string;
+        sshKeyPath?: string;
+    });
+    /**
+     * Initialize a workspace for a task:
+     *  1. Ensure bare clone exists
+     *  2. Fetch (throttled)
+     *  3. Create worktree
+     *  4. Configure git identity
+     *  5. Run setup commands
+     *  6. Return workspace info with PTY env
+     */
+    initialize(params: WorkspaceInitParams): Promise<WorkspaceInfo>;
+    /**
+     * Clean up a workspace:
+     *  1. Remove the git worktree
+     *  2. Remove leftover directory
+     *  3. Prune stale worktree metadata
+     */
+    cleanup(taskId: number, workspacePath: string): Promise<void>;
+    /**
+     * Scan the workspace root and remove worktrees older than 24 hours.
+     */
+    cleanupExpired(activeTaskIds?: Set<number>): Promise<void>;
+    /**
+     * Check whether a bare clone exists for a given project key.
+     */
+    hasBaseRepo(projectKey: string): boolean;
+}
+//# sourceMappingURL=workspace-manager.d.ts.map

package/dist/workspace-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace-manager.d.ts","sourceRoot":"","sources":["../src/workspace-manager.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAY3D,MAAM,WAAW,mBAAmB;IAClC,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,cAAc,CAAC;IAC/B,SAAS,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;CAClD;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC7B;AAID,qBAAa,gBAAgB;IAKf,OAAO,CAAC,MAAM;IAJ1B,OAAO,CAAC,GAAG,CAAgB;IAC3B,OAAO,CAAC,KAAK,CAAe;IAC5B,OAAO,CAAC,cAAc,CAA6B;gBAE/B,MAAM,EAAE;QAAE,aAAa,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE;IAK1E;;;;;;;;OAQG;IACG,UAAU,CAAC,MAAM,EAAE,mBAAmB,GAAG,OAAO,CAAC,aAAa,CAAC;IAqErE;;;;;OAKG;IACG,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA4BnE;;OAEG;IACG,cAAc,CAAC,aAAa,GAAE,GAAG,CAAC,MAAM,CAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IAiD3E;;OAEG;IACH,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO;CAIzC"}

package/dist/workspace-manager.js

@@ -0,0 +1,276 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkspaceManager = void 0;
+const node_child_process_1 = require("node:child_process");
+const fs = __importStar(require("node:fs"));
+const path = __importStar(require("node:path"));
+const pino_1 = __importDefault(require("pino"));
+const git_operations_js_1 = require("./git-operations.js");
+const project_mutex_js_1 = require("./project-mutex.js");
+const safe_env_js_1 = require("./safe-env.js");
+const logger = (0, pino_1.default)({ name: 'workspace-manager' });
+const FETCH_THROTTLE_MS = 60_000; // 60 seconds
+const WORKTREE_MAX_AGE_MS = 24 * 60 * 60 * 1000; // 24 hours
+// ─── WorkspaceManager ────────────────────────────────────────
+class WorkspaceManager {
+    config;
+    git;
+    mutex;
+    lastFetchTimes = new Map();
+    constructor(config) {
+        this.config = config;
+        this.git = new git_operations_js_1.GitOperations(config.sshKeyPath);
+        this.mutex = new project_mutex_js_1.ProjectMutex();
+    }
+    /**
+     * Initialize a workspace for a task:
+     *  1. Ensure bare clone exists
+     *  2. Fetch (throttled)
+     *  3. Create worktree
+     *  4. Configure git identity
+     *  5. Run setup commands
+     *  6. Return workspace info with PTY env
+     */
+    async initialize(params) {
+        const { taskId, projectKey, configSnapshot, developer } = params;
+        const projectDir = path.join(this.config.workspaceRoot, projectKey);
+        const baseDir = path.join(projectDir, '_base');
+        const branch = `feat/overlord-${taskId}`;
+        const worktreePath = path.join(projectDir, `task-${taskId}`);
+        // 1–5. Acquire per-project mutex for bare clone + fetch + worktree add
+        const release = await this.mutex.acquire(projectKey);
+        try {
+            // 1. Clone bare repo if it doesn't exist (inside mutex to avoid race)
+            if (!fs.existsSync(baseDir)) {
+                logger.info({ projectKey, baseDir }, 'Base repo missing — cloning bare');
+                fs.mkdirSync(projectDir, { recursive: true });
+                this.git.cloneBare(configSnapshot.repoUrl, baseDir);
+            }
+            // 3. Fetch with throttle (prune stale entries to bound memory)
+            if (this.lastFetchTimes.size > 500) {
+                const oldest = [...this.lastFetchTimes.entries()]
+                    .sort((a, b) => a[1] - b[1])
+                    .slice(0, Math.floor(this.lastFetchTimes.size / 2));
+                for (const [k] of oldest)
+                    this.lastFetchTimes.delete(k);
+            }
+            const lastFetch = this.lastFetchTimes.get(projectKey) ?? 0;
+            if (Date.now() - lastFetch >= FETCH_THROTTLE_MS) {
+                this.git.fetch(baseDir);
+                this.lastFetchTimes.set(projectKey, Date.now());
+            }
+            else {
+                logger.debug({ projectKey }, 'Skipping fetch — throttled');
+            }
+            // 4. Create worktree
+            const startPoint = `origin/${configSnapshot.defaultBranch}`;
+            this.git.addWorktree(baseDir, worktreePath, branch, startPoint);
+        }
+        finally {
+            // 5. Release mutex
+            release();
+        }
+        // 6. Configure git user in worktree
+        this.git.setLocalConfig(worktreePath, 'user.name', developer.gitName);
+        this.git.setLocalConfig(worktreePath, 'user.email', developer.gitEmail);
+        // 7. Run setup commands (JSON array, sequential)
+        if (configSnapshot.setupCommands) {
+            const commands = JSON.parse(configSnapshot.setupCommands);
+            for (const cmd of commands) {
+                logger.info({ taskId, cmd }, 'Running setup command');
+                try {
+                    (0, node_child_process_1.execFileSync)('sh', ['-c', cmd], {
+                        cwd: worktreePath,
+                        timeout: 120_000,
+                        stdio: 'pipe',
+                        env: (0, safe_env_js_1.buildSafeEnv)(),
+                    });
+                }
+                catch (err) {
+                    logger.error({ taskId, cmd, err }, 'Setup command failed');
+                    throw new Error(`Setup command failed: ${cmd} — ${err.message}`);
+                }
+            }
+        }
+        // 8–9. Build PTY env and return
+        const env = buildPtyEnv(worktreePath, taskId, configSnapshot, developer, this.config.sshKeyPath);
+        return { path: worktreePath, branch, env };
+    }
+    /**
+     * Clean up a workspace:
+     *  1. Remove the git worktree
+     *  2. Remove leftover directory
+     *  3. Prune stale worktree metadata
+     */
+    async cleanup(taskId, workspacePath) {
+        const projectDir = path.dirname(workspacePath);
+        const baseDir = path.join(projectDir, '_base');
+        const projectKey = path.basename(projectDir);
+        const release = await this.mutex.acquire(projectKey);
+        try {
+            // Try git worktree remove first
+            try {
+                this.git.removeWorktree(baseDir, workspacePath);
+            }
+            catch (err) {
+                logger.warn({ taskId, workspacePath, err }, 'git worktree remove failed — will force-remove directory');
+            }
+            // Force-remove directory if it still exists
+            if (fs.existsSync(workspacePath)) {
+                fs.rmSync(workspacePath, { recursive: true, force: true });
+            }
+            // Prune stale worktree refs
+            this.git.pruneWorktrees(baseDir);
+        }
+        finally {
+            release();
+        }
+        logger.info({ taskId, workspacePath }, 'Workspace cleaned up');
+    }
+    /**
+     * Scan the workspace root and remove worktrees older than 24 hours.
+     */
+    async cleanupExpired(activeTaskIds = new Set()) {
+        const root = this.config.workspaceRoot;
+        if (!fs.existsSync(root)) {
+            return;
+        }
+        const projectDirs = fs.readdirSync(root, { withFileTypes: true })
+            .filter((d) => d.isDirectory());
+        for (const projectDirent of projectDirs) {
+            const projectDir = path.join(root, projectDirent.name);
+            const baseDir = path.join(projectDir, '_base');
+            if (!fs.existsSync(baseDir)) {
+                continue;
+            }
+            const entries = fs.readdirSync(projectDir, { withFileTypes: true })
+                .filter((d) => d.isDirectory() && d.name.startsWith('task-'));
+            for (const entry of entries) {
+                const worktreePath = path.join(projectDir, entry.name);
+                try {
+                    // Extract task ID from directory name
+                    const taskIdMatch = entry.name.match(/^task-(\d+)$/);
+                    const taskId = taskIdMatch ? Number(taskIdMatch[1]) : 0;
+                    // Skip directories whose task is still active
+                    if (taskId > 0 && activeTaskIds.has(taskId)) {
+                        logger.debug({ worktreePath, taskId }, 'Skipping cleanup — task is still active');
+                        continue;
+                    }
+                    const stat = fs.statSync(worktreePath);
+                    const ageMs = Date.now() - stat.mtimeMs;
+                    if (ageMs > WORKTREE_MAX_AGE_MS) {
+                        logger.info({ worktreePath, ageMs }, 'Removing expired worktree');
+                        await this.cleanup(taskId, worktreePath);
+                    }
+                }
+                catch (err) {
+                    logger.error({ worktreePath, err }, 'Error checking/cleaning expired worktree');
+                }
+            }
+        }
+    }
+    /**
+     * Check whether a bare clone exists for a given project key.
+     */
+    hasBaseRepo(projectKey) {
+        const baseDir = path.join(this.config.workspaceRoot, projectKey, '_base');
+        return fs.existsSync(baseDir);
+    }
+}
+exports.WorkspaceManager = WorkspaceManager;
+// ─── PTY Environment Builder ─────────────────────────────────
+function buildPtyEnv(workspacePath, taskId, config, developer, sshKeyPath) {
+    const env = {
+        HOME: workspacePath,
+        PATH: process.env.PATH ?? '/usr/local/bin:/usr/bin:/bin',
+        TERM: 'xterm-256color',
+        COLORTERM: 'truecolor',
+        GIT_AUTHOR_NAME: developer.gitName,
+        GIT_AUTHOR_EMAIL: developer.gitEmail,
+        GIT_COMMITTER_NAME: developer.gitName,
+        GIT_COMMITTER_EMAIL: developer.gitEmail,
+        OVERLORD_TASK_ID: String(taskId),
+    };
+    if (sshKeyPath) {
+        // Shell-safe single-quote escaping: replace ' with '\'' (end quote, escaped quote, start quote)
+        const escapedPath = sshKeyPath.replace(/'/g, "'\\''");
+        env.GIT_SSH_COMMAND = `ssh -i '${escapedPath}' -o IdentitiesOnly=yes`;
+    }
+    if (config.agentEnv) {
+        // Allowlist approach: only permit keys that look like safe user-defined env vars.
+        // This prevents injection via dangerous vars like BASH_ENV, LD_AUDIT,
+        // PYTHONSTARTUP, RUBYOPT, NODE_EXTRA_CA_CERTS, etc.
+        const SAFE_KEY_PATTERN = /^[A-Z][A-Z0-9_]*$/;
+        const RESERVED_PREFIXES = ['LD_', 'DYLD_', 'BASH_', 'GIT_', 'OVERLORD_', 'NODE_', 'PYTHON', 'RUBY', 'PERL'];
+        const RESERVED_EXACT = new Set([
+            'PATH', 'HOME', 'SHELL', 'USER', 'LOGNAME', 'TERM', 'COLORTERM',
+            'ENV', 'IFS', 'CDPATH', 'GLOBIGNORE', 'SHELLOPTS', 'BASHOPTS',
+            'PROMPT_COMMAND', 'MAIL', 'MAILPATH', 'MAILCHECK',
+            'HOSTALIASES', 'LOCALDOMAIN', 'RES_OPTIONS',
+            'LD_PRELOAD', 'LD_LIBRARY_PATH', 'LD_AUDIT',
+        ]);
+        const isAllowed = (key) => {
+            if (!SAFE_KEY_PATTERN.test(key))
+                return false;
+            if (RESERVED_EXACT.has(key))
+                return false;
+            for (const prefix of RESERVED_PREFIXES) {
+                if (key.startsWith(prefix))
+                    return false;
+            }
+            return true;
+        };
+        try {
+            const parsed = JSON.parse(config.agentEnv);
+            for (const [key, value] of Object.entries(parsed)) {
+                if (isAllowed(key) && typeof value === 'string') {
+                    env[key] = value;
+                }
+                else {
+                    logger.warn({ taskId, key }, 'agentEnv key blocked by allowlist');
+                }
+            }
+        }
+        catch (err) {
+            logger.warn({ taskId, err }, 'Failed to parse agentEnv JSON, using empty object');
+        }
+    }
+    return env;
+}
+//# sourceMappingURL=workspace-manager.js.map

package/dist/workspace-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace-manager.js","sourceRoot":"","sources":["../src/workspace-manager.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2DAAkD;AAClD,4CAA8B;AAC9B,gDAAkC;AAClC,gDAAwB;AAExB,2DAAoD;AACpD,yDAAkD;AAClD,+CAA6C;AAE7C,MAAM,MAAM,GAAG,IAAA,cAAI,EAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;AAEnD,MAAM,iBAAiB,GAAG,MAAM,CAAC,CAAC,aAAa;AAC/C,MAAM,mBAAmB,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,WAAW;AAiB5D,gEAAgE;AAEhE,MAAa,gBAAgB;IAKP;IAJZ,GAAG,CAAgB;IACnB,KAAK,CAAe;IACpB,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEnD,YAAoB,MAAsD;QAAtD,WAAM,GAAN,MAAM,CAAgD;QACxE,IAAI,CAAC,GAAG,GAAG,IAAI,iCAAa,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAChD,IAAI,CAAC,KAAK,GAAG,IAAI,+BAAY,EAAE,CAAC;IAClC,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,UAAU,CAAC,MAA2B;QAC1C,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC;QACjE,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;QACpE,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC/C,MAAM,MAAM,GAAG,iBAAiB,MAAM,EAAE,CAAC;QACzC,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,MAAM,EAAE,CAAC,CAAC;QAE7D,uEAAuE;QACvE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACrD,IAAI,CAAC;YACH,sEAAsE;YACtE,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC5B,MAAM,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE,kCAAkC,CAAC,CAAC;gBACzE,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC9C,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,cAAc,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACtD,CAAC;YAED,+DAA+D;YAC/D,IAAI,IAAI,CAAC,cAAc,CAAC,IAAI,GAAG,GAAG,EAAE,CAAC;gBACnC,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;qBAC9C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;qBAC3B,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;gBACtD,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,MAAM;oBAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC1D,CAAC;YACD,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAC3D,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,IAAI,iBAAiB,EAAE,CAAC;gBAChD,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBACxB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;YAClD,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,EAAE,4BAA4B,CAAC,CAAC;YAC7D,CAAC;YAED,qBAAqB;YACrB,MAAM,UAAU,GAAG,UAAU,cAAc,CAAC,aAAa,EAAE,CAAC;YAC5D,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QAClE,CAAC;gBAAS,CAAC;YACT,mBAAmB;YACnB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,oCAAoC;QACpC,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,WAAW,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC;QACtE,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,YAAY,EAAE,SAAS,CAAC,QAAQ,CAAC,CAAC;QAExE,iDAAiD;QACjD,IAAI,cAAc,CAAC,aAAa,EAAE,CAAC;YACjC,MAAM,QAAQ,GAAa,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YACpE,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,uBAAuB,CAAC,CAAC;gBACtD,IAAI,CAAC;oBACH,IAAA,iCAAY,EAAC,IAAI,EAAE,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE;wBAC9B,GAAG,EAAE,YAAY;wBACjB,OAAO,EAAE,OAAO;wBAChB,KAAK,EAAE,MAAM;wBACb,GAAG,EAAE,IAAA,0BAAY,GAAE;qBACpB,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,sBAAsB,CAAC,CAAC;oBAC3D,MAAM,IAAI,KAAK,CAAC,yBAAyB,GAAG,MAAO,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC9E,CAAC;YACH,CAAC;QACH,CAAC;QAED,gCAAgC;QAChC,MAAM,GAAG,GAAG,WAAW,CAAC,YAAY,EAAE,MAAM,EAAE,cAAc,EAAE,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAEjG,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;IAC7C,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,aAAqB;QACjD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC/C,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAE7C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACrD,IAAI,CAAC;YACH,gCAAgC;YAChC,IAAI,CAAC;gBACH,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YAClD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,EAAE,EAAE,0DAA0D,CAAC,CAAC;YAC1G,CAAC;YAED,4CAA4C;YAC5C,IAAI,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;gBACjC,EAAE,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAC7D,CAAC;YAED,4BAA4B;YAC5B,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACnC,CAAC;gBAAS,CAAC;YACT,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,EAAE,sBAAsB,CAAC,CAAC;IACjE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,gBAA6B,IAAI,GAAG,EAAE;QACzD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;QAEvC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,OAAO;QACT,CAAC;QAED,MAAM,WAAW,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;aAC9D,MAAM,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;QAE7C,KAAK,MAAM,aAAa,IAAI,WAAW,EAAE,CAAC;YACxC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;YACvD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YAE/C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YAED,MAAM,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;iBAChE,MAAM,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC;YAE3E,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEvD,IAAI,CAAC;oBACH,sCAAsC;oBACtC,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;oBACrD,MAAM,MAAM,GAAG,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;oBAExD,8CAA8C;oBAC9C,IAAI,MAAM,GAAG,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5C,MAAM,CAAC,KAAK,CAAC,EAAE,YAAY,EAAE,MAAM,EAAE,EAAE,yCAAyC,CAAC,CAAC;wBAClF,SAAS;oBACX,CAAC;oBAED,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;oBACvC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;oBAExC,IAAI,KAAK,GAAG,mBAAmB,EAAE,CAAC;wBAChC,MAAM,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,2BAA2B,CAAC,CAAC;wBAClE,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;oBAC3C,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,KAAK,CAAC,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,0CAA0C,CAAC,CAAC;gBAClF,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,UAAkB;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAC1E,OAAO,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;CACF;AArLD,4CAqLC;AAED,gEAAgE;AAEhE,SAAS,WAAW,CAClB,aAAqB,EACrB,MAAc,EACd,MAAsB,EACtB,SAAgD,EAChD,UAAmB;IAEnB,MAAM,GAAG,GAA2B;QAClC,IAAI,EAAE,aAAa;QACnB,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,8BAA8B;QACxD,IAAI,EAAE,gBAAgB;QACtB,SAAS,EAAE,WAAW;QACtB,eAAe,EAAE,SAAS,CAAC,OAAO;QAClC,gBAAgB,EAAE,SAAS,CAAC,QAAQ;QACpC,kBAAkB,EAAE,SAAS,CAAC,OAAO;QACrC,mBAAmB,EAAE,SAAS,CAAC,QAAQ;QACvC,gBAAgB,EAAE,MAAM,CAAC,MAAM,CAAC;KACjC,CAAC;IAEF,IAAI,UAAU,EAAE,CAAC;QACf,gGAAgG;QAChG,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACtD,GAAG,CAAC,eAAe,GAAG,WAAW,WAAW,yBAAyB,CAAC;IACxE,CAAC;IAED,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpB,kFAAkF;QAClF,sEAAsE;QACtE,oDAAoD;QACpD,MAAM,gBAAgB,GAAG,mBAAmB,CAAC;QAC7C,MAAM,iBAAiB,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAC5G,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC;YAC7B,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW;YAC/D,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,WAAW,EAAE,UAAU;YAC7D,gBAAgB,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW;YACjD,aAAa,EAAE,aAAa,EAAE,aAAa;YAC3C,YAAY,EAAE,iBAAiB,EAAE,UAAU;SAC5C,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,CAAC,GAAW,EAAW,EAAE;YACzC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC9C,IAAI,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC1C,KAAK,MAAM,MAAM,IAAI,iBAAiB,EAAE,CAAC;gBACvC,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC;oBAAE,OAAO,KAAK,CAAC;YAC3C,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAA2B,CAAC;YACrE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAClD,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBAChD,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;gBACnB,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,mCAAmC,CAAC,CAAC;gBACpE,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,mDAAmD,CAAC,CAAC;QACpF,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC"}