gencow 0.1.144 → 0.1.146

package/core/index.js

@@ -1502,7 +1502,17 @@ var DEFAULT_GROUNDING_BUDGET = {
 };
 
 // ../core/src/rag-schema.ts
-import { bigint, boolean, doublePrecision, integer, jsonb, pgTable, text, timestamp, vector } from "drizzle-orm/pg-core";
+import {
+  bigint,
+  boolean,
+  doublePrecision,
+  integer,
+  jsonb,
+  pgTable,
+  text,
+  timestamp,
+  vector
+} from "drizzle-orm/pg-core";
 function buildScopeColumns() {
   return {
     id: text("id").primaryKey(),
@@ -2105,7 +2115,7 @@ function createScheduler(options) {
   }
   return {
     runAfter(ms, action, args, scheduleOpts) {
-      const id = generateId();
+      const id = scheduleOpts?.id ?? generateId();
       const jobEntry = {
         id,
         action,
@@ -2207,10 +2217,17 @@ function createScheduler(options) {
 }
 
 // ../core/src/workflow.ts
-import { sql as sql2 } from "drizzle-orm";
+import { sql as sql3 } from "drizzle-orm";
 
-// ../core/src/workflows-api.ts
+// ../core/src/workflow-json.ts
 import { sql } from "drizzle-orm";
+function workflowJsonb(value) {
+  const json = JSON.stringify(value ?? {});
+  return sql`${json}::text::jsonb`;
+}
+
+// ../core/src/workflows-api.ts
+import { sql as sql2 } from "drizzle-orm";
 
 // ../core/src/v.ts
 var GencowValidationError = class extends Error {
@@ -2419,7 +2436,7 @@ function toWorkflowStatusFilter(status) {
 async function ensureWorkflowRealtimeToken(db, workflowId, currentToken) {
   if (currentToken && currentToken.trim() !== "") return currentToken;
   const nextToken = createWorkflowRealtimeToken();
-  const updateResult = await db.execute(sql`
+  const updateResult = await db.execute(sql2`
         UPDATE _gencow_workflows
         SET realtime_token = ${nextToken}
         WHERE id = ${workflowId}
@@ -2428,7 +2445,7 @@ async function ensureWorkflowRealtimeToken(db, workflowId, currentToken) {
     `);
   const updatedToken = rowsFromResult(updateResult)[0]?.realtime_token ?? null;
   if (updatedToken && updatedToken.trim() !== "") return updatedToken;
-  const rereadResult = await db.execute(sql`
+  const rereadResult = await db.execute(sql2`
         SELECT realtime_token
         FROM _gencow_workflows
         WHERE id = ${workflowId}
@@ -2438,7 +2455,7 @@ async function ensureWorkflowRealtimeToken(db, workflowId, currentToken) {
   return rereadToken && rereadToken.trim() !== "" ? rereadToken : null;
 }
 async function loadWorkflowSignalTarget(db, workflowId) {
-  const result = await db.execute(sql`
+  const result = await db.execute(sql2`
         SELECT
             id,
             name,
@@ -2460,7 +2477,7 @@ function isMissingWorkflowV2SignalSchemaError(error) {
 }
 async function tryRecordWorkflowV2Signal(options) {
   try {
-    await options.db.execute(sql`
+    await options.db.execute(sql2`
       WITH inserted AS (
         INSERT INTO _gencow_workflow_signals_v2 (
           id,
@@ -2473,7 +2490,7 @@ async function tryRecordWorkflowV2Signal(options) {
           ${crypto.randomUUID()},
           ${options.workflowId},
           ${options.event},
-          ${JSON.stringify(options.payload)}::jsonb,
+          ${workflowJsonb(options.payload)},
           ${crypto.randomUUID()}
         )
         RETURNING run_id
@@ -2499,7 +2516,7 @@ async function tryRecordWorkflowV2Signal(options) {
   }
 }
 async function loadWorkflowSnapshot(db, workflowId, options) {
-  const workflowResult = await db.execute(sql`
+  const workflowResult = await db.execute(sql2`
         SELECT
             id,
             name,
@@ -2529,7 +2546,7 @@ async function loadWorkflowSnapshot(db, workflowId, options) {
   }
   const realtimeToken = await ensureWorkflowRealtimeToken(db, workflowId, row.realtime_token);
   if (!realtimeToken) return null;
-  const stepsResult = await db.execute(sql`
+  const stepsResult = await db.execute(sql2`
         SELECT
             step_name,
             status,
@@ -2594,7 +2611,7 @@ function registerWorkflowsApi() {
         };
       }
       const persistedPayload = serializeWorkflowValue(args.payload);
-      await ctx.unsafeDb.execute(sql`
+      await ctx.unsafeDb.execute(sql2`
                 INSERT INTO _gencow_workflow_events (
                     id,
                     workflow_id,
@@ -2605,7 +2622,7 @@ function registerWorkflowsApi() {
                     ${crypto.randomUUID()},
                     ${workflow2.id},
                     ${normalizedEvent},
-                  ${JSON.stringify(persistedPayload)}::jsonb
+                  ${workflowJsonb(persistedPayload)}
                 )
             `);
       await tryRecordWorkflowV2Signal({
@@ -2642,7 +2659,7 @@ function registerWorkflowsApi() {
       const limit = normalizeListLimit(args.limit);
       const requestedStatus = normalizeDerivedStatus(args.status);
       const status = toWorkflowStatusFilter(requestedStatus);
-      const result = status == null ? await ctx.unsafeDb.execute(sql`
+      const result = status == null ? await ctx.unsafeDb.execute(sql2`
                     SELECT
                         id,
                         name,
@@ -2663,7 +2680,7 @@ function registerWorkflowsApi() {
                     WHERE user_id = ${userId}
                     ORDER BY started_at DESC
                     LIMIT ${limit}
-                `) : await ctx.unsafeDb.execute(sql`
+                `) : await ctx.unsafeDb.execute(sql2`
                     SELECT
                         id,
                         name,
@@ -2769,61 +2786,60 @@ function isMissingWorkflowV2SchemaError(error) {
   const message = error instanceof Error ? error.message : String(error);
   return code === "42P01" || code === "42703" || causeCode === "42P01" || causeCode === "42703" || message.includes('relation "_gencow_workflow_runs_v2" does not exist') || message.includes("relation _gencow_workflow_runs_v2 does not exist") || message.includes('relation "_gencow_workflow_outbox_v2" does not exist') || message.includes("relation _gencow_workflow_outbox_v2 does not exist") || message.includes('column "max_active_duration_ms"') || message.includes('column "retry_count"') || message.includes('column "user_id"');
 }
-async function tryInsertWorkflowV2WakeOutbox(db, workflowId) {
-  try {
-    await db.execute(sql2`
-      INSERT INTO _gencow_workflow_outbox_v2 (
-        id,
-        run_id,
-        kind,
-        available_at,
-        status
-      )
-      VALUES (
-        ${`start:${workflowId}`},
-        ${workflowId},
-        'wake_run',
-        NOW(),
-        'pending'
-      )
-      ON CONFLICT (id) DO NOTHING
-    `);
-  } catch (error) {
-    if (!isMissingWorkflowV2SchemaError(error)) throw error;
-  }
-}
 async function tryInsertWorkflowV2Run(options) {
   try {
-    await options.db.execute(sql2`
-      INSERT INTO _gencow_workflow_runs_v2 (
-        id,
-        workflow_name,
-        workflow_version,
-        args_json,
-        user_id,
-        max_active_duration_ms,
-        lifecycle_deadline_at,
-        retry_count,
-        max_retries,
-        max_attempts
-      )
-      VALUES (
-        ${options.workflowId},
-        ${options.workflowName},
-        ${options.workflowVersion ?? null},
-        ${JSON.stringify(options.args)}::jsonb,
-        ${options.userId},
-        ${options.maxActiveDurationMs},
-        CASE
-          WHEN ${options.lifecycleTimeoutMs}::bigint IS NULL THEN NULL
-          ELSE NOW() + (${options.lifecycleTimeoutMs}::bigint * INTERVAL '1 millisecond')
-        END,
-        0,
-        ${options.maxRetries},
-        ${options.maxRetries + 1}
+    await options.db.execute(sql3`
+      WITH inserted_run AS (
+        INSERT INTO _gencow_workflow_runs_v2 (
+          id,
+          workflow_name,
+          workflow_version,
+          args_json,
+          user_id,
+          max_active_duration_ms,
+          lifecycle_deadline_at,
+          retry_count,
+          max_retries,
+          max_attempts
+        )
+        VALUES (
+          ${options.workflowId},
+          ${options.workflowName},
+          ${options.workflowVersion ?? null},
+          ${workflowJsonb(options.args)},
+          ${options.userId},
+          ${options.maxActiveDurationMs},
+          CASE
+            WHEN ${options.lifecycleTimeoutMs}::bigint IS NULL THEN NULL
+            ELSE NOW() + (${options.lifecycleTimeoutMs}::bigint * INTERVAL '1 millisecond')
+          END,
+          0,
+          ${options.maxRetries},
+          ${options.maxRetries + 1}
+        )
+        RETURNING id
+      ),
+      inserted_outbox AS (
+        INSERT INTO _gencow_workflow_outbox_v2 (
+          id,
+          run_id,
+          kind,
+          available_at,
+          status
+        )
+        SELECT
+          'start:' || inserted_run.id,
+          inserted_run.id,
+          'wake_run',
+          NOW(),
+          'pending'
+        FROM inserted_run
+        ON CONFLICT (id) DO NOTHING
+        RETURNING id
       )
+      SELECT id
+      FROM inserted_run
     `);
-    await tryInsertWorkflowV2WakeOutbox(options.db, options.workflowId);
     return true;
   } catch (error) {
     if (isMissingWorkflowV2SchemaError(error)) return false;
@@ -2832,7 +2848,7 @@ async function tryInsertWorkflowV2Run(options) {
 }
 async function tryDeleteWorkflowV2Run(db, workflowId) {
   try {
-    await db.execute(sql2`
+    await db.execute(sql3`
       DELETE FROM _gencow_workflow_runs_v2
       WHERE id = ${workflowId}
     `);
@@ -2888,7 +2904,7 @@ function workflow(name, options) {
       const ownerId = ctx.auth.getUserIdentity()?.id ?? null;
       const persistedArgs = serializeWorkflowValue(args ?? {});
       const realtimeToken = createWorkflowRealtimeToken();
-      await ctx.unsafeDb.execute(sql2`
+      await ctx.unsafeDb.execute(sql3`
                 INSERT INTO _gencow_workflows (
                     id,
                     name,
@@ -2903,7 +2919,7 @@ function workflow(name, options) {
                 VALUES (
                     ${workflowId},
                     ${name},
-                    ${JSON.stringify(persistedArgs)}::jsonb,
+                    ${workflowJsonb(persistedArgs)},
                   ${realtimeToken},
                   'pending',
                   0,
@@ -2925,7 +2941,7 @@ function workflow(name, options) {
           lifecycleTimeoutMs,
           maxRetries
         });
-        const scheduledJobId = ctx.scheduler.runAfter(0, resumeAction, { workflowId });
+        const scheduledJobId = insertedWorkflowV2 ? `start:${workflowId}` : ctx.scheduler.runAfter(0, resumeAction, { workflowId });
         return {
           id: workflowId,
           name,
@@ -2933,7 +2949,7 @@ function workflow(name, options) {
           scheduledJobId
         };
       } catch (error) {
-        await ctx.unsafeDb.execute(sql2`
+        await ctx.unsafeDb.execute(sql3`
                     DELETE FROM _gencow_workflows
                     WHERE id = ${workflowId}
                 `);
@@ -3019,12 +3035,19 @@ var RESERVED_TENANT_RUNTIME_ENV_KEYS = /* @__PURE__ */ new Set([
   "GENCOW_SHUTDOWN_MARKER_PATH",
   "GENCOW_SKIP_MIGRATION",
   "GENCOW_DB_MAX_CONNECTIONS",
+  "GENCOW_DB_IDLE_TIMEOUT_SECONDS",
+  "GENCOW_DB_CONNECTION_TIMEOUT_SECONDS",
   "GENCOW_MEMORY_MB",
   "BUN_JSC_forceRAMSize",
   "MIMALLOC_PURGE_DELAY",
   "NODE_PATH"
 ]);
-var RESERVED_TENANT_RUNTIME_ENV_PREFIXES = ["__GENCOW_", "GENCOW_DOCUMENT_", "GENCOW_TEMPLATE_", "GENCOW_WARM_"];
+var RESERVED_TENANT_RUNTIME_ENV_PREFIXES = [
+  "__GENCOW_",
+  "GENCOW_DOCUMENT_",
+  "GENCOW_TEMPLATE_",
+  "GENCOW_WARM_"
+];
 function isReservedTenantRuntimeEnvKey(key) {
   const normalized = key.trim();
   return RESERVED_TENANT_RUNTIME_ENV_KEYS.has(normalized) || RESERVED_TENANT_RUNTIME_ENV_PREFIXES.some((prefix) => normalized.startsWith(prefix));
@@ -3088,7 +3111,7 @@ function intervalToPattern(options) {
 
 // ../core/src/auth-config.ts
 function defineAuth(config) {
-  return config;
+  return { ...config, provider: config.provider ?? "better-auth" };
 }
 
 // ../core/src/config.ts
@@ -3097,7 +3120,7 @@ function defineConfig(config) {
 }
 
 // ../core/src/rls.ts
-import { sql as sql3 } from "drizzle-orm";
+import { sql as sql4 } from "drizzle-orm";
 import { pgPolicy } from "drizzle-orm/pg-core";
 var _ownerRlsRegistry = /* @__PURE__ */ new WeakMap();
 function getOwnerRlsMeta(table) {
@@ -3113,13 +3136,13 @@ function ownerRls(userIdColumn, options) {
       "[ownerRls] userIdColumn must have a .name property. Ensure you pass a valid Drizzle column reference (e.g. t.userId)."
     );
   }
-  const isOwner = sql3`${userIdColumn} = current_setting('app.current_user_id', true)`;
+  const isOwner = sql4`${userIdColumn} = current_setting('app.current_user_id', true)`;
   const meta = {
     columnName: colName,
     readPublic: options?.read === "public"
   };
   const policies = [
-    pgPolicy("rls-select", { for: "select", using: options?.read === "public" ? sql3`true` : isOwner }),
+    pgPolicy("rls-select", { for: "select", using: options?.read === "public" ? sql4`true` : isOwner }),
     pgPolicy("rls-insert", { for: "insert", withCheck: isOwner }),
     pgPolicy("rls-update", { for: "update", using: isOwner, withCheck: isOwner }),
     pgPolicy("rls-delete", { for: "delete", using: isOwner })
@@ -3129,7 +3152,7 @@ function ownerRls(userIdColumn, options) {
 
 // ../core/src/rls-db.ts
 import { AsyncLocalStorage } from "node:async_hooks";
-import { sql as sql4 } from "drizzle-orm";
+import { sql as sql5 } from "drizzle-orm";
 var gucNameRe = /^app\.[a-z][a-z0-9_]*(?:\.[a-z][a-z0-9_]*)*$/;
 var RESERVED_VARS_KEYS = /* @__PURE__ */ new Set(["app.current_user_id", "app.current_user_role", "app.tenant_id"]);
 function assertSafeGucName(key) {
@@ -3189,7 +3212,7 @@ async function applyRlsSessionVars(client, rls) {
   await forEachSetConfig(rls, (name, value) => execSetConfig(client, name, value));
 }
 async function injectRlsVarsOnTx(tx, rls) {
-  await forEachSetConfig(rls, (name, value) => tx.execute(sql4`SELECT set_config(${name}, ${value}, true)`));
+  await forEachSetConfig(rls, (name, value) => tx.execute(sql5`SELECT set_config(${name}, ${value}, true)`));
 }
 async function withRlsLeasedConnection(leased, rls, fn) {
   try {