@overlordai/worker 1.0.0

package/dist/task-handler.d.ts

@@ -0,0 +1,56 @@
+import type { ExecuteTaskFrame, CancelTaskFrame, CleanupWorkspaceFrame, ResumeTaskFrame, ConfigSnapshot } from '@overlordai/protocol';
+import { WorkerConfig } from './config.js';
+import type { WorkspaceManager, WorkspaceInfo } from './workspace-manager.js';
+import type { PtyManager, PtySession } from './pty-manager.js';
+import { PipelineRunner, type LeaseChecker } from './pipeline-runner.js';
+import type { StageConfirmMode, StageChoiceOption } from '@overlordai/protocol';
+export interface TaskContext {
+    taskId: number;
+    description: string;
+    projectKey: string;
+    stage: string;
+    workspaceInfo: WorkspaceInfo | null;
+    ptySession: PtySession | null;
+    pipelineRunner: PipelineRunner | null;
+    channelToken: string;
+    configSnapshot: ConfigSnapshot;
+    abortController: AbortController;
+}
+type ProgressCallback = (taskId: number, stage: string, status: string, message?: string) => void;
+type AckCallback = (msgId: string, payload?: Record<string, unknown>) => void;
+type ConfirmRequestCallback = (taskId: number, stageIndex: number, mode: StageConfirmMode, timeout: number, prompt?: string, choices?: StageChoiceOption[]) => void;
+export declare class TaskHandler {
+    private activeTasks;
+    private config;
+    private onProgress;
+    private onAck;
+    private onConfirmRequest;
+    private workspaceManager;
+    private ptyManager;
+    private leaseChecker;
+    constructor(config: WorkerConfig, onProgress: ProgressCallback, onAck: AckCallback, onConfirmRequest: ConfirmRequestCallback, workspaceManager: WorkspaceManager, ptyManager: PtyManager, leaseChecker: LeaseChecker);
+    getActiveTaskCount(): number;
+    getActiveTaskIds(): number[];
+    /**
+     * Wait until all active tasks have finished.
+     * Resolves immediately if no tasks are active.
+     */
+    waitForActiveTasks(): Promise<void>;
+    getActiveTaskStatuses(): Array<{
+        taskId: number;
+        localStatus: 'still_running' | 'completed_locally' | 'failed_locally';
+        currentStage?: string;
+    }>;
+    handleExecute(frame: ExecuteTaskFrame): Promise<void>;
+    handleCancel(frame: CancelTaskFrame): Promise<void>;
+    handleCleanupWorkspace(frame: CleanupWorkspaceFrame): Promise<void>;
+    handleStageConfirmResponse(taskId: number, stageIndex: number, result: string): void;
+    handleResume(frame: ResumeTaskFrame): Promise<void>;
+    /**
+     * Run the pipeline for a task using PipelineRunner.
+     * Creates a PipelineRunner instance, wires up events, and waits for completion.
+     */
+    private runPipeline;
+}
+export {};
+//# sourceMappingURL=task-handler.d.ts.map

package/dist/task-handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-handler.d.ts","sourceRoot":"","sources":["../src/task-handler.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,gBAAgB,EAChB,eAAe,EACf,qBAAqB,EACrB,eAAe,EACf,cAAc,EACf,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAC9E,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,KAAK,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACzE,OAAO,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAIhF,MAAM,WAAW,WAAW;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,EAAE,aAAa,GAAG,IAAI,CAAC;IACpC,UAAU,EAAE,UAAU,GAAG,IAAI,CAAC;IAC9B,cAAc,EAAE,cAAc,GAAG,IAAI,CAAC;IACtC,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,cAAc,CAAC;IAC/B,eAAe,EAAE,eAAe,CAAC;CAClC;AAED,KAAK,gBAAgB,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;AAClG,KAAK,WAAW,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;AAC9E,KAAK,sBAAsB,GAAG,CAC5B,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,MAAM,EAClB,IAAI,EAAE,gBAAgB,EACtB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,iBAAiB,EAAE,KAC1B,IAAI,CAAC;AAEV,qBAAa,WAAW;IACtB,OAAO,CAAC,WAAW,CAAkC;IACrD,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,UAAU,CAAmB;IACrC,OAAO,CAAC,KAAK,CAAc;IAC3B,OAAO,CAAC,gBAAgB,CAAyB;IACjD,OAAO,CAAC,gBAAgB,CAAmB;IAC3C,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,YAAY,CAAe;gBAGjC,MAAM,EAAE,YAAY,EACpB,UAAU,EAAE,gBAAgB,EAC5B,KAAK,EAAE,WAAW,EAClB,gBAAgB,EAAE,sBAAsB,EACxC,gBAAgB,EAAE,gBAAgB,EAClC,UAAU,EAAE,UAAU,EACtB,YAAY,EAAE,YAAY;IAW5B,kBAAkB,IAAI,MAAM;IAI5B,gBAAgB,IAAI,MAAM,EAAE;IAI5B;;;OAGG;IACH,kBAAkB,IAAI,OAAO,CAAC,IAAI,CAAC;IAgBnC,qBAAqB,IAAI,KAAK,CAAC;QAC7B,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,eAAe,GAAG,mBAAmB,GAAG,gBAAgB,CAAC;QACtE,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,CAAC;IAgBI,aAAa,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAqFrD,YAAY,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;IAoBnD,sBAAsB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;IAuBzE,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAgB9E,YAAY,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBzD;;;OAGG;IACH,OAAO,CAAC,WAAW;CAoEpB"}

package/dist/task-handler.js

@@ -0,0 +1,296 @@
+"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.TaskHandler = void 0;
+const fs = __importStar(require("node:fs"));
+const pino_1 = __importDefault(require("pino"));
+const pipeline_runner_js_1 = require("./pipeline-runner.js");
+const log = (0, pino_1.default)({ name: 'task-handler' });
+class TaskHandler {
+    activeTasks = new Map();
+    config;
+    onProgress;
+    onAck;
+    onConfirmRequest;
+    workspaceManager;
+    ptyManager;
+    leaseChecker;
+    constructor(config, onProgress, onAck, onConfirmRequest, workspaceManager, ptyManager, leaseChecker) {
+        this.config = config;
+        this.onProgress = onProgress;
+        this.onAck = onAck;
+        this.onConfirmRequest = onConfirmRequest;
+        this.workspaceManager = workspaceManager;
+        this.ptyManager = ptyManager;
+        this.leaseChecker = leaseChecker;
+    }
+    getActiveTaskCount() {
+        return this.activeTasks.size;
+    }
+    getActiveTaskIds() {
+        return Array.from(this.activeTasks.keys());
+    }
+    /**
+     * Wait until all active tasks have finished.
+     * Resolves immediately if no tasks are active.
+     */
+    waitForActiveTasks() {
+        if (this.activeTasks.size === 0) {
+            return Promise.resolve();
+        }
+        return new Promise((resolve) => {
+            const check = setInterval(() => {
+                if (this.activeTasks.size === 0) {
+                    clearInterval(check);
+                    resolve();
+                }
+            }, 500);
+            if (check.unref)
+                check.unref();
+        });
+    }
+    getActiveTaskStatuses() {
+        const statuses = [];
+        for (const [taskId, ctx] of this.activeTasks) {
+            statuses.push({
+                taskId,
+                localStatus: ctx.ptySession ? 'still_running' : 'completed_locally',
+                currentStage: ctx.stage,
+            });
+        }
+        return statuses;
+    }
+    async handleExecute(frame) {
+        const { taskId, description, projectKey, configSnapshot, developer, channelToken, msgId } = frame;
+        if (this.activeTasks.has(taskId)) {
+            log.warn({ taskId }, 'Task already active, ignoring duplicate execute');
+            this.onAck(msgId, { status: 'already_running' });
+            return;
+        }
+        log.info({ taskId, agentType: configSnapshot.agentType }, 'Executing task');
+        const abortController = new AbortController();
+        const ctx = {
+            taskId,
+            description,
+            projectKey,
+            stage: 'init',
+            workspaceInfo: null,
+            ptySession: null,
+            pipelineRunner: null,
+            channelToken,
+            configSnapshot,
+            abortController,
+        };
+        this.activeTasks.set(taskId, ctx);
+        // Ack the execute message
+        this.onAck(msgId, { status: 'accepted' });
+        this.onProgress(taskId, 'init', 'running', 'Initializing workspace');
+        try {
+            // 1. Initialize workspace via WorkspaceManager (bare clone, worktree, git config, setup commands)
+            const workspaceInfo = await this.workspaceManager.initialize({
+                taskId,
+                projectKey,
+                configSnapshot,
+                developer,
+            });
+            ctx.workspaceInfo = workspaceInfo;
+            if (abortController.signal.aborted) {
+                throw new Error('Task cancelled');
+            }
+            // 2. Create PTY session via PtyManager
+            ctx.stage = 'agent';
+            this.onProgress(taskId, 'agent', 'running', 'Starting agent');
+            const ptySession = this.ptyManager.createSession({
+                taskId,
+                workspacePath: workspaceInfo.path,
+                env: workspaceInfo.env,
+                configSnapshot,
+                developer,
+            });
+            ctx.ptySession = ptySession;
+            // 3. Run pipeline via PipelineRunner
+            await this.runPipeline(ctx, ptySession, description);
+        }
+        catch (err) {
+            if (abortController.signal.aborted) {
+                log.info({ taskId }, 'Task was cancelled');
+                this.onProgress(taskId, ctx.stage, 'cancelled', 'Task cancelled');
+            }
+            else {
+                const message = err instanceof Error ? err.message : String(err);
+                log.error({ taskId, err }, 'Task failed');
+                this.onProgress(taskId, ctx.stage, 'failed', message);
+            }
+        }
+        finally {
+            // Cleanup PTY session
+            if (ctx.ptySession) {
+                await this.ptyManager.closeSession(taskId);
+                ctx.ptySession = null;
+            }
+            // Destroy pipeline runner
+            if (ctx.pipelineRunner) {
+                ctx.pipelineRunner.destroy();
+                ctx.pipelineRunner = null;
+            }
+            this.activeTasks.delete(taskId);
+        }
+    }
+    async handleCancel(frame) {
+        const { taskId, msgId } = frame;
+        const ctx = this.activeTasks.get(taskId);
+        if (!ctx) {
+            log.warn({ taskId }, 'Cancel requested for unknown task');
+            this.onAck(msgId, { status: 'not_found' });
+            return;
+        }
+        log.info({ taskId }, 'Cancelling task');
+        this.onAck(msgId, { status: 'cancelling' });
+        // Signal the abort controller
+        ctx.abortController.abort();
+        // Cancel PTY session (SIGTERM -> grace -> SIGKILL)
+        await this.ptyManager.cancel(taskId);
+    }
+    async handleCleanupWorkspace(frame) {
+        const { taskId, workspacePath, msgId } = frame;
+        log.info({ taskId, workspacePath }, 'Cleaning up workspace');
+        try {
+            await this.workspaceManager.cleanup(taskId, workspacePath);
+            this.onAck(msgId, { status: 'cleaned' });
+        }
+        catch (err) {
+            // Fallback: try direct removal if worktree cleanup fails
+            log.warn({ taskId, workspacePath, err }, 'WorkspaceManager cleanup failed, trying direct removal');
+            try {
+                if (fs.existsSync(workspacePath)) {
+                    fs.rmSync(workspacePath, { recursive: true, force: true });
+                    log.info({ taskId, workspacePath }, 'Workspace force-removed');
+                }
+                this.onAck(msgId, { status: 'cleaned' });
+            }
+            catch (rmErr) {
+                log.error({ taskId, workspacePath, err: rmErr }, 'Failed to cleanup workspace');
+                this.onAck(msgId, { status: 'error', error: rmErr instanceof Error ? rmErr.message : String(rmErr) });
+            }
+        }
+    }
+    handleStageConfirmResponse(taskId, stageIndex, result) {
+        const ctx = this.activeTasks.get(taskId);
+        if (!ctx) {
+            log.warn({ taskId }, 'Stage confirm response for unknown task');
+            return;
+        }
+        if (!ctx.pipelineRunner) {
+            log.warn({ taskId }, 'Stage confirm response but no pipeline runner');
+            return;
+        }
+        ctx.pipelineRunner.confirmStage(taskId, stageIndex, result);
+    }
+    async handleResume(frame) {
+        const { taskId, msgId } = frame;
+        const ctx = this.activeTasks.get(taskId);
+        if (!ctx) {
+            log.warn({ taskId }, 'Resume requested for unknown task');
+            this.onAck(msgId, { status: 'not_found' });
+            return;
+        }
+        log.info({ taskId }, 'Resuming task');
+        this.onAck(msgId, { status: 'resuming' });
+        // Resume the pipeline runner if it exists
+        if (ctx.pipelineRunner) {
+            ctx.pipelineRunner.resume();
+        }
+        this.onProgress(taskId, ctx.stage, 'running', 'Task resumed');
+    }
+    /**
+     * Run the pipeline for a task using PipelineRunner.
+     * Creates a PipelineRunner instance, wires up events, and waits for completion.
+     */
+    runPipeline(ctx, ptySession, taskDescription) {
+        return new Promise((resolve, reject) => {
+            const { taskId, configSnapshot, abortController } = ctx;
+            // Adapt PtySession to the minimal interface PipelineRunner expects
+            const pipelinePtySession = {
+                write: (data) => ptySession.pty.write(data),
+                getCleanOutput: () => ptySession.cleanOutput,
+                resetCleanOutput: () => ptySession.resetCleanOutput(),
+                onData: (handler) => ptySession.onData(handler),
+                onExit: (handler) => ptySession.onExit(handler),
+            };
+            // Use the executor as a prompt detector
+            const promptDetector = {
+                detectPrompt: (output) => ptySession.executor.detectPrompt(output),
+            };
+            const runner = new pipeline_runner_js_1.PipelineRunner(taskId, ptySession.sessionId, pipelinePtySession, this.leaseChecker, promptDetector);
+            ctx.pipelineRunner = runner;
+            // Wire up pipeline events
+            runner.onStageChanged((tid, stageName) => {
+                ctx.stage = stageName;
+                this.onProgress(tid, stageName, 'running', `Stage: ${stageName}`);
+            });
+            runner.onCompleted((tid) => {
+                ctx.stage = 'completed';
+                this.onProgress(tid, 'completed', 'completed', 'Task completed successfully');
+                resolve();
+            });
+            runner.onFailed((tid, error) => {
+                if (abortController.signal.aborted) {
+                    reject(new Error('Task cancelled'));
+                }
+                else {
+                    reject(new Error(error));
+                }
+            });
+            runner.onConfirmRequired((tid, _stageName, mode, prompt, choices, timeout) => {
+                const stageIndex = runner.getState().currentStageIndex;
+                this.onConfirmRequest(tid, stageIndex, mode, timeout ?? 0, prompt, choices);
+            });
+            // Listen for abort
+            const onAbort = () => {
+                runner.destroy();
+            };
+            abortController.signal.addEventListener('abort', onAbort, { once: true });
+            // Start the pipeline
+            runner.start(configSnapshot.pipeline, taskDescription).catch((err) => {
+                reject(err);
+            });
+        });
+    }
+}
+exports.TaskHandler = TaskHandler;
+//# sourceMappingURL=task-handler.js.map

package/dist/task-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-handler.js","sourceRoot":"","sources":["../src/task-handler.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAA8B;AAQ9B,gDAAwB;AAIxB,6DAAyE;AAGzE,MAAM,GAAG,GAAG,IAAA,cAAI,EAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;AA0B3C,MAAa,WAAW;IACd,WAAW,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC7C,MAAM,CAAe;IACrB,UAAU,CAAmB;IAC7B,KAAK,CAAc;IACnB,gBAAgB,CAAyB;IACzC,gBAAgB,CAAmB;IACnC,UAAU,CAAa;IACvB,YAAY,CAAe;IAEnC,YACE,MAAoB,EACpB,UAA4B,EAC5B,KAAkB,EAClB,gBAAwC,EACxC,gBAAkC,EAClC,UAAsB,EACtB,YAA0B;QAE1B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IAED,kBAAkB;QAChB,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;IAC/B,CAAC;IAED,gBAAgB;QACd,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED;;;OAGG;IACH,kBAAkB;QAChB,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAChC,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC;QAED,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YACnC,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;gBAC7B,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;oBAChC,aAAa,CAAC,KAAK,CAAC,CAAC;oBACrB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,EAAE,GAAG,CAAC,CAAC;YACR,IAAI,KAAK,CAAC,KAAK;gBAAE,KAAK,CAAC,KAAK,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,qBAAqB;QAKnB,MAAM,QAAQ,GAIT,EAAE,CAAC;QACR,KAAK,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAC7C,QAAQ,CAAC,IAAI,CAAC;gBACZ,MAAM;gBACN,WAAW,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,mBAAmB;gBACnE,YAAY,EAAE,GAAG,CAAC,KAAK;aACxB,CAAC,CAAC;QACL,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,KAAuB;QACzC,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,cAAc,EAAE,SAAS,EAAE,YAAY,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;QAElG,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACjC,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,iDAAiD,CAAC,CAAC;YACxE,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC;YACjD,OAAO;QACT,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,CAAC,SAAS,EAAE,EAAE,gBAAgB,CAAC,CAAC;QAE5E,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAE9C,MAAM,GAAG,GAAgB;YACvB,MAAM;YACN,WAAW;YACX,UAAU;YACV,KAAK,EAAE,MAAM;YACb,aAAa,EAAE,IAAI;YACnB,UAAU,EAAE,IAAI;YAChB,cAAc,EAAE,IAAI;YACpB,YAAY;YACZ,cAAc;YACd,eAAe;SAChB,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAElC,0BAA0B;QAC1B,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,wBAAwB,CAAC,CAAC;QAErE,IAAI,CAAC;YACH,kGAAkG;YAClG,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC;gBAC3D,MAAM;gBACN,UAAU;gBACV,cAAc;gBACd,SAAS;aACV,CAAC,CAAC;YACH,GAAG,CAAC,aAAa,GAAG,aAAa,CAAC;YAElC,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnC,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;YACpC,CAAC;YAED,uCAAuC;YACvC,GAAG,CAAC,KAAK,GAAG,OAAO,CAAC;YACpB,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,gBAAgB,CAAC,CAAC;YAE9D,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC;gBAC/C,MAAM;gBACN,aAAa,EAAE,aAAa,CAAC,IAAI;gBACjC,GAAG,EAAE,aAAa,CAAC,GAAG;gBACtB,cAAc;gBACd,SAAS;aACV,CAAC,CAAC;YACH,GAAG,CAAC,UAAU,GAAG,UAAU,CAAC;YAE5B,qCAAqC;YACrC,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;QAEvD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnC,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,oBAAoB,CAAC,CAAC;gBAC3C,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,EAAE,WAAW,EAAE,gBAAgB,CAAC,CAAC;YACpE,CAAC;iBAAM,CAAC;gBACN,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACjE,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,aAAa,CAAC,CAAC;gBAC1C,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,sBAAsB;YACtB,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;gBACnB,MAAM,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;gBAC3C,GAAG,CAAC,UAAU,GAAG,IAAI,CAAC;YACxB,CAAC;YACD,0BAA0B;YAC1B,IAAI,GAAG,CAAC,cAAc,EAAE,CAAC;gBACvB,GAAG,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;gBAC7B,GAAG,CAAC,cAAc,GAAG,IAAI,CAAC;YAC5B,CAAC;YACD,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAsB;QACvC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;QAChC,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAEzC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,mCAAmC,CAAC,CAAC;YAC1D,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;YAC3C,OAAO;QACT,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;QAE5C,8BAA8B;QAC9B,GAAG,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAE5B,mDAAmD;QACnD,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,KAA4B;QACvD,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;QAC/C,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,EAAE,uBAAuB,CAAC,CAAC;QAE7D,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YAC3D,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QAC3C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,yDAAyD;YACzD,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,EAAE,EAAE,wDAAwD,CAAC,CAAC;YACnG,IAAI,CAAC;gBACH,IAAI,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;oBACjC,EAAE,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;oBAC3D,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,EAAE,yBAAyB,CAAC,CAAC;gBACjE,CAAC;gBACD,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;YAC3C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,6BAA6B,CAAC,CAAC;gBAChF,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACxG,CAAC;QACH,CAAC;IACH,CAAC;IAED,0BAA0B,CAAC,MAAc,EAAE,UAAkB,EAAE,MAAc;QAC3E,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAEzC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,yCAAyC,CAAC,CAAC;YAChE,OAAO;QACT,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;YACxB,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,+CAA+C,CAAC,CAAC;YACtE,OAAO;QACT,CAAC;QAED,GAAG,CAAC,cAAc,CAAC,YAAY,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAsB;QACvC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;QAChC,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAEzC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,mCAAmC,CAAC,CAAC;YAC1D,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;YAC3C,OAAO;QACT,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,eAAe,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;QAE1C,0CAA0C;QAC1C,IAAI,GAAG,CAAC,cAAc,EAAE,CAAC;YACvB,GAAG,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC;QAC9B,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;IAChE,CAAC;IAED;;;OAGG;IACK,WAAW,CACjB,GAAgB,EAChB,UAAsB,EACtB,eAAuB;QAEvB,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3C,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,eAAe,EAAE,GAAG,GAAG,CAAC;YAExD,mEAAmE;YACnE,MAAM,kBAAkB,GAAG;gBACzB,KAAK,EAAE,CAAC,IAAY,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;gBACnD,cAAc,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,WAAW;gBAC5C,gBAAgB,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,gBAAgB,EAAE;gBACrD,MAAM,EAAE,CAAC,OAA+B,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC;gBACvE,MAAM,EAAE,CAAC,OAAmC,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC;aAC5E,CAAC;YAEF,wCAAwC;YACxC,MAAM,cAAc,GAAG;gBACrB,YAAY,EAAE,CAAC,MAAc,EAAE,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,YAAY,CAAC,MAAM,CAAC;aAC3E,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,mCAAc,CAC/B,MAAM,EACN,UAAU,CAAC,SAAS,EACpB,kBAAkB,EAClB,IAAI,CAAC,YAAY,EACjB,cAAc,CACf,CAAC;YACF,GAAG,CAAC,cAAc,GAAG,MAAM,CAAC;YAE5B,0BAA0B;YAC1B,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;gBACvC,GAAG,CAAC,KAAK,GAAG,SAAS,CAAC;gBACtB,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU,SAAS,EAAE,CAAC,CAAC;YACpE,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE;gBACzB,GAAG,CAAC,KAAK,GAAG,WAAW,CAAC;gBACxB,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,WAAW,EAAE,WAAW,EAAE,6BAA6B,CAAC,CAAC;gBAC9E,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;gBAC7B,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnC,MAAM,CAAC,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC;gBACtC,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE;gBAC3E,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC,iBAAiB,CAAC;gBACvD,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,IAAI,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAC9E,CAAC,CAAC,CAAC;YAEH,mBAAmB;YACnB,MAAM,OAAO,GAAG,GAAG,EAAE;gBACnB,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,CAAC,CAAC;YACF,eAAe,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAE1E,qBAAqB;YACrB,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACnE,MAAM,CAAC,GAAG,CAAC,CAAC;YACd,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAzTD,kCAyTC"}

package/dist/tunnel-manager.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Manages Cursor remote tunnel processes for active tasks.
+ *
+ * TODO: Full implementation should:
+ * - Parse the tunnel URL from the `cursor --remote-tunnel` stdout
+ * - Handle tunnel process crashes and automatic restarts
+ * - Enforce tunnel TTL / expiry
+ * - Allocate and track ports
+ */
+export declare class TunnelManager {
+    private tunnels;
+    /**
+     * Start a Cursor remote tunnel for the given task workspace.
+     * Returns the tunnel URL on success.
+     */
+    start(taskId: number, workspacePath: string): Promise<string>;
+    /**
+     * Stop the tunnel for the given task.
+     */
+    stop(taskId: number): Promise<void>;
+    /**
+     * Stop all active tunnels. Called during worker shutdown.
+     */
+    stopAll(): void;
+    /**
+     * Check whether a tunnel is active for the given task.
+     */
+    hasTunnel(taskId: number): boolean;
+    /**
+     * Get the tunnel URL for a task, or null if not available.
+     */
+    getTunnelUrl(taskId: number): string | null;
+}
+//# sourceMappingURL=tunnel-manager.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tunnel-manager.d.ts","sourceRoot":"","sources":["../src/tunnel-manager.ts"],"names":[],"mappings":"AAaA;;;;;;;;GAQG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,OAAO,CAAkC;IAEjD;;;OAGG;IACG,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAmFnE;;OAEG;IACG,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAsCzC;;OAEG;IACH,OAAO,IAAI,IAAI;IAYf;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAIlC;;OAEG;IACH,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;CAG5C"}

package/dist/tunnel-manager.js

@@ -0,0 +1,165 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TunnelManager = void 0;
+const node_child_process_1 = require("node:child_process");
+const pino_1 = __importDefault(require("pino"));
+const safe_env_js_1 = require("./safe-env.js");
+const log = (0, pino_1.default)({ name: 'tunnel-manager' });
+/**
+ * Manages Cursor remote tunnel processes for active tasks.
+ *
+ * TODO: Full implementation should:
+ * - Parse the tunnel URL from the `cursor --remote-tunnel` stdout
+ * - Handle tunnel process crashes and automatic restarts
+ * - Enforce tunnel TTL / expiry
+ * - Allocate and track ports
+ */
+class TunnelManager {
+    tunnels = new Map();
+    /**
+     * Start a Cursor remote tunnel for the given task workspace.
+     * Returns the tunnel URL on success.
+     */
+    async start(taskId, workspacePath) {
+        // If a tunnel already exists for this task, stop it first
+        if (this.tunnels.has(taskId)) {
+            log.warn({ taskId }, 'Tunnel already exists for task, stopping existing tunnel');
+            await this.stop(taskId);
+        }
+        log.info({ taskId, workspacePath }, 'Starting cursor remote tunnel');
+        return new Promise((resolve, reject) => {
+            let resolved = false;
+            let output = '';
+            // TODO: Replace with actual `cursor --remote-tunnel` invocation once
+            // the cursor CLI tunnel feature is available and tested.
+            // For now, spawn the process and attempt to extract a URL from stdout.
+            const child = (0, node_child_process_1.spawn)('cursor', ['--remote-tunnel', '--folder', workspacePath], {
+                stdio: ['ignore', 'pipe', 'pipe'],
+                env: (0, safe_env_js_1.buildSafeEnv)(),
+            });
+            const entry = {
+                taskId,
+                workspacePath,
+                process: child,
+                tunnelUrl: null,
+            };
+            this.tunnels.set(taskId, entry);
+            const timeout = setTimeout(() => {
+                if (!resolved) {
+                    resolved = true;
+                    child.kill('SIGTERM');
+                    this.tunnels.delete(taskId);
+                    log.warn({ taskId }, 'Tunnel start timed out, killing child process');
+                    reject(new Error('Tunnel start timed out waiting for URL'));
+                }
+            }, 30_000);
+            child.stdout?.on('data', (data) => {
+                if (resolved)
+                    return;
+                const chunk = data.toString();
+                output += chunk;
+                log.debug({ taskId, chunk: chunk.trim() }, 'Tunnel stdout');
+                // Attempt to extract tunnel URL from output
+                // Common patterns: https://... or vscode://...
+                const urlMatch = output.match(/https?:\/\/[^\s"'<>)\]},;]+/);
+                if (urlMatch && !resolved) {
+                    resolved = true;
+                    clearTimeout(timeout);
+                    entry.tunnelUrl = urlMatch[0];
+                    log.info({ taskId, tunnelUrl: entry.tunnelUrl }, 'Tunnel URL extracted');
+                    resolve(entry.tunnelUrl);
+                }
+            });
+            child.stderr?.on('data', (data) => {
+                log.warn({ taskId, stderr: data.toString().trim() }, 'Tunnel stderr');
+            });
+            child.on('error', (err) => {
+                clearTimeout(timeout);
+                this.tunnels.delete(taskId);
+                if (!resolved) {
+                    resolved = true;
+                    log.error({ taskId, err }, 'Tunnel process failed to start');
+                    reject(new Error(`Tunnel process error: ${err.message}`));
+                }
+            });
+            child.on('exit', (code, signal) => {
+                clearTimeout(timeout);
+                log.info({ taskId, code, signal }, 'Tunnel process exited');
+                this.tunnels.delete(taskId);
+                if (!resolved) {
+                    resolved = true;
+                    reject(new Error(`Tunnel process exited prematurely with code ${code}`));
+                }
+            });
+        });
+    }
+    /**
+     * Stop the tunnel for the given task.
+     */
+    async stop(taskId) {
+        const entry = this.tunnels.get(taskId);
+        if (!entry) {
+            log.debug({ taskId }, 'No tunnel found for task, nothing to stop');
+            return;
+        }
+        log.info({ taskId }, 'Stopping tunnel');
+        this.tunnels.delete(taskId);
+        return new Promise((resolve) => {
+            const child = entry.process;
+            // Give the process a chance to exit gracefully
+            const forceKillTimeout = setTimeout(() => {
+                try {
+                    child.kill('SIGKILL');
+                }
+                catch {
+                    // Process may already be dead
+                }
+                resolve();
+            }, 5_000);
+            child.once('exit', () => {
+                clearTimeout(forceKillTimeout);
+                resolve();
+            });
+            try {
+                child.kill('SIGTERM');
+            }
+            catch {
+                // Process may already be dead
+                clearTimeout(forceKillTimeout);
+                resolve();
+            }
+        });
+    }
+    /**
+     * Stop all active tunnels. Called during worker shutdown.
+     */
+    stopAll() {
+        for (const [taskId, entry] of this.tunnels) {
+            log.info({ taskId }, 'Stopping tunnel during shutdown');
+            try {
+                entry.process.kill('SIGTERM');
+            }
+            catch {
+                // Process may already be dead
+            }
+        }
+        this.tunnels.clear();
+    }
+    /**
+     * Check whether a tunnel is active for the given task.
+     */
+    hasTunnel(taskId) {
+        return this.tunnels.has(taskId);
+    }
+    /**
+     * Get the tunnel URL for a task, or null if not available.
+     */
+    getTunnelUrl(taskId) {
+        return this.tunnels.get(taskId)?.tunnelUrl ?? null;
+    }
+}
+exports.TunnelManager = TunnelManager;
+//# sourceMappingURL=tunnel-manager.js.map

package/dist/tunnel-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tunnel-manager.js","sourceRoot":"","sources":["../src/tunnel-manager.ts"],"names":[],"mappings":";;;;;;AAAA,2DAA8D;AAC9D,gDAAwB;AACxB,+CAA6C;AAE7C,MAAM,GAAG,GAAG,IAAA,cAAI,EAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;AAS7C;;;;;;;;GAQG;AACH,MAAa,aAAa;IAChB,OAAO,GAAG,IAAI,GAAG,EAAuB,CAAC;IAEjD;;;OAGG;IACH,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,aAAqB;QAC/C,0DAA0D;QAC1D,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7B,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,0DAA0D,CAAC,CAAC;YACjF,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1B,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,EAAE,+BAA+B,CAAC,CAAC;QAErE,OAAO,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC7C,IAAI,QAAQ,GAAG,KAAK,CAAC;YACrB,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,qEAAqE;YACrE,yDAAyD;YACzD,uEAAuE;YACvE,MAAM,KAAK,GAAG,IAAA,0BAAK,EAAC,QAAQ,EAAE,CAAC,iBAAiB,EAAE,UAAU,EAAE,aAAa,CAAC,EAAE;gBAC5E,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;gBACjC,GAAG,EAAE,IAAA,0BAAY,GAAE;aACpB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAgB;gBACzB,MAAM;gBACN,aAAa;gBACb,OAAO,EAAE,KAAK;gBACd,SAAS,EAAE,IAAI;aAChB,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAEhC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC9B,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,QAAQ,GAAG,IAAI,CAAC;oBAChB,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBACtB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBAC5B,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,+CAA+C,CAAC,CAAC;oBACtE,MAAM,CAAC,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC,EAAE,MAAM,CAAC,CAAC;YAEX,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;gBACxC,IAAI,QAAQ;oBAAE,OAAO;gBACrB,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC;gBAChB,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE,EAAE,eAAe,CAAC,CAAC;gBAE5D,4CAA4C;gBAC5C,+CAA+C;gBAC/C,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;gBAC7D,IAAI,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC;oBAC1B,QAAQ,GAAG,IAAI,CAAC;oBAChB,YAAY,CAAC,OAAO,CAAC,CAAC;oBACtB,KAAK,CAAC,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;oBAC9B,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,EAAE,sBAAsB,CAAC,CAAC;oBACzE,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;gBACxC,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE,EAAE,eAAe,CAAC,CAAC;YACxE,CAAC,CAAC,CAAC;YAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACxB,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBAC5B,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,QAAQ,GAAG,IAAI,CAAC;oBAChB,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,gCAAgC,CAAC,CAAC;oBAC7D,MAAM,CAAC,IAAI,KAAK,CAAC,yBAAyB,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBAC5D,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;gBAChC,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC;gBAC5D,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBAC5B,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,QAAQ,GAAG,IAAI,CAAC;oBAChB,MAAM,CAAC,IAAI,KAAK,CAAC,+CAA+C,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC3E,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,MAAc;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,EAAE,2CAA2C,CAAC,CAAC;YACnE,OAAO;QACT,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAE5B,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YACnC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC;YAE5B,+CAA+C;YAC/C,MAAM,gBAAgB,GAAG,UAAU,CAAC,GAAG,EAAE;gBACvC,IAAI,CAAC;oBACH,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACxB,CAAC;gBAAC,MAAM,CAAC;oBACP,8BAA8B;gBAChC,CAAC;gBACD,OAAO,EAAE,CAAC;YACZ,CAAC,EAAE,KAAK,CAAC,CAAC;YAEV,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE;gBACtB,YAAY,CAAC,gBAAgB,CAAC,CAAC;gBAC/B,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC;gBACH,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACxB,CAAC;YAAC,MAAM,CAAC;gBACP,8BAA8B;gBAC9B,YAAY,CAAC,gBAAgB,CAAC,CAAC;gBAC/B,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,OAAO;QACL,KAAK,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC3C,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,iCAAiC,CAAC,CAAC;YACxD,IAAI,CAAC;gBACH,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAChC,CAAC;YAAC,MAAM,CAAC;gBACP,8BAA8B;YAChC,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,MAAc;QACtB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,MAAc;QACzB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,SAAS,IAAI,IAAI,CAAC;IACrD,CAAC;CACF;AA/JD,sCA+JC"}

package/dist/worker-client.d.ts

@@ -0,0 +1,62 @@
+import type { ExecuteTaskFrame, CancelTaskFrame, CleanupWorkspaceFrame, ResumeTaskFrame, HeartbeatAckFrame, ServerShuttingDownFrame, RecoverySecretBootstrapFrame, StartTunnelFrame, StopTunnelFrame, StageConfirmResponseFrame } from '@overlordai/protocol';
+import type { StageConfirmMode, StageChoiceOption } from '@overlordai/protocol';
+export interface WorkerClientOptions {
+    overlordHost: string;
+    machineId: string;
+    getToken: () => string;
+    getCapabilities: () => string[];
+    getActiveSlots: () => number;
+}
+export interface WorkerClientHandlers {
+    onExecute: (frame: ExecuteTaskFrame) => void;
+    onCancel: (frame: CancelTaskFrame) => void;
+    onCleanupWorkspace: (frame: CleanupWorkspaceFrame) => void;
+    onResume: (frame: ResumeTaskFrame) => void;
+    onServerShuttingDown: (frame: ServerShuttingDownFrame) => void;
+    onRecoverySecretBootstrap: (frame: RecoverySecretBootstrapFrame) => void;
+    onHeartbeatAck: (frame: HeartbeatAckFrame) => void;
+    onStartTunnel: (frame: StartTunnelFrame) => void;
+    onStopTunnel: (frame: StopTunnelFrame) => void;
+    onStageConfirmResponse: (frame: StageConfirmResponseFrame) => void;
+    onConnected: () => void;
+    onDisconnected: () => void;
+}
+export declare class WorkerClient {
+    private ws;
+    private options;
+    private handlers;
+    private heartbeatTimer;
+    private reconnectTimer;
+    private reconnectDelay;
+    private connected;
+    private intentionalClose;
+    private dedupSet;
+    private dedupCleanupTimer;
+    constructor(options: WorkerClientOptions, handlers: WorkerClientHandlers);
+    isConnected(): boolean;
+    connect(): void;
+    disconnect(): void;
+    sendHeartbeat(): void;
+    sendAck(msgId: string, payload?: Record<string, unknown>): void;
+    sendProgress(taskId: number, stage: string, status: string, message?: string): void;
+    sendReconnect(tasks: Array<{
+        taskId: number;
+        localStatus: 'still_running' | 'completed_locally' | 'failed_locally';
+        currentStage?: string;
+        branch?: string;
+        mrUrl?: string;
+        errorMessage?: string;
+    }>): void;
+    sendRecoverySecretAck(): void;
+    sendStageConfirmRequest(taskId: number, stageIndex: number, mode: StageConfirmMode, timeout: number, prompt?: string, choices?: StageChoiceOption[]): void;
+    private sendFrame;
+    private handleMessage;
+    private buildWsUrl;
+    private startHeartbeat;
+    private stopHeartbeat;
+    private startDedupCleanup;
+    private stopDedupCleanup;
+    private scheduleReconnect;
+    private cleanup;
+}
+//# sourceMappingURL=worker-client.d.ts.map

package/dist/worker-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-client.d.ts","sourceRoot":"","sources":["../src/worker-client.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAGV,gBAAgB,EAChB,eAAe,EACf,qBAAqB,EACrB,eAAe,EACf,iBAAiB,EACjB,uBAAuB,EACvB,4BAA4B,EAC5B,gBAAgB,EAChB,eAAe,EACf,yBAAyB,EAE1B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAehF,MAAM,WAAW,mBAAmB;IAClC,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,MAAM,CAAC;IACvB,eAAe,EAAE,MAAM,MAAM,EAAE,CAAC;IAChC,cAAc,EAAE,MAAM,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,CAAC,KAAK,EAAE,gBAAgB,KAAK,IAAI,CAAC;IAC7C,QAAQ,EAAE,CAAC,KAAK,EAAE,eAAe,KAAK,IAAI,CAAC;IAC3C,kBAAkB,EAAE,CAAC,KAAK,EAAE,qBAAqB,KAAK,IAAI,CAAC;IAC3D,QAAQ,EAAE,CAAC,KAAK,EAAE,eAAe,KAAK,IAAI,CAAC;IAC3C,oBAAoB,EAAE,CAAC,KAAK,EAAE,uBAAuB,KAAK,IAAI,CAAC;IAC/D,yBAAyB,EAAE,CAAC,KAAK,EAAE,4BAA4B,KAAK,IAAI,CAAC;IACzE,cAAc,EAAE,CAAC,KAAK,EAAE,iBAAiB,KAAK,IAAI,CAAC;IACnD,aAAa,EAAE,CAAC,KAAK,EAAE,gBAAgB,KAAK,IAAI,CAAC;IACjD,YAAY,EAAE,CAAC,KAAK,EAAE,eAAe,KAAK,IAAI,CAAC;IAC/C,sBAAsB,EAAE,CAAC,KAAK,EAAE,yBAAyB,KAAK,IAAI,CAAC;IACnE,WAAW,EAAE,MAAM,IAAI,CAAC;IACxB,cAAc,EAAE,MAAM,IAAI,CAAC;CAC5B;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,EAAE,CAA0B;IACpC,OAAO,CAAC,OAAO,CAAsB;IACrC,OAAO,CAAC,QAAQ,CAAuB;IACvC,OAAO,CAAC,cAAc,CAA+C;IACrE,OAAO,CAAC,cAAc,CAA8C;IACpE,OAAO,CAAC,cAAc,CAA8B;IACpD,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,QAAQ,CAA6B;IAC7C,OAAO,CAAC,iBAAiB,CAA+C;gBAE5D,OAAO,EAAE,mBAAmB,EAAE,QAAQ,EAAE,oBAAoB;IAKxE,WAAW,IAAI,OAAO;IAItB,OAAO,IAAI,IAAI;IAgEf,UAAU,IAAI,IAAI;IASlB,aAAa,IAAI,IAAI;IAiBrB,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI;IAS/D,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI;IAWnF,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC;QACzB,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,eAAe,GAAG,mBAAmB,GAAG,gBAAgB,CAAC;QACtE,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,CAAC,GAAG,IAAI;IAST,qBAAqB,IAAI,IAAI;IAO7B,uBAAuB,CACrB,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,MAAM,EAClB,IAAI,EAAE,gBAAgB,EACtB,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,iBAAiB,EAAE,GAC5B,IAAI;IAaP,OAAO,CAAC,SAAS;IAYjB,OAAO,CAAC,aAAa;IA0DrB,OAAO,CAAC,UAAU;IAiBlB,OAAO,CAAC,cAAc;IAYtB,OAAO,CAAC,aAAa;IAOrB,OAAO,CAAC,iBAAiB;IAezB,OAAO,CAAC,gBAAgB;IAOxB,OAAO,CAAC,iBAAiB;IAgBzB,OAAO,CAAC,OAAO;CAUhB"}