@sudocode-ai/local-server 0.1.10 → 0.1.11

package/dist/workflow/services/wakeup-service.js

@@ -0,0 +1,809 @@
+/**
+ * WorkflowWakeupService
+ *
+ * Handles event recording and orchestrator wakeups.
+ * Events are batched within a configurable window before triggering wakeups.
+ */
+import { randomUUID } from "crypto";
+/**
+ * Default wakeup configuration
+ */
+export const DEFAULT_WAKEUP_CONFIG = {
+    batchWindowMs: 5000, // 5 seconds
+};
+// =============================================================================
+// Helper Functions
+// =============================================================================
+/**
+ * Convert a database row to a WorkflowEvent object
+ */
+function rowToEvent(row) {
+    return {
+        id: row.id,
+        workflowId: row.workflow_id,
+        type: row.type,
+        stepId: row.step_id ?? undefined,
+        executionId: row.execution_id ?? undefined,
+        payload: JSON.parse(row.payload),
+        createdAt: row.created_at,
+        processedAt: row.processed_at ?? undefined,
+    };
+}
+// =============================================================================
+// WorkflowWakeupService
+// =============================================================================
+/**
+ * Service for recording workflow events and triggering orchestrator wakeups.
+ *
+ * Events are batched within a configurable window. When the window expires,
+ * a wakeup is triggered by creating a follow-up execution for the orchestrator.
+ */
+export class WorkflowWakeupService {
+    db;
+    executionService;
+    promptBuilder;
+    eventEmitter;
+    config;
+    /** Pending wakeup timers by workflow ID */
+    pendingWakeups = new Map();
+    /** Active execution timeout timers by execution ID */
+    executionTimeouts = new Map();
+    /** Pending await conditions by workflow ID */
+    pendingAwaits = new Map();
+    /** Await timeout timers by await ID */
+    awaitTimeouts = new Map();
+    /** Last resolved await for each workflow (for wakeup message context) */
+    resolvedAwaits = new Map();
+    /** Whether the service is running (for timeout monitoring) */
+    _isRunning = false;
+    /** Check if the service is running */
+    get isRunning() {
+        return this._isRunning;
+    }
+    constructor(deps) {
+        this.db = deps.db;
+        this.executionService = deps.executionService;
+        this.promptBuilder = deps.promptBuilder;
+        this.eventEmitter = deps.eventEmitter;
+        this.config = { ...DEFAULT_WAKEUP_CONFIG, ...deps.config };
+    }
+    // ===========================================================================
+    // Event Recording
+    // ===========================================================================
+    /**
+     * Record a workflow event.
+     * Automatically schedules a wakeup after the batch window.
+     * If an await condition is satisfied, triggers immediate wakeup.
+     */
+    async recordEvent(params) {
+        const eventId = randomUUID();
+        const now = new Date().toISOString();
+        // Insert event into database
+        this.db
+            .prepare(`
+        INSERT INTO workflow_events (id, workflow_id, type, step_id, execution_id, payload, created_at)
+        VALUES (?, ?, ?, ?, ?, ?, ?)
+      `)
+            .run(eventId, params.workflowId, params.type, params.stepId ?? null, params.executionId ?? null, JSON.stringify(params.payload), now);
+        // Check if this event satisfies a pending await condition
+        const event = {
+            id: eventId,
+            workflowId: params.workflowId,
+            type: params.type,
+            stepId: params.stepId,
+            executionId: params.executionId,
+            payload: params.payload,
+            createdAt: now,
+        };
+        if (this.checkAwaitCondition(params.workflowId, event)) {
+            // Resolve the await - this stores context for wakeup message
+            this.resolveAwait(params.workflowId, params.type);
+            // Trigger immediate wakeup (skip debounce)
+            await this.triggerWakeup(params.workflowId);
+            return;
+        }
+        // Schedule wakeup (debounced)
+        this.scheduleWakeup(params.workflowId);
+    }
+    /**
+     * Get all unprocessed events for a workflow.
+     */
+    getUnprocessedEvents(workflowId) {
+        const rows = this.db
+            .prepare(`
+        SELECT * FROM workflow_events
+        WHERE workflow_id = ? AND processed_at IS NULL
+        ORDER BY created_at ASC
+      `)
+            .all(workflowId);
+        return rows.map(rowToEvent);
+    }
+    /**
+     * Mark events as processed.
+     */
+    markEventsProcessed(eventIds) {
+        if (eventIds.length === 0)
+            return;
+        const now = new Date().toISOString();
+        const placeholders = eventIds.map(() => "?").join(", ");
+        this.db
+            .prepare(`UPDATE workflow_events SET processed_at = ? WHERE id IN (${placeholders})`)
+            .run(now, ...eventIds);
+    }
+    // ===========================================================================
+    // Wakeup Scheduling
+    // ===========================================================================
+    /**
+     * Schedule a wakeup for a workflow.
+     * Debounced within the batch window - multiple events will be batched.
+     */
+    scheduleWakeup(workflowId) {
+        // Cancel any existing pending wakeup
+        const existing = this.pendingWakeups.get(workflowId);
+        if (existing) {
+            clearTimeout(existing);
+        }
+        // Schedule new wakeup after batch window
+        const timeout = setTimeout(() => {
+            this.pendingWakeups.delete(workflowId);
+            this.triggerWakeup(workflowId).catch((err) => {
+                console.error(`Failed to trigger wakeup for workflow ${workflowId}:`, err);
+            });
+        }, this.config.batchWindowMs);
+        this.pendingWakeups.set(workflowId, timeout);
+    }
+    /**
+     * Cancel a pending wakeup for a workflow.
+     */
+    cancelPendingWakeup(workflowId) {
+        const existing = this.pendingWakeups.get(workflowId);
+        if (existing) {
+            clearTimeout(existing);
+            this.pendingWakeups.delete(workflowId);
+        }
+    }
+    // ===========================================================================
+    // Execution Timeout Tracking
+    // ===========================================================================
+    /**
+     * Start a timeout for an execution.
+     *
+     * When the timeout fires, the execution is cancelled and a step_failed
+     * event is recorded with reason "timeout".
+     *
+     * The timeout deadline is persisted to the database so it can be recovered
+     * after server restart.
+     *
+     * @param executionId - The execution to track
+     * @param workflowId - The workflow the execution belongs to
+     * @param stepId - The workflow step ID
+     * @param timeoutMs - Timeout duration in milliseconds
+     */
+    startExecutionTimeout(executionId, workflowId, stepId, timeoutMs) {
+        // Clear any existing timeout for this execution
+        this.clearExecutionTimeout(executionId);
+        // Calculate absolute timeout deadline
+        const timeoutAt = new Date(Date.now() + timeoutMs).toISOString();
+        // Persist the timeout to the database for recovery after restart
+        // Use INSERT OR REPLACE to handle replacing an existing timeout for the same execution
+        const now = new Date().toISOString();
+        this.db
+            .prepare(`
+        INSERT OR REPLACE INTO workflow_events (id, workflow_id, type, execution_id, step_id, payload, created_at)
+        VALUES (?, ?, 'execution_timeout', ?, ?, ?, ?)
+      `)
+            .run(`timeout-${executionId}`, workflowId, executionId, stepId, JSON.stringify({ timeoutAt }), now);
+        const timeout = setTimeout(() => {
+            this.executionTimeouts.delete(executionId);
+            this.handleExecutionTimeout(executionId, workflowId, stepId).catch((err) => {
+                console.error(`[WakeupService] Error handling execution timeout:`, err);
+            });
+        }, timeoutMs);
+        this.executionTimeouts.set(executionId, { timeout, workflowId, stepId });
+        console.log(`[WakeupService] Started ${timeoutMs}ms timeout for execution ${executionId} (deadline: ${timeoutAt})`);
+    }
+    /**
+     * Clear timeout for an execution.
+     *
+     * Call this when an execution completes normally to prevent
+     * the timeout from firing. Also marks the persisted timeout
+     * event as processed in the database.
+     *
+     * @param executionId - The execution to clear timeout for
+     */
+    clearExecutionTimeout(executionId) {
+        const entry = this.executionTimeouts.get(executionId);
+        if (entry) {
+            clearTimeout(entry.timeout);
+            this.executionTimeouts.delete(executionId);
+        }
+        // Mark the persisted timeout event as processed
+        const result = this.db
+            .prepare(`
+        UPDATE workflow_events
+        SET processed_at = ?
+        WHERE id = ? AND type = 'execution_timeout' AND processed_at IS NULL
+      `)
+            .run(new Date().toISOString(), `timeout-${executionId}`);
+        if (entry || result.changes > 0) {
+            console.log(`[WakeupService] Cleared timeout for execution ${executionId}`);
+        }
+    }
+    /**
+     * Handle an execution timeout.
+     *
+     * Cancels the execution and records a step_failed event.
+     *
+     * @param executionId - The timed-out execution
+     * @param workflowId - The workflow containing the execution
+     * @param stepId - The workflow step that timed out
+     */
+    async handleExecutionTimeout(executionId, workflowId, stepId) {
+        console.warn(`[WakeupService] Execution ${executionId} timed out`);
+        // Mark the persisted timeout event as processed
+        this.db
+            .prepare(`
+        UPDATE workflow_events
+        SET processed_at = ?
+        WHERE id = ? AND type = 'execution_timeout' AND processed_at IS NULL
+      `)
+            .run(new Date().toISOString(), `timeout-${executionId}`);
+        // Try to cancel the execution
+        try {
+            await this.executionService.cancelExecution(executionId);
+        }
+        catch (err) {
+            console.error(`[WakeupService] Failed to cancel timed-out execution ${executionId}:`, err);
+            // Continue to record the timeout event anyway
+        }
+        // Record timeout event (will trigger wakeup via scheduleWakeup)
+        await this.recordEvent({
+            workflowId,
+            type: "step_failed",
+            executionId,
+            stepId,
+            payload: {
+                reason: "timeout",
+                message: "Execution exceeded configured timeout",
+            },
+        });
+    }
+    // ===========================================================================
+    // Wakeup Triggering
+    // ===========================================================================
+    /**
+     * Trigger an immediate wakeup for a workflow.
+     * Collects unprocessed events, builds a wakeup message, and creates
+     * a follow-up execution for the orchestrator.
+     */
+    async triggerWakeup(workflowId) {
+        // 1. Get workflow
+        const workflow = this.getWorkflow(workflowId);
+        if (!workflow) {
+            console.warn(`Cannot trigger wakeup: workflow ${workflowId} not found`);
+            return;
+        }
+        // 2. Check if workflow has an orchestrator execution
+        if (!workflow.orchestratorExecutionId) {
+            console.warn(`Cannot trigger wakeup: workflow ${workflowId} has no orchestrator execution`);
+            return;
+        }
+        // 3. Check workflow status - don't wake if paused/cancelled/completed
+        if (workflow.status === "paused" ||
+            workflow.status === "cancelled" ||
+            workflow.status === "completed" ||
+            workflow.status === "failed") {
+            console.debug(`Skipping wakeup for workflow ${workflowId}: status is ${workflow.status}`);
+            return;
+        }
+        // 4. Get unprocessed events
+        const events = this.getUnprocessedEvents(workflowId);
+        // 5. Get resolved await context (if any) - check before events check
+        // This allows timeout wakeups to proceed even with no events
+        const resolvedAwait = this.getAndClearResolvedAwait(workflowId);
+        // 6. Skip if no events AND no resolved await (nothing to report)
+        if (events.length === 0 && !resolvedAwait) {
+            console.debug(`No unprocessed events for workflow ${workflowId}`);
+            return;
+        }
+        // 7. Get executions referenced by events
+        const executions = this.getExecutionsForEvents(events);
+        // 8. Build wakeup message with await context
+        const message = this.promptBuilder.buildWakeupMessage(events, executions, resolvedAwait);
+        // 9. Stop the previous orchestrator execution before creating follow-up
+        // This prevents parallel execution of orchestrator and follow-up
+        // TODO: Migrate to using agent executor 'sendMessage' for inter-execution messages.
+        try {
+            const prevExecution = this.db
+                .prepare("SELECT status FROM executions WHERE id = ?")
+                .get(workflow.orchestratorExecutionId);
+            if (prevExecution &&
+                ["running", "pending", "preparing"].includes(prevExecution.status)) {
+                console.log(`[WakeupService] Stopping previous orchestrator execution ${workflow.orchestratorExecutionId} before creating follow-up`);
+                await this.executionService.cancelExecution(workflow.orchestratorExecutionId);
+            }
+        }
+        catch (err) {
+            console.warn(`[WakeupService] Failed to stop previous orchestrator execution ${workflow.orchestratorExecutionId}:`, err);
+            // Continue with follow-up creation anyway
+        }
+        // 10. Create follow-up execution
+        try {
+            const followUp = await this.executionService.createFollowUp(workflow.orchestratorExecutionId, message);
+            // 11. Update workflow with new orchestrator execution ID
+            this.updateOrchestratorExecution(workflowId, followUp.id, followUp.session_id);
+            // 12. Mark events as processed
+            this.markEventsProcessed(events.map((e) => e.id));
+            // 13. Emit wakeup event
+            this.eventEmitter.emit({
+                type: "orchestrator_wakeup",
+                workflowId,
+                payload: {
+                    eventCount: events.length,
+                    executionId: followUp.id,
+                },
+                timestamp: Date.now(),
+            });
+        }
+        catch (err) {
+            console.error(`Failed to create follow-up execution for workflow ${workflowId}:`, err);
+            // Don't mark events as processed - they'll be retried on next wakeup
+            throw err;
+        }
+    }
+    // ===========================================================================
+    // Lifecycle
+    // ===========================================================================
+    /**
+     * Start the wakeup service (enables timeout monitoring).
+     */
+    start() {
+        this._isRunning = true;
+        // TODO: Implement idle timeout monitoring if configured
+        // TODO: Implement execution timeout monitoring if configured
+    }
+    /**
+     * Recover state from the database after server restart.
+     *
+     * This method:
+     * 1. Recovers pending await conditions from workflow_events
+     * 2. Reschedules await timeouts (calculating remaining time)
+     * 3. Triggers immediate wakeup for expired awaits
+     * 4. Schedules wakeups for workflows with unprocessed events
+     * 5. Recovers execution timeout timers
+     *
+     * Should be called during server startup after the service is constructed.
+     */
+    async recoverState() {
+        console.log("[WakeupService] Starting state recovery...");
+        // 1. Recover pending await conditions
+        await this.recoverPendingAwaits();
+        // 2. Schedule wakeups for workflows with unprocessed events
+        await this.recoverPendingWakeups();
+        // 3. Recover execution timeouts
+        await this.recoverExecutionTimeouts();
+        console.log("[WakeupService] State recovery complete");
+    }
+    /**
+     * Recover pending await conditions from the database.
+     */
+    async recoverPendingAwaits() {
+        // Find unprocessed await events
+        const awaitRows = this.db
+            .prepare(`
+        SELECT we.*, w.status as workflow_status
+        FROM workflow_events we
+        JOIN workflows w ON we.workflow_id = w.id
+        WHERE we.type = 'orchestrator_wakeup'
+        AND we.processed_at IS NULL
+        AND json_extract(we.payload, '$.awaitType') = 'pending'
+        AND w.status = 'running'
+      `)
+            .all();
+        console.log(`[WakeupService] Found ${awaitRows.length} pending awaits to recover`);
+        for (const row of awaitRows) {
+            try {
+                const payload = JSON.parse(row.payload);
+                // Reconstruct pendingAwaits Map entry
+                const pendingAwait = {
+                    id: row.id,
+                    workflowId: row.workflow_id,
+                    eventTypes: payload.eventTypes,
+                    executionIds: payload.executionIds,
+                    timeoutAt: payload.timeoutAt,
+                    message: payload.message,
+                    createdAt: row.created_at,
+                };
+                this.pendingAwaits.set(row.workflow_id, pendingAwait);
+                // Handle timeout
+                if (payload.timeoutAt) {
+                    const remainingMs = new Date(payload.timeoutAt).getTime() - Date.now();
+                    if (remainingMs > 0) {
+                        // Reschedule timeout with remaining time
+                        this.scheduleAwaitTimeout(row.workflow_id, row.id, remainingMs / 1000);
+                        console.log(`[WakeupService] Rescheduled await ${row.id} timeout (${Math.round(remainingMs / 1000)}s remaining)`);
+                    }
+                    else {
+                        // Already expired - trigger immediately
+                        console.log(`[WakeupService] Await ${row.id} expired during downtime, triggering wakeup`);
+                        this.resolveAwait(row.workflow_id, "timeout");
+                        await this.triggerWakeup(row.workflow_id);
+                    }
+                }
+                else {
+                    console.log(`[WakeupService] Recovered await ${row.id} (no timeout)`);
+                }
+            }
+            catch (error) {
+                console.error(`[WakeupService] Failed to recover await for workflow ${row.workflow_id}:`, error);
+            }
+        }
+    }
+    /**
+     * Schedule wakeups for workflows with unprocessed events.
+     */
+    async recoverPendingWakeups() {
+        // Find workflows with unprocessed events (excluding await events and execution timeouts
+        // which are handled separately)
+        const workflowRows = this.db
+            .prepare(`
+        SELECT DISTINCT we.workflow_id
+        FROM workflow_events we
+        JOIN workflows w ON we.workflow_id = w.id
+        WHERE we.processed_at IS NULL
+        AND we.type NOT IN ('orchestrator_wakeup', 'execution_timeout')
+        AND w.status = 'running'
+      `)
+            .all();
+        console.log(`[WakeupService] Found ${workflowRows.length} workflows with unprocessed events`);
+        for (const row of workflowRows) {
+            // Schedule a wakeup for this workflow
+            this.scheduleWakeup(row.workflow_id);
+            console.log(`[WakeupService] Scheduled wakeup for workflow ${row.workflow_id}`);
+        }
+    }
+    /**
+     * Recover execution timeout timers from the database.
+     *
+     * Finds unprocessed execution_timeout events and reschedules the timers
+     * with the remaining time. If the timeout has already expired, handles
+     * the timeout immediately.
+     */
+    async recoverExecutionTimeouts() {
+        // Find unprocessed execution timeout events for running executions
+        const timeoutRows = this.db
+            .prepare(`
+        SELECT we.*, e.status as execution_status
+        FROM workflow_events we
+        JOIN executions e ON we.execution_id = e.id
+        JOIN workflows w ON we.workflow_id = w.id
+        WHERE we.type = 'execution_timeout'
+        AND we.processed_at IS NULL
+        AND e.status = 'running'
+        AND w.status = 'running'
+      `)
+            .all();
+        console.log(`[WakeupService] Found ${timeoutRows.length} execution timeouts to recover`);
+        for (const row of timeoutRows) {
+            try {
+                const payload = JSON.parse(row.payload);
+                const remainingMs = new Date(payload.timeoutAt).getTime() - Date.now();
+                if (remainingMs > 0) {
+                    // Reschedule timeout with remaining time
+                    const timeout = setTimeout(() => {
+                        this.executionTimeouts.delete(row.execution_id);
+                        this.handleExecutionTimeout(row.execution_id, row.workflow_id, row.step_id).catch((err) => {
+                            console.error(`[WakeupService] Error handling recovered execution timeout:`, err);
+                        });
+                    }, remainingMs);
+                    this.executionTimeouts.set(row.execution_id, {
+                        timeout,
+                        workflowId: row.workflow_id,
+                        stepId: row.step_id,
+                    });
+                    console.log(`[WakeupService] Rescheduled execution ${row.execution_id} timeout (${Math.round(remainingMs / 1000)}s remaining)`);
+                }
+                else {
+                    // Already expired - handle immediately
+                    console.log(`[WakeupService] Execution ${row.execution_id} timeout expired during downtime, handling now`);
+                    await this.handleExecutionTimeout(row.execution_id, row.workflow_id, row.step_id);
+                }
+            }
+            catch (error) {
+                console.error(`[WakeupService] Failed to recover timeout for execution ${row.execution_id}:`, error);
+            }
+        }
+    }
+    /**
+     * Stop the wakeup service.
+     * Cancels all pending wakeups, execution timeouts, and await conditions.
+     */
+    stop() {
+        this._isRunning = false;
+        // Cancel all pending wakeups
+        for (const [, timeout] of this.pendingWakeups) {
+            clearTimeout(timeout);
+        }
+        this.pendingWakeups.clear();
+        // Cancel all execution timeouts
+        for (const [, entry] of this.executionTimeouts) {
+            clearTimeout(entry.timeout);
+        }
+        this.executionTimeouts.clear();
+        // Cancel all await timeouts
+        for (const [, timeout] of this.awaitTimeouts) {
+            clearTimeout(timeout);
+        }
+        this.awaitTimeouts.clear();
+        this.pendingAwaits.clear();
+        this.resolvedAwaits.clear();
+    }
+    // ===========================================================================
+    // Await Condition Management
+    // ===========================================================================
+    /**
+     * Register a new await condition.
+     * Called by the await-events API endpoint.
+     *
+     * Persists the await to workflow_events for recovery after server restart.
+     */
+    registerAwait(params) {
+        const awaitId = randomUUID();
+        const now = new Date().toISOString();
+        const timeoutAt = params.timeoutSeconds
+            ? new Date(Date.now() + params.timeoutSeconds * 1000).toISOString()
+            : undefined;
+        const pendingAwait = {
+            id: awaitId,
+            workflowId: params.workflowId,
+            eventTypes: params.eventTypes,
+            executionIds: params.executionIds,
+            timeoutAt,
+            message: params.message,
+            createdAt: now,
+        };
+        // Store in memory (replaces any existing await for this workflow)
+        this.pendingAwaits.set(params.workflowId, pendingAwait);
+        // Persist to workflow_events for recovery
+        // Using 'orchestrator_wakeup' type with awaitType in payload
+        this.db
+            .prepare(`
+        INSERT INTO workflow_events (id, workflow_id, type, payload, created_at)
+        VALUES (?, ?, 'orchestrator_wakeup', ?, ?)
+      `)
+            .run(awaitId, params.workflowId, JSON.stringify({
+            awaitType: "pending",
+            eventTypes: params.eventTypes,
+            executionIds: params.executionIds,
+            timeoutAt,
+            message: params.message,
+        }), now);
+        // Schedule timeout if specified
+        if (params.timeoutSeconds) {
+            this.scheduleAwaitTimeout(params.workflowId, awaitId, params.timeoutSeconds);
+        }
+        console.log(`[WakeupService] Registered await ${awaitId} for workflow ${params.workflowId}`, { eventTypes: params.eventTypes, executionIds: params.executionIds });
+        return { id: awaitId, timeoutAt };
+    }
+    /**
+     * Check if an event satisfies the pending await for a workflow.
+     * Called internally when recording events.
+     */
+    checkAwaitCondition(workflowId, event) {
+        const pendingAwait = this.pendingAwaits.get(workflowId);
+        if (!pendingAwait)
+            return false;
+        // Map workflow event types to awaitable event types
+        const eventType = this.mapToAwaitableEventType(event.type);
+        if (!eventType)
+            return false;
+        // Check event type matches
+        if (!pendingAwait.eventTypes.includes(eventType)) {
+            return false;
+        }
+        // Check execution ID filter if specified
+        if (pendingAwait.executionIds && pendingAwait.executionIds.length > 0) {
+            if (!event.executionId ||
+                !pendingAwait.executionIds.includes(event.executionId)) {
+                return false;
+            }
+        }
+        return true;
+    }
+    /**
+     * Map workflow event type to awaitable event type.
+     */
+    mapToAwaitableEventType(eventType) {
+        switch (eventType) {
+            case "step_completed":
+                return "step_completed";
+            case "step_failed":
+                return "step_failed";
+            case "user_response":
+                return "user_response";
+            case "escalation_resolved":
+                return "escalation_resolved";
+            default:
+                return null;
+        }
+    }
+    /**
+     * Resolve an await condition (internal).
+     *
+     * Marks the await event as processed in the database for recovery tracking.
+     */
+    resolveAwait(workflowId, resolvedBy) {
+        const pendingAwait = this.pendingAwaits.get(workflowId);
+        if (!pendingAwait)
+            return;
+        // Store for wakeup message context
+        this.resolvedAwaits.set(workflowId, { ...pendingAwait, resolvedBy });
+        // Clear from pending
+        this.pendingAwaits.delete(workflowId);
+        // Mark event as processed in database
+        const now = new Date().toISOString();
+        this.db
+            .prepare(`
+        UPDATE workflow_events
+        SET processed_at = ?,
+            payload = json_set(payload, '$.resolvedBy', ?)
+        WHERE id = ?
+      `)
+            .run(now, resolvedBy, pendingAwait.id);
+        // Clear timeout if exists
+        const timeoutId = this.awaitTimeouts.get(pendingAwait.id);
+        if (timeoutId) {
+            clearTimeout(timeoutId);
+            this.awaitTimeouts.delete(pendingAwait.id);
+        }
+        console.log(`[WakeupService] Resolved await ${pendingAwait.id} for workflow ${workflowId}`, { resolvedBy });
+    }
+    /**
+     * Schedule timeout for an await.
+     */
+    scheduleAwaitTimeout(workflowId, awaitId, seconds) {
+        // Clear any existing timeout for this await
+        const existingTimeout = this.awaitTimeouts.get(awaitId);
+        if (existingTimeout) {
+            clearTimeout(existingTimeout);
+        }
+        const timeout = setTimeout(async () => {
+            this.awaitTimeouts.delete(awaitId);
+            // Check if this await is still pending
+            const pendingAwait = this.pendingAwaits.get(workflowId);
+            if (!pendingAwait || pendingAwait.id !== awaitId) {
+                return; // Already resolved
+            }
+            // Resolve the await with "timeout" as the trigger
+            this.resolveAwait(workflowId, "timeout");
+            // Trigger wakeup immediately
+            try {
+                await this.triggerWakeup(workflowId);
+            }
+            catch (err) {
+                console.error(`[WakeupService] Failed to trigger timeout wakeup for workflow ${workflowId}:`, err);
+            }
+        }, seconds * 1000);
+        this.awaitTimeouts.set(awaitId, timeout);
+        console.log(`[WakeupService] Scheduled ${seconds}s timeout for await ${awaitId}`);
+    }
+    /**
+     * Get the last resolved await for a workflow (for wakeup message).
+     * Clears after retrieval.
+     */
+    getAndClearResolvedAwait(workflowId) {
+        const resolved = this.resolvedAwaits.get(workflowId);
+        if (resolved) {
+            this.resolvedAwaits.delete(workflowId);
+        }
+        return resolved;
+    }
+    /**
+     * Clear all await state for a workflow (on cancel/complete).
+     * Marks any pending await events as processed in the database.
+     */
+    clearAwaitState(workflowId) {
+        const pendingAwait = this.pendingAwaits.get(workflowId);
+        if (pendingAwait) {
+            // Mark as processed (cancelled) in database
+            const now = new Date().toISOString();
+            this.db
+                .prepare(`
+          UPDATE workflow_events
+          SET processed_at = ?,
+              payload = json_set(payload, '$.resolvedBy', 'cancelled')
+          WHERE id = ?
+        `)
+                .run(now, pendingAwait.id);
+            const timeoutId = this.awaitTimeouts.get(pendingAwait.id);
+            if (timeoutId) {
+                clearTimeout(timeoutId);
+                this.awaitTimeouts.delete(pendingAwait.id);
+            }
+            this.pendingAwaits.delete(workflowId);
+        }
+        this.resolvedAwaits.delete(workflowId);
+    }
+    /**
+     * Check if a workflow has a pending await condition.
+     */
+    hasPendingAwait(workflowId) {
+        return this.pendingAwaits.has(workflowId);
+    }
+    /**
+     * Get the pending await for a workflow (if any).
+     */
+    getPendingAwait(workflowId) {
+        return this.pendingAwaits.get(workflowId);
+    }
+    // ===========================================================================
+    // Private Helpers
+    // ===========================================================================
+    /**
+     * Get a workflow by ID.
+     */
+    getWorkflow(workflowId) {
+        const row = this.db
+            .prepare("SELECT * FROM workflows WHERE id = ?")
+            .get(workflowId);
+        if (!row) {
+            return null;
+        }
+        return {
+            id: row.id,
+            title: row.title,
+            source: JSON.parse(row.source),
+            status: row.status,
+            steps: JSON.parse(row.steps),
+            worktreePath: row.worktree_path ?? undefined,
+            branchName: row.branch_name ?? undefined,
+            baseBranch: row.base_branch,
+            currentStepIndex: row.current_step_index,
+            orchestratorExecutionId: row.orchestrator_execution_id ?? undefined,
+            orchestratorSessionId: row.orchestrator_session_id ?? undefined,
+            config: JSON.parse(row.config),
+            createdAt: row.created_at,
+            updatedAt: row.updated_at,
+            startedAt: row.started_at ?? undefined,
+            completedAt: row.completed_at ?? undefined,
+        };
+    }
+    /**
+     * Get executions referenced by events.
+     */
+    getExecutionsForEvents(events) {
+        const executions = new Map();
+        // Collect unique execution IDs
+        const executionIds = new Set();
+        for (const event of events) {
+            if (event.executionId) {
+                executionIds.add(event.executionId);
+            }
+        }
+        // Fetch executions
+        for (const executionId of executionIds) {
+            const row = this.db
+                .prepare("SELECT * FROM executions WHERE id = ?")
+                .get(executionId);
+            if (row) {
+                executions.set(executionId, row);
+            }
+        }
+        return executions;
+    }
+    /**
+     * Update the workflow's orchestrator execution ID.
+     */
+    updateOrchestratorExecution(workflowId, executionId, sessionId) {
+        this.db
+            .prepare(`
+        UPDATE workflows
+        SET orchestrator_execution_id = ?,
+            orchestrator_session_id = ?,
+            updated_at = ?
+        WHERE id = ?
+      `)
+            .run(executionId, sessionId, new Date().toISOString(), workflowId);
+    }
+}
+//# sourceMappingURL=wakeup-service.js.map