@sixb/pg 0.1.1 → 0.1.3

package/src/pg-workflow-run-storage.ts

@@ -1,10 +1,17 @@
-import type { Principal, WorkflowRunSource } from "@sixb/core"
+import type { JsonValue } from "@sixb/core"
+import { normalizeRequesterGroupIds } from "@sixb/core/internal/auth"
+import { parseSixbFailure, serializeSixbFailure } from "@sixb/core/internal/errors"
+import {
+  assertWorkflowAgentNodeRunExecution,
+  assertWorkflowRunExecution,
+} from "@sixb/core/internal/workflow-run-storage-provider"
 import type { WorkflowIOSnapshot } from "@sixb/core/internal/workflows"
 import type {
   CancelWorkflowAgentNodeRunInput,
   ConfirmWorkflowAgentNodeRunExecutionOwnershipInput,
   ConfirmWorkflowRunExecutionOwnershipInput,
   CreateWorkflowAgentNodeRunInput,
+  ExecutionStorage,
   FinishWorkflowAgentNodeRunInput,
   FinishWorkflowNodeRunInput,
   FinishWorkflowRunInput,
@@ -32,7 +39,11 @@ import type {
   WorkflowRunRecord,
   WorkflowRunStorage,
 } from "@sixb/core/storage"
-import { WorkflowRunError } from "@sixb/core/storage"
+import {
+  AGENT_RUN_FAILURE_CODES,
+  WORKFLOW_RUN_FAILURE_CODES,
+  WorkflowRunError,
+} from "@sixb/core/storage"
 import { queryLatestRunsByOwnerId } from "./latest-run-query"
 import type { SqlParameter } from "./pg-client"
 import { appendRunListFilters, hasEmptyStatuses, queryRunList } from "./run-list-query"
@@ -43,39 +54,47 @@ export class PgWorkflowRunStorage implements WorkflowRunStorage {
   readonly nodes: PgWorkflowNodeRunStorage
   readonly agentNodes: PgWorkflowAgentNodeRunStorage
 
-  constructor(private readonly sql: PgStoreClient) {
+  constructor(
+    private readonly sql: PgStoreClient,
+    private readonly executions: ExecutionStorage
+  ) {
     this.nodes = new PgWorkflowNodeRunStorage(sql)
-    this.agentNodes = new PgWorkflowAgentNodeRunStorage(sql)
+    this.agentNodes = new PgWorkflowAgentNodeRunStorage(sql, executions)
   }
 
   async queue(input: QueueWorkflowRunInput): Promise<WorkflowRunRecord> {
     const queuedAt = input.queuedAt ?? new Date()
+    await assertWorkflowRunExecution({
+      executions: this.executions,
+      projectId: input.projectId,
+      executionId: input.executionId,
+      runId: input.id,
+      workflowId: input.workflowId,
+    })
 
     try {
       const [row] = await this.sql<WorkflowRunDatabaseRow[]>`
         INSERT INTO workflow_runs (
           project_id,
           id,
+          execution_id,
           workflow_id,
           status,
           input,
           queued_at,
           started_at,
-          source,
-          requested_by_principal_type,
-          requested_by_principal_id,
+          requester_group_ids,
           attempt
         ) VALUES (
           ${input.projectId},
           ${input.id},
+          ${input.executionId},
           ${input.workflowId},
           ${"queued"},
           ${serializeRecord(input.input)}::text::jsonb,
           ${queuedAt},
           ${queuedAt},
-          ${input.source ? JSON.stringify(input.source) : null}::text::jsonb,
-          ${input.requestedByPrincipal?.type ?? "system"},
-          ${input.requestedByPrincipal?.id ?? "system"},
+          ${JSON.stringify(normalizeRequesterGroupIds(input.requesterGroupIds))}::text::jsonb,
           ${0}
         )
         RETURNING *
@@ -85,7 +104,7 @@ export class PgWorkflowRunStorage implements WorkflowRunStorage {
     } catch (error) {
       if (isUniqueViolation(error)) {
         throw new WorkflowRunError(
-          `[SixbPg] Workflow run '${input.id}' already exists for project '${input.projectId}'.`
+          `[SixbPg] Workflow run '${input.id}' or execution '${input.executionId}' is already linked for project '${input.projectId}'.`
         )
       }
 
@@ -102,83 +121,29 @@ export class PgWorkflowRunStorage implements WorkflowRunStorage {
         FOR UPDATE
       `
 
-      if (existing) {
-        if (existing.status !== "queued") {
-          throw new WorkflowRunError(
-            `[SixbPg] Workflow run '${input.id}' already exists for project '${input.projectId}'.`
-          )
-        }
-
-        if (existing.workflow_id !== input.workflowId) {
-          throw new WorkflowRunError(
-            `[SixbPg] Workflow run '${input.id}' workflow '${input.workflowId}' does not match existing workflow '${existing.workflow_id}'.`
-          )
-        }
-
-        const [updated] = await tx<WorkflowRunDatabaseRow[]>`
-          UPDATE workflow_runs
-          SET
-            status = ${"running"},
-            input = ${serializeRecord(input.input)}::text::jsonb,
-            started_at = ${startedAt},
-            finished_at = ${null},
-            error = ${null},
-            source = COALESCE(
-              source,
-              ${input.source ? JSON.stringify(input.source) : null}::text::jsonb
-            ),
-            attempt = attempt + 1,
-            execution_token = ${input.execution?.token ?? null},
-            execution_queue_lease_expires_at = ${input.execution?.queueLeaseExpiresAt ?? null}
-          WHERE project_id = ${input.projectId} AND id = ${input.id}
-          RETURNING *
-        `
-
-        return rowToWorkflowRunRecord(updated)
+      if (!existing || existing.status !== "queued") {
+        throw new WorkflowRunError(
+          existing
+            ? `[SixbPg] Workflow run '${input.id}' cannot start from status '${existing.status}'.`
+            : `[SixbPg] Workflow run '${input.id}' not found for project '${input.projectId}'.`
+        )
       }
 
-      try {
-        const [row] = await tx<WorkflowRunDatabaseRow[]>`
-          INSERT INTO workflow_runs (
-            project_id,
-            id,
-            workflow_id,
-            status,
-            input,
-            started_at,
-            source,
-            requested_by_principal_type,
-            requested_by_principal_id,
-            attempt,
-            execution_token,
-            execution_queue_lease_expires_at
-          ) VALUES (
-            ${input.projectId},
-            ${input.id},
-            ${input.workflowId},
-            ${"running"},
-            ${serializeRecord(input.input)}::text::jsonb,
-            ${startedAt},
-            ${input.source ? JSON.stringify(input.source) : null}::text::jsonb,
-            ${input.requestedByPrincipal?.type ?? "system"},
-            ${input.requestedByPrincipal?.id ?? "system"},
-            ${1},
-            ${input.execution?.token ?? null},
-            ${input.execution?.queueLeaseExpiresAt ?? null}
-          )
-          RETURNING *
-        `
-
-        return rowToWorkflowRunRecord(row)
-      } catch (error) {
-        if (isUniqueViolation(error)) {
-          throw new WorkflowRunError(
-            `[SixbPg] Workflow run '${input.id}' already exists for project '${input.projectId}'.`
-          )
-        }
+      const [updated] = await tx<WorkflowRunDatabaseRow[]>`
+        UPDATE workflow_runs
+        SET
+          status = ${"running"},
+          started_at = ${startedAt},
+          finished_at = ${null},
+          error = ${null},
+          attempt = attempt + 1,
+          execution_token = ${input.execution?.token ?? null},
+          execution_queue_lease_expires_at = ${input.execution?.queueLeaseExpiresAt ?? null}
+        WHERE project_id = ${input.projectId} AND id = ${input.id}
+        RETURNING *
+      `
 
-        throw error
-      }
+      return rowToWorkflowRunRecord(updated)
     })
   }
 
@@ -258,7 +223,7 @@ export class PgWorkflowRunStorage implements WorkflowRunStorage {
                 status = ${input.status},
                 finished_at = ${input.finishedAt ?? new Date()},
                 output = ${null},
-                error = ${input.error ?? null},
+                error = ${input.error === undefined ? null : serializeSixbFailure(input.error, WORKFLOW_RUN_FAILURE_CODES)}::text::jsonb,
                 execution_token = ${null},
                 execution_queue_lease_expires_at = ${null}
               WHERE project_id = ${input.projectId} AND id = ${input.id}
@@ -416,16 +381,43 @@ export class PgWorkflowRunStorage implements WorkflowRunStorage {
 }
 
 export class PgWorkflowAgentNodeRunStorage implements WorkflowAgentNodeRunStorage {
-  constructor(private readonly sql: PgStoreClient) {}
+  constructor(
+    private readonly sql: PgStoreClient,
+    private readonly executions: ExecutionStorage
+  ) {}
 
   async create(input: CreateWorkflowAgentNodeRunInput): Promise<WorkflowAgentNodeRunRecord> {
+    const [parent] = await this.sql<{ execution_id: string }[]>`
+      SELECT workflow_runs.execution_id
+      FROM workflow_node_runs
+      JOIN workflow_runs
+        ON workflow_runs.project_id = workflow_node_runs.project_id
+        AND workflow_runs.id = workflow_node_runs.workflow_run_id
+      WHERE workflow_node_runs.project_id = ${input.projectId}
+        AND workflow_node_runs.id = ${input.nodeRunId}
+        AND workflow_node_runs.node_type = ${"agent"}
+    `
+    if (!parent) {
+      throw new WorkflowRunError(
+        `[SixbPg] Agent workflow node run '${input.nodeRunId}' was not found.`
+      )
+    }
+    await assertWorkflowAgentNodeRunExecution({
+      executions: this.executions,
+      projectId: input.projectId,
+      executionId: input.executionId,
+      nodeRunId: input.nodeRunId,
+      agentId: input.agentId,
+      workflowExecutionId: parent.execution_id,
+    })
+
     try {
       const [row] = await this.sql<WorkflowAgentNodeRunDatabaseRow[]>`
         INSERT INTO workflow_agent_node_runs (
-          project_id, node_run_id, agent_id, status, prompt, attempt, created_at
+          project_id, node_run_id, execution_id, agent_id, status, prompt, attempt, created_at
         ) VALUES (
-          ${input.projectId}, ${input.nodeRunId}, ${input.agentId}, ${"queued"},
-          ${input.prompt}, ${0}, ${input.createdAt ?? new Date()}
+          ${input.projectId}, ${input.nodeRunId}, ${input.executionId}, ${input.agentId},
+          ${"queued"}, ${input.prompt}, ${0}, ${input.createdAt ?? new Date()}
         ) RETURNING *
       `
       return rowToWorkflowAgentNodeRunRecord(row)
@@ -445,8 +437,6 @@ export class PgWorkflowAgentNodeRunStorage implements WorkflowAgentNodeRunStorag
       const [updated] = await tx<WorkflowAgentNodeRunDatabaseRow[]>`
         UPDATE workflow_agent_node_runs SET
           status = ${"running"},
-          execution_principal_type = ${input.executionPrincipal?.type ?? null},
-          execution_principal_id = ${input.executionPrincipal?.id ?? null},
           model_id = ${input.modelId ?? null},
           attempt = ${1},
           execution_token = ${input.execution.token},
@@ -504,10 +494,9 @@ export class PgWorkflowAgentNodeRunStorage implements WorkflowAgentNodeRunStorag
           status = ${input.status},
           model_id = COALESCE(${input.modelId ?? null}, model_id),
           finish_reason = ${input.finishReason ?? null},
-          usage = ${input.usage ? JSON.stringify(input.usage) : null}::text::jsonb,
           trace = ${input.trace ? JSON.stringify(input.trace) : null}::text::jsonb,
           diagnostics = ${input.diagnostics ? JSON.stringify(input.diagnostics) : null}::text::jsonb,
-          error = ${input.status === "succeeded" ? null : (input.error ?? null)},
+          error = ${input.status === "succeeded" || input.error === undefined ? null : serializeSixbFailure(input.error, AGENT_RUN_FAILURE_CODES)}::text::jsonb,
           execution_token = ${null},
           execution_queue_lease_expires_at = ${null},
           completed_at = ${input.completedAt ?? new Date()}
@@ -538,7 +527,7 @@ export class PgWorkflowAgentNodeRunStorage implements WorkflowAgentNodeRunStorag
       const [updated] = await tx<WorkflowAgentNodeRunDatabaseRow[]>`
         UPDATE workflow_agent_node_runs SET
           status = ${"cancelled"},
-          error = ${input.error ?? null},
+          error = ${input.error === undefined ? null : serializeSixbFailure(input.error, AGENT_RUN_FAILURE_CODES)}::text::jsonb,
           execution_token = ${null},
           execution_queue_lease_expires_at = ${null},
           completed_at = ${input.completedAt ?? new Date()}
@@ -698,7 +687,7 @@ export class PgWorkflowNodeRunStorage implements WorkflowNodeRunStorage {
                 status = ${input.status},
                 finished_at = ${input.finishedAt ?? new Date()},
                 output = ${null},
-                error = ${input.error ?? null}
+                error = ${input.error === undefined ? null : serializeSixbFailure(input.error, WORKFLOW_RUN_FAILURE_CODES)}::text::jsonb
               WHERE project_id = ${input.projectId} AND id = ${input.id}
               RETURNING *
             `
@@ -824,6 +813,7 @@ function rowToWorkflowRunRecord(row: WorkflowRunDatabaseRow): WorkflowRunRecord
   return {
     id: row.id,
     projectId: row.project_id,
+    executionId: row.execution_id,
     workflowId: row.workflow_id,
     status: row.status,
     input: parseRecord(row.input),
@@ -831,12 +821,8 @@ function rowToWorkflowRunRecord(row: WorkflowRunDatabaseRow): WorkflowRunRecord
     queuedAt: row.queued_at ? new Date(row.queued_at) : undefined,
     startedAt: new Date(row.started_at),
     finishedAt: row.finished_at ? new Date(row.finished_at) : undefined,
-    error: row.error ?? undefined,
-    source: parseSource(row.source),
-    requestedByPrincipal: {
-      type: row.requested_by_principal_type,
-      id: row.requested_by_principal_id,
-    },
+    error: row.error === null ? undefined : parseSixbFailure(row.error, WORKFLOW_RUN_FAILURE_CODES),
+    requesterGroupIds: parseJson(row.requester_group_ids),
     attempt: Number(row.attempt),
     ...(row.execution_token && row.execution_queue_lease_expires_at
       ? {
@@ -849,14 +835,6 @@ function rowToWorkflowRunRecord(row: WorkflowRunDatabaseRow): WorkflowRunRecord
   }
 }
 
-function parseSource(value: WorkflowRunSource | string | null): WorkflowRunSource | undefined {
-  if (value === null || value === undefined) {
-    return undefined
-  }
-
-  return typeof value === "string" ? (JSON.parse(value) as WorkflowRunSource) : value
-}
-
 function rowToWorkflowNodeRunRecord(row: WorkflowNodeRunDatabaseRow): WorkflowNodeRunRecord {
   return {
     id: row.id,
@@ -872,7 +850,7 @@ function rowToWorkflowNodeRunRecord(row: WorkflowNodeRunDatabaseRow): WorkflowNo
     startedAt: new Date(row.started_at),
     finishedAt: row.finished_at ? new Date(row.finished_at) : undefined,
     output: row.output ? parseRecord(row.output) : undefined,
-    error: row.error ?? undefined,
+    error: row.error === null ? undefined : parseSixbFailure(row.error, WORKFLOW_RUN_FAILURE_CODES),
   }
 }
 
@@ -896,6 +874,7 @@ function canFinishWorkflowRun(
 interface WorkflowRunDatabaseRow {
   project_id: string
   id: string
+  execution_id: string
   workflow_id: string
   status: WorkflowRunRecord["status"]
   input: WorkflowIOSnapshot | string
@@ -903,10 +882,8 @@ interface WorkflowRunDatabaseRow {
   queued_at: Date | string | null
   started_at: Date | string
   finished_at: Date | string | null
-  error: string | null
-  source: WorkflowRunSource | string | null
-  requested_by_principal_type: Principal["type"]
-  requested_by_principal_id: string
+  error: JsonValue | null
+  requester_group_ids: string[] | string
   attempt: number | string
   execution_token: string | null
   execution_queue_lease_expires_at: Date | string | null
@@ -926,23 +903,21 @@ interface WorkflowNodeRunDatabaseRow {
   started_at: Date | string
   finished_at: Date | string | null
   output: WorkflowIOSnapshot | string | null
-  error: string | null
+  error: JsonValue | null
 }
 
 interface WorkflowAgentNodeRunDatabaseRow {
   project_id: string
   node_run_id: string
+  execution_id: string
   agent_id: string
   status: WorkflowAgentNodeRunRecord["status"]
   prompt: string
-  execution_principal_type: "serviceAccount" | null
-  execution_principal_id: string | null
   model_id: string | null
   finish_reason: WorkflowAgentNodeRunRecord["finishReason"] | null
-  usage: WorkflowAgentNodeRunRecord["usage"] | string | null
   trace: WorkflowAgentNodeRunRecord["trace"] | string | null
   diagnostics: WorkflowAgentNodeRunRecord["diagnostics"] | string | null
-  error: string | null
+  error: JsonValue | null
   attempt: number | string
   execution_token: string | null
   execution_queue_lease_expires_at: Date | string | null
@@ -992,23 +967,15 @@ function rowToWorkflowAgentNodeRunRecord(
   return {
     projectId: row.project_id,
     nodeRunId: row.node_run_id,
+    executionId: row.execution_id,
     agentId: row.agent_id,
     status: row.status,
     prompt: row.prompt,
-    ...(row.execution_principal_type && row.execution_principal_id
-      ? {
-          executionPrincipal: {
-            type: row.execution_principal_type,
-            id: row.execution_principal_id,
-          },
-        }
-      : {}),
     ...(row.model_id ? { modelId: row.model_id } : {}),
     ...(row.finish_reason ? { finishReason: row.finish_reason } : {}),
-    ...(row.usage ? { usage: parseJson(row.usage) } : {}),
     ...(row.trace ? { trace: parseJson(row.trace) } : {}),
     ...(row.diagnostics ? { diagnostics: parseJson(row.diagnostics) } : {}),
-    ...(row.error ? { error: row.error } : {}),
+    ...(row.error ? { error: parseSixbFailure(row.error, AGENT_RUN_FAILURE_CODES) } : {}),
     attempt: Number(row.attempt),
     ...(row.execution_token && row.execution_queue_lease_expires_at
       ? {

package/src/run-list-query.ts

@@ -12,7 +12,10 @@ type PgRunListTable =
 
 /** Column projection — `sync_runs` needs the derived `checkpoint_present` flag. */
 type PgRunListSelectList = "*" | "*, checkpoint IS NOT NULL AS checkpoint_present"
-type StartedAtExpression = "started_at" | "COALESCE(started_at, created_at)"
+type StartedAtExpression =
+  | "started_at"
+  | "COALESCE(started_at, created_at)"
+  | "COALESCE(started_at, queued_at)"
 
 export function hasEmptyStatuses(input: { readonly statuses?: readonly unknown[] }): boolean {
   return input.statuses !== undefined && input.statuses.length === 0
@@ -78,9 +81,11 @@ export async function queryRunList<TRow>(input: {
   const orderColumn =
     input.tableName === "agent_runs"
       ? "COALESCE(started_at, created_at)"
-      : input.tableName === "workflow_agent_node_runs"
-        ? "created_at"
-        : "started_at"
+      : input.tableName === "sync_runs" || input.tableName === "pipeline_runs"
+        ? "COALESCE(started_at, queued_at)"
+        : input.tableName === "workflow_agent_node_runs"
+          ? "created_at"
+          : "started_at"
   const idColumn = input.tableName === "workflow_agent_node_runs" ? "node_run_id" : "id"
   let query = `
     SELECT ${input.selectList ?? "*"} FROM ${input.tableName}

package/src/testing.ts

@@ -0,0 +1,18 @@
+export interface PostgresStorageTestingAdapter {
+  advanceConnectorConnectionTime(durationMs: number): void
+}
+
+const adapters = new WeakMap<object, PostgresStorageTestingAdapter>()
+
+export function registerPostgresStorageTestingAdapter(
+  storage: object,
+  advanceConnectorConnectionTime: (durationMs: number) => void
+): void {
+  adapters.set(storage, { advanceConnectorConnectionTime })
+}
+
+export function getPostgresStorageTestingAdapter(storage: object): PostgresStorageTestingAdapter {
+  const adapter = adapters.get(storage)
+  if (!adapter) throw new Error("[SixbPg] Storage testing adapter is unavailable.")
+  return adapter
+}

package/dist/pg-webhook-delivery-storage.d.ts

@@ -1,17 +0,0 @@
-import type { WebhookDeliveryClaimRecord, WebhookDeliveryKey, WebhookDeliveryRecord, WebhookDeliveryStorage } from "@sixb/core/storage";
-import { type PgStoreClient } from "./transactions";
-export declare class PgWebhookDeliveryStorage implements WebhookDeliveryStorage {
-    private readonly sql;
-    constructor(sql: PgStoreClient);
-    claim(input: WebhookDeliveryKey & {
-        receivedAt: string;
-    }): Promise<WebhookDeliveryClaimRecord>;
-    complete(input: WebhookDeliveryKey & {
-        completedAt: string;
-    }): Promise<WebhookDeliveryRecord>;
-    fail(input: WebhookDeliveryKey & {
-        failedAt: string;
-        error: string;
-    }): Promise<WebhookDeliveryRecord>;
-}
-//# sourceMappingURL=pg-webhook-delivery-storage.d.ts.map

package/dist/pg-webhook-delivery-storage.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"pg-webhook-delivery-storage.d.ts","sourceRoot":"","sources":["../src/pg-webhook-delivery-storage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,0BAA0B,EAE1B,kBAAkB,EAClB,qBAAqB,EACrB,sBAAsB,EACvB,MAAM,oBAAoB,CAAA;AAC3B,OAAO,EAAE,KAAK,aAAa,EAAoB,MAAM,gBAAgB,CAAA;AAErE,qBAAa,wBAAyB,YAAW,sBAAsB;IACzD,OAAO,CAAC,QAAQ,CAAC,GAAG;gBAAH,GAAG,EAAE,aAAa;IAEzC,KAAK,CACT,KAAK,EAAE,kBAAkB,GAAG;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GACjD,OAAO,CAAC,0BAA0B,CAAC;IAyEhC,QAAQ,CACZ,KAAK,EAAE,kBAAkB,GAAG;QAAE,WAAW,EAAE,MAAM,CAAA;KAAE,GAClD,OAAO,CAAC,qBAAqB,CAAC;IAuB3B,IAAI,CACR,KAAK,EAAE,kBAAkB,GAAG;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,GAC9D,OAAO,CAAC,qBAAqB,CAAC;CAqBlC"}

package/src/pg-webhook-delivery-storage.ts

@@ -1,180 +0,0 @@
-import type {
-  WebhookDeliveryClaimRecord,
-  WebhookDeliveryClaimResult,
-  WebhookDeliveryKey,
-  WebhookDeliveryRecord,
-  WebhookDeliveryStorage,
-} from "@sixb/core/storage"
-import { type PgStoreClient, runPgTransaction } from "./transactions"
-
-export class PgWebhookDeliveryStorage implements WebhookDeliveryStorage {
-  constructor(private readonly sql: PgStoreClient) {}
-
-  async claim(
-    input: WebhookDeliveryKey & { receivedAt: string }
-  ): Promise<WebhookDeliveryClaimRecord> {
-    return runPgTransaction(this.sql, async (tx) => {
-      // Atomic first claim: only one transaction can insert this scoped delivery key.
-      const [inserted] = await tx<WebhookDeliveryRow[]>`
-        INSERT INTO webhook_deliveries (
-          project_id,
-          connector_id,
-          webhook_id,
-          idempotency_key,
-          status,
-          received_at
-        ) VALUES (
-          ${input.projectId},
-          ${input.connectorId},
-          ${input.webhookId},
-          ${input.idempotencyKey},
-          ${"in_progress"},
-          ${input.receivedAt}
-        )
-        ON CONFLICT (project_id, connector_id, webhook_id, idempotency_key) DO NOTHING
-        RETURNING *
-      `
-
-      if (inserted) {
-        return toWebhookDeliveryClaimRecord(inserted, "claimed")
-      }
-
-      const [existing] = await tx<WebhookDeliveryRow[]>`
-        SELECT * FROM webhook_deliveries
-        WHERE project_id = ${input.projectId}
-          AND connector_id = ${input.connectorId}
-          AND webhook_id = ${input.webhookId}
-          AND idempotency_key = ${input.idempotencyKey}
-        FOR UPDATE
-      `
-
-      if (existing?.status === "completed") {
-        return toWebhookDeliveryClaimRecord(existing, "duplicate")
-      }
-
-      if (existing?.status === "in_progress") {
-        return toWebhookDeliveryClaimRecord(existing, "in_progress")
-      }
-
-      if (existing?.status !== "failed") {
-        throw new Error("[SixbPg] Webhook delivery row disappeared during claim.")
-      }
-
-      // Failed deliveries are intentionally claimable so provider retries can run again.
-      const [updated] = await tx<WebhookDeliveryRow[]>`
-        UPDATE webhook_deliveries
-        SET status = ${"in_progress"},
-            received_at = ${input.receivedAt},
-            completed_at = ${null},
-            failed_at = ${null},
-            error = ${null}
-        WHERE project_id = ${input.projectId}
-          AND connector_id = ${input.connectorId}
-          AND webhook_id = ${input.webhookId}
-          AND idempotency_key = ${input.idempotencyKey}
-        RETURNING *
-      `
-
-      if (!updated) {
-        throw new Error(
-          `[SixbPg] Missing webhook delivery row after claim for ${serializeKey(input)}.`
-        )
-      }
-
-      return toWebhookDeliveryClaimRecord(updated, "claimed")
-    })
-  }
-
-  async complete(
-    input: WebhookDeliveryKey & { completedAt: string }
-  ): Promise<WebhookDeliveryRecord> {
-    const [updated] = await this.sql<WebhookDeliveryRow[]>`
-      UPDATE webhook_deliveries
-      SET status = ${"completed"},
-          completed_at = ${input.completedAt},
-          failed_at = ${null},
-          error = ${null}
-      WHERE project_id = ${input.projectId}
-        AND connector_id = ${input.connectorId}
-        AND webhook_id = ${input.webhookId}
-        AND idempotency_key = ${input.idempotencyKey}
-      RETURNING *
-    `
-
-    if (!updated) {
-      throw new Error(
-        `[SixbPg] Missing webhook delivery row after complete for ${serializeKey(input)}.`
-      )
-    }
-
-    return toWebhookDeliveryRecord(updated)
-  }
-
-  async fail(
-    input: WebhookDeliveryKey & { failedAt: string; error: string }
-  ): Promise<WebhookDeliveryRecord> {
-    const [updated] = await this.sql<WebhookDeliveryRow[]>`
-      UPDATE webhook_deliveries
-      SET status = ${"failed"},
-          failed_at = ${input.failedAt},
-          error = ${input.error}
-      WHERE project_id = ${input.projectId}
-        AND connector_id = ${input.connectorId}
-        AND webhook_id = ${input.webhookId}
-        AND idempotency_key = ${input.idempotencyKey}
-      RETURNING *
-    `
-
-    if (!updated) {
-      throw new Error(
-        `[SixbPg] Missing webhook delivery row after fail for ${serializeKey(input)}.`
-      )
-    }
-
-    return toWebhookDeliveryRecord(updated)
-  }
-}
-
-function toWebhookDeliveryClaimRecord(
-  row: WebhookDeliveryRow,
-  claimResult: WebhookDeliveryClaimResult
-): WebhookDeliveryClaimRecord {
-  return {
-    ...toWebhookDeliveryRecord(row),
-    claimResult,
-  }
-}
-
-function toWebhookDeliveryRecord(row: WebhookDeliveryRow): WebhookDeliveryRecord {
-  return {
-    projectId: row.project_id,
-    connectorId: row.connector_id,
-    webhookId: row.webhook_id,
-    idempotencyKey: row.idempotency_key,
-    status: row.status,
-    receivedAt: toIsoString(row.received_at),
-    completedAt: row.completed_at ? toIsoString(row.completed_at) : undefined,
-    failedAt: row.failed_at ? toIsoString(row.failed_at) : undefined,
-    error: row.error ?? undefined,
-  }
-}
-
-function toIsoString(value: Date | string): string {
-  return value instanceof Date ? value.toISOString() : new Date(value).toISOString()
-}
-
-function serializeKey(key: WebhookDeliveryKey): string {
-  return JSON.stringify([key.projectId, key.connectorId, key.webhookId, key.idempotencyKey])
-}
-
-interface WebhookDeliveryRow {
-  readonly project_id: string
-  readonly connector_id: string
-  readonly webhook_id: string
-  readonly idempotency_key: string
-  readonly status: WebhookDeliveryRecord["status"]
-  readonly received_at: Date | string
-  readonly completed_at: Date | string | null
-  readonly failed_at: Date | string | null
-  readonly error: string | null
-}