patchrelay 0.1.0

package/dist/db/authoritative-ledger-store.js

@@ -0,0 +1,437 @@
+import { isoNow } from "./shared.js";
+export class AuthoritativeLedgerStore {
+    connection;
+    constructor(connection) {
+        this.connection = connection;
+    }
+    insertEventReceipt(params) {
+        const inserted = this.connection
+            .prepare(`
+        INSERT OR IGNORE INTO event_receipts (
+          source, external_id, event_type, received_at, acceptance_status, processing_status,
+          project_id, linear_issue_id, headers_json, payload_json
+        ) VALUES (?, ?, ?, ?, ?, 'pending', ?, ?, ?, ?)
+        `)
+            .run(params.source, params.externalId, params.eventType, params.receivedAt, params.acceptanceStatus, params.projectId ?? null, params.linearIssueId ?? null, params.headersJson ?? null, params.payloadJson ?? null);
+        const row = this.connection
+            .prepare("SELECT * FROM event_receipts WHERE source = ? AND external_id = ?")
+            .get(params.source, params.externalId);
+        if (!row) {
+            throw new Error(`Failed to load event receipt for ${params.source}:${params.externalId}`);
+        }
+        if (!inserted.changes) {
+            this.connection
+                .prepare("UPDATE event_receipts SET acceptance_status = 'duplicate' WHERE id = ? AND acceptance_status = 'accepted'")
+                .run(row.id);
+        }
+        return { id: Number(row.id), inserted: Boolean(inserted.changes) };
+    }
+    markEventReceiptProcessed(id, status) {
+        this.connection.prepare("UPDATE event_receipts SET processing_status = ? WHERE id = ?").run(status, id);
+    }
+    assignEventReceiptContext(id, params) {
+        this.connection
+            .prepare(`
+        UPDATE event_receipts
+        SET project_id = COALESCE(?, project_id),
+            linear_issue_id = COALESCE(?, linear_issue_id)
+        WHERE id = ?
+        `)
+            .run(params.projectId ?? null, params.linearIssueId ?? null, id);
+    }
+    getEventReceipt(id) {
+        const row = this.connection.prepare("SELECT * FROM event_receipts WHERE id = ?").get(id);
+        return row ? mapEventReceipt(row) : undefined;
+    }
+    getEventReceiptBySourceExternalId(source, externalId) {
+        const row = this.connection
+            .prepare("SELECT * FROM event_receipts WHERE source = ? AND external_id = ?")
+            .get(source, externalId);
+        return row ? mapEventReceipt(row) : undefined;
+    }
+    upsertIssueControl(params) {
+        const now = isoNow();
+        this.connection
+            .prepare(`
+        INSERT INTO issue_control (
+          project_id, linear_issue_id, desired_stage, desired_receipt_id, active_workspace_ownership_id,
+          active_run_lease_id, service_owned_comment_id, active_agent_session_id, lifecycle_status, updated_at
+        ) VALUES (
+          @projectId, @linearIssueId, @desiredStage, @desiredReceiptId, @activeWorkspaceOwnershipId,
+          @activeRunLeaseId, @serviceOwnedCommentId, @activeAgentSessionId, @lifecycleStatus, @updatedAt
+        )
+        ON CONFLICT(project_id, linear_issue_id) DO UPDATE SET
+          desired_stage = CASE WHEN @setDesiredStage = 1 THEN @desiredStage ELSE issue_control.desired_stage END,
+          desired_receipt_id = CASE WHEN @setDesiredReceiptId = 1 THEN @desiredReceiptId ELSE issue_control.desired_receipt_id END,
+          active_workspace_ownership_id = CASE WHEN @setActiveWorkspaceOwnershipId = 1 THEN @activeWorkspaceOwnershipId ELSE issue_control.active_workspace_ownership_id END,
+          active_run_lease_id = CASE WHEN @setActiveRunLeaseId = 1 THEN @activeRunLeaseId ELSE issue_control.active_run_lease_id END,
+          service_owned_comment_id = CASE WHEN @setServiceOwnedCommentId = 1 THEN @serviceOwnedCommentId ELSE issue_control.service_owned_comment_id END,
+          active_agent_session_id = CASE WHEN @setActiveAgentSessionId = 1 THEN @activeAgentSessionId ELSE issue_control.active_agent_session_id END,
+          lifecycle_status = @lifecycleStatus,
+          updated_at = @updatedAt
+        `)
+            .run({
+            projectId: params.projectId,
+            linearIssueId: params.linearIssueId,
+            desiredStage: params.desiredStage ?? null,
+            desiredReceiptId: params.desiredReceiptId ?? null,
+            activeWorkspaceOwnershipId: params.activeWorkspaceOwnershipId ?? null,
+            activeRunLeaseId: params.activeRunLeaseId ?? null,
+            serviceOwnedCommentId: params.serviceOwnedCommentId ?? null,
+            activeAgentSessionId: params.activeAgentSessionId ?? null,
+            lifecycleStatus: params.lifecycleStatus,
+            updatedAt: now,
+            setDesiredStage: Number("desiredStage" in params),
+            setDesiredReceiptId: Number("desiredReceiptId" in params),
+            setActiveWorkspaceOwnershipId: Number("activeWorkspaceOwnershipId" in params),
+            setActiveRunLeaseId: Number("activeRunLeaseId" in params),
+            setServiceOwnedCommentId: Number("serviceOwnedCommentId" in params),
+            setActiveAgentSessionId: Number("activeAgentSessionId" in params),
+        });
+        return this.getIssueControl(params.projectId, params.linearIssueId);
+    }
+    getIssueControl(projectId, linearIssueId) {
+        const row = this.connection
+            .prepare("SELECT * FROM issue_control WHERE project_id = ? AND linear_issue_id = ?")
+            .get(projectId, linearIssueId);
+        return row ? mapIssueControl(row) : undefined;
+    }
+    listIssueControlsReadyForLaunch() {
+        const rows = this.connection
+            .prepare("SELECT * FROM issue_control WHERE desired_stage IS NOT NULL AND active_run_lease_id IS NULL ORDER BY id")
+            .all();
+        return rows.map((row) => mapIssueControl(row));
+    }
+    upsertWorkspaceOwnership(params) {
+        const now = isoNow();
+        this.connection
+            .prepare(`
+        INSERT INTO workspace_ownership (
+          project_id, linear_issue_id, branch_name, worktree_path, status, current_run_lease_id, created_at, updated_at
+        ) VALUES (@projectId, @linearIssueId, @branchName, @worktreePath, @status, @currentRunLeaseId, @createdAt, @updatedAt)
+        ON CONFLICT(project_id, linear_issue_id) DO UPDATE SET
+          branch_name = @branchName,
+          worktree_path = @worktreePath,
+          status = @status,
+          current_run_lease_id = CASE WHEN @setCurrentRunLeaseId = 1 THEN @currentRunLeaseId ELSE workspace_ownership.current_run_lease_id END,
+          updated_at = @updatedAt
+        `)
+            .run({
+            projectId: params.projectId,
+            linearIssueId: params.linearIssueId,
+            branchName: params.branchName,
+            worktreePath: params.worktreePath,
+            status: params.status,
+            currentRunLeaseId: params.currentRunLeaseId ?? null,
+            createdAt: now,
+            updatedAt: now,
+            setCurrentRunLeaseId: Number("currentRunLeaseId" in params),
+        });
+        return this.getWorkspaceOwnershipForIssue(params.projectId, params.linearIssueId);
+    }
+    getWorkspaceOwnership(id) {
+        const row = this.connection
+            .prepare("SELECT * FROM workspace_ownership WHERE id = ?")
+            .get(id);
+        return row ? mapWorkspaceOwnership(row) : undefined;
+    }
+    getWorkspaceOwnershipForIssue(projectId, linearIssueId) {
+        const row = this.connection
+            .prepare("SELECT * FROM workspace_ownership WHERE project_id = ? AND linear_issue_id = ?")
+            .get(projectId, linearIssueId);
+        return row ? mapWorkspaceOwnership(row) : undefined;
+    }
+    createRunLease(params) {
+        const result = this.connection
+            .prepare(`
+        INSERT INTO run_leases (
+          issue_control_id, project_id, linear_issue_id, workspace_ownership_id, stage, status, trigger_receipt_id, workflow_file, prompt_text, started_at
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        `)
+            .run(params.issueControlId, params.projectId, params.linearIssueId, params.workspaceOwnershipId, params.stage, params.status ?? "queued", params.triggerReceiptId ?? null, params.workflowFile ?? "", params.promptText ?? "", isoNow());
+        return this.getRunLease(Number(result.lastInsertRowid));
+    }
+    getRunLease(id) {
+        const row = this.connection.prepare("SELECT * FROM run_leases WHERE id = ?").get(id);
+        return row ? mapRunLease(row) : undefined;
+    }
+    getRunLeaseByThreadId(threadId) {
+        const row = this.connection
+            .prepare("SELECT * FROM run_leases WHERE thread_id = ? ORDER BY id DESC LIMIT 1")
+            .get(threadId);
+        return row ? mapRunLease(row) : undefined;
+    }
+    listActiveRunLeases() {
+        const rows = this.connection
+            .prepare("SELECT * FROM run_leases WHERE status IN ('queued', 'running', 'paused') ORDER BY id")
+            .all();
+        return rows.map((row) => mapRunLease(row));
+    }
+    listRunLeasesForIssue(projectId, linearIssueId) {
+        const rows = this.connection
+            .prepare("SELECT * FROM run_leases WHERE project_id = ? AND linear_issue_id = ? ORDER BY id")
+            .all(projectId, linearIssueId);
+        return rows.map((row) => mapRunLease(row));
+    }
+    updateRunLeaseThread(params) {
+        this.connection
+            .prepare(`
+        UPDATE run_leases
+        SET thread_id = CASE WHEN @setThreadId = 1 THEN @threadId ELSE thread_id END,
+            parent_thread_id = CASE WHEN @setParentThreadId = 1 THEN @parentThreadId ELSE parent_thread_id END,
+            turn_id = CASE WHEN @setTurnId = 1 THEN @turnId ELSE turn_id END
+        WHERE id = @runLeaseId
+        `)
+            .run({
+            runLeaseId: params.runLeaseId,
+            threadId: params.threadId ?? null,
+            parentThreadId: params.parentThreadId ?? null,
+            turnId: params.turnId ?? null,
+            setThreadId: Number("threadId" in params),
+            setParentThreadId: Number("parentThreadId" in params),
+            setTurnId: Number("turnId" in params),
+        });
+    }
+    finishRunLease(params) {
+        const now = isoNow();
+        this.connection
+            .prepare(`
+        UPDATE run_leases
+        SET status = @status,
+            thread_id = CASE WHEN @setThreadId = 1 THEN @threadId ELSE thread_id END,
+            turn_id = CASE WHEN @setTurnId = 1 THEN @turnId ELSE turn_id END,
+            failure_reason = CASE WHEN @setFailureReason = 1 THEN @failureReason ELSE failure_reason END,
+            ended_at = CASE WHEN @status IN ('completed', 'failed', 'released') THEN @endedAt ELSE ended_at END
+        WHERE id = @runLeaseId
+        `)
+            .run({
+            runLeaseId: params.runLeaseId,
+            status: params.status,
+            threadId: params.threadId ?? null,
+            turnId: params.turnId ?? null,
+            failureReason: params.failureReason ?? null,
+            endedAt: now,
+            setThreadId: Number("threadId" in params),
+            setTurnId: Number("turnId" in params),
+            setFailureReason: Number("failureReason" in params),
+        });
+    }
+    enqueueObligation(params) {
+        const now = isoNow();
+        const result = this.connection
+            .prepare(`
+        INSERT OR IGNORE INTO obligations (
+          project_id, linear_issue_id, kind, status, source, payload_json, run_lease_id, thread_id, turn_id, dedupe_key, created_at, updated_at
+        ) VALUES (?, ?, ?, 'pending', ?, ?, ?, ?, ?, ?, ?, ?)
+        `)
+            .run(params.projectId, params.linearIssueId, params.kind, params.source, params.payloadJson, params.runLeaseId ?? null, params.threadId ?? null, params.turnId ?? null, params.dedupeKey ?? null, now, now);
+        if (result.changes) {
+            return this.getObligation(Number(result.lastInsertRowid));
+        }
+        if (params.dedupeKey) {
+            const existing = params.runLeaseId === undefined || params.runLeaseId === null
+                ? undefined
+                : this.getObligationByDedupeKey({
+                    runLeaseId: params.runLeaseId,
+                    kind: params.kind,
+                    dedupeKey: params.dedupeKey,
+                });
+            if (existing) {
+                return existing;
+            }
+        }
+        throw new Error(`Failed to persist obligation for ${params.projectId}:${params.linearIssueId}:${params.kind}`);
+    }
+    listPendingObligations(params) {
+        const statuses = params?.includeInProgress ? ["pending", "in_progress"] : ["pending"];
+        const clauses = [`status IN (${statuses.map(() => "?").join(", ")})`];
+        const values = [...statuses];
+        if (params?.runLeaseId !== undefined) {
+            clauses.push("run_lease_id = ?");
+            values.push(params.runLeaseId);
+        }
+        if (params?.kind) {
+            clauses.push("kind = ?");
+            values.push(params.kind);
+        }
+        const rows = this.connection
+            .prepare(`SELECT * FROM obligations WHERE ${clauses.join(" AND ")} ORDER BY id`)
+            .all(...values);
+        return rows.map((row) => mapObligation(row));
+    }
+    claimPendingObligation(id, params) {
+        const result = this.connection
+            .prepare(`
+        UPDATE obligations
+        SET status = 'in_progress',
+            run_lease_id = CASE WHEN @setRunLeaseId = 1 THEN @runLeaseId ELSE run_lease_id END,
+            thread_id = CASE WHEN @setThreadId = 1 THEN @threadId ELSE thread_id END,
+            turn_id = CASE WHEN @setTurnId = 1 THEN @turnId ELSE turn_id END,
+            last_error = NULL,
+            updated_at = @updatedAt
+        WHERE id = @id
+          AND status = 'pending'
+        `)
+            .run({
+            id,
+            runLeaseId: params?.runLeaseId ?? null,
+            threadId: params?.threadId ?? null,
+            turnId: params?.turnId ?? null,
+            updatedAt: isoNow(),
+            setRunLeaseId: Number("runLeaseId" in (params ?? {})),
+            setThreadId: Number("threadId" in (params ?? {})),
+            setTurnId: Number("turnId" in (params ?? {})),
+        });
+        return result.changes > 0;
+    }
+    updateObligationPayloadJson(id, payloadJson) {
+        this.connection
+            .prepare(`
+        UPDATE obligations
+        SET payload_json = ?,
+            updated_at = ?
+        WHERE id = ?
+        `)
+            .run(payloadJson, isoNow(), id);
+    }
+    updateObligationRouting(id, params) {
+        this.connection
+            .prepare(`
+        UPDATE obligations
+        SET run_lease_id = CASE WHEN @setRunLeaseId = 1 THEN @runLeaseId ELSE run_lease_id END,
+            thread_id = CASE WHEN @setThreadId = 1 THEN @threadId ELSE thread_id END,
+            turn_id = CASE WHEN @setTurnId = 1 THEN @turnId ELSE turn_id END,
+            updated_at = @updatedAt
+        WHERE id = @id
+        `)
+            .run({
+            id,
+            runLeaseId: params.runLeaseId ?? null,
+            threadId: params.threadId ?? null,
+            turnId: params.turnId ?? null,
+            updatedAt: isoNow(),
+            setRunLeaseId: Number("runLeaseId" in params),
+            setThreadId: Number("threadId" in params),
+            setTurnId: Number("turnId" in params),
+        });
+    }
+    markObligationStatus(id, status, lastError) {
+        const now = isoNow();
+        this.connection
+            .prepare(`
+        UPDATE obligations
+        SET status = @status,
+            last_error = CASE WHEN @setLastError = 1 THEN @lastError ELSE last_error END,
+            updated_at = @updatedAt,
+            completed_at = CASE WHEN @status = 'completed' THEN @completedAt ELSE completed_at END
+        WHERE id = @id
+        `)
+            .run({
+            id,
+            status,
+            lastError: lastError ?? null,
+            updatedAt: now,
+            completedAt: now,
+            setLastError: Number(lastError !== undefined),
+        });
+    }
+    getObligation(id) {
+        const row = this.connection.prepare("SELECT * FROM obligations WHERE id = ?").get(id);
+        return row ? mapObligation(row) : undefined;
+    }
+    getObligationByDedupeKey(params) {
+        const row = this.connection
+            .prepare(`
+        SELECT * FROM obligations
+        WHERE run_lease_id IS ?
+          AND kind = ?
+          AND dedupe_key = ?
+        ORDER BY id DESC
+        LIMIT 1
+        `)
+            .get(params.runLeaseId, params.kind, params.dedupeKey);
+        return row ? mapObligation(row) : undefined;
+    }
+}
+function mapEventReceipt(row) {
+    return {
+        id: Number(row.id),
+        source: String(row.source),
+        externalId: String(row.external_id),
+        eventType: String(row.event_type),
+        receivedAt: String(row.received_at),
+        acceptanceStatus: row.acceptance_status,
+        processingStatus: row.processing_status,
+        ...(row.project_id === null ? {} : { projectId: String(row.project_id) }),
+        ...(row.linear_issue_id === null ? {} : { linearIssueId: String(row.linear_issue_id) }),
+        ...(row.headers_json === null ? {} : { headersJson: String(row.headers_json) }),
+        ...(row.payload_json === null ? {} : { payloadJson: String(row.payload_json) }),
+    };
+}
+function mapIssueControl(row) {
+    return {
+        id: Number(row.id),
+        projectId: String(row.project_id),
+        linearIssueId: String(row.linear_issue_id),
+        ...(row.desired_stage === null ? {} : { desiredStage: row.desired_stage }),
+        ...(row.desired_receipt_id === null ? {} : { desiredReceiptId: Number(row.desired_receipt_id) }),
+        ...(row.active_run_lease_id === null ? {} : { activeRunLeaseId: Number(row.active_run_lease_id) }),
+        ...(row.active_workspace_ownership_id === null ? {} : { activeWorkspaceOwnershipId: Number(row.active_workspace_ownership_id) }),
+        ...(row.service_owned_comment_id === null ? {} : { serviceOwnedCommentId: String(row.service_owned_comment_id) }),
+        ...(row.active_agent_session_id === null ? {} : { activeAgentSessionId: String(row.active_agent_session_id) }),
+        lifecycleStatus: row.lifecycle_status,
+        updatedAt: String(row.updated_at),
+    };
+}
+function mapWorkspaceOwnership(row) {
+    return {
+        id: Number(row.id),
+        projectId: String(row.project_id),
+        linearIssueId: String(row.linear_issue_id),
+        branchName: String(row.branch_name),
+        worktreePath: String(row.worktree_path),
+        status: row.status,
+        ...(row.current_run_lease_id === null ? {} : { currentRunLeaseId: Number(row.current_run_lease_id) }),
+        createdAt: String(row.created_at),
+        updatedAt: String(row.updated_at),
+    };
+}
+function mapRunLease(row) {
+    return {
+        id: Number(row.id),
+        issueControlId: Number(row.issue_control_id),
+        projectId: String(row.project_id),
+        linearIssueId: String(row.linear_issue_id),
+        workspaceOwnershipId: Number(row.workspace_ownership_id),
+        stage: row.stage,
+        status: row.status,
+        ...(row.trigger_receipt_id === null ? {} : { triggerReceiptId: Number(row.trigger_receipt_id) }),
+        workflowFile: String(row.workflow_file ?? ""),
+        promptText: String(row.prompt_text ?? ""),
+        ...(row.thread_id === null ? {} : { threadId: String(row.thread_id) }),
+        ...(row.parent_thread_id === null ? {} : { parentThreadId: String(row.parent_thread_id) }),
+        ...(row.turn_id === null ? {} : { turnId: String(row.turn_id) }),
+        startedAt: String(row.started_at),
+        ...(row.ended_at === null ? {} : { endedAt: String(row.ended_at) }),
+        ...(row.failure_reason === null ? {} : { failureReason: String(row.failure_reason) }),
+    };
+}
+function mapObligation(row) {
+    return {
+        id: Number(row.id),
+        projectId: String(row.project_id),
+        linearIssueId: String(row.linear_issue_id),
+        kind: String(row.kind),
+        status: row.status,
+        source: String(row.source),
+        payloadJson: String(row.payload_json),
+        ...(row.run_lease_id === null ? {} : { runLeaseId: Number(row.run_lease_id) }),
+        ...(row.thread_id === null ? {} : { threadId: String(row.thread_id) }),
+        ...(row.turn_id === null ? {} : { turnId: String(row.turn_id) }),
+        ...(row.dedupe_key === null ? {} : { dedupeKey: String(row.dedupe_key) }),
+        ...(row.last_error === null ? {} : { lastError: String(row.last_error) }),
+        createdAt: String(row.created_at),
+        updatedAt: String(row.updated_at),
+        ...(row.completed_at === null ? {} : { completedAt: String(row.completed_at) }),
+    };
+}