@opengeni/db 1.0.2 → 1.2.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (86) hide show
  1. package/dist/artifact-catalog.d.ts +12 -0
  2. package/dist/canonical-human-identities.js +3 -3
  3. package/dist/capability-reconciliation.js +2 -2
  4. package/dist/{chunk-KH3TYDGY.js → chunk-3GAR5UN3.js} +2 -2
  5. package/dist/{chunk-UEQQ2WWU.js → chunk-3TGYFFZR.js} +3 -26
  6. package/dist/chunk-3TGYFFZR.js.map +1 -0
  7. package/dist/{chunk-F3YMVRF4.js → chunk-6EIEGHPK.js} +4 -4
  8. package/dist/{chunk-OQZOGCLG.js → chunk-CONWNUSL.js} +2 -2
  9. package/dist/{chunk-7S53WLPI.js → chunk-CYP72JTQ.js} +11 -4
  10. package/dist/{chunk-7S53WLPI.js.map → chunk-CYP72JTQ.js.map} +1 -1
  11. package/dist/{chunk-LDTQGYD2.js → chunk-FWQNHNOU.js} +54 -2
  12. package/dist/chunk-FWQNHNOU.js.map +1 -0
  13. package/dist/{chunk-KEDWSSNP.js → chunk-GDN5TSZP.js} +2 -2
  14. package/dist/{chunk-PQGTWFRF.js → chunk-JICUP7ZR.js} +2 -2
  15. package/dist/{chunk-NLSG7GGA.js → chunk-JLVWBIEM.js} +3 -3
  16. package/dist/{chunk-Y77PVZZD.js → chunk-LXPFNG4L.js} +2 -2
  17. package/dist/{chunk-56MYWKQV.js → chunk-O2Y5E5M3.js} +68 -32
  18. package/dist/chunk-O2Y5E5M3.js.map +1 -0
  19. package/dist/{chunk-7POTLB24.js → chunk-QFJNEXFL.js} +2 -2
  20. package/dist/{chunk-MACELCXG.js → chunk-UCIMHNII.js} +101 -21
  21. package/dist/chunk-UCIMHNII.js.map +1 -0
  22. package/dist/{chunk-OMHBVHX7.js → chunk-VDPCIFKL.js} +5 -5
  23. package/dist/{chunk-N5QCKHYL.js → chunk-YOYILDLY.js} +87 -19
  24. package/dist/chunk-YOYILDLY.js.map +1 -0
  25. package/dist/connect-attempts.js +3 -3
  26. package/dist/credit-balances.d.ts +18 -0
  27. package/dist/credit-promotion-policy.d.ts +10 -0
  28. package/dist/editable-artifact-durable-export.js +2 -2
  29. package/dist/editable-artifacts.js +3 -3
  30. package/dist/index.d.ts +35 -10
  31. package/dist/index.js +2432 -2255
  32. package/dist/index.js.map +1 -1
  33. package/dist/managed-auth-session-sets.js +3 -3
  34. package/dist/managed-user-setup.d.ts +1 -0
  35. package/dist/mcp-operations.js +2 -2
  36. package/dist/organization-integration-policy.js +3 -3
  37. package/dist/organization-membership-lifecycle.d.ts +1 -0
  38. package/dist/persistence-errors.d.ts +9 -6
  39. package/dist/provision-roles.js +1 -1
  40. package/dist/retained-provider-commands.js +6 -6
  41. package/dist/runtime-posture.d.ts +7 -6
  42. package/dist/schema.d.ts +214 -2
  43. package/dist/schema.js +5 -1
  44. package/dist/session-background-commands.js +4 -4
  45. package/dist/session-command-output.js +5 -5
  46. package/dist/session-event-slices.js +3 -3
  47. package/dist/session-mcp-credential-rotation.js +5 -5
  48. package/dist/session-tenancy.d.ts +6 -1
  49. package/dist/session-tenancy.js +3 -5
  50. package/dist/video-generation.js +3 -3
  51. package/dist/workspace-tool-defaults.js +4 -4
  52. package/drizzle/0610_artifact_catalog_pins.sql +165 -0
  53. package/drizzle/0611_universal_session_tenancy_activation.sql +166 -0
  54. package/drizzle/0612_scheduled_model_unavailable_refusal.sql +21 -0
  55. package/drizzle/0613_model_scoped_promotional_credits.sql +238 -0
  56. package/drizzle/0614_quiescence_command_containment.sql +7 -0
  57. package/drizzle/0615_credit_promotion_policy_validation.sql +63 -0
  58. package/package.json +7 -7
  59. package/src/artifact-catalog.ts +31 -5
  60. package/src/credit-balances.ts +109 -0
  61. package/src/credit-promotion-policy.ts +12 -0
  62. package/src/database.ts +81 -42
  63. package/src/index.ts +295 -79
  64. package/src/managed-user-setup.ts +8 -0
  65. package/src/organization-membership-lifecycle.ts +8 -0
  66. package/src/persistence-errors.ts +17 -6
  67. package/src/provision-roles.ts +25 -0
  68. package/src/runtime-posture-cli.ts +0 -2
  69. package/src/runtime-posture.ts +86 -27
  70. package/src/schema.ts +57 -1
  71. package/src/session-tenancy.ts +9 -31
  72. package/src/video-generation.ts +2 -7
  73. package/dist/chunk-56MYWKQV.js.map +0 -1
  74. package/dist/chunk-LDTQGYD2.js.map +0 -1
  75. package/dist/chunk-MACELCXG.js.map +0 -1
  76. package/dist/chunk-N5QCKHYL.js.map +0 -1
  77. package/dist/chunk-UEQQ2WWU.js.map +0 -1
  78. /package/dist/{chunk-KH3TYDGY.js.map → chunk-3GAR5UN3.js.map} +0 -0
  79. /package/dist/{chunk-F3YMVRF4.js.map → chunk-6EIEGHPK.js.map} +0 -0
  80. /package/dist/{chunk-OQZOGCLG.js.map → chunk-CONWNUSL.js.map} +0 -0
  81. /package/dist/{chunk-KEDWSSNP.js.map → chunk-GDN5TSZP.js.map} +0 -0
  82. /package/dist/{chunk-PQGTWFRF.js.map → chunk-JICUP7ZR.js.map} +0 -0
  83. /package/dist/{chunk-NLSG7GGA.js.map → chunk-JLVWBIEM.js.map} +0 -0
  84. /package/dist/{chunk-Y77PVZZD.js.map → chunk-LXPFNG4L.js.map} +0 -0
  85. /package/dist/{chunk-7POTLB24.js.map → chunk-QFJNEXFL.js.map} +0 -0
  86. /package/dist/{chunk-OMHBVHX7.js.map → chunk-VDPCIFKL.js.map} +0 -0
@@ -1,7 +1,7 @@
1
1
  import {
2
2
  nestedPostgresSqlState,
3
3
  rawRows
4
- } from "./chunk-56MYWKQV.js";
4
+ } from "./chunk-O2Y5E5M3.js";
5
5
 
6
6
  // src/canonical-human-identities.ts
7
7
  import { createHash } from "crypto";
@@ -318,4 +318,4 @@ export {
318
318
  getCanonicalHumanIdentityProjection,
319
319
  applyCanonicalHumanIdentityOperation
320
320
  };
321
- //# sourceMappingURL=chunk-KEDWSSNP.js.map
321
+ //# sourceMappingURL=chunk-GDN5TSZP.js.map
@@ -6,7 +6,7 @@ import {
6
6
  rlsContextForWorkspace,
7
7
  setSubjectRlsContext,
8
8
  withRlsContext
9
- } from "./chunk-56MYWKQV.js";
9
+ } from "./chunk-O2Y5E5M3.js";
10
10
 
11
11
  // src/organization-integration-policy.ts
12
12
  import {
@@ -153,4 +153,4 @@ export {
153
153
  withOrganizationIntegrationPolicyFence,
154
154
  withOrganizationIntegrationAcquisition
155
155
  };
156
- //# sourceMappingURL=chunk-PQGTWFRF.js.map
156
+ //# sourceMappingURL=chunk-JICUP7ZR.js.map
@@ -1,11 +1,11 @@
1
1
  import {
2
2
  rawRows,
3
3
  withWorkspaceRls
4
- } from "./chunk-56MYWKQV.js";
4
+ } from "./chunk-O2Y5E5M3.js";
5
5
  import {
6
6
  sessionEvents,
7
7
  sessionTurns
8
- } from "./chunk-LDTQGYD2.js";
8
+ } from "./chunk-FWQNHNOU.js";
9
9
  import {
10
10
  LOSSLESS_JSON_STRING_PREFIX,
11
11
  fromPostgresLosslessJson,
@@ -254,4 +254,4 @@ export {
254
254
  readSessionMessageSliceInScope,
255
255
  listSessionEventSlices
256
256
  };
257
- //# sourceMappingURL=chunk-NLSG7GGA.js.map
257
+ //# sourceMappingURL=chunk-JLVWBIEM.js.map
@@ -1,7 +1,7 @@
1
1
  import {
2
2
  rawRows,
3
3
  withRlsContext
4
- } from "./chunk-56MYWKQV.js";
4
+ } from "./chunk-O2Y5E5M3.js";
5
5
 
6
6
  // src/editable-artifacts.ts
7
7
  import { createHash } from "crypto";
@@ -3592,4 +3592,4 @@ export {
3592
3592
  PostgresEditableArtifactLiveReadStore,
3593
3593
  transactionallyAuthorizeEditableArtifactActor
3594
3594
  };
3595
- //# sourceMappingURL=chunk-Y77PVZZD.js.map
3595
+ //# sourceMappingURL=chunk-LXPFNG4L.js.map
@@ -3,7 +3,7 @@ import {
3
3
  sessions,
4
4
  workspaceSessionActivityRevisions,
5
5
  workspaces
6
- } from "./chunk-LDTQGYD2.js";
6
+ } from "./chunk-FWQNHNOU.js";
7
7
  import {
8
8
  LOSSLESS_CONTENT_WRITER_APPLICATION_NAME
9
9
  } from "./chunk-P7YDM2RI.js";
@@ -163,6 +163,14 @@ function safeDatabaseErrorFacts(error) {
163
163
  }
164
164
  return facts;
165
165
  }
166
+ var DatabaseTransactionError = class extends Error {
167
+ constructor(stage, cause, original) {
168
+ super(`Database transaction ${stage} failed`, { cause });
169
+ this.stage = stage;
170
+ this.original = original;
171
+ }
172
+ name = "DatabaseTransactionError";
173
+ };
166
174
  var SessionEventPersistenceError = class extends Error {
167
175
  constructor(details, cause) {
168
176
  const label = details.code === "db_deadlock" ? "Database deadlock" : details.code === "db_serialization_failure" ? "Database serialization failure" : "Database failure";
@@ -468,43 +476,70 @@ async function restoreRlsContextSettings(db, settings) {
468
476
  throw new Error("RLS context could not be restored after a nested scope");
469
477
  }
470
478
  }
471
- async function withRlsContext(db, context, fn, transactionConfig, sessionTenancyFence = "shared") {
472
- const restoreParentScope = isTransactionHandle(db);
473
- const admission = startDatabaseTiming(
474
- restoreParentScope ? "savepoint_admission" : "transaction_admission"
475
- );
479
+ async function withDatabaseTransactionProvenance(db, fn, transactionConfig) {
480
+ let entered = false;
481
+ let callbackFailure;
476
482
  try {
477
483
  return await db.transaction(async (tx) => {
478
- admission("completed");
479
- const setup = startDatabaseTiming("rls_setup");
480
- const scoped = tx;
481
- let parentScope;
484
+ entered = true;
482
485
  try {
483
- parentScope = restoreParentScope ? await readRlsContextSettings(scoped) : null;
484
- await setRlsContext(scoped, context);
485
- if (context.workspaceId && sessionTenancyFence === "shared") {
486
- await scoped.execute(
487
- sql`select pg_advisory_xact_lock_shared(hashtextextended(${`session-tenancy:${context.workspaceId}`}, 0))`
488
- );
489
- }
490
- await assertRlsContextApplied(scoped, context);
491
- setup("completed");
492
- } catch (error) {
493
- setup("failed");
494
- throw error;
495
- }
496
- const callback = startDatabaseTiming("scoped_callback");
497
- let value;
498
- try {
499
- value = await fn(scoped);
500
- callback("completed");
486
+ return await fn(tx);
501
487
  } catch (error) {
502
- callback("failed");
488
+ callbackFailure = { error };
503
489
  throw error;
504
490
  }
505
- if (parentScope) await restoreRlsContextSettings(scoped, parentScope);
506
- return value;
507
491
  }, transactionConfig);
492
+ } catch (error) {
493
+ if (callbackFailure && callbackFailure.error === error) throw error;
494
+ if (!isDatabasePersistenceFailure(error)) throw error;
495
+ throw new DatabaseTransactionError(
496
+ entered ? "settlement" : "admission",
497
+ error,
498
+ callbackFailure?.error
499
+ );
500
+ }
501
+ }
502
+ async function withRlsContext(db, context, fn, transactionConfig, sessionTenancyFence = "shared") {
503
+ const restoreParentScope = isTransactionHandle(db);
504
+ const admission = startDatabaseTiming(
505
+ restoreParentScope ? "savepoint_admission" : "transaction_admission"
506
+ );
507
+ try {
508
+ return await withDatabaseTransactionProvenance(
509
+ db,
510
+ async (tx) => {
511
+ admission("completed");
512
+ const setup = startDatabaseTiming("rls_setup");
513
+ const scoped = tx;
514
+ let parentScope;
515
+ try {
516
+ parentScope = restoreParentScope ? await readRlsContextSettings(scoped) : null;
517
+ await setRlsContext(scoped, context);
518
+ if (context.workspaceId && sessionTenancyFence === "shared") {
519
+ await scoped.execute(
520
+ sql`select pg_advisory_xact_lock_shared(hashtextextended(${`session-tenancy:${context.workspaceId}`}, 0))`
521
+ );
522
+ }
523
+ await assertRlsContextApplied(scoped, context);
524
+ setup("completed");
525
+ } catch (error) {
526
+ setup("failed");
527
+ throw error;
528
+ }
529
+ const callback = startDatabaseTiming("scoped_callback");
530
+ let value;
531
+ try {
532
+ value = await fn(scoped);
533
+ callback("completed");
534
+ } catch (error) {
535
+ callback("failed");
536
+ throw error;
537
+ }
538
+ if (parentScope) await restoreRlsContextSettings(scoped, parentScope);
539
+ return value;
540
+ },
541
+ transactionConfig
542
+ );
508
543
  } catch (error) {
509
544
  admission("failed");
510
545
  throw error;
@@ -821,6 +856,7 @@ export {
821
856
  isRetryableDatabaseTransportFailure,
822
857
  isDatabasePersistenceFailure,
823
858
  safeDatabaseErrorFacts,
859
+ DatabaseTransactionError,
824
860
  SessionEventPersistenceError,
825
861
  isSessionEventPersistenceError,
826
862
  runIdempotentPersistenceTransaction,
@@ -853,4 +889,4 @@ export {
853
889
  withAccountRls,
854
890
  dbBindingFor
855
891
  };
856
- //# sourceMappingURL=chunk-56MYWKQV.js.map
892
+ //# sourceMappingURL=chunk-O2Y5E5M3.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/persistence-errors.ts","../src/database-timing.ts","../src/database.ts"],"sourcesContent":["export type DatabaseFailureCode = \"db_deadlock\" | \"db_serialization_failure\" | \"db_failure\";\n\nexport type PersistenceRetryOutcome = \"not_retryable\" | \"exhausted\";\n\nexport type SafeDatabaseErrorFacts = {\n severity?: string;\n schema?: string;\n table?: string;\n column?: string;\n dataType?: string;\n constraint?: string;\n routine?: string;\n};\n\nexport type PersistenceFailureDetails = {\n code: DatabaseFailureCode;\n sqlState: string | null;\n stage: string;\n eventTypes: string[];\n correlationId: string;\n attempts: number;\n retryOutcome: PersistenceRetryOutcome;\n database: SafeDatabaseErrorFacts;\n};\n\nconst SQLSTATE_KEYS = [\"sqlState\", \"sqlstate\", \"code\"] as const;\nconst NESTED_ERROR_KEYS = [\"cause\", \"original\", \"driverError\", \"error\", \"errors\"] as const;\nconst RETRYABLE_DATABASE_TRANSPORT_CODES = new Set([\n // postgres.js connection lifecycle failures.\n \"CONNECTION_CLOSED\",\n \"CONNECTION_DESTROYED\",\n \"CONNECTION_ENDED\",\n \"CONNECT_TIMEOUT\",\n // Node socket/DNS failures surfaced unchanged by postgres.js.\n \"EAI_AGAIN\",\n \"ECONNABORTED\",\n \"ECONNREFUSED\",\n \"ECONNRESET\",\n \"EHOSTDOWN\",\n \"EHOSTUNREACH\",\n \"ENETDOWN\",\n \"ENETRESET\",\n \"ENETUNREACH\",\n \"ENOTFOUND\",\n \"EPIPE\",\n \"ETIMEDOUT\",\n]);\nconst DATABASE_ERROR_NAMES = new Set([\"DatabaseError\", \"DrizzleQueryError\", \"PostgresError\"]);\nconst DATABASE_DIAGNOSTIC_KEYS = [\n \"severity\",\n \"schema_name\",\n \"table_name\",\n \"column_name\",\n \"data_type_name\",\n \"constraint_name\",\n \"routine\",\n] as const;\nconst SAFE_FACT_KEYS = [\n [\"severity\", \"severity\"],\n [\"schema_name\", \"schema\"],\n [\"schema\", \"schema\"],\n [\"table_name\", \"table\"],\n [\"table\", \"table\"],\n [\"column_name\", \"column\"],\n [\"column\", \"column\"],\n [\"data_type_name\", \"dataType\"],\n [\"dataType\", \"dataType\"],\n [\"constraint_name\", \"constraint\"],\n [\"constraint\", \"constraint\"],\n [\"routine\", \"routine\"],\n] as const;\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n return Boolean(value && typeof value === \"object\");\n}\n\nfunction safeFact(value: unknown): string | undefined {\n if (typeof value !== \"string\" || value.length === 0) return undefined;\n return value;\n}\n\n/** Find the driver SQLSTATE even when Drizzle wrapped it under nested causes. */\nexport function nestedPostgresSqlState(error: unknown): string | null {\n const queue: unknown[] = [error];\n const seen = new Set<unknown>();\n let fallback: string | null = null;\n while (queue.length > 0 && seen.size < 64) {\n const current = queue.shift();\n if (!isRecord(current) || seen.has(current)) continue;\n seen.add(current);\n for (const key of SQLSTATE_KEYS) {\n const value = current[key];\n if (typeof value !== \"string\" || !/^[0-9A-Z]{5}$/i.test(value)) continue;\n const normalized = value.toUpperCase();\n // Node transport codes such as EPIPE happen to be five characters but\n // are not PostgreSQL SQLSTATEs.\n if (RETRYABLE_DATABASE_TRANSPORT_CODES.has(normalized)) continue;\n if (normalized === \"40P01\" || normalized === \"40001\") return normalized;\n fallback ??= normalized;\n }\n for (const key of NESTED_ERROR_KEYS) {\n const nested = current[key];\n if (Array.isArray(nested)) queue.push(...nested);\n else if (nested !== undefined) queue.push(nested);\n }\n }\n return fallback;\n}\n\nexport function databaseFailureCode(sqlState: string | null): DatabaseFailureCode {\n if (sqlState === \"40P01\") return \"db_deadlock\";\n if (sqlState === \"40001\") return \"db_serialization_failure\";\n return \"db_failure\";\n}\n\nexport function isRetryablePersistenceSqlState(sqlState: string | null): boolean {\n return sqlState === \"40P01\" || sqlState === \"40001\";\n}\n\n/**\n * Recognize only explicit postgres.js/Node transport codes, including nested\n * driver causes. These failures can arrive as plain Errors with no SQLSTATE or\n * database diagnostic fields when the connection dies before PostgreSQL can\n * answer. Messages are intentionally ignored: they are unstable and may\n * contain connection detail.\n */\nexport function isRetryableDatabaseTransportFailure(error: unknown): boolean {\n const queue: unknown[] = [error];\n const seen = new Set<unknown>();\n while (queue.length > 0 && seen.size < 64) {\n const current = queue.shift();\n if (!isRecord(current) || seen.has(current)) continue;\n seen.add(current);\n\n for (const key of [\"code\", \"errno\"] as const) {\n const value = current[key];\n if (\n typeof value === \"string\" &&\n RETRYABLE_DATABASE_TRANSPORT_CODES.has(value.toUpperCase())\n ) {\n return true;\n }\n }\n\n for (const key of NESTED_ERROR_KEYS) {\n const nested = current[key];\n if (Array.isArray(nested)) queue.push(...nested);\n else if (nested !== undefined) queue.push(nested);\n }\n }\n return false;\n}\n\n/**\n * Distinguish database/ORM failures from expected domain exceptions when a\n * driver omitted SQLSTATE. This checks shape only; callers retain the original\n * failure independently as canonical error evidence.\n */\nexport function isDatabasePersistenceFailure(error: unknown): boolean {\n if (isRetryableDatabaseTransportFailure(error) || nestedPostgresSqlState(error) !== null) {\n return true;\n }\n\n const queue: unknown[] = [error];\n const seen = new Set<unknown>();\n while (queue.length > 0 && seen.size < 64) {\n const current = queue.shift();\n if (!isRecord(current) || seen.has(current)) continue;\n seen.add(current);\n\n if (typeof current.name === \"string\" && DATABASE_ERROR_NAMES.has(current.name)) {\n return true;\n }\n if (\n typeof current.query === \"string\" &&\n (Object.hasOwn(current, \"params\") || Object.hasOwn(current, \"parameters\"))\n ) {\n return true;\n }\n if (DATABASE_DIAGNOSTIC_KEYS.some((key) => Object.hasOwn(current, key))) {\n return true;\n }\n\n for (const key of NESTED_ERROR_KEYS) {\n const nested = current[key];\n if (Array.isArray(nested)) queue.push(...nested);\n else if (nested !== undefined) queue.push(nested);\n }\n }\n return false;\n}\n\n/** Extract only PostgreSQL diagnostic identifiers; never query text/parameters. */\nexport function safeDatabaseErrorFacts(error: unknown): SafeDatabaseErrorFacts {\n const queue: unknown[] = [error];\n const seen = new Set<unknown>();\n const facts: SafeDatabaseErrorFacts = {};\n while (queue.length > 0 && seen.size < 64) {\n const current = queue.shift();\n if (!isRecord(current) || seen.has(current)) continue;\n seen.add(current);\n for (const [source, destination] of SAFE_FACT_KEYS) {\n if (facts[destination] !== undefined) continue;\n const value = safeFact(current[source]);\n if (value !== undefined) facts[destination] = value;\n }\n for (const key of NESTED_ERROR_KEYS) {\n const nested = current[key];\n if (Array.isArray(nested)) queue.push(...nested);\n else if (nested !== undefined) queue.push(nested);\n }\n }\n return facts;\n}\n\n/** Own-driver transaction boundary; callback errors never establish provenance. */\nexport class DatabaseTransactionError extends Error {\n readonly name = \"DatabaseTransactionError\";\n\n constructor(\n readonly stage: \"admission\" | \"settlement\",\n cause: unknown,\n // A rollback failure must not erase a callback's no-replay/permanent\n // evidence. The recovery classifier inspects both branches for vetoes.\n readonly original?: unknown,\n ) {\n super(`Database transaction ${stage} failed`, { cause });\n }\n}\n\n/** Typed persistence classification retaining the original cause internally.\n * Its ordinary message is stable and contains no SQL or driver parameters. */\nexport class SessionEventPersistenceError extends Error {\n readonly name = \"SessionEventPersistenceError\";\n\n constructor(\n readonly details: PersistenceFailureDetails,\n cause?: unknown,\n ) {\n const label =\n details.code === \"db_deadlock\"\n ? \"Database deadlock\"\n : details.code === \"db_serialization_failure\"\n ? \"Database serialization failure\"\n : \"Database failure\";\n const operation = `${label} while persisting ${details.eventTypes.join(\", \") || \"session events\"}`;\n super(operation, cause === undefined ? undefined : { cause });\n }\n\n get code(): DatabaseFailureCode {\n return this.details.code;\n }\n}\n\nexport function isSessionEventPersistenceError(\n error: unknown,\n): error is SessionEventPersistenceError {\n return error instanceof SessionEventPersistenceError;\n}\n\nexport type IdempotentPersistenceTransactionOptions = {\n stage: string;\n eventTypes?: string[];\n maxAttempts?: number;\n correlationId?: string;\n onRetry?: (input: { attempt: number; sqlState: \"40P01\" | \"40001\" }) => void | Promise<void>;\n};\n\n/**\n * Retry only the supplied idempotent database transaction/savepoint. Provider\n * inference, tools, NATS, and all other external effects must remain outside\n * this function. A single correlation ID follows every persistence attempt.\n */\nexport async function runIdempotentPersistenceTransaction<T>(\n options: IdempotentPersistenceTransactionOptions,\n transaction: (attempt: number) => Promise<T>,\n): Promise<T> {\n const maxAttempts = options.maxAttempts ?? 3;\n if (!Number.isInteger(maxAttempts) || maxAttempts < 1) {\n throw new Error(\"Persistence maxAttempts must be a positive integer\");\n }\n const correlationId = options.correlationId ?? crypto.randomUUID();\n const eventTypes = [...new Set(options.eventTypes ?? [])].sort();\n for (let attempt = 1; attempt <= maxAttempts; attempt += 1) {\n try {\n return await transaction(attempt);\n } catch (error) {\n const sqlState = nestedPostgresSqlState(error);\n const retryable = isRetryablePersistenceSqlState(sqlState);\n if (retryable && attempt < maxAttempts) {\n await options.onRetry?.({\n attempt,\n sqlState: sqlState as \"40P01\" | \"40001\",\n });\n continue;\n }\n if (!isDatabasePersistenceFailure(error)) throw error;\n throw new SessionEventPersistenceError(\n {\n code: databaseFailureCode(sqlState),\n sqlState,\n stage: options.stage,\n eventTypes,\n correlationId,\n attempts: attempt,\n retryOutcome: retryable ? \"exhausted\" : \"not_retryable\",\n database: safeDatabaseErrorFacts(error),\n },\n error,\n );\n }\n }\n throw new Error(\"Unreachable persistence retry state\");\n}\n","import { AsyncLocalStorage } from \"node:async_hooks\";\n\nexport type DatabaseTimingStage =\n | \"transaction_admission\"\n | \"savepoint_admission\"\n | \"rls_setup\"\n | \"scoped_callback\";\nexport type DatabaseTimingObservation = {\n stage: DatabaseTimingStage;\n outcome: \"completed\" | \"failed\";\n durationMs: number;\n};\nexport type DatabaseTimingObserver = (observation: DatabaseTimingObservation) => void;\n\nconst timingObserver = new AsyncLocalStorage<DatabaseTimingObserver>();\n\n/** Opt-in, async-local diagnostics only; no identity or query payload is retained. */\nexport function withDatabaseTimingObserver<T>(\n observer: DatabaseTimingObserver,\n work: () => Promise<T>,\n): Promise<T> {\n return timingObserver.run(observer, work);\n}\n\nexport function startDatabaseTiming(\n stage: DatabaseTimingStage,\n): (outcome: DatabaseTimingObservation[\"outcome\"]) => void {\n const observer = timingObserver.getStore();\n if (!observer) return () => undefined;\n const started = performance.now();\n let finished = false;\n return (outcome) => {\n if (finished) return;\n finished = true;\n try {\n // Do not await observers or let rejected async observers escape either.\n const result = observer({ stage, outcome, durationMs: performance.now() - started });\n void Promise.resolve(result).catch(() => undefined);\n } catch {\n // Diagnostics must not replace database results or the original error.\n }\n };\n}\n","import type { SessionAttachmentReadAccess } from \"./session-file-attachments\";\nimport { AsyncLocalStorage } from \"node:async_hooks\";\nimport { eq, sql, type SQL } from \"drizzle-orm\";\nimport type { PgDatabase, PgTransactionConfig } from \"drizzle-orm/pg-core\";\nimport { drizzle } from \"drizzle-orm/postgres-js\";\nimport postgres from \"postgres\";\n\nimport {\n DatabaseTransactionError,\n isDatabasePersistenceFailure,\n runIdempotentPersistenceTransaction,\n type IdempotentPersistenceTransactionOptions,\n} from \"./persistence-errors\";\nimport { LOSSLESS_CONTENT_WRITER_APPLICATION_NAME } from \"./lossless-json\";\nimport * as schema from \"./schema\";\nimport { startDatabaseTiming } from \"./database-timing\";\n\n// §7.7 driver widening (Step I). `Database` is the structural, cross-driver\n// query-layer port: every helper in this file accepts `db: Database` and uses\n// only the methods present on drizzle's base `PgDatabase` (select/insert/update/\n// delete/transaction/execute). Widening from the concrete\n// `PostgresJsDatabase<typeof schema>` to `PgDatabase<any, typeof schema>` is a\n// pure TYPE change — no runtime behavior changes — that lets an embedded host\n// inject ANY drizzle pg driver handle (node-postgres, neon-http, etc.) bound to\n// OpenGeni's schema, not just the postgres-js handle `createDb` builds. The\n// `any` for the query-result HKT is deliberate: it keeps `db.execute(sql\\`…\\`)`\n// callable across drivers whose raw-result shapes differ (postgres-js returns a\n// row array; node-postgres returns `{ rows }`). The three raw `db.execute(…)`\n// reads that index a row array (`getManagedUserByEmail` here is the only\n// host-facing one — see `userLookup`) stay postgres-js-shaped for standalone;\n// `userLookup` is the injection seam for hosts on a different driver.\n// `PostgresJsDatabase<typeof schema>` is assignable to this, so standalone is\n// unaffected.\n\nexport type Database = PgDatabase<any, typeof schema>;\n\ndeclare const sessionActivityDatabaseBrand: unique symbol;\n\n/**\n * A workspace-scoped transaction whose session activity commit gate is open.\n * Session-row writers accept this narrower handle so new call sites cannot\n * accidentally bypass the once-per-transaction revision finalizer.\n */\nexport type SessionActivityDatabase = Database & {\n readonly [sessionActivityDatabaseBrand]: true;\n};\n\nexport type DbClient = {\n db: Database;\n close: () => Promise<void>;\n};\n\nexport type RlsContext = {\n accountId: string;\n workspaceId?: string | null;\n};\n\nexport type SessionRlsActorContext = {\n /** Server-authorized session attachment scope; never taken from request JSON. */\n sessionAttachmentReadAccess?: SessionAttachmentReadAccess;\n subjectId: string;\n /** Host-verified personal file ownership, never a request-supplied subject. */\n privateFileOwnerSubjectId?: string | null;\n initiatingHumanSubjectId?: string | null;\n};\n\nconst sessionRlsActorContext = new AsyncLocalStorage<SessionRlsActorContext>();\nexport function currentSessionAttachmentReadAccess(): SessionAttachmentReadAccess | undefined {\n return sessionRlsActorContext.getStore()?.sessionAttachmentReadAccess;\n}\n\n/**\n * Stable identity of the ambient session RLS actor: every field that\n * `setRlsContext` turns into database-visible GUCs or that scoped reads\n * consult. Two callers with equal keys see exactly the same rows, so the key\n * is safe for sharing a read result between them; `null` means no actor, in\n * which case the subject GUC stays empty for every such caller.\n */\nexport function currentSessionRlsActorIdentityKey(): string | null {\n const actor = sessionRlsActorContext.getStore();\n if (!actor) return null;\n return JSON.stringify([\n actor.subjectId,\n actor.initiatingHumanSubjectId ?? null,\n actor.privateFileOwnerSubjectId ?? null,\n actor.sessionAttachmentReadAccess ?? null,\n ]);\n}\n\nexport async function withSessionRlsActorContext<T>(\n actor: SessionRlsActorContext,\n fn: () => Promise<T>,\n): Promise<T> {\n if (!actor.subjectId.trim()) {\n throw new Error(\"withSessionRlsActorContext: a non-empty subjectId is required\");\n }\n if (\n actor.initiatingHumanSubjectId !== undefined &&\n actor.initiatingHumanSubjectId !== null &&\n !actor.initiatingHumanSubjectId.trim()\n ) {\n throw new Error(\n \"withSessionRlsActorContext: initiatingHumanSubjectId must be null or non-empty\",\n );\n }\n if (\n actor.privateFileOwnerSubjectId !== undefined &&\n actor.privateFileOwnerSubjectId !== null &&\n !actor.privateFileOwnerSubjectId.trim()\n ) {\n throw new Error(\n \"withSessionRlsActorContext: privateFileOwnerSubjectId must be null or non-empty\",\n );\n }\n return await sessionRlsActorContext.run(actor, fn);\n}\n\ntype RlsContextSettings = {\n accountId: string;\n workspaceId: string;\n subjectId: string;\n privateFileOwnerSubjectId: string;\n initiatingHumanSubjectId: string;\n personalResourceHumanSubjectId: string;\n personalResourceActorSubjectId: string;\n};\n\n/**\n * RLS posture for the connection OpenGeni's query layer runs over (Step I, §7.7).\n *\n * - `\"force\"` (DEFAULT — today's standalone behavior, byte-for-byte): OpenGeni\n * connects as a NON-OWNER role (`opengeni_app`) and every table carries\n * `FORCE ROW LEVEL SECURITY`, so the workspace/account GUCs set by\n * `setRlsContext` are the ONLY thing that admits rows — even the table owner\n * is subject to RLS. This is the Fork-A isolation guarantee.\n * - `\"scoped\"` (embedded Fork-B opt-in): the host runs OpenGeni's queries over a\n * role that OWNS the dedicated schema (RLS need not be forced for that role),\n * relying on the host's own tenant boundary. OpenGeni STILL emits the\n * `set_config('opengeni.account_id'/'workspace_id', …)` GUCs defensively on\n * every scoped query, so the application query path is byte-identical between\n * the two strategies and the app code is RLS-mode-agnostic. The strategy is a\n * declared posture (consumed by `provisionRoles` and as a documented\n * invariant), NOT a query-path branch — there is deliberately no `if\n * (strategy === …)` anywhere in the helpers below. Picking `\"scoped\"` does not\n * relax any GUC; it only changes which DB role the host provisions/connects as\n * and asserts that the host accepts owning the isolation boundary.\n */\nexport type RlsStrategy = \"force\" | \"scoped\";\n\n/**\n * Resolve a host-IdP/Better-Auth user *identifier* by email. Injected via\n * `createDb({ userLookup })` (Step I). UNSET → today's raw parameterized select\n * against Better Auth's `auth_users` table (see `getManagedUserByEmail`), which\n * relies on the postgres-js array-shaped `db.execute` result. An embedded host\n * whose identity lives elsewhere (a different IdP table, a different driver, or\n * a non-`auth_users` user store) injects this closure so OpenGeni never touches\n * `auth_users` directly. Returns the user id, or null when no such user exists.\n */\nexport type UserLookup = (db: Database, email: string) => Promise<string | null>;\n\nexport type ManagedUserProfile = {\n id: string;\n name: string | null;\n email: string;\n};\n\n/** Host identity projection used by organization administration rosters. */\nexport type UserProfileLookup = (\n db: Database,\n userIds: readonly string[],\n) => Promise<ManagedUserProfile[]>;\n\nexport type CreateDbOptions = {\n /**\n * The Postgres `search_path` for this connection (Step I, §7.8 runtime half).\n * UNSET → today's behavior: NO `search_path` startup parameter is sent, so the\n * server default applies (`public` for standalone, where every table + the\n * `vector` extension + `gen_random_uuid()` live). For an embedded dedicated\n * schema, pass e.g. `\"opengeni,opengeni_private,public\"` — postgres-js sends\n * it as a per-session startup parameter (the supported, query-param-free way;\n * URL `?search_path=` is IGNORED by postgres-js). Keep `public` LAST so the\n * `vector` type and `gen_random_uuid()` (which live in `public` on the\n * pgvector image) still resolve — the schema-isolation contract live footgun.\n */\n searchPath?: string;\n /** RLS posture; defaults to `\"force\"` (today's standalone). */\n rlsStrategy?: RlsStrategy;\n /** Host-provided user-by-email resolver; unset → today's raw `auth_users` query. */\n userLookup?: UserLookup;\n /** Host-provided user profile projection; unset → standalone `auth_users`. */\n userProfileLookup?: UserProfileLookup;\n /** postgres-js pool size; defaults to today's `10`. */\n max?: number;\n /**\n * Connection-local default transaction isolation sent in the postgres-js\n * startup parameters. This is intentionally not a role/database default:\n * tests and embedded callers can exercise a different ambient isolation\n * without mutating a shared PostgreSQL role or affecting other connections.\n */\n isolationLevel?: postgres.ConnectionParameters[\"default_transaction_isolation\"];\n};\n\n/**\n * The active RLS strategy + userLookup for an injected `Database`, recorded in a\n * side WeakMap so helpers (and `getManagedUserByEmail`) can consult the host's\n * binding without changing every call signature. A handle with no recorded\n * config (e.g. one built outside `createDb`, or in a test) falls back to the\n * standalone defaults: `rlsStrategy: \"force\"`, raw `auth_users` lookup.\n */\ntype DbBinding = {\n rlsStrategy: RlsStrategy;\n userLookup?: UserLookup;\n userProfileLookup?: UserProfileLookup;\n};\n\nconst dbBindings = new WeakMap<object, DbBinding>();\n\n/** The strategy bound to a handle (or the `\"force\"` default). */\nexport function rlsStrategyFor(db: Database): RlsStrategy {\n return dbBindings.get(db as unknown as object)?.rlsStrategy ?? \"force\";\n}\n\n/**\n * Run a raw SQL query and read its rows as a typed array.\n *\n * Why this exists: the Step I driver widening (`Database = PgDatabase<any, …>`)\n * deliberately sets the query-result HKT to `any` so `db.execute(…)` is callable\n * across drivers whose raw-result shapes differ (postgres-js → row array;\n * node-postgres → `{ rows }`). A side effect is that `db.execute<T>(…)` now\n * resolves to `any`, erasing the per-row element type at the call site. OpenGeni's\n * OWN internal raw queries usually run over the postgres-js handle `createDb`\n * builds (array result), while an embedded host may inject a node-postgres style\n * driver (`{ rows }`). Normalize those two standard shapes in one place; reject\n * an unknown driver result rather than silently treating it as an empty query.\n */\nexport async function rawRows<T extends Record<string, unknown>>(\n executor: Pick<Database, \"execute\">,\n query: SQL,\n): Promise<T[]> {\n const result = await executor.execute<T>(query);\n if (Array.isArray(result)) {\n return result as unknown as T[];\n }\n const rows = (result as unknown as { rows?: unknown }).rows;\n if (Array.isArray(rows)) {\n return rows as T[];\n }\n throw new Error(\"Unsupported database execute() result shape\");\n}\n\nexport type SandboxProviderReadLockIdentity = {\n workspaceId: string;\n sandboxGroupId: string;\n leaseEpoch: number;\n instanceId: string;\n};\n\nexport class SandboxProviderReadLockUnavailableError extends Error {\n constructor() {\n super(\"Sandbox provider reads are temporarily busy. Retry the request.\");\n this.name = \"SandboxProviderReadLockUnavailableError\";\n }\n}\n\nconst SANDBOX_PROVIDER_READ_LOCK_WAIT_MS = 15_000;\nconst SANDBOX_PROVIDER_READ_LOCK_POLL_MAX_MS = 200;\n\nfunction sandboxProviderReadLockKey(identity: SandboxProviderReadLockIdentity): string {\n return [\n \"sandbox-provider-read\",\n identity.workspaceId,\n identity.sandboxGroupId,\n identity.leaseEpoch,\n identity.instanceId,\n ].join(\":\");\n}\n\nasync function waitForSandboxProviderReadLockPoll(\n delayMs: number,\n signal?: AbortSignal | undefined,\n): Promise<void> {\n if (signal?.aborted) {\n throw signal.reason ?? new DOMException(\"The operation was aborted\", \"AbortError\");\n }\n await new Promise<void>((resolve, reject) => {\n let settled = false;\n let timer: ReturnType<typeof setTimeout>;\n const cleanup = () => signal?.removeEventListener(\"abort\", abort);\n const finish = () => {\n if (settled) return;\n settled = true;\n cleanup();\n resolve();\n };\n const abort = () => {\n if (settled) return;\n settled = true;\n clearTimeout(timer);\n cleanup();\n reject(signal?.reason ?? new DOMException(\"The operation was aborted\", \"AbortError\"));\n };\n timer = setTimeout(finish, delayMs);\n signal?.addEventListener(\"abort\", abort, { once: true });\n if (signal?.aborted) abort();\n });\n}\n\n/**\n * Cross-process exclusive boundary for one exact live provider instance's\n * API-direct read request. Modal accepts concurrent commands from one attached\n * handle, but independent API replicas reconstruct different handles; issuing\n * overlapping command groups through those handles can make an otherwise valid\n * read fail transiently. Keep concurrency inside one batched request while\n * serializing only separate requests that target the same lease identity.\n *\n * The callback intentionally runs while one transaction-pinned DB connection\n * holds the advisory lock. The direct lease holder separately prevents drain,\n * and the abort-aware bounded poll prevents a disconnected HTTP request from\n * occupying the connection indefinitely.\n */\nexport async function withSandboxProviderReadLock<T>(\n db: Database,\n identity: SandboxProviderReadLockIdentity,\n signal: AbortSignal | undefined,\n fn: () => Promise<T>,\n): Promise<T> {\n const key = sandboxProviderReadLockKey(identity);\n const deadline = Date.now() + SANDBOX_PROVIDER_READ_LOCK_WAIT_MS;\n return await db.transaction(async (lockedDb) => {\n let pollMs = 20;\n while (true) {\n if (signal?.aborted) {\n throw signal.reason ?? new DOMException(\"The operation was aborted\", \"AbortError\");\n }\n const [row] = await rawRows<{ acquired: boolean }>(\n lockedDb,\n sql`select pg_try_advisory_xact_lock(hashtextextended(${key}, 0)) as acquired`,\n );\n if (row?.acquired === true) return await fn();\n const remainingMs = deadline - Date.now();\n if (remainingMs <= 0) throw new SandboxProviderReadLockUnavailableError();\n await waitForSandboxProviderReadLockPoll(Math.min(pollMs, remainingMs), signal);\n pollMs = Math.min(pollMs * 2, SANDBOX_PROVIDER_READ_LOCK_POLL_MAX_MS);\n }\n });\n}\n\nexport function createDb(databaseUrl: string, options: CreateDbOptions = {}): DbClient {\n // `prepare: false` is REQUIRED for Azure Database for PostgreSQL Flexible\n // Server's transaction-pooling PgBouncer: postgres-js's default named prepared\n // statements (`s_N`) are bound to one backend, but a transaction pooler hands\n // each transaction a different backend, so a later `execute` intermittently\n // throws `prepared statement \"s_N\" does not exist`. Every RLS read in this\n // module (set_config + SELECT inside one db.transaction) rides on this pool, so\n // the failure surfaces as a \"worked, then didn't\" credential/permission read.\n // idle_timeout + max_lifetime recycle connections so a pooler-recycled backend\n // is never reused indefinitely; application_name aids server-side diagnostics.\n const client = postgres(databaseUrl, {\n max: options.max ?? 10,\n prepare: false,\n idle_timeout: 30,\n max_lifetime: 1800,\n // `connection` carries per-session Postgres STARTUP parameters. The exact\n // `application_name` is also the PgBouncer-compatible current-image receipt\n // for migration 0352's restrictive sessions policy; arbitrary custom startup\n // parameters are not portable through transaction poolers. `search_path`\n // (embedded only) is the supported, query-param-free way to scope a connection\n // to a dedicated schema — postgres-js IGNORES a URL `?search_path=`. Unset\n // searchPath → omit it so the server default (`public`) is unchanged for\n // standalone.\n connection: {\n application_name: LOSSLESS_CONTENT_WRITER_APPLICATION_NAME,\n ...(options.searchPath ? { search_path: options.searchPath } : {}),\n ...(options.isolationLevel ? { default_transaction_isolation: options.isolationLevel } : {}),\n },\n });\n const db = drizzle(client, { schema });\n dbBindings.set(db as unknown as object, {\n rlsStrategy: options.rlsStrategy ?? \"force\",\n ...(options.userLookup ? { userLookup: options.userLookup } : {}),\n ...(options.userProfileLookup ? { userProfileLookup: options.userProfileLookup } : {}),\n });\n return {\n db,\n close: async () => {\n await client.end();\n },\n };\n}\n\n/**\n * Register a host's `rlsStrategy`/`userLookup` against an externally-constructed\n * `Database` handle (e.g. one the embedded host built from its own driver and\n * injected, rather than via `createDb`). Lets the same WeakMap-backed lookups\n * work for injected handles. Standalone never calls this (it uses `createDb`).\n */\nexport function registerDbBinding(\n db: Database,\n binding: {\n rlsStrategy?: RlsStrategy;\n userLookup?: UserLookup;\n userProfileLookup?: UserProfileLookup;\n },\n): void {\n dbBindings.set(db as unknown as object, {\n rlsStrategy: binding.rlsStrategy ?? \"force\",\n ...(binding.userLookup ? { userLookup: binding.userLookup } : {}),\n ...(binding.userProfileLookup ? { userProfileLookup: binding.userProfileLookup } : {}),\n });\n}\n\nexport async function setRlsContext(db: Database, context: RlsContext): Promise<void> {\n // Fail loud on an empty/blank account id: a \"\" account would set an RLS GUC\n // that matches no tenant row, silently returning zero rows from every scoped\n // read (a phantom \"not found\" / \"no active subscription\"). An RLS context with\n // no account is always a bug at the call site, never a valid query scope.\n if (typeof context.accountId !== \"string\" || context.accountId.trim() === \"\") {\n throw new Error(\"setRlsContext: a non-empty accountId is required to establish an RLS context\");\n }\n // Transaction-local writer identity covers supported injected/embedded\n // database handles whose connection-level application_name is host-owned.\n // Standalone createDb connections also carry version receipts in their exact\n // application_name, while old OpenGeni binaries set neither current receipt.\n await db.execute(sql`select\n set_config('opengeni.account_id', ${context.accountId}, true),\n set_config('opengeni.workspace_id', ${context.workspaceId ?? \"\"}, true),\n set_config('opengeni.lossless_content_writer', '1', true),\n set_config('opengeni.sandbox_recovery_protocol_v2', '1', true),\n set_config('opengeni.pending_tool_event_output_v1', '1', true),\n set_config('opengeni.session_variable_set_attachments_v1', '1', true)`);\n const sessionActor = sessionRlsActorContext.getStore();\n if (sessionActor) {\n await setSubjectRlsContext(db, sessionActor.subjectId);\n await db.execute(\n sql`select set_config('opengeni.private_file_owner', ${sessionActor.privateFileOwnerSubjectId ?? \"\"}, true), set_config(\n 'opengeni.initiating_human_subject_id',\n ${sessionActor.initiatingHumanSubjectId ?? \"\"},\n true\n )`,\n );\n }\n}\n\nasync function readRlsContextSettings(db: Database): Promise<RlsContextSettings> {\n const [settings] = await rawRows<{\n account_id: string | null;\n workspace_id: string | null;\n subject_id: string | null;\n private_file_owner: string | null;\n initiating_human_subject_id: string | null;\n personal_resource_human_subject_id: string | null;\n personal_resource_actor_subject_id: string | null;\n }>(\n db,\n sql`select\n current_setting('opengeni.account_id', true) as account_id,\n current_setting('opengeni.workspace_id', true) as workspace_id,\n current_setting('opengeni.subject_id', true) as subject_id,\n current_setting('opengeni.private_file_owner', true) as private_file_owner,\n current_setting('opengeni.initiating_human_subject_id', true) as initiating_human_subject_id,\n current_setting('opengeni.personal_resource_human_subject_id', true) as personal_resource_human_subject_id,\n current_setting('opengeni.personal_resource_actor_subject_id', true) as personal_resource_actor_subject_id`,\n );\n return {\n accountId: settings?.account_id ?? \"\",\n workspaceId: settings?.workspace_id ?? \"\",\n subjectId: settings?.subject_id ?? \"\",\n privateFileOwnerSubjectId: settings?.private_file_owner ?? \"\",\n initiatingHumanSubjectId: settings?.initiating_human_subject_id ?? \"\",\n personalResourceHumanSubjectId: settings?.personal_resource_human_subject_id ?? \"\",\n personalResourceActorSubjectId: settings?.personal_resource_actor_subject_id ?? \"\",\n };\n}\n\nasync function assertRlsContextApplied(db: Database, context: RlsContext): Promise<void> {\n const applied = await readRlsContextSettings(db);\n const expectedWorkspaceId = context.workspaceId ?? \"\";\n if (applied.accountId !== context.accountId) {\n throw new Error(\n `RLS context not applied on the active backend: expected account ${context.accountId}, got \"${applied.accountId}\"`,\n );\n }\n if (applied.workspaceId !== expectedWorkspaceId) {\n throw new Error(\n `RLS context not applied on the active backend: expected workspace \"${expectedWorkspaceId}\", got \"${applied.workspaceId}\"`,\n );\n }\n}\n\nasync function restoreRlsContextSettings(\n db: Database,\n settings: RlsContextSettings,\n): Promise<void> {\n const [restored] = await rawRows<{\n account_id: string;\n workspace_id: string;\n subject_id: string;\n private_file_owner: string;\n initiating_human_subject_id: string;\n personal_resource_human_subject_id: string;\n personal_resource_actor_subject_id: string;\n }>(\n db,\n sql`select\n set_config('opengeni.account_id', ${settings.accountId}, true) as account_id,\n set_config('opengeni.workspace_id', ${settings.workspaceId}, true) as workspace_id,\n set_config('opengeni.subject_id', ${settings.subjectId}, true) as subject_id,\n set_config('opengeni.private_file_owner', ${settings.privateFileOwnerSubjectId}, true) as private_file_owner,\n set_config('opengeni.initiating_human_subject_id', ${settings.initiatingHumanSubjectId}, true) as initiating_human_subject_id,\n set_config('opengeni.personal_resource_human_subject_id', ${settings.personalResourceHumanSubjectId}, true) as personal_resource_human_subject_id,\n set_config('opengeni.personal_resource_actor_subject_id', ${settings.personalResourceActorSubjectId}, true) as personal_resource_actor_subject_id`,\n );\n if (\n restored?.account_id !== settings.accountId ||\n restored.workspace_id !== settings.workspaceId ||\n restored.subject_id !== settings.subjectId ||\n restored.private_file_owner !== settings.privateFileOwnerSubjectId ||\n restored.initiating_human_subject_id !== settings.initiatingHumanSubjectId ||\n restored.personal_resource_human_subject_id !== settings.personalResourceHumanSubjectId ||\n restored.personal_resource_actor_subject_id !== settings.personalResourceActorSubjectId\n ) {\n throw new Error(\"RLS context could not be restored after a nested scope\");\n }\n}\n\n/** Attribute only failures outside the application callback to our transaction\n * driver. BEGIN/COMMIT/ROLLBACK can throw raw postgres.js errors, unlike ORM\n * queries. This supplies provenance, never transaction retry permission. */\nasync function withDatabaseTransactionProvenance<T>(\n db: Database,\n fn: (db: Database) => Promise<T>,\n transactionConfig?: PgTransactionConfig,\n): Promise<T> {\n let entered = false;\n let callbackFailure: { error: unknown } | undefined;\n try {\n return await db.transaction(async (tx) => {\n entered = true;\n try {\n return await fn(tx as unknown as Database);\n } catch (error) {\n callbackFailure = { error };\n throw error;\n }\n }, transactionConfig);\n } catch (error) {\n // A provider/tool rejection propagated by a successful rollback is still\n // the provider/tool's error, even if it has a connection-looking code.\n if (callbackFailure && callbackFailure.error === error) throw error;\n if (!isDatabasePersistenceFailure(error)) throw error;\n throw new DatabaseTransactionError(\n entered ? \"settlement\" : \"admission\",\n error,\n callbackFailure?.error,\n );\n }\n}\n\nexport async function withRlsContext<T>(\n db: Database,\n context: RlsContext,\n fn: (db: Database) => Promise<T>,\n transactionConfig?: PgTransactionConfig,\n sessionTenancyFence: \"shared\" | \"none\" = \"shared\",\n): Promise<T> {\n const restoreParentScope = isTransactionHandle(db);\n // Callback entry includes connection admission AND BEGIN/SAVEPOINT round-trip.\n // This is deliberately not advertised as pure pool wait.\n const admission = startDatabaseTiming(\n restoreParentScope ? \"savepoint_admission\" : \"transaction_admission\",\n );\n try {\n return await withDatabaseTransactionProvenance(\n db,\n async (tx) => {\n admission(\"completed\");\n const setup = startDatabaseTiming(\"rls_setup\");\n const scoped = tx as unknown as Database;\n let parentScope: RlsContextSettings | null;\n try {\n parentScope = restoreParentScope ? await readRlsContextSettings(scoped) : null;\n await setRlsContext(scoped, context);\n if (context.workspaceId && sessionTenancyFence === \"shared\") {\n await scoped.execute(\n sql`select pg_advisory_xact_lock_shared(hashtextextended(${`session-tenancy:${context.workspaceId}`}, 0))`,\n );\n }\n // Defense-in-depth: read the LOCAL GUC back on THIS backend BEFORE running\n // the scoped query. The set_config and this read share one db.transaction,\n // which a transaction pooler pins to a single backend — so a mismatch here\n // means the context was genuinely lost (a torn transaction / pooler backend\n // swap), not normal operation. Without this guard such an event runs the\n // scoped read with an empty account_id and returns zero RLS-visible rows,\n // manufacturing a phantom \"no active subscription\" from a credential that is\n // in fact active. Convert that silent false into a loud, root-cause-bearing\n // error so the caller can retry rather than permanently mis-decide.\n await assertRlsContextApplied(scoped, context);\n setup(\"completed\");\n } catch (error) {\n setup(\"failed\");\n throw error;\n }\n const callback = startDatabaseTiming(\"scoped_callback\");\n let value: T;\n try {\n value = await fn(scoped);\n callback(\"completed\");\n } catch (error) {\n callback(\"failed\");\n throw error;\n }\n // A nested transaction is a savepoint, and SET LOCAL survives successful\n // savepoint release. Restore the parent tenant and actor proof together;\n // writer/protocol capabilities intentionally remain transaction-wide.\n if (parentScope) await restoreRlsContextSettings(scoped, parentScope);\n return value;\n },\n transactionConfig,\n );\n } catch (error) {\n // No-op after callback entry; only admission failures are counted here.\n admission(\"failed\");\n throw error;\n }\n}\n\n/**\n * Run one bounded database operation on a transaction-pinned backend.\n *\n * Callers that also have an application deadline should check their abort\n * signal before returning from `fn`; throwing there rolls the transaction back\n * even when the application deadline won a surrounding Promise race.\n */\nexport async function withDatabaseStatementTimeout<T>(\n db: Database,\n timeoutMs: number,\n fn: (db: Database) => Promise<T>,\n): Promise<T> {\n if (!Number.isFinite(timeoutMs) || timeoutMs <= 0) {\n throw new Error(\"withDatabaseStatementTimeout requires a positive timeout\");\n }\n const boundedTimeoutMs = Math.max(1, Math.floor(timeoutMs));\n return await db.transaction(async (tx) => {\n const scoped = tx as unknown as Database;\n await scoped.execute(\n sql`select set_config('statement_timeout', ${`${boundedTimeoutMs}ms`}, true)`,\n );\n return await fn(scoped);\n });\n}\n\nexport async function rlsContextForWorkspace(\n db: Database,\n workspaceId: string,\n): Promise<RlsContext> {\n const [row] = await db\n .select({ accountId: schema.workspaces.accountId })\n .from(schema.workspaces)\n .where(eq(schema.workspaces.id, workspaceId))\n .limit(1);\n if (!row) {\n throw new Error(`Workspace not found: ${workspaceId}`);\n }\n return { accountId: row.accountId, workspaceId };\n}\n\nexport async function withWorkspaceRls<T>(\n db: Database,\n workspaceId: string,\n fn: (db: Database) => Promise<T>,\n transactionConfig?: PgTransactionConfig,\n): Promise<T> {\n return await withRlsContext(\n db,\n await rlsContextForWorkspace(db, workspaceId),\n fn,\n transactionConfig,\n );\n}\n\ntype SessionActivityGate = {\n db: SessionActivityDatabase;\n owner: boolean;\n};\n\nfunction isTransactionHandle(db: Database): boolean {\n return typeof (db as Database & { rollback?: unknown }).rollback === \"function\";\n}\n\nasync function assertSessionActivityGateEntry(db: Database): Promise<void> {\n if (!isTransactionHandle(db)) return;\n const [gate] = await rawRows<{ state: string | null }>(\n db,\n sql`\n select nullif(\n current_setting('opengeni.session_activity_gate_state', true),\n ''\n ) as state\n `,\n );\n if (gate?.state === null || gate === undefined) {\n throw new Error(\"Session activity commit gate must start at the outer transaction boundary\");\n }\n}\n\nasync function beginSessionActivityGate(\n db: Database,\n workspaceId: string,\n): Promise<SessionActivityGate> {\n const [gate] = await rawRows<{\n prior_state: string | null;\n prior_workspace_id: string | null;\n }>(\n db,\n sql`\n with current_gate as materialized (\n select\n nullif(current_setting('opengeni.session_activity_gate_state', true), '') as prior_state,\n nullif(current_setting('opengeni.session_activity_gate_workspace_id', true), '') as prior_workspace_id\n )\n select\n prior_state,\n prior_workspace_id,\n case\n when prior_state is null\n then set_config('opengeni.session_activity_gate_state', 'open', true)\n else prior_state\n end as effective_state,\n case\n when prior_state is null\n then set_config('opengeni.session_activity_gate_workspace_id', ${workspaceId}, true)\n else prior_workspace_id\n end as effective_workspace_id\n from current_gate\n `,\n );\n if (!gate) {\n throw new Error(\"Session activity commit gate could not be opened\");\n }\n if (gate.prior_state === null) {\n return { db: db as SessionActivityDatabase, owner: true };\n }\n if (gate.prior_state === \"open\" && gate.prior_workspace_id === workspaceId) {\n return { db: db as SessionActivityDatabase, owner: false };\n }\n throw new Error(\"Session activity commit gate is already owned by another transaction scope\");\n}\n\nasync function finalizeSessionActivityGate(\n db: SessionActivityDatabase,\n workspaceId: string,\n): Promise<void> {\n // Resolve every deferred FK/constraint before taking the shared workspace\n // counter. The only row writes after that point stamp session rows already\n // modified and therefore already owned by this transaction.\n await db.execute(\n sql`select set_config('opengeni.session_activity_gate_state', 'preparing', true)`,\n );\n await db.execute(sql`set constraints all immediate`);\n // The first flush proves every pre-existing deferred obligation. Re-defer\n // only our commit guard so the marker-clearing writes below queue a fresh\n // proof against the final row state, then force that proof before returning.\n await db.execute(\n sql`set constraints sessions_activity_insert_commit_guard, sessions_activity_update_commit_guard deferred`,\n );\n const [result] = await rawRows<{\n had_pending: boolean;\n pending_count: number;\n revision: string | null;\n stamped_count: number;\n }>(\n db,\n sql`\n with gate as materialized (\n select set_config('opengeni.session_activity_gate_state', 'finalizing', true) as state\n ),\n pending as materialized (\n select ${schema.sessions.id}\n from ${schema.sessions}\n cross join gate\n where ${schema.sessions.workspaceId} = ${workspaceId}\n and ${schema.sessions.activityRevisionPendingXid}\n = pg_current_xact_id()::text::bigint\n ),\n advanced as (\n update ${schema.workspaceSessionActivityRevisions} as counter\n set revision = counter.revision + 1\n where counter.workspace_id = ${workspaceId}\n and exists (select 1 from pending)\n returning counter.revision\n ),\n stamped as (\n update ${schema.sessions} as session\n set\n activity_revision = advanced.revision,\n activity_revision_pending_xid = null\n from advanced\n where session.workspace_id = ${workspaceId}\n and session.activity_revision_pending_xid\n = pg_current_xact_id()::text::bigint\n returning session.id\n ),\n stats as materialized (\n select\n exists (select 1 from pending) as had_pending,\n (select count(*)::integer from pending) as pending_count,\n (select revision::text from advanced) as revision,\n (select count(*)::integer from stamped) as stamped_count\n ),\n finalized as materialized (\n select\n had_pending,\n pending_count,\n revision,\n stamped_count,\n set_config('opengeni.session_activity_gate_state', 'finalized', true) as state\n from stats\n )\n select had_pending, pending_count, revision, stamped_count\n from finalized\n `,\n );\n if (!result) {\n throw new Error(\"Session activity commit gate did not return a finalization result\");\n }\n if (result.had_pending && result.revision === null) {\n throw new Error(\"Workspace session activity counter is unavailable\");\n }\n if (result.stamped_count !== result.pending_count) {\n throw new Error(\"Session activity commit gate did not stamp every pending row\");\n }\n await db.execute(\n sql`set constraints sessions_activity_insert_commit_guard, sessions_activity_update_commit_guard immediate`,\n );\n // A restored activity scope can be followed by another workspace scope, or\n // by a SECURITY DEFINER lifecycle that opens and finalizes its own gate in\n // the same outer transaction. Keep the guards deferred after proving this\n // scope's final state so those later writes reach their own finalizer instead\n // of firing the commit guard at the first session update.\n await db.execute(\n sql`set constraints sessions_activity_insert_commit_guard, sessions_activity_update_commit_guard deferred`,\n );\n}\n\n/**\n * Run one workspace transaction with explicit once-per-transaction session\n * activity finalization. Nested callers reuse the outer gate; only its owner\n * finalizes immediately before the enclosing RLS transaction commits.\n */\nexport async function withSessionActivityRlsContext<T>(\n db: Database,\n context: RlsContext & { workspaceId: string },\n fn: (db: SessionActivityDatabase) => Promise<T>,\n transactionConfig?: PgTransactionConfig,\n fenceMode: \"shared\" | \"none\" = \"shared\",\n organizationMembershipFence = false,\n): Promise<T> {\n await assertSessionActivityGateEntry(db);\n return await withRlsContext(\n db,\n context,\n async (scopedDb) => {\n if (organizationMembershipFence) {\n // Claim can inherit membership-fenced authority after locking sessions.\n // Acquire membership before even a shared tenancy fence: an exclusive\n // tenancy/control waiter can otherwise complete the same lock cycle.\n await scopedDb.execute(sql`select pg_advisory_xact_lock(\n hashtextextended(${`organization-membership:${context.accountId}`}, 0))`);\n if (fenceMode === \"shared\") {\n await scopedDb.execute(sql`select pg_advisory_xact_lock_shared(\n hashtextextended(${`session-tenancy:${context.workspaceId}`}, 0))`);\n }\n }\n const gate = await beginSessionActivityGate(scopedDb, context.workspaceId);\n const value = await fn(gate.db);\n // The finalizer must never trust tenant GUCs that arbitrary callback code\n // could have changed. Nested RLS helpers restore their parent scope; this\n // assertion is the fail-closed commit-boundary proof for every other path.\n await assertRlsContextApplied(gate.db, context);\n if (gate.owner) {\n await finalizeSessionActivityGate(gate.db, context.workspaceId);\n }\n return value;\n },\n transactionConfig,\n organizationMembershipFence ? \"none\" : fenceMode,\n );\n}\n\nexport async function withWorkspaceSessionActivityRls<T>(\n db: Database,\n workspaceId: string,\n fn: (db: SessionActivityDatabase) => Promise<T>,\n transactionConfig?: PgTransactionConfig,\n): Promise<T> {\n return await withSessionActivityRlsContext(\n db,\n { ...(await rlsContextForWorkspace(db, workspaceId)), workspaceId },\n fn,\n transactionConfig,\n );\n}\n\n/**\n * Run one of several ordered workspace activity scopes inside a caller-owned\n * transaction. Each scope is fully finalized before restoring the caller's\n * gate GUCs, so the next workspace can reuse the same outer transaction.\n */\nexport async function withRestoredSessionActivityRlsContext<T>(\n db: Database,\n context: RlsContext & { workspaceId: string },\n fn: (db: SessionActivityDatabase) => Promise<T>,\n): Promise<T> {\n if (!isTransactionHandle(db)) {\n throw new Error(\"Sequential workspace activity scopes require a caller-owned transaction\");\n }\n const [prior] = await rawRows<{ state: string; workspace_id: string }>(\n db,\n sql`select\n coalesce(current_setting('opengeni.session_activity_gate_state', true), '') as state,\n coalesce(\n current_setting('opengeni.session_activity_gate_workspace_id', true), ''\n ) as workspace_id`,\n );\n try {\n return await withRlsContext(db, context, async (scopedDb) => {\n const gate = await beginSessionActivityGate(scopedDb, context.workspaceId);\n if (!gate.owner) {\n throw new Error(\"Sequential workspace activity scope unexpectedly reused an open gate\");\n }\n const value = await fn(gate.db);\n await assertRlsContextApplied(gate.db, context);\n await finalizeSessionActivityGate(gate.db, context.workspaceId);\n return value;\n });\n } finally {\n await db.execute(sql`select\n set_config('opengeni.session_activity_gate_state', ${prior?.state ?? \"\"}, true),\n set_config(\n 'opengeni.session_activity_gate_workspace_id', ${prior?.workspace_id ?? \"\"}, true\n )`);\n }\n}\n\n/**\n * Open one nested savepoint without losing the gate brand. Drizzle correctly\n * preserves the backend-local gate GUCs, but its structural transaction type\n * cannot express that provenance; keep the single trusted rebrand here.\n */\nexport async function withSessionActivitySavepoint<T>(\n db: SessionActivityDatabase,\n fn: (db: SessionActivityDatabase) => Promise<T>,\n): Promise<T> {\n return await db.transaction(async (tx) => await fn(tx as unknown as SessionActivityDatabase));\n}\n\nexport async function retrySessionActivityRls<T>(\n db: Database,\n workspaceId: string,\n options: IdempotentPersistenceTransactionOptions & {\n /** Claim can inherit membership-fenced causal authority after locking the\n * session. Acquire that fence before tenancy/control/session instead. An\n * enclosing transaction must preserve this same lock prefix. */\n organizationMembershipFence?: boolean;\n },\n fn: (db: SessionActivityDatabase) => Promise<T>,\n): Promise<T> {\n return await runIdempotentPersistenceTransaction(options, async () => {\n if (options.organizationMembershipFence) {\n const context = { ...(await rlsContextForWorkspace(db, workspaceId)), workspaceId };\n return await withSessionActivityRlsContext(db, context, fn, undefined, \"shared\", true);\n }\n return await withWorkspaceSessionActivityRls(db, workspaceId, fn);\n });\n}\n\nexport async function retryWorkspacePersistence<T>(\n db: Database,\n workspaceId: string,\n options: IdempotentPersistenceTransactionOptions,\n fn: (db: Database) => Promise<T>,\n): Promise<T> {\n return await runIdempotentPersistenceTransaction(options, async () => {\n return await withWorkspaceRls(db, workspaceId, fn);\n });\n}\n\nexport async function retryRlsPersistence<T>(\n db: Database,\n context: RlsContext,\n options: IdempotentPersistenceTransactionOptions,\n fn: (db: Database) => Promise<T>,\n): Promise<T> {\n return await runIdempotentPersistenceTransaction(options, async () => {\n return await withRlsContext(db, context, fn);\n });\n}\n\n/**\n * Personal workspace data needs both tenant and authenticated-principal GUCs.\n * `session_pins` uses this helper so FORCE RLS rejects another member's rows\n * even if a future query accidentally omits its explicit subject predicate.\n */\nexport async function withWorkspaceSubjectRls<T>(\n db: Database,\n workspaceId: string,\n subjectId: string,\n fn: (db: Database) => Promise<T>,\n transactionConfig?: PgTransactionConfig,\n sessionTenancyFence: \"shared\" | \"none\" = \"shared\",\n): Promise<T> {\n if (!subjectId.trim()) {\n throw new Error(\"withWorkspaceSubjectRls: a non-empty subjectId is required\");\n }\n const context = await rlsContextForWorkspace(db, workspaceId);\n return await withRlsContext(\n db,\n context,\n async (scopedDb) => {\n await setSubjectRlsContext(scopedDb, subjectId);\n return await fn(scopedDb);\n },\n transactionConfig,\n sessionTenancyFence,\n );\n}\n\nexport async function withWorkspaceSubjectSessionActivityRls<T>(\n db: Database,\n workspaceId: string,\n subjectId: string,\n fn: (db: SessionActivityDatabase) => Promise<T>,\n transactionConfig?: PgTransactionConfig,\n fenceMode: \"shared\" | \"none\" = \"shared\",\n organizationMembershipFence = false,\n): Promise<T> {\n if (!subjectId.trim()) {\n throw new Error(\"withWorkspaceSubjectSessionActivityRls: a non-empty subjectId is required\");\n }\n const context = await rlsContextForWorkspace(db, workspaceId);\n return await withSessionActivityRlsContext(\n db,\n { ...context, workspaceId },\n async (scopedDb) => {\n await setSubjectRlsContext(scopedDb, subjectId);\n return await fn(scopedDb);\n },\n transactionConfig,\n fenceMode,\n organizationMembershipFence,\n );\n}\n\n/** Apply and verify actor-private RLS on an already transaction-pinned handle. */\nexport async function setSubjectRlsContext(db: Database, subjectId: string): Promise<void> {\n if (!subjectId.trim()) {\n throw new Error(\"setSubjectRlsContext: a non-empty subjectId is required\");\n }\n await db.execute(sql`select set_config('opengeni.subject_id', ${subjectId}, true)`);\n const applied = await db.execute<{ subject_id: string | null }>(\n sql`select current_setting('opengeni.subject_id', true) as subject_id`,\n );\n if ((applied[0]?.subject_id ?? \"\") !== subjectId) {\n throw new Error(\"Authenticated subject RLS context was not applied on the active backend\");\n }\n}\n\nexport async function withWorkspaceUsageLock<T>(\n db: Database,\n workspaceId: string,\n fn: (db: Database) => Promise<T>,\n): Promise<T> {\n const context = await rlsContextForWorkspace(db, workspaceId);\n return await withRlsContext(db, context, async (scopedDb) => {\n await scopedDb.execute(sql`select pg_advisory_xact_lock(hashtext(${`usage:${workspaceId}`}))`);\n return await fn(scopedDb);\n });\n}\n\nexport async function withAccountRls<T>(\n db: Database,\n accountId: string,\n fn: (db: Database) => Promise<T>,\n): Promise<T> {\n return await withRlsContext(db, { accountId, workspaceId: null }, fn);\n}\n\n/** Internal lookup for the host binding attached to a database handle. */\nexport function dbBindingFor(db: Database): DbBinding | undefined {\n return dbBindings.get(db as unknown as object);\n}\n"],"mappings":";;;;;;;;;;;AAyBA,IAAM,gBAAgB,CAAC,YAAY,YAAY,MAAM;AACrD,IAAM,oBAAoB,CAAC,SAAS,YAAY,eAAe,SAAS,QAAQ;AAChF,IAAM,qCAAqC,oBAAI,IAAI;AAAA;AAAA,EAEjD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EAEA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AACD,IAAM,uBAAuB,oBAAI,IAAI,CAAC,iBAAiB,qBAAqB,eAAe,CAAC;AAC5F,IAAM,2BAA2B;AAAA,EAC/B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AACA,IAAM,iBAAiB;AAAA,EACrB,CAAC,YAAY,UAAU;AAAA,EACvB,CAAC,eAAe,QAAQ;AAAA,EACxB,CAAC,UAAU,QAAQ;AAAA,EACnB,CAAC,cAAc,OAAO;AAAA,EACtB,CAAC,SAAS,OAAO;AAAA,EACjB,CAAC,eAAe,QAAQ;AAAA,EACxB,CAAC,UAAU,QAAQ;AAAA,EACnB,CAAC,kBAAkB,UAAU;AAAA,EAC7B,CAAC,YAAY,UAAU;AAAA,EACvB,CAAC,mBAAmB,YAAY;AAAA,EAChC,CAAC,cAAc,YAAY;AAAA,EAC3B,CAAC,WAAW,SAAS;AACvB;AAEA,SAAS,SAAS,OAAkD;AAClE,SAAO,QAAQ,SAAS,OAAO,UAAU,QAAQ;AACnD;AAEA,SAAS,SAAS,OAAoC;AACpD,MAAI,OAAO,UAAU,YAAY,MAAM,WAAW,EAAG,QAAO;AAC5D,SAAO;AACT;AAGO,SAAS,uBAAuB,OAA+B;AACpE,QAAM,QAAmB,CAAC,KAAK;AAC/B,QAAM,OAAO,oBAAI,IAAa;AAC9B,MAAI,WAA0B;AAC9B,SAAO,MAAM,SAAS,KAAK,KAAK,OAAO,IAAI;AACzC,UAAM,UAAU,MAAM,MAAM;AAC5B,QAAI,CAAC,SAAS,OAAO,KAAK,KAAK,IAAI,OAAO,EAAG;AAC7C,SAAK,IAAI,OAAO;AAChB,eAAW,OAAO,eAAe;AAC/B,YAAM,QAAQ,QAAQ,GAAG;AACzB,UAAI,OAAO,UAAU,YAAY,CAAC,iBAAiB,KAAK,KAAK,EAAG;AAChE,YAAM,aAAa,MAAM,YAAY;AAGrC,UAAI,mCAAmC,IAAI,UAAU,EAAG;AACxD,UAAI,eAAe,WAAW,eAAe,QAAS,QAAO;AAC7D,mBAAa;AAAA,IACf;AACA,eAAW,OAAO,mBAAmB;AACnC,YAAM,SAAS,QAAQ,GAAG;AAC1B,UAAI,MAAM,QAAQ,MAAM,EAAG,OAAM,KAAK,GAAG,MAAM;AAAA,eACtC,WAAW,OAAW,OAAM,KAAK,MAAM;AAAA,IAClD;AAAA,EACF;AACA,SAAO;AACT;AAEO,SAAS,oBAAoB,UAA8C;AAChF,MAAI,aAAa,QAAS,QAAO;AACjC,MAAI,aAAa,QAAS,QAAO;AACjC,SAAO;AACT;AAEO,SAAS,+BAA+B,UAAkC;AAC/E,SAAO,aAAa,WAAW,aAAa;AAC9C;AASO,SAAS,oCAAoC,OAAyB;AAC3E,QAAM,QAAmB,CAAC,KAAK;AAC/B,QAAM,OAAO,oBAAI,IAAa;AAC9B,SAAO,MAAM,SAAS,KAAK,KAAK,OAAO,IAAI;AACzC,UAAM,UAAU,MAAM,MAAM;AAC5B,QAAI,CAAC,SAAS,OAAO,KAAK,KAAK,IAAI,OAAO,EAAG;AAC7C,SAAK,IAAI,OAAO;AAEhB,eAAW,OAAO,CAAC,QAAQ,OAAO,GAAY;AAC5C,YAAM,QAAQ,QAAQ,GAAG;AACzB,UACE,OAAO,UAAU,YACjB,mCAAmC,IAAI,MAAM,YAAY,CAAC,GAC1D;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAEA,eAAW,OAAO,mBAAmB;AACnC,YAAM,SAAS,QAAQ,GAAG;AAC1B,UAAI,MAAM,QAAQ,MAAM,EAAG,OAAM,KAAK,GAAG,MAAM;AAAA,eACtC,WAAW,OAAW,OAAM,KAAK,MAAM;AAAA,IAClD;AAAA,EACF;AACA,SAAO;AACT;AAOO,SAAS,6BAA6B,OAAyB;AACpE,MAAI,oCAAoC,KAAK,KAAK,uBAAuB,KAAK,MAAM,MAAM;AACxF,WAAO;AAAA,EACT;AAEA,QAAM,QAAmB,CAAC,KAAK;AAC/B,QAAM,OAAO,oBAAI,IAAa;AAC9B,SAAO,MAAM,SAAS,KAAK,KAAK,OAAO,IAAI;AACzC,UAAM,UAAU,MAAM,MAAM;AAC5B,QAAI,CAAC,SAAS,OAAO,KAAK,KAAK,IAAI,OAAO,EAAG;AAC7C,SAAK,IAAI,OAAO;AAEhB,QAAI,OAAO,QAAQ,SAAS,YAAY,qBAAqB,IAAI,QAAQ,IAAI,GAAG;AAC9E,aAAO;AAAA,IACT;AACA,QACE,OAAO,QAAQ,UAAU,aACxB,OAAO,OAAO,SAAS,QAAQ,KAAK,OAAO,OAAO,SAAS,YAAY,IACxE;AACA,aAAO;AAAA,IACT;AACA,QAAI,yBAAyB,KAAK,CAAC,QAAQ,OAAO,OAAO,SAAS,GAAG,CAAC,GAAG;AACvE,aAAO;AAAA,IACT;AAEA,eAAW,OAAO,mBAAmB;AACnC,YAAM,SAAS,QAAQ,GAAG;AAC1B,UAAI,MAAM,QAAQ,MAAM,EAAG,OAAM,KAAK,GAAG,MAAM;AAAA,eACtC,WAAW,OAAW,OAAM,KAAK,MAAM;AAAA,IAClD;AAAA,EACF;AACA,SAAO;AACT;AAGO,SAAS,uBAAuB,OAAwC;AAC7E,QAAM,QAAmB,CAAC,KAAK;AAC/B,QAAM,OAAO,oBAAI,IAAa;AAC9B,QAAM,QAAgC,CAAC;AACvC,SAAO,MAAM,SAAS,KAAK,KAAK,OAAO,IAAI;AACzC,UAAM,UAAU,MAAM,MAAM;AAC5B,QAAI,CAAC,SAAS,OAAO,KAAK,KAAK,IAAI,OAAO,EAAG;AAC7C,SAAK,IAAI,OAAO;AAChB,eAAW,CAAC,QAAQ,WAAW,KAAK,gBAAgB;AAClD,UAAI,MAAM,WAAW,MAAM,OAAW;AACtC,YAAM,QAAQ,SAAS,QAAQ,MAAM,CAAC;AACtC,UAAI,UAAU,OAAW,OAAM,WAAW,IAAI;AAAA,IAChD;AACA,eAAW,OAAO,mBAAmB;AACnC,YAAM,SAAS,QAAQ,GAAG;AAC1B,UAAI,MAAM,QAAQ,MAAM,EAAG,OAAM,KAAK,GAAG,MAAM;AAAA,eACtC,WAAW,OAAW,OAAM,KAAK,MAAM;AAAA,IAClD;AAAA,EACF;AACA,SAAO;AACT;AAGO,IAAM,2BAAN,cAAuC,MAAM;AAAA,EAGlD,YACW,OACT,OAGS,UACT;AACA,UAAM,wBAAwB,KAAK,WAAW,EAAE,MAAM,CAAC;AAN9C;AAIA;AAAA,EAGX;AAAA,EAVS,OAAO;AAWlB;AAIO,IAAM,+BAAN,cAA2C,MAAM;AAAA,EAGtD,YACW,SACT,OACA;AACA,UAAM,QACJ,QAAQ,SAAS,gBACb,sBACA,QAAQ,SAAS,6BACf,mCACA;AACR,UAAM,YAAY,GAAG,KAAK,qBAAqB,QAAQ,WAAW,KAAK,IAAI,KAAK,gBAAgB;AAChG,UAAM,WAAW,UAAU,SAAY,SAAY,EAAE,MAAM,CAAC;AAVnD;AAAA,EAWX;AAAA,EAdS,OAAO;AAAA,EAgBhB,IAAI,OAA4B;AAC9B,WAAO,KAAK,QAAQ;AAAA,EACtB;AACF;AAEO,SAAS,+BACd,OACuC;AACvC,SAAO,iBAAiB;AAC1B;AAeA,eAAsB,oCACpB,SACA,aACY;AACZ,QAAM,cAAc,QAAQ,eAAe;AAC3C,MAAI,CAAC,OAAO,UAAU,WAAW,KAAK,cAAc,GAAG;AACrD,UAAM,IAAI,MAAM,oDAAoD;AAAA,EACtE;AACA,QAAM,gBAAgB,QAAQ,iBAAiB,OAAO,WAAW;AACjE,QAAM,aAAa,CAAC,GAAG,IAAI,IAAI,QAAQ,cAAc,CAAC,CAAC,CAAC,EAAE,KAAK;AAC/D,WAAS,UAAU,GAAG,WAAW,aAAa,WAAW,GAAG;AAC1D,QAAI;AACF,aAAO,MAAM,YAAY,OAAO;AAAA,IAClC,SAAS,OAAO;AACd,YAAM,WAAW,uBAAuB,KAAK;AAC7C,YAAM,YAAY,+BAA+B,QAAQ;AACzD,UAAI,aAAa,UAAU,aAAa;AACtC,cAAM,QAAQ,UAAU;AAAA,UACtB;AAAA,UACA;AAAA,QACF,CAAC;AACD;AAAA,MACF;AACA,UAAI,CAAC,6BAA6B,KAAK,EAAG,OAAM;AAChD,YAAM,IAAI;AAAA,QACR;AAAA,UACE,MAAM,oBAAoB,QAAQ;AAAA,UAClC;AAAA,UACA,OAAO,QAAQ;AAAA,UACf;AAAA,UACA;AAAA,UACA,UAAU;AAAA,UACV,cAAc,YAAY,cAAc;AAAA,UACxC,UAAU,uBAAuB,KAAK;AAAA,QACxC;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACA,QAAM,IAAI,MAAM,qCAAqC;AACvD;;;ACzTA,SAAS,yBAAyB;AAclC,IAAM,iBAAiB,IAAI,kBAA0C;AAG9D,SAAS,2BACd,UACA,MACY;AACZ,SAAO,eAAe,IAAI,UAAU,IAAI;AAC1C;AAEO,SAAS,oBACd,OACyD;AACzD,QAAM,WAAW,eAAe,SAAS;AACzC,MAAI,CAAC,SAAU,QAAO,MAAM;AAC5B,QAAM,UAAU,YAAY,IAAI;AAChC,MAAI,WAAW;AACf,SAAO,CAAC,YAAY;AAClB,QAAI,SAAU;AACd,eAAW;AACX,QAAI;AAEF,YAAM,SAAS,SAAS,EAAE,OAAO,SAAS,YAAY,YAAY,IAAI,IAAI,QAAQ,CAAC;AACnF,WAAK,QAAQ,QAAQ,MAAM,EAAE,MAAM,MAAM,MAAS;AAAA,IACpD,QAAQ;AAAA,IAER;AAAA,EACF;AACF;;;ACzCA,SAAS,qBAAAA,0BAAyB;AAClC,SAAS,IAAI,WAAqB;AAElC,SAAS,eAAe;AACxB,OAAO,cAAc;AA6DrB,IAAM,yBAAyB,IAAIC,mBAA0C;AACtE,SAAS,qCAA8E;AAC5F,SAAO,uBAAuB,SAAS,GAAG;AAC5C;AASO,SAAS,oCAAmD;AACjE,QAAM,QAAQ,uBAAuB,SAAS;AAC9C,MAAI,CAAC,MAAO,QAAO;AACnB,SAAO,KAAK,UAAU;AAAA,IACpB,MAAM;AAAA,IACN,MAAM,4BAA4B;AAAA,IAClC,MAAM,6BAA6B;AAAA,IACnC,MAAM,+BAA+B;AAAA,EACvC,CAAC;AACH;AAEA,eAAsB,2BACpB,OACA,IACY;AACZ,MAAI,CAAC,MAAM,UAAU,KAAK,GAAG;AAC3B,UAAM,IAAI,MAAM,+DAA+D;AAAA,EACjF;AACA,MACE,MAAM,6BAA6B,UACnC,MAAM,6BAA6B,QACnC,CAAC,MAAM,yBAAyB,KAAK,GACrC;AACA,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,MACE,MAAM,8BAA8B,UACpC,MAAM,8BAA8B,QACpC,CAAC,MAAM,0BAA0B,KAAK,GACtC;AACA,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,SAAO,MAAM,uBAAuB,IAAI,OAAO,EAAE;AACnD;AAoGA,IAAM,aAAa,oBAAI,QAA2B;AAG3C,SAAS,eAAe,IAA2B;AACxD,SAAO,WAAW,IAAI,EAAuB,GAAG,eAAe;AACjE;AAeA,eAAsB,QACpB,UACA,OACc;AACd,QAAM,SAAS,MAAM,SAAS,QAAW,KAAK;AAC9C,MAAI,MAAM,QAAQ,MAAM,GAAG;AACzB,WAAO;AAAA,EACT;AACA,QAAM,OAAQ,OAAyC;AACvD,MAAI,MAAM,QAAQ,IAAI,GAAG;AACvB,WAAO;AAAA,EACT;AACA,QAAM,IAAI,MAAM,6CAA6C;AAC/D;AASO,IAAM,0CAAN,cAAsD,MAAM;AAAA,EACjE,cAAc;AACZ,UAAM,iEAAiE;AACvE,SAAK,OAAO;AAAA,EACd;AACF;AAEA,IAAM,qCAAqC;AAC3C,IAAM,yCAAyC;AAE/C,SAAS,2BAA2B,UAAmD;AACrF,SAAO;AAAA,IACL;AAAA,IACA,SAAS;AAAA,IACT,SAAS;AAAA,IACT,SAAS;AAAA,IACT,SAAS;AAAA,EACX,EAAE,KAAK,GAAG;AACZ;AAEA,eAAe,mCACb,SACA,QACe;AACf,MAAI,QAAQ,SAAS;AACnB,UAAM,OAAO,UAAU,IAAI,aAAa,6BAA6B,YAAY;AAAA,EACnF;AACA,QAAM,IAAI,QAAc,CAAC,SAAS,WAAW;AAC3C,QAAI,UAAU;AACd,QAAI;AACJ,UAAM,UAAU,MAAM,QAAQ,oBAAoB,SAAS,KAAK;AAChE,UAAM,SAAS,MAAM;AACnB,UAAI,QAAS;AACb,gBAAU;AACV,cAAQ;AACR,cAAQ;AAAA,IACV;AACA,UAAM,QAAQ,MAAM;AAClB,UAAI,QAAS;AACb,gBAAU;AACV,mBAAa,KAAK;AAClB,cAAQ;AACR,aAAO,QAAQ,UAAU,IAAI,aAAa,6BAA6B,YAAY,CAAC;AAAA,IACtF;AACA,YAAQ,WAAW,QAAQ,OAAO;AAClC,YAAQ,iBAAiB,SAAS,OAAO,EAAE,MAAM,KAAK,CAAC;AACvD,QAAI,QAAQ,QAAS,OAAM;AAAA,EAC7B,CAAC;AACH;AAeA,eAAsB,4BACpB,IACA,UACA,QACA,IACY;AACZ,QAAM,MAAM,2BAA2B,QAAQ;AAC/C,QAAM,WAAW,KAAK,IAAI,IAAI;AAC9B,SAAO,MAAM,GAAG,YAAY,OAAO,aAAa;AAC9C,QAAI,SAAS;AACb,WAAO,MAAM;AACX,UAAI,QAAQ,SAAS;AACnB,cAAM,OAAO,UAAU,IAAI,aAAa,6BAA6B,YAAY;AAAA,MACnF;AACA,YAAM,CAAC,GAAG,IAAI,MAAM;AAAA,QAClB;AAAA,QACA,wDAAwD,GAAG;AAAA,MAC7D;AACA,UAAI,KAAK,aAAa,KAAM,QAAO,MAAM,GAAG;AAC5C,YAAM,cAAc,WAAW,KAAK,IAAI;AACxC,UAAI,eAAe,EAAG,OAAM,IAAI,wCAAwC;AACxE,YAAM,mCAAmC,KAAK,IAAI,QAAQ,WAAW,GAAG,MAAM;AAC9E,eAAS,KAAK,IAAI,SAAS,GAAG,sCAAsC;AAAA,IACtE;AAAA,EACF,CAAC;AACH;AAEO,SAAS,SAAS,aAAqB,UAA2B,CAAC,GAAa;AAUrF,QAAM,SAAS,SAAS,aAAa;AAAA,IACnC,KAAK,QAAQ,OAAO;AAAA,IACpB,SAAS;AAAA,IACT,cAAc;AAAA,IACd,cAAc;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,IASd,YAAY;AAAA,MACV,kBAAkB;AAAA,MAClB,GAAI,QAAQ,aAAa,EAAE,aAAa,QAAQ,WAAW,IAAI,CAAC;AAAA,MAChE,GAAI,QAAQ,iBAAiB,EAAE,+BAA+B,QAAQ,eAAe,IAAI,CAAC;AAAA,IAC5F;AAAA,EACF,CAAC;AACD,QAAM,KAAK,QAAQ,QAAQ,EAAE,uBAAO,CAAC;AACrC,aAAW,IAAI,IAAyB;AAAA,IACtC,aAAa,QAAQ,eAAe;AAAA,IACpC,GAAI,QAAQ,aAAa,EAAE,YAAY,QAAQ,WAAW,IAAI,CAAC;AAAA,IAC/D,GAAI,QAAQ,oBAAoB,EAAE,mBAAmB,QAAQ,kBAAkB,IAAI,CAAC;AAAA,EACtF,CAAC;AACD,SAAO;AAAA,IACL;AAAA,IACA,OAAO,YAAY;AACjB,YAAM,OAAO,IAAI;AAAA,IACnB;AAAA,EACF;AACF;AAQO,SAAS,kBACd,IACA,SAKM;AACN,aAAW,IAAI,IAAyB;AAAA,IACtC,aAAa,QAAQ,eAAe;AAAA,IACpC,GAAI,QAAQ,aAAa,EAAE,YAAY,QAAQ,WAAW,IAAI,CAAC;AAAA,IAC/D,GAAI,QAAQ,oBAAoB,EAAE,mBAAmB,QAAQ,kBAAkB,IAAI,CAAC;AAAA,EACtF,CAAC;AACH;AAEA,eAAsB,cAAc,IAAc,SAAoC;AAKpF,MAAI,OAAO,QAAQ,cAAc,YAAY,QAAQ,UAAU,KAAK,MAAM,IAAI;AAC5E,UAAM,IAAI,MAAM,8EAA8E;AAAA,EAChG;AAKA,QAAM,GAAG,QAAQ;AAAA,wCACqB,QAAQ,SAAS;AAAA,0CACf,QAAQ,eAAe,EAAE;AAAA;AAAA;AAAA;AAAA,0EAIO;AACxE,QAAM,eAAe,uBAAuB,SAAS;AACrD,MAAI,cAAc;AAChB,UAAM,qBAAqB,IAAI,aAAa,SAAS;AACrD,UAAM,GAAG;AAAA,MACP,uDAAuD,aAAa,6BAA6B,EAAE;AAAA;AAAA,UAE/F,aAAa,4BAA4B,EAAE;AAAA;AAAA;AAAA,IAGjD;AAAA,EACF;AACF;AAEA,eAAe,uBAAuB,IAA2C;AAC/E,QAAM,CAAC,QAAQ,IAAI,MAAM;AAAA,IASvB;AAAA,IACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQF;AACA,SAAO;AAAA,IACL,WAAW,UAAU,cAAc;AAAA,IACnC,aAAa,UAAU,gBAAgB;AAAA,IACvC,WAAW,UAAU,cAAc;AAAA,IACnC,2BAA2B,UAAU,sBAAsB;AAAA,IAC3D,0BAA0B,UAAU,+BAA+B;AAAA,IACnE,gCAAgC,UAAU,sCAAsC;AAAA,IAChF,gCAAgC,UAAU,sCAAsC;AAAA,EAClF;AACF;AAEA,eAAe,wBAAwB,IAAc,SAAoC;AACvF,QAAM,UAAU,MAAM,uBAAuB,EAAE;AAC/C,QAAM,sBAAsB,QAAQ,eAAe;AACnD,MAAI,QAAQ,cAAc,QAAQ,WAAW;AAC3C,UAAM,IAAI;AAAA,MACR,mEAAmE,QAAQ,SAAS,UAAU,QAAQ,SAAS;AAAA,IACjH;AAAA,EACF;AACA,MAAI,QAAQ,gBAAgB,qBAAqB;AAC/C,UAAM,IAAI;AAAA,MACR,sEAAsE,mBAAmB,WAAW,QAAQ,WAAW;AAAA,IACzH;AAAA,EACF;AACF;AAEA,eAAe,0BACb,IACA,UACe;AACf,QAAM,CAAC,QAAQ,IAAI,MAAM;AAAA,IASvB;AAAA,IACA;AAAA,0CACsC,SAAS,SAAS;AAAA,4CAChB,SAAS,WAAW;AAAA,0CACtB,SAAS,SAAS;AAAA,kDACV,SAAS,yBAAyB;AAAA,2DACzB,SAAS,wBAAwB;AAAA,kEAC1B,SAAS,8BAA8B;AAAA,kEACvC,SAAS,8BAA8B;AAAA,EACvG;AACA,MACE,UAAU,eAAe,SAAS,aAClC,SAAS,iBAAiB,SAAS,eACnC,SAAS,eAAe,SAAS,aACjC,SAAS,uBAAuB,SAAS,6BACzC,SAAS,gCAAgC,SAAS,4BAClD,SAAS,uCAAuC,SAAS,kCACzD,SAAS,uCAAuC,SAAS,gCACzD;AACA,UAAM,IAAI,MAAM,wDAAwD;AAAA,EAC1E;AACF;AAKA,eAAe,kCACb,IACA,IACA,mBACY;AACZ,MAAI,UAAU;AACd,MAAI;AACJ,MAAI;AACF,WAAO,MAAM,GAAG,YAAY,OAAO,OAAO;AACxC,gBAAU;AACV,UAAI;AACF,eAAO,MAAM,GAAG,EAAyB;AAAA,MAC3C,SAAS,OAAO;AACd,0BAAkB,EAAE,MAAM;AAC1B,cAAM;AAAA,MACR;AAAA,IACF,GAAG,iBAAiB;AAAA,EACtB,SAAS,OAAO;AAGd,QAAI,mBAAmB,gBAAgB,UAAU,MAAO,OAAM;AAC9D,QAAI,CAAC,6BAA6B,KAAK,EAAG,OAAM;AAChD,UAAM,IAAI;AAAA,MACR,UAAU,eAAe;AAAA,MACzB;AAAA,MACA,iBAAiB;AAAA,IACnB;AAAA,EACF;AACF;AAEA,eAAsB,eACpB,IACA,SACA,IACA,mBACA,sBAAyC,UAC7B;AACZ,QAAM,qBAAqB,oBAAoB,EAAE;AAGjD,QAAM,YAAY;AAAA,IAChB,qBAAqB,wBAAwB;AAAA,EAC/C;AACA,MAAI;AACF,WAAO,MAAM;AAAA,MACX;AAAA,MACA,OAAO,OAAO;AACZ,kBAAU,WAAW;AACrB,cAAM,QAAQ,oBAAoB,WAAW;AAC7C,cAAM,SAAS;AACf,YAAI;AACJ,YAAI;AACF,wBAAc,qBAAqB,MAAM,uBAAuB,MAAM,IAAI;AAC1E,gBAAM,cAAc,QAAQ,OAAO;AACnC,cAAI,QAAQ,eAAe,wBAAwB,UAAU;AAC3D,kBAAM,OAAO;AAAA,cACX,2DAA2D,mBAAmB,QAAQ,WAAW,EAAE;AAAA,YACrG;AAAA,UACF;AAUA,gBAAM,wBAAwB,QAAQ,OAAO;AAC7C,gBAAM,WAAW;AAAA,QACnB,SAAS,OAAO;AACd,gBAAM,QAAQ;AACd,gBAAM;AAAA,QACR;AACA,cAAM,WAAW,oBAAoB,iBAAiB;AACtD,YAAI;AACJ,YAAI;AACF,kBAAQ,MAAM,GAAG,MAAM;AACvB,mBAAS,WAAW;AAAA,QACtB,SAAS,OAAO;AACd,mBAAS,QAAQ;AACjB,gBAAM;AAAA,QACR;AAIA,YAAI,YAAa,OAAM,0BAA0B,QAAQ,WAAW;AACpE,eAAO;AAAA,MACT;AAAA,MACA;AAAA,IACF;AAAA,EACF,SAAS,OAAO;AAEd,cAAU,QAAQ;AAClB,UAAM;AAAA,EACR;AACF;AASA,eAAsB,6BACpB,IACA,WACA,IACY;AACZ,MAAI,CAAC,OAAO,SAAS,SAAS,KAAK,aAAa,GAAG;AACjD,UAAM,IAAI,MAAM,0DAA0D;AAAA,EAC5E;AACA,QAAM,mBAAmB,KAAK,IAAI,GAAG,KAAK,MAAM,SAAS,CAAC;AAC1D,SAAO,MAAM,GAAG,YAAY,OAAO,OAAO;AACxC,UAAM,SAAS;AACf,UAAM,OAAO;AAAA,MACX,6CAA6C,GAAG,gBAAgB,IAAI;AAAA,IACtE;AACA,WAAO,MAAM,GAAG,MAAM;AAAA,EACxB,CAAC;AACH;AAEA,eAAsB,uBACpB,IACA,aACqB;AACrB,QAAM,CAAC,GAAG,IAAI,MAAM,GACjB,OAAO,EAAE,WAAkB,WAAW,UAAU,CAAC,EACjD,KAAY,UAAU,EACtB,MAAM,GAAU,WAAW,IAAI,WAAW,CAAC,EAC3C,MAAM,CAAC;AACV,MAAI,CAAC,KAAK;AACR,UAAM,IAAI,MAAM,wBAAwB,WAAW,EAAE;AAAA,EACvD;AACA,SAAO,EAAE,WAAW,IAAI,WAAW,YAAY;AACjD;AAEA,eAAsB,iBACpB,IACA,aACA,IACA,mBACY;AACZ,SAAO,MAAM;AAAA,IACX;AAAA,IACA,MAAM,uBAAuB,IAAI,WAAW;AAAA,IAC5C;AAAA,IACA;AAAA,EACF;AACF;AAOA,SAAS,oBAAoB,IAAuB;AAClD,SAAO,OAAQ,GAAyC,aAAa;AACvE;AAEA,eAAe,+BAA+B,IAA6B;AACzE,MAAI,CAAC,oBAAoB,EAAE,EAAG;AAC9B,QAAM,CAAC,IAAI,IAAI,MAAM;AAAA,IACnB;AAAA,IACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAMF;AACA,MAAI,MAAM,UAAU,QAAQ,SAAS,QAAW;AAC9C,UAAM,IAAI,MAAM,2EAA2E;AAAA,EAC7F;AACF;AAEA,eAAe,yBACb,IACA,aAC8B;AAC9B,QAAM,CAAC,IAAI,IAAI,MAAM;AAAA,IAInB;AAAA,IACA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,6EAgByE,WAAW;AAAA;AAAA;AAAA;AAAA;AAAA,EAKtF;AACA,MAAI,CAAC,MAAM;AACT,UAAM,IAAI,MAAM,kDAAkD;AAAA,EACpE;AACA,MAAI,KAAK,gBAAgB,MAAM;AAC7B,WAAO,EAAE,IAAmC,OAAO,KAAK;AAAA,EAC1D;AACA,MAAI,KAAK,gBAAgB,UAAU,KAAK,uBAAuB,aAAa;AAC1E,WAAO,EAAE,IAAmC,OAAO,MAAM;AAAA,EAC3D;AACA,QAAM,IAAI,MAAM,4EAA4E;AAC9F;AAEA,eAAe,4BACb,IACA,aACe;AAIf,QAAM,GAAG;AAAA,IACP;AAAA,EACF;AACA,QAAM,GAAG,QAAQ,kCAAkC;AAInD,QAAM,GAAG;AAAA,IACP;AAAA,EACF;AACA,QAAM,CAAC,MAAM,IAAI,MAAM;AAAA,IAMrB;AAAA,IACA;AAAA;AAAA;AAAA;AAAA;AAAA,iBAKoB,SAAS,EAAE;AAAA,eACb,QAAQ;AAAA;AAAA,gBAEP,SAAS,WAAW,MAAM,WAAW;AAAA,gBACrC,SAAS,0BAA0B;AAAA;AAAA;AAAA;AAAA,iBAIlC,iCAAiC;AAAA;AAAA,uCAElB,WAAW;AAAA;AAAA;AAAA;AAAA;AAAA,iBAK1B,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA,uCAKO,WAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAwBhD;AACA,MAAI,CAAC,QAAQ;AACX,UAAM,IAAI,MAAM,mEAAmE;AAAA,EACrF;AACA,MAAI,OAAO,eAAe,OAAO,aAAa,MAAM;AAClD,UAAM,IAAI,MAAM,mDAAmD;AAAA,EACrE;AACA,MAAI,OAAO,kBAAkB,OAAO,eAAe;AACjD,UAAM,IAAI,MAAM,8DAA8D;AAAA,EAChF;AACA,QAAM,GAAG;AAAA,IACP;AAAA,EACF;AAMA,QAAM,GAAG;AAAA,IACP;AAAA,EACF;AACF;AAOA,eAAsB,8BACpB,IACA,SACA,IACA,mBACA,YAA+B,UAC/B,8BAA8B,OAClB;AACZ,QAAM,+BAA+B,EAAE;AACvC,SAAO,MAAM;AAAA,IACX;AAAA,IACA;AAAA,IACA,OAAO,aAAa;AAClB,UAAI,6BAA6B;AAI/B,cAAM,SAAS,QAAQ;AAAA,6BACF,2BAA2B,QAAQ,SAAS,EAAE,OAAO;AAC1E,YAAI,cAAc,UAAU;AAC1B,gBAAM,SAAS,QAAQ;AAAA,+BACF,mBAAmB,QAAQ,WAAW,EAAE,OAAO;AAAA,QACtE;AAAA,MACF;AACA,YAAM,OAAO,MAAM,yBAAyB,UAAU,QAAQ,WAAW;AACzE,YAAM,QAAQ,MAAM,GAAG,KAAK,EAAE;AAI9B,YAAM,wBAAwB,KAAK,IAAI,OAAO;AAC9C,UAAI,KAAK,OAAO;AACd,cAAM,4BAA4B,KAAK,IAAI,QAAQ,WAAW;AAAA,MAChE;AACA,aAAO;AAAA,IACT;AAAA,IACA;AAAA,IACA,8BAA8B,SAAS;AAAA,EACzC;AACF;AAEA,eAAsB,gCACpB,IACA,aACA,IACA,mBACY;AACZ,SAAO,MAAM;AAAA,IACX;AAAA,IACA,EAAE,GAAI,MAAM,uBAAuB,IAAI,WAAW,GAAI,YAAY;AAAA,IAClE;AAAA,IACA;AAAA,EACF;AACF;AAOA,eAAsB,sCACpB,IACA,SACA,IACY;AACZ,MAAI,CAAC,oBAAoB,EAAE,GAAG;AAC5B,UAAM,IAAI,MAAM,yEAAyE;AAAA,EAC3F;AACA,QAAM,CAAC,KAAK,IAAI,MAAM;AAAA,IACpB;AAAA,IACA;AAAA;AAAA;AAAA;AAAA;AAAA,EAKF;AACA,MAAI;AACF,WAAO,MAAM,eAAe,IAAI,SAAS,OAAO,aAAa;AAC3D,YAAM,OAAO,MAAM,yBAAyB,UAAU,QAAQ,WAAW;AACzE,UAAI,CAAC,KAAK,OAAO;AACf,cAAM,IAAI,MAAM,sEAAsE;AAAA,MACxF;AACA,YAAM,QAAQ,MAAM,GAAG,KAAK,EAAE;AAC9B,YAAM,wBAAwB,KAAK,IAAI,OAAO;AAC9C,YAAM,4BAA4B,KAAK,IAAI,QAAQ,WAAW;AAC9D,aAAO;AAAA,IACT,CAAC;AAAA,EACH,UAAE;AACA,UAAM,GAAG,QAAQ;AAAA,2DACsC,OAAO,SAAS,EAAE;AAAA;AAAA,yDAEpB,OAAO,gBAAgB,EAAE;AAAA,QAC1E;AAAA,EACN;AACF;AAOA,eAAsB,6BACpB,IACA,IACY;AACZ,SAAO,MAAM,GAAG,YAAY,OAAO,OAAO,MAAM,GAAG,EAAwC,CAAC;AAC9F;AAEA,eAAsB,wBACpB,IACA,aACA,SAMA,IACY;AACZ,SAAO,MAAM,oCAAoC,SAAS,YAAY;AACpE,QAAI,QAAQ,6BAA6B;AACvC,YAAM,UAAU,EAAE,GAAI,MAAM,uBAAuB,IAAI,WAAW,GAAI,YAAY;AAClF,aAAO,MAAM,8BAA8B,IAAI,SAAS,IAAI,QAAW,UAAU,IAAI;AAAA,IACvF;AACA,WAAO,MAAM,gCAAgC,IAAI,aAAa,EAAE;AAAA,EAClE,CAAC;AACH;AAEA,eAAsB,0BACpB,IACA,aACA,SACA,IACY;AACZ,SAAO,MAAM,oCAAoC,SAAS,YAAY;AACpE,WAAO,MAAM,iBAAiB,IAAI,aAAa,EAAE;AAAA,EACnD,CAAC;AACH;AAEA,eAAsB,oBACpB,IACA,SACA,SACA,IACY;AACZ,SAAO,MAAM,oCAAoC,SAAS,YAAY;AACpE,WAAO,MAAM,eAAe,IAAI,SAAS,EAAE;AAAA,EAC7C,CAAC;AACH;AAOA,eAAsB,wBACpB,IACA,aACA,WACA,IACA,mBACA,sBAAyC,UAC7B;AACZ,MAAI,CAAC,UAAU,KAAK,GAAG;AACrB,UAAM,IAAI,MAAM,4DAA4D;AAAA,EAC9E;AACA,QAAM,UAAU,MAAM,uBAAuB,IAAI,WAAW;AAC5D,SAAO,MAAM;AAAA,IACX;AAAA,IACA;AAAA,IACA,OAAO,aAAa;AAClB,YAAM,qBAAqB,UAAU,SAAS;AAC9C,aAAO,MAAM,GAAG,QAAQ;AAAA,IAC1B;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEA,eAAsB,uCACpB,IACA,aACA,WACA,IACA,mBACA,YAA+B,UAC/B,8BAA8B,OAClB;AACZ,MAAI,CAAC,UAAU,KAAK,GAAG;AACrB,UAAM,IAAI,MAAM,2EAA2E;AAAA,EAC7F;AACA,QAAM,UAAU,MAAM,uBAAuB,IAAI,WAAW;AAC5D,SAAO,MAAM;AAAA,IACX;AAAA,IACA,EAAE,GAAG,SAAS,YAAY;AAAA,IAC1B,OAAO,aAAa;AAClB,YAAM,qBAAqB,UAAU,SAAS;AAC9C,aAAO,MAAM,GAAG,QAAQ;AAAA,IAC1B;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAGA,eAAsB,qBAAqB,IAAc,WAAkC;AACzF,MAAI,CAAC,UAAU,KAAK,GAAG;AACrB,UAAM,IAAI,MAAM,yDAAyD;AAAA,EAC3E;AACA,QAAM,GAAG,QAAQ,+CAA+C,SAAS,SAAS;AAClF,QAAM,UAAU,MAAM,GAAG;AAAA,IACvB;AAAA,EACF;AACA,OAAK,QAAQ,CAAC,GAAG,cAAc,QAAQ,WAAW;AAChD,UAAM,IAAI,MAAM,yEAAyE;AAAA,EAC3F;AACF;AAEA,eAAsB,uBACpB,IACA,aACA,IACY;AACZ,QAAM,UAAU,MAAM,uBAAuB,IAAI,WAAW;AAC5D,SAAO,MAAM,eAAe,IAAI,SAAS,OAAO,aAAa;AAC3D,UAAM,SAAS,QAAQ,4CAA4C,SAAS,WAAW,EAAE,IAAI;AAC7F,WAAO,MAAM,GAAG,QAAQ;AAAA,EAC1B,CAAC;AACH;AAEA,eAAsB,eACpB,IACA,WACA,IACY;AACZ,SAAO,MAAM,eAAe,IAAI,EAAE,WAAW,aAAa,KAAK,GAAG,EAAE;AACtE;AAGO,SAAS,aAAa,IAAqC;AAChE,SAAO,WAAW,IAAI,EAAuB;AAC/C;","names":["AsyncLocalStorage","AsyncLocalStorage"]}
@@ -1,7 +1,7 @@
1
1
  import {
2
2
  setSubjectRlsContext,
3
3
  withRlsContext
4
- } from "./chunk-56MYWKQV.js";
4
+ } from "./chunk-O2Y5E5M3.js";
5
5
 
6
6
  // src/connection-token-resolver.ts
7
7
  import { normalizeSlackScopes, OFFICIAL_SLACK_MCP_URL } from "@opengeni/contracts/slack-rest-mcp";
@@ -857,4 +857,4 @@ export {
857
857
  refreshOAuthConnectionCredential,
858
858
  isPrivateAddress
859
859
  };
860
- //# sourceMappingURL=chunk-7POTLB24.js.map
860
+ //# sourceMappingURL=chunk-QFJNEXFL.js.map
@@ -4,11 +4,13 @@ import {
4
4
  import {
5
5
  rawRows,
6
6
  setSubjectRlsContext,
7
+ withAccountRls,
7
8
  withRlsContext,
8
9
  withWorkspaceRls
9
- } from "./chunk-56MYWKQV.js";
10
+ } from "./chunk-O2Y5E5M3.js";
10
11
  import {
11
12
  connections,
13
+ creditDebitAllocations,
12
14
  creditLedgerEntries,
13
15
  files,
14
16
  generatedVideoArtifacts,
@@ -18,7 +20,7 @@ import {
18
20
  workspaceVideoGenerationPolicies,
19
21
  workspaceVideoGenerationQuotas,
20
22
  workspaces
21
- } from "./chunk-LDTQGYD2.js";
23
+ } from "./chunk-FWQNHNOU.js";
22
24
 
23
25
  // src/video-generation.ts
24
26
  import {
@@ -30,7 +32,7 @@ import {
30
32
  VideoGenerationOperationSummary,
31
33
  retainedGeneratedVideoReferenceFromFile
32
34
  } from "@opengeni/contracts";
33
- import { and, eq, inArray, sql as sql3 } from "drizzle-orm";
35
+ import { and, eq, inArray, sql as sql4 } from "drizzle-orm";
34
36
 
35
37
  // src/usage-allowances.ts
36
38
  import { sql as sql2 } from "drizzle-orm";
@@ -318,6 +320,83 @@ async function maintainWorkspaceAllowances(db, input = {}) {
318
320
  return Number(row?.processed ?? 0);
319
321
  }
320
322
 
323
+ // src/credit-balances.ts
324
+ import { sql as sql3 } from "drizzle-orm";
325
+ async function getBillingBalance(db, accountId, policyRevision) {
326
+ return await withAccountRls(db, accountId, async (tx) => {
327
+ const [row] = await rawRows(
328
+ tx,
329
+ sql3`
330
+ with current_policy as (
331
+ select revision, policy from opengeni_private.credit_promotion_policy_revisions
332
+ where (${policyRevision ?? null}::bigint is null or revision <= ${policyRevision ?? null}::bigint)
333
+ order by revision desc limit 1
334
+ )
335
+ select
336
+ coalesce((select revision from current_policy), 0)::int as revision,
337
+ (select coalesce(sum(amount_micros), 0)::text
338
+ from ${creditLedgerEntries} where account_id = ${accountId}) as balance,
339
+ coalesce((select jsonb_agg(jsonb_build_object(
340
+ 'grantId', g.id,
341
+ 'label', coalesce(g.metadata->>'creditOfferLabel', 'Promotional credits'),
342
+ 'eligibleModelIds', coalesce(
343
+ (select coalesce(
344
+ case when g.source_type = 'verified_signup_trial' then policy->'signupModelIds'
345
+ else policy->'offers'->(g.metadata->>'creditOfferId')->'eligibleModelIds' end,
346
+ policy->'defaultModelIds'
347
+ ) from current_policy), to_jsonb(g.eligible_model_ids)),
348
+ 'remainingMicros', g.amount_micros - coalesce(used.amount, 0)
349
+ ) order by g.created_at, g.id)
350
+ from ${creditLedgerEntries} g
351
+ left join (
352
+ select grant_entry_id, sum(amount_micros) as amount
353
+ from ${creditDebitAllocations} where account_id = ${accountId}
354
+ group by grant_entry_id
355
+ ) used on used.grant_entry_id = g.id
356
+ where g.account_id = ${accountId} and g.eligible_model_ids is not null
357
+ and g.amount_micros > coalesce(used.amount, 0)
358
+ ), '[]'::jsonb) as grants
359
+ `
360
+ );
361
+ const balanceMicros = Number(row?.balance ?? 0);
362
+ const promotionalCredits = row?.grants ?? [];
363
+ return {
364
+ accountId,
365
+ creditPolicyRevision: row?.revision ?? 0,
366
+ balanceMicros,
367
+ generalBalanceMicros: balanceMicros - promotionalCredits.reduce((sum, grant) => sum + grant.remainingMicros, 0),
368
+ promotionalCredits,
369
+ currency: "usd",
370
+ updatedAt: (/* @__PURE__ */ new Date()).toISOString()
371
+ };
372
+ });
373
+ }
374
+ function spendableCreditMicros(balance, modelId) {
375
+ return Math.max(0, balance.generalBalanceMicros ?? balance.balanceMicros) + (balance.promotionalCredits ?? []).reduce(
376
+ (sum, grant) => sum + (modelId && grant.eligibleModelIds.includes(modelId) ? grant.remainingMicros : 0),
377
+ 0
378
+ );
379
+ }
380
+ async function getSpendableCreditBalance(db, accountId, modelId, policyRevision) {
381
+ const balance = await getBillingBalance(db, accountId, policyRevision);
382
+ return { ...balance, balanceMicros: spendableCreditMicros(balance, modelId) };
383
+ }
384
+ function planCreditDebit(balance, requestedMicros, modelId) {
385
+ if (!Number.isSafeInteger(requestedMicros) || requestedMicros <= 0) {
386
+ throw new Error("credit debit requires a positive, safe integer micro amount");
387
+ }
388
+ const debitedMicros = Math.min(requestedMicros, spendableCreditMicros(balance, modelId));
389
+ let remaining = debitedMicros;
390
+ const allocations = [];
391
+ for (const grant of balance.promotionalCredits ?? []) {
392
+ if (!remaining || !modelId || !grant.eligibleModelIds.includes(modelId)) continue;
393
+ const amountMicros = Math.min(remaining, grant.remainingMicros);
394
+ allocations.push({ grantEntryId: grant.grantId, amountMicros });
395
+ remaining -= amountMicros;
396
+ }
397
+ return { debitedMicros, allocations };
398
+ }
399
+
321
400
  // src/video-generation.ts
322
401
  var ACTIVE_VIDEO_GENERATION_STATUSES = [
323
402
  "preparing",
@@ -400,7 +479,7 @@ async function admitVideoGenerationOperation(db, input) {
400
479
  { accountId: input.accountId, workspaceId: input.workspaceId },
401
480
  async (scopedDb) => await scopedDb.transaction(async (tx) => {
402
481
  await tx.execute(
403
- sql3`select pg_advisory_xact_lock(hashtextextended(${`video-generation:${input.admissionKey}`}, 0))`
482
+ sql4`select pg_advisory_xact_lock(hashtextextended(${`video-generation:${input.admissionKey}`}, 0))`
404
483
  );
405
484
  const [existing] = await tx.select().from(videoGenerationOperations).where(
406
485
  and(
@@ -487,7 +566,7 @@ async function admitVideoGenerationOperation(db, input) {
487
566
  );
488
567
  }
489
568
  await tx.update(workspaceVideoGenerationQuotas).set({
490
- reservedBytes: sql3`${workspaceVideoGenerationQuotas.reservedBytes} + ${reservedBytes}`,
569
+ reservedBytes: sql4`${workspaceVideoGenerationQuotas.reservedBytes} + ${reservedBytes}`,
491
570
  updatedAt: /* @__PURE__ */ new Date()
492
571
  }).where(eq(workspaceVideoGenerationQuotas.workspaceId, input.workspaceId));
493
572
  return { operation, created: true };
@@ -615,7 +694,7 @@ async function cancelUnacceptedVideoGenerationsForToolCallsInTransaction(tx, inp
615
694
  inArray(videoGenerationOperations.toolCallId, toolCallIds),
616
695
  eq(videoGenerationOperations.admissionOutputState, "pending"),
617
696
  inArray(videoGenerationOperations.status, ["preparing", "prepared"]),
618
- sql3`${videoGenerationOperations.providerRequestSentAt} is null`
697
+ sql4`${videoGenerationOperations.providerRequestSentAt} is null`
619
698
  )
620
699
  ).orderBy(videoGenerationOperations.id).for("update");
621
700
  const cancelled = [];
@@ -648,7 +727,7 @@ async function cancelUnacceptedVideoGenerationsForToolCallsInTransaction(tx, inp
648
727
  async function claimVideoGenerationOperations(db, input) {
649
728
  const rows = await rawRows(
650
729
  db,
651
- sql3`select * from opengeni_private.claim_video_generation_operations(
730
+ sql4`select * from opengeni_private.claim_video_generation_operations(
652
731
  ${input.owner}, ${input.leaseSeconds}, ${input.limit}
653
732
  )`
654
733
  );
@@ -839,7 +918,7 @@ async function settleVideoGenerationReady(db, input) {
839
918
  throw new Error("Video artifact and File identities must be distinct");
840
919
  }
841
920
  await tx.execute(
842
- sql3`SELECT set_config('opengeni.private_file_owner',${operation.privateFileOwnerSubjectId ?? ""},true)`
921
+ sql4`SELECT set_config('opengeni.private_file_owner',${operation.privateFileOwnerSubjectId ?? ""},true)`
843
922
  );
844
923
  const now = /* @__PURE__ */ new Date();
845
924
  await tx.insert(files).values({
@@ -883,8 +962,8 @@ async function settleVideoGenerationReady(db, input) {
883
962
  }).returning();
884
963
  if (!artifact) throw new Error("Generated video artifact was not persisted");
885
964
  await tx.update(workspaceVideoGenerationQuotas).set({
886
- reservedBytes: sql3`${workspaceVideoGenerationQuotas.reservedBytes} - ${operation.reservedBytes}`,
887
- readyBytes: sql3`${workspaceVideoGenerationQuotas.readyBytes} + ${input.sizeBytes}`,
965
+ reservedBytes: sql4`${workspaceVideoGenerationQuotas.reservedBytes} - ${operation.reservedBytes}`,
966
+ readyBytes: sql4`${workspaceVideoGenerationQuotas.readyBytes} + ${input.sizeBytes}`,
888
967
  updatedAt: now
889
968
  }).where(eq(workspaceVideoGenerationQuotas.workspaceId, input.workspaceId));
890
969
  const [updated] = await tx.update(videoGenerationOperations).set({
@@ -933,7 +1012,7 @@ async function listSessionGeneratedVideoArtifacts(db, workspaceId, sessionId) {
933
1012
  and(
934
1013
  eq(generatedVideoArtifacts.workspaceId, workspaceId),
935
1014
  eq(generatedVideoArtifacts.sessionId, sessionId),
936
- sql3`${generatedVideoArtifacts.deletedAt} is null`
1015
+ sql4`${generatedVideoArtifacts.deletedAt} is null`
937
1016
  )
938
1017
  ).orderBy(generatedVideoArtifacts.createdAt, generatedVideoArtifacts.id)
939
1018
  );
@@ -1076,7 +1155,7 @@ async function getVideoGenerationOperationSummary(db, workspaceId, operationId)
1076
1155
  workspaceId,
1077
1156
  (tx) => rawRows(
1078
1157
  tx,
1079
- sql3`SELECT nullif(current_setting('opengeni.private_file_owner',true),'') AS owner`
1158
+ sql4`SELECT nullif(current_setting('opengeni.private_file_owner',true),'') AS owner`
1080
1159
  )
1081
1160
  );
1082
1161
  if (context?.owner !== operation.privateFileOwnerSubjectId) return null;
@@ -1119,9 +1198,9 @@ async function mutateOperation(db, input, mutate) {
1119
1198
  async function releaseReservedQuota(tx, operation, ready) {
1120
1199
  if (operation.quotaState !== "reserved") return;
1121
1200
  await tx.update(workspaceVideoGenerationQuotas).set({
1122
- reservedBytes: sql3`${workspaceVideoGenerationQuotas.reservedBytes} - ${operation.reservedBytes}`,
1201
+ reservedBytes: sql4`${workspaceVideoGenerationQuotas.reservedBytes} - ${operation.reservedBytes}`,
1123
1202
  ...ready ? {
1124
- readyBytes: sql3`${workspaceVideoGenerationQuotas.readyBytes} + ${operation.reservedBytes}`
1203
+ readyBytes: sql4`${workspaceVideoGenerationQuotas.readyBytes} + ${operation.reservedBytes}`
1125
1204
  } : {},
1126
1205
  updatedAt: /* @__PURE__ */ new Date()
1127
1206
  }).where(eq(workspaceVideoGenerationQuotas.workspaceId, operation.workspaceId));
@@ -1133,7 +1212,7 @@ async function debitVideoGenerationCredits(tx, operation) {
1133
1212
  if (!operation.turnId) {
1134
1213
  throw new VideoGenerationCreditError("Paid video initiating turn is unavailable");
1135
1214
  }
1136
- await tx.execute(sql3`select pg_advisory_xact_lock(hashtext(${operation.accountId}))`);
1215
+ await tx.execute(sql4`select pg_advisory_xact_lock(hashtext(${operation.accountId}))`);
1137
1216
  const attribution = await creditDebitAttributionForTurn(tx, {
1138
1217
  accountId: operation.accountId,
1139
1218
  workspaceId: operation.workspaceId,
@@ -1145,10 +1224,7 @@ async function debitVideoGenerationCredits(tx, operation) {
1145
1224
  subjectId: attribution.kind === "turn" ? attribution.initiatingHumanSubjectId : null
1146
1225
  });
1147
1226
  if (refusal) throw Object.assign(new VideoGenerationCreditError(refusal.message), refusal);
1148
- const [balanceRow] = await tx.select({
1149
- balanceMicros: sql3`coalesce(sum(${creditLedgerEntries.amountMicros}), 0)`
1150
- }).from(creditLedgerEntries).where(eq(creditLedgerEntries.accountId, operation.accountId));
1151
- const balanceMicros = Number(balanceRow?.balanceMicros ?? 0);
1227
+ const { balanceMicros } = await getSpendableCreditBalance(tx, operation.accountId);
1152
1228
  if (!Number.isSafeInteger(balanceMicros) || balanceMicros < operation.pricedCostMicros) {
1153
1229
  throw new VideoGenerationCreditError();
1154
1230
  }
@@ -1192,7 +1268,7 @@ async function refundVideoGenerationCredits(tx, operation, now) {
1192
1268
  if (operation.fundingSource !== "opengeni_credits" || operation.pricedCostMicros <= 0) {
1193
1269
  throw new Error("Video generation credit refund has an invalid funding binding");
1194
1270
  }
1195
- await tx.execute(sql3`select pg_advisory_xact_lock(hashtext(${operation.accountId}))`);
1271
+ await tx.execute(sql4`select pg_advisory_xact_lock(hashtext(${operation.accountId}))`);
1196
1272
  const inserted = await tx.insert(creditLedgerEntries).values({
1197
1273
  accountId: operation.accountId,
1198
1274
  workspaceId: operation.workspaceId,
@@ -1301,6 +1377,10 @@ export {
1301
1377
  getWorkspaceUsage,
1302
1378
  checkWorkspaceAllowance,
1303
1379
  maintainWorkspaceAllowances,
1380
+ getBillingBalance,
1381
+ spendableCreditMicros,
1382
+ getSpendableCreditBalance,
1383
+ planCreditDebit,
1304
1384
  ACTIVE_VIDEO_GENERATION_STATUSES,
1305
1385
  VideoGenerationConflictError,
1306
1386
  VideoGenerationCapacityError,
@@ -1332,4 +1412,4 @@ export {
1332
1412
  mediaGenerationResultForStoredOperation,
1333
1413
  getVideoGenerationOperationSummary
1334
1414
  };
1335
- //# sourceMappingURL=chunk-MACELCXG.js.map
1415
+ //# sourceMappingURL=chunk-UCIMHNII.js.map