@gencow/core 0.1.28 → 0.1.30

package/src/workflows-api.ts

@@ -1,5 +1,6 @@
 import { sql } from "drizzle-orm";
-import { mutation, query } from "./reactive.js";
+import { query } from "./reactive-query.js";
+import { mutation } from "./reactive-mutation.js";
 import {
   createWorkflowRealtimeToken,
   deserializeWorkflowValue,
@@ -7,6 +8,7 @@ import {
   getWorkflowRealtimeKey,
   serializeWorkflowValue,
 } from "./workflow.js";
+import { workflowJsonb } from "./workflow-json.js";
 import { GencowValidationError, v } from "./v.js";
 import type {
   WorkflowDerivedStatus,
@@ -31,6 +33,7 @@ type WorkflowRow = {
   current_step: string | null;
   result: unknown;
   error: string | null;
+  error_code: string | null;
   retry_count: number;
   max_retries: number;
   max_duration_ms: number;
@@ -101,6 +104,7 @@ function mapWorkflowSummary(row: WorkflowRow): WorkflowSummary {
     derivedStatus: deriveWorkflowStatus(row.status, row.current_step),
     currentStep: row.current_step,
     error: row.error,
+    errorCode: row.error_code,
     retryCount: row.retry_count,
     maxRetries: row.max_retries,
     maxDurationMs: Number(row.max_duration_ms),
@@ -198,10 +202,77 @@ async function loadWorkflowSignalTarget(
         FROM _gencow_workflows
         WHERE id = ${workflowId}
         LIMIT 1
-    `);
+  `);
   return rowsFromResult<WorkflowSignalTargetRow>(result)[0] ?? null;
 }
 
+function isMissingWorkflowV2SignalSchemaError(error: unknown): boolean {
+  const code =
+    error && typeof error === "object" && "code" in error ? String((error as { code?: unknown }).code) : "";
+  const cause =
+    error && typeof error === "object" && "cause" in error ? (error as { cause?: unknown }).cause : null;
+  const causeCode =
+    cause && typeof cause === "object" && "code" in cause ? String((cause as { code?: unknown }).code) : "";
+  const message = error instanceof Error ? error.message : String(error);
+  return (
+    code === "42P01" ||
+    code === "42703" ||
+    causeCode === "42P01" ||
+    causeCode === "42703" ||
+    ((code === "23503" || causeCode === "23503") && message.includes("_gencow_workflow_signals_v2")) ||
+    message.includes('relation "_gencow_workflow_signals_v2" does not exist') ||
+    message.includes("relation _gencow_workflow_signals_v2 does not exist") ||
+    message.includes('relation "_gencow_workflow_runs_v2" does not exist') ||
+    message.includes("relation _gencow_workflow_runs_v2 does not exist")
+  );
+}
+
+async function tryRecordWorkflowV2Signal(options: {
+  db: WorkflowDbLike;
+  workflowId: string;
+  event: string;
+  payload: unknown;
+}): Promise<boolean> {
+  try {
+    await options.db.execute(sql`
+      WITH inserted AS (
+        INSERT INTO _gencow_workflow_signals_v2 (
+          id,
+          run_id,
+          event_name,
+          payload_json,
+          idempotency_key
+        )
+        VALUES (
+          ${crypto.randomUUID()},
+          ${options.workflowId},
+          ${options.event},
+          ${workflowJsonb(options.payload)},
+          ${crypto.randomUUID()}
+        )
+        RETURNING run_id
+      )
+      UPDATE _gencow_workflow_runs_v2 run
+      SET
+        status = 'queued',
+        runnable_at = NOW(),
+        lease_owner = NULL,
+        lease_expires_at = NULL,
+        heartbeat_at = NULL,
+        updated_at = NOW()
+      FROM inserted
+      WHERE run.id = inserted.run_id
+        AND run.status = 'waiting'
+        AND run.completed_at IS NULL
+        AND run.cancel_requested_at IS NULL
+    `);
+    return true;
+  } catch (error) {
+    if (isMissingWorkflowV2SignalSchemaError(error)) return false;
+    throw error;
+  }
+}
+
 export async function loadWorkflowSnapshot(
   db: WorkflowDbLike,
   workflowId: string,
@@ -219,6 +290,7 @@ export async function loadWorkflowSnapshot(
             current_step,
             result,
             error,
+            error_code,
             retry_count,
             max_retries,
             max_duration_ms,
@@ -326,9 +398,15 @@ export function registerWorkflowsApi(): void {
                     ${crypto.randomUUID()},
                     ${workflow.id},
                     ${normalizedEvent},
-                    ${JSON.stringify(persistedPayload)}::jsonb
+                  ${workflowJsonb(persistedPayload)}
                 )
             `);
+      await tryRecordWorkflowV2Signal({
+        db: ctx.unsafeDb,
+        workflowId: workflow.id,
+        event: normalizedEvent,
+        payload: persistedPayload,
+      });
 
       let scheduledJobId: string | null = null;
       if (workflow.status === "pending" && workflow.current_step?.startsWith("wait:")) {
@@ -372,6 +450,7 @@ export function registerWorkflowsApi(): void {
                         current_step,
                         result,
                         error,
+                        error_code,
                         retry_count,
                         max_retries,
                         max_duration_ms,
@@ -393,6 +472,7 @@ export function registerWorkflowsApi(): void {
                         current_step,
                         result,
                         error,
+                        error_code,
                         retry_count,
                         max_retries,
                         max_duration_ms,