@vrplatform/kysely 1.3.45-6561 → 1.3.45-736

package/build/module/control-plane/index.d.ts

@@ -1,4 +1,5 @@
 import { type ColumnType, type Kysely } from 'kysely';
+import type { TenantDataRegion } from '../regional';
 export type Generated<T> = T extends ColumnType<infer S, infer I, infer U> ? ColumnType<S, I | undefined, U> : ColumnType<T, T | undefined, T>;
 export type Json = JsonValue;
 export type JsonArray = JsonValue[];
@@ -9,6 +10,11 @@ export type JsonPrimitive = boolean | number | string | null;
 export type JsonValue = JsonArray | JsonObject | JsonPrimitive;
 export type Int8 = ColumnType<string, number | string | bigint, number | string | bigint>;
 export type Timestamp = ColumnType<Date | string, Date | string, Date | string>;
+export type ControlPlaneOperationStatus = 'completed' | 'failed' | 'queued' | 'running';
+export type ControlPlaneOperationType = 'ach-verification' | 'bank-rule-run' | 'calendar-block' | 'calendar-unblock' | 'connection-archive' | 'connection-connect' | 'connection-extract' | 'connection-sync' | 'csv-import' | 'csv-import-preview' | 'export' | 'flow-run' | 'historical-gl-import' | 'historical-statement-import' | 'listing-journal-refresh' | 'plaid-initial-sync' | 'pms-cutover' | 'pms-cutover-preview' | 'provider-payment' | 'provider-payment-recovery' | 'recurring-transaction-run' | 'reservation-journal-refresh' | 'team-delete' | 'team-demo-data' | 'team-duplicate' | 'team-initialize' | 'transaction-journal-refresh' | 'vri-statement-import' | 'vri-team-migration' | 'webhook-test';
+export type ControlPlaneOperationResourceRole = 'result' | 'source' | 'target';
+export type ControlPlaneOperationResourceType = 'bank-rule' | 'connection' | 'contact' | 'file' | 'flow' | 'listing' | 'recurring-fee' | 'recurring-transaction-template' | 'reservation' | 'statement' | 'team' | 'transaction' | 'webhook';
+export type ControlPlaneOperationSourceType = 'destructive-job' | 'regional-operation';
 export interface ControlPlaneApp {
     authentication: Generated<Json>;
     category: string | null;
@@ -104,6 +110,92 @@ export interface ControlPlaneBookingChannelIconCandidate {
 export interface ControlPlaneCurrency {
     name: string;
 }
+export type DestructiveJobStep = 'completed' | 'completion_effects' | 'control_plane_deleted' | 'control_plane_deleting' | 'queued' | 'regional_deleted' | 'regional_deleting' | 'validated';
+export interface ControlPlaneDestructiveJob {
+    attempts: Generated<number>;
+    authorizedByTenantId: string;
+    availableAt: Generated<Timestamp>;
+    claimedAt: Timestamp | null;
+    claimedBy: string | null;
+    completedAt: Timestamp | null;
+    completionData: Json | null;
+    completionWarningCode: string | null;
+    completionWarningDetails: Json | null;
+    createdAt: Generated<Timestamp>;
+    dataRegion: TenantDataRegion;
+    failureCode: string | null;
+    failureDetails: Json | null;
+    force: Generated<boolean>;
+    id: Generated<string>;
+    jobType: 'tenant.delete';
+    maxAttempts: Generated<number>;
+    requestId: string | null;
+    requestedByActorType: 'api_key' | 'system' | 'user';
+    requestedByTenantId: string;
+    requestedByUserId: string | null;
+    startedAt: Timestamp | null;
+    status: Generated<'canceled' | 'completed' | 'failed' | 'queued' | 'running'>;
+    step: Generated<DestructiveJobStep>;
+    targetId: string;
+    triggerRunId: string | null;
+    updatedAt: Generated<Timestamp>;
+}
+export interface ControlPlaneDestructiveJobAttempt {
+    attemptNumber: number;
+    failureCode: string | null;
+    failureDetails: Json | null;
+    finishedAt: Timestamp | null;
+    id: Generated<string>;
+    jobId: string;
+    startedAt: Generated<Timestamp>;
+    status: Generated<'completed' | 'failed' | 'running' | 'superseded'>;
+    step: DestructiveJobStep;
+    triggerRunId: string | null;
+}
+export interface ControlPlaneDestructiveJobOperation {
+    action: 'abandon' | 'cancel' | 'retry';
+    actor: string;
+    createdAt: Generated<Timestamp>;
+    id: Generated<string>;
+    jobId: string;
+    previousStatus: string;
+    previousStep: string;
+    reasonCode: string;
+}
+export interface ControlPlaneOperation {
+    createdAt: Generated<Timestamp>;
+    failureCode: string | null;
+    failureMessage: string | null;
+    finishedAt: Timestamp | null;
+    id: Generated<string>;
+    initiatedByUserId: string | null;
+    requesterTenantId: string;
+    retainUntil: Timestamp | null;
+    sourceRegion: TenantDataRegion | null;
+    sourceVersion: Generated<number>;
+    startedAt: Timestamp | null;
+    status: Generated<ControlPlaneOperationStatus>;
+    type: ControlPlaneOperationType;
+    updatedAt: Generated<Timestamp>;
+}
+export interface ControlPlaneOperationResource {
+    createdAt: Generated<Timestamp>;
+    isPrimary: Generated<boolean>;
+    operationId: string;
+    resourceId: string;
+    resourceType: ControlPlaneOperationResourceType;
+    role: ControlPlaneOperationResourceRole;
+}
+export interface ControlPlaneOperationSource {
+    finishedAt: Timestamp | null;
+    operationId: string;
+    required: Generated<boolean>;
+    sourceId: string;
+    sourceType: ControlPlaneOperationSourceType;
+    sourceVersion: Generated<number>;
+    startedAt: Timestamp | null;
+    status: Generated<ControlPlaneOperationStatus>;
+}
 export interface ControlPlaneEvent {
     createdAt: Generated<Timestamp>;
     id: Generated<string>;
@@ -115,6 +207,27 @@ export interface ControlPlaneEvent {
     trackedAt: Generated<Timestamp | null>;
     userId: string | null;
 }
+export interface ControlPlaneTrackingQueueFailure {
+    bodyFingerprint: string | null;
+    capturedAt: Generated<Timestamp>;
+    dispositionAt: Timestamp | null;
+    dispositionHistory: Generated<Json[]>;
+    dispositionReasonCode: string | null;
+    dispositionActor: string | null;
+    environment: 'production' | 'staging';
+    event: Json | null;
+    eventId: string | null;
+    expiresAt: Timestamp;
+    failureCode: string;
+    id: Generated<string>;
+    lastReplayedAt: Timestamp | null;
+    queueMessageId: string;
+    replayCount: Generated<number>;
+    recurrenceCount: Generated<number>;
+    status: Generated<'captured' | 'discarded' | 'replayed'>;
+    updatedAt: Generated<Timestamp>;
+    validationIssues: Json | null;
+}
 export interface ControlPlaneConnectionRoute {
     connectionId: string;
     createdAt: Generated<Timestamp>;
@@ -262,6 +375,22 @@ export interface ControlPlanePartnerManagedTeamAssignment {
     partnerTenantId: string;
     userId: string;
 }
+export interface ControlPlaneOwnerPortal {
+    hostname: string;
+    tenantId: string;
+    status: Generated<'pending' | 'active' | 'disabled'>;
+    name: string;
+    logo: string | null;
+    colorPrimary: string;
+    showPlatformBranding: Generated<boolean>;
+    linksEnabled: Generated<boolean>;
+    sender: {
+        from: string;
+        replyTo: string;
+    } | null;
+    revision: Generated<number>;
+    updatedAt: Generated<Timestamp>;
+}
 export interface ControlPlaneTenant {
     billingPartnerId: string | null;
     billingPlan: string | null;
@@ -345,6 +474,9 @@ export interface ControlPlaneDB {
     'controlPlane.connectionRoute': ControlPlaneConnectionRoute;
     'controlPlane.controlPlaneOutbox': ControlPlaneOutbox;
     'controlPlane.currency': ControlPlaneCurrency;
+    'controlPlane.destructiveJob': ControlPlaneDestructiveJob;
+    'controlPlane.destructiveJobAttempt': ControlPlaneDestructiveJobAttempt;
+    'controlPlane.destructiveJobOperation': ControlPlaneDestructiveJobOperation;
     'controlPlane.events': ControlPlaneEvent;
     'controlPlane.feature': ControlPlaneFeature;
     'controlPlane.featureApproval': ControlPlaneFeatureApproval;
@@ -353,12 +485,17 @@ export interface ControlPlaneDB {
     'controlPlane.generalLedgerDefaultVersion': ControlPlaneGeneralLedgerDefaultVersion;
     'controlPlane.lineClassification': ControlPlaneLineClassification;
     'controlPlane.lineClassificationTeamUsage': ControlPlaneLineClassificationTeamUsage;
+    'controlPlane.operation': ControlPlaneOperation;
+    'controlPlane.operationResource': ControlPlaneOperationResource;
+    'controlPlane.operationSource': ControlPlaneOperationSource;
     'controlPlane.partnerManagedTeamAssignment': ControlPlanePartnerManagedTeamAssignment;
     'controlPlane.reservationLineDefault': ControlPlaneReservationLineDefault;
     'controlPlane.rolloutMonitorSnapshot': ControlPlaneRolloutMonitorSnapshot;
+    'controlPlane.ownerPortal': ControlPlaneOwnerPortal;
     'controlPlane.tenant': ControlPlaneTenant;
     'controlPlane.tenantRegionMigration': ControlPlaneTenantRegionMigration;
     'controlPlane.tenantUser': ControlPlaneTenantUser;
+    'controlPlane.trackingQueueFailure': ControlPlaneTrackingQueueFailure;
     'controlPlane.token': ControlPlaneToken;
     'controlPlane.transactionLineMapping': ControlPlaneTransactionLineMapping;
     'controlPlane.user': ControlPlaneUser;

package/build/module/control-plane/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["control-plane/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,cAAc,EAGd,wBAAwB,GAQzB,MAAM,QAAQ,CAAC;AA4ZhB,MAAM,6BAA8B,SAAQ,wBAAwB;IAEvD;IACA;IAFX,YACW,MAAc,EACd,2BAAmC;QAE5C,KAAK,EAAE,CAAC;QAHC,WAAM,GAAN,MAAM,CAAQ;QACd,gCAA2B,GAA3B,2BAA2B,CAAQ;IAG9C,CAAC;IAEkB,4BAA4B,CAC7C,IAA6B,EAC7B,OAAgB;QAEhB,MAAM,WAAW,GAAG,KAAK,CAAC,4BAA4B,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACtE,IACE,WAAW,CAAC,MAAM,EAAE,IAAI,KAAK,cAAc;YAC3C,WAAW,CAAC,MAAM,EAAE,IAAI,KAAK,eAAe,EAC5C,CAAC;YACD,OAAO,EAAE,GAAG,WAAW,EAAE,MAAM,EAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACxE,CAAC;QACD,IACE,WAAW,CAAC,MAAM,EAAE,IAAI,KAAK,kBAAkB;YAC/C,WAAW,CAAC,MAAM,EAAE,IAAI,KAAK,oBAAoB,EACjD,CAAC;YACD,OAAO;gBACL,GAAG,WAAW;gBACd,MAAM,EAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,CAAC;aAChE,CAAC;QACJ,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAEkB,iBAAiB,CAClC,IAAkB,EAClB,OAAgB;QAEhB,MAAM,WAAW,GAAG,KAAK,CAAC,iBAAiB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC3D,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC;YAC1C,CAAC,CAAC;gBACE,GAAG,WAAW;gBACd,IAAI,EAAE,GAAG,IAAI,CAAC,MAAM,IAAI,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE;aACzE;YACH,CAAC,CAAC,WAAW,CAAC;IAClB,CAAC;CACF;AAED,MAAM,wBAAwB;IACnB,YAAY,CAAgC;IAErD,YAAY,MAAc,EAAE,2BAAmC;QAC7D,IAAI,CAAC,YAAY,GAAG,IAAI,6BAA6B,CACnD,MAAM,EACN,2BAA2B,CAC5B,CAAC;IACJ,CAAC;IAED,cAAc,CAAC,IAA8B;QAC3C,OAAO,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,IAA+B;QAE/B,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;CACF;AAED,MAAM,UAAU,sBAAsB,CACpC,MAA8B,EAC9B,MAAc,EACd,2BAA2B,GAAG,MAAM;IAEpC,OAAO,MAAM,CAAC,UAAU,CACtB,IAAI,wBAAwB,CAAC,MAAM,EAAE,2BAA2B,CAAC,CAClE,CAAC;AACJ,CAAC","sourcesContent":["import {\n  type ColumnType,\n  type FunctionNode,\n  IdentifierNode,\n  type Kysely,\n  type KyselyPlugin,\n  OperationNodeTransformer,\n  type PluginTransformQueryArgs,\n  type PluginTransformResultArgs,\n  type QueryId,\n  type QueryResult,\n  type RootOperationNode,\n  type SchemableIdentifierNode,\n  type UnknownRow,\n} from 'kysely';\n\nexport type Generated<T> =\n  T extends ColumnType<infer S, infer I, infer U>\n    ? ColumnType<S, I | undefined, U>\n    : ColumnType<T, T | undefined, T>;\n\nexport type Json = JsonValue;\n\nexport type JsonArray = JsonValue[];\n\nexport type JsonObject = {\n  [x: string]: JsonValue | undefined;\n};\n\nexport type JsonPrimitive = boolean | number | string | null;\n\nexport type JsonValue = JsonArray | JsonObject | JsonPrimitive;\n\nexport type Int8 = ColumnType<\n  string,\n  number | string | bigint,\n  number | string | bigint\n>;\n\nexport type Timestamp = ColumnType<Date | string, Date | string, Date | string>;\n\nexport interface ControlPlaneApp {\n  authentication: Generated<Json>;\n  category: string | null;\n  color: string | null;\n  createdAt: Generated<Timestamp>;\n  icon: string | null;\n  iconRound: string | null;\n  id: Generated<string>;\n  info: Json | null;\n  name: string;\n  type: string;\n  version: Generated<number>;\n}\n\nexport interface ControlPlaneAccountingIntegrityEvaluation {\n  caseCount: Generated<Int8>;\n  completedAt: Timestamp | null;\n  completedRunCount: Generated<number>;\n  createdAt: Generated<Timestamp>;\n  environment: string;\n  error: Json | null;\n  expectedRunCount: number;\n  failedRunCount: Generated<number>;\n  findingRowCount: Generated<Int8>;\n  gitRevision: string;\n  id: string;\n  results: Generated<Json>;\n  rowsScanned: Generated<Int8>;\n  skippedRunCount: Generated<number>;\n  source: string;\n  startedAt: Timestamp;\n  status: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneAccountingIntegrityDefinitionDraft {\n  artifact: Json;\n  artifactHash: string;\n  code: string;\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  ownerId: string;\n  previewResult: Json | null;\n  rationale: string;\n  revision: Generated<number>;\n  status: string;\n  templateId: string;\n  templateVersion: number;\n  title: string;\n  updatedAt: Generated<Timestamp>;\n  validationResult: Json | null;\n}\n\nexport interface ControlPlaneAccountingIntegrityDefinitionVersion {\n  artifact: Json;\n  artifactHash: string;\n  code: string;\n  draftId: string;\n  id: Generated<string>;\n  publishedAt: Generated<Timestamp>;\n  publishedBy: string;\n  rollout: string;\n  templateId: string;\n  templateVersion: number;\n  version: number;\n}\n\nexport interface ControlPlaneBookingChannel {\n  channelRef: string | null;\n  color: string | null;\n  createdAt: Generated<Timestamp>;\n  icon: string | null;\n  iconProvider: string | null;\n  id: Generated<string>;\n  lastGenerateRun: Timestamp | null;\n  logo: string | null;\n  logoProvider: string | null;\n  selectedBookingChannelIconCandidateId: string | null;\n  selectedBookingChannelLogoCandidateId: string | null;\n  uniqueRef: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneBookingChannelIconCandidate {\n  bookingChannelId: string;\n  cloudflareImageId: string | null;\n  color: string | null;\n  comment: string | null;\n  createdAt: Generated<Timestamp>;\n  externalIcon: string | null;\n  height: number | null;\n  id: Generated<string>;\n  provider: Generated<string>;\n  source: string | null;\n  updatedAt: Generated<Timestamp>;\n  width: number | null;\n}\n\nexport interface ControlPlaneCurrency {\n  name: string;\n}\n\nexport interface ControlPlaneEvent {\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  name: string;\n  packageName: string;\n  props: Json;\n  tenantId: string | null;\n  timestamp: Generated<Timestamp>;\n  trackedAt: Generated<Timestamp | null>;\n  userId: string | null;\n}\n\nexport interface ControlPlaneConnectionRoute {\n  connectionId: string;\n  createdAt: Generated<Timestamp>;\n  tenantId: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneFeature {\n  createdAt: Generated<Timestamp>;\n  description: string | null;\n  id: Generated<string>;\n  issueUrl: string | null;\n  requiredApprovals: Generated<Json | null>;\n  status: Generated<string | null>;\n  tenantId: string | null;\n  title: string;\n  updatedAt: Generated<Timestamp>;\n  url: string | null;\n}\n\nexport interface ControlPlaneFeatureEnabledTeam {\n  createdAt: Generated<Timestamp>;\n  featureId: string;\n  id: Generated<string>;\n  status: Generated<string | null>;\n  tenantId: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneFeatureApproval {\n  createdAt: Generated<Timestamp>;\n  featureId: string;\n  id: Generated<string>;\n  status: Generated<string | null>;\n  updatedAt: Generated<Timestamp>;\n  userId: string;\n}\n\nexport interface ControlPlaneFlow {\n  appId: string;\n  createdAt: Generated<Timestamp>;\n  description: string | null;\n  eventListening: Generated<Json>;\n  id: Generated<string>;\n  isPublic: Generated<boolean>;\n  leftType: string | null;\n  mappingSchema: Generated<Json>;\n  revision: Generated<number>;\n  rightType: string | null;\n  runSchema: Generated<Json>;\n  settingSchema: Generated<Json>;\n  title: string;\n  type: 'pull' | 'push' | null;\n  uniqueRef: string;\n  updatedAt: Generated<Timestamp>;\n  useGlobalMapping: Generated<boolean>;\n}\n\nexport interface ControlPlaneIssueMessageOverwrite {\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  message: string;\n  pattern: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneLineClassification {\n  appConflict: Generated<boolean>;\n  appId: string | null;\n  createdAt: Generated<Timestamp>;\n  firstSeenAt: Timestamp;\n  id: Generated<string>;\n  kind: string;\n  lastSeenAt: Timestamp;\n  name: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneLineClassificationTeamUsage {\n  classificationId: string;\n  createdAt: Generated<Timestamp>;\n  dataRegion: string;\n  firstSeenAt: Timestamp;\n  lastSeenAt: Timestamp;\n  mappingStatus: string | null;\n  revision: number;\n  tenantId: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneReservationLineDefault {\n  accountKey: string | null;\n  bookingChannel: string | null;\n  bookingChannelKey: Generated<string>;\n  classificationId: string;\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  partnerTenantId: string | null;\n  revenueRecognition: string | null;\n  revision: Generated<number>;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneRolloutMonitorSnapshot {\n  monitorName: string;\n  result: Json;\n  snapshotDate: Timestamp;\n}\n\nexport interface ControlPlaneTransactionLineMapping {\n  accountAssignmentType: string | null;\n  createdAt: Generated<Timestamp>;\n  deletedAt: Timestamp | null;\n  id: Generated<string>;\n  lineType: string;\n  revision: Generated<number>;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneGeneralLedgerDefaultSet {\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  partnerTenantId: string | null;\n  publishedVersion: number | null;\n  sourceDataRegion: string;\n  sourceDigest: string | null;\n  sourceTenantId: string;\n  templateKey: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneGeneralLedgerDefaultVersion {\n  dataJson: Json;\n  defaultSetId: string;\n  id: Generated<string>;\n  publishedAt: Generated<Timestamp>;\n  sourceDigest: string;\n  sourceTenantId: string;\n  version: number;\n}\n\nexport interface ControlPlaneOutbox {\n  aggregateId: string;\n  aggregateType: string;\n  attempts: Generated<number>;\n  createdAt: Generated<Timestamp>;\n  dataRegion: string | null;\n  eventType: string;\n  eventVersion: number;\n  id: Generated<string>;\n  lastError: string | null;\n  payload: Generated<Json>;\n  publishedAt: Timestamp | null;\n}\n\nexport interface ControlPlanePartnerManagedTeamAssignment {\n  createdAt: Generated<Timestamp>;\n  managedTeamId: string;\n  partnerTenantId: string;\n  userId: string;\n}\n\nexport interface ControlPlaneTenant {\n  billingPartnerId: string | null;\n  billingPlan: string | null;\n  createdAt: Generated<Timestamp>;\n  dataRegion: Generated<string>;\n  id: Generated<string>;\n  migrationStatus: Generated<string>;\n  name: string;\n  partnerId: string | null;\n  partnerDeniedPermissions: Generated<string[]>;\n  slug: string;\n  status: Generated<string | null>;\n  storageRealm: Generated<string>;\n  type: Generated<string | null>;\n  uniqueRef: string | null;\n  updatedAt: Generated<Timestamp | null>;\n}\n\nexport interface ControlPlaneTenantRegionMigration {\n  actor: string;\n  completedAt: Timestamp | null;\n  error: Json | null;\n  id: string;\n  reason: string;\n  sourceManifest: Json | null;\n  sourceRegion: string;\n  startedAt: Generated<Timestamp>;\n  status: Generated<string>;\n  targetManifest: Json | null;\n  targetRegion: string;\n  tenantId: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneTenantUser {\n  createdAt: Generated<Timestamp | null>;\n  id: Generated<string>;\n  lastInvitedAt: Timestamp | null;\n  managedTeamAccess: Generated<'all' | 'assigned'>;\n  permissionBundles: Generated<string[]>;\n  role: Generated<string | null>;\n  status: Generated<string | null>;\n  tenantId: string;\n  updatedAt: Generated<Timestamp | null>;\n  userId: string;\n}\n\nexport interface ControlPlaneToken {\n  createdAt: Generated<Timestamp>;\n  expiresAt: Timestamp | null;\n  id: Generated<string>;\n  nanoId: Generated<string>;\n  payload: Generated<Json>;\n  tenantId: string | null;\n  type: string | null;\n  userId: string | null;\n}\n\nexport interface ControlPlaneUser {\n  clerkId: string | null;\n  createdAt: Generated<Timestamp | null>;\n  email: string | null;\n  firstName: string | null;\n  id: Generated<string>;\n  internalOperationGrants: Generated<string[]>;\n  isAdmin: Generated<boolean>;\n  lastName: string | null;\n  lastSeen: Timestamp | null;\n  name: string | null;\n  partnerId: string | null;\n  phone: string | null;\n  secondaryEmails: Generated<Json>;\n  status: Generated<string | null>;\n  sub: string | null;\n  type: Generated<string | null>;\n  updatedAt: Generated<Timestamp | null>;\n}\n\nexport interface ControlPlaneDB {\n  'controlPlane.accountingIntegrityDefinitionDraft': ControlPlaneAccountingIntegrityDefinitionDraft;\n  'controlPlane.accountingIntegrityDefinitionVersion': ControlPlaneAccountingIntegrityDefinitionVersion;\n  'controlPlane.accountingIntegrityEvaluation': ControlPlaneAccountingIntegrityEvaluation;\n  'controlPlane.app': ControlPlaneApp;\n  'controlPlane.bookingChannel': ControlPlaneBookingChannel;\n  'controlPlane.bookingChannelIconCandidate': ControlPlaneBookingChannelIconCandidate;\n  'controlPlane.connectionRoute': ControlPlaneConnectionRoute;\n  'controlPlane.controlPlaneOutbox': ControlPlaneOutbox;\n  'controlPlane.currency': ControlPlaneCurrency;\n  'controlPlane.events': ControlPlaneEvent;\n  'controlPlane.feature': ControlPlaneFeature;\n  'controlPlane.featureApproval': ControlPlaneFeatureApproval;\n  'controlPlane.featureEnabledTeam': ControlPlaneFeatureEnabledTeam;\n  'controlPlane.generalLedgerDefaultSet': ControlPlaneGeneralLedgerDefaultSet;\n  'controlPlane.generalLedgerDefaultVersion': ControlPlaneGeneralLedgerDefaultVersion;\n  'controlPlane.lineClassification': ControlPlaneLineClassification;\n  'controlPlane.lineClassificationTeamUsage': ControlPlaneLineClassificationTeamUsage;\n  'controlPlane.partnerManagedTeamAssignment': ControlPlanePartnerManagedTeamAssignment;\n  'controlPlane.reservationLineDefault': ControlPlaneReservationLineDefault;\n  'controlPlane.rolloutMonitorSnapshot': ControlPlaneRolloutMonitorSnapshot;\n  'controlPlane.tenant': ControlPlaneTenant;\n  'controlPlane.tenantRegionMigration': ControlPlaneTenantRegionMigration;\n  'controlPlane.tenantUser': ControlPlaneTenantUser;\n  'controlPlane.token': ControlPlaneToken;\n  'controlPlane.transactionLineMapping': ControlPlaneTransactionLineMapping;\n  'controlPlane.user': ControlPlaneUser;\n  'controlPlaneCore.flow': ControlPlaneFlow;\n  'controlPlaneCore.issueMessageOverwrite': ControlPlaneIssueMessageOverwrite;\n}\n\nclass ControlPlaneSchemaTransformer extends OperationNodeTransformer {\n  constructor(\n    readonly schema: string,\n    readonly issueMessageOverwriteSchema: string\n  ) {\n    super();\n  }\n\n  protected override transformSchemableIdentifier(\n    node: SchemableIdentifierNode,\n    queryId: QueryId\n  ): SchemableIdentifierNode {\n    const transformed = super.transformSchemableIdentifier(node, queryId);\n    if (\n      transformed.schema?.name === 'controlPlane' ||\n      transformed.schema?.name === 'control_plane'\n    ) {\n      return { ...transformed, schema: IdentifierNode.create(this.schema) };\n    }\n    if (\n      transformed.schema?.name === 'controlPlaneCore' ||\n      transformed.schema?.name === 'control_plane_core'\n    ) {\n      return {\n        ...transformed,\n        schema: IdentifierNode.create(this.issueMessageOverwriteSchema),\n      };\n    }\n    return transformed;\n  }\n\n  protected override transformFunction(\n    node: FunctionNode,\n    queryId: QueryId\n  ): FunctionNode {\n    const transformed = super.transformFunction(node, queryId);\n    return node.func.startsWith('controlPlane.')\n      ? {\n          ...transformed,\n          func: `${this.schema}.${transformed.func.slice('controlPlane.'.length)}`,\n        }\n      : transformed;\n  }\n}\n\nclass ControlPlaneSchemaPlugin implements KyselyPlugin {\n  readonly #transformer: ControlPlaneSchemaTransformer;\n\n  constructor(schema: string, issueMessageOverwriteSchema: string) {\n    this.#transformer = new ControlPlaneSchemaTransformer(\n      schema,\n      issueMessageOverwriteSchema\n    );\n  }\n\n  transformQuery(args: PluginTransformQueryArgs): RootOperationNode {\n    return this.#transformer.transformNode(args.node, args.queryId);\n  }\n\n  async transformResult(\n    args: PluginTransformResultArgs\n  ): Promise<QueryResult<UnknownRow>> {\n    return args.result;\n  }\n}\n\nexport function withControlPlaneSchema(\n  kysely: Kysely<ControlPlaneDB>,\n  schema: string,\n  issueMessageOverwriteSchema = 'core'\n) {\n  return kysely.withPlugin(\n    new ControlPlaneSchemaPlugin(schema, issueMessageOverwriteSchema)\n  );\n}\n"]}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["control-plane/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,cAAc,EAGd,wBAAwB,GAQzB,MAAM,QAAQ,CAAC;AAymBhB,MAAM,6BAA8B,SAAQ,wBAAwB;IAEvD;IACA;IAFX,YACW,MAAc,EACd,2BAAmC;QAE5C,KAAK,EAAE,CAAC;QAHC,WAAM,GAAN,MAAM,CAAQ;QACd,gCAA2B,GAA3B,2BAA2B,CAAQ;IAG9C,CAAC;IAEkB,4BAA4B,CAC7C,IAA6B,EAC7B,OAAgB;QAEhB,MAAM,WAAW,GAAG,KAAK,CAAC,4BAA4B,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACtE,IACE,WAAW,CAAC,MAAM,EAAE,IAAI,KAAK,cAAc;YAC3C,WAAW,CAAC,MAAM,EAAE,IAAI,KAAK,eAAe,EAC5C,CAAC;YACD,OAAO,EAAE,GAAG,WAAW,EAAE,MAAM,EAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACxE,CAAC;QACD,IACE,WAAW,CAAC,MAAM,EAAE,IAAI,KAAK,kBAAkB;YAC/C,WAAW,CAAC,MAAM,EAAE,IAAI,KAAK,oBAAoB,EACjD,CAAC;YACD,OAAO;gBACL,GAAG,WAAW;gBACd,MAAM,EAAE,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,CAAC;aAChE,CAAC;QACJ,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAEkB,iBAAiB,CAClC,IAAkB,EAClB,OAAgB;QAEhB,MAAM,WAAW,GAAG,KAAK,CAAC,iBAAiB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC3D,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC;YAC1C,CAAC,CAAC;gBACE,GAAG,WAAW;gBACd,IAAI,EAAE,GAAG,IAAI,CAAC,MAAM,IAAI,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE;aACzE;YACH,CAAC,CAAC,WAAW,CAAC;IAClB,CAAC;CACF;AAED,MAAM,wBAAwB;IACnB,YAAY,CAAgC;IAErD,YAAY,MAAc,EAAE,2BAAmC;QAC7D,IAAI,CAAC,YAAY,GAAG,IAAI,6BAA6B,CACnD,MAAM,EACN,2BAA2B,CAC5B,CAAC;IACJ,CAAC;IAED,cAAc,CAAC,IAA8B;QAC3C,OAAO,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,IAA+B;QAE/B,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;CACF;AAED,MAAM,UAAU,sBAAsB,CACpC,MAA8B,EAC9B,MAAc,EACd,2BAA2B,GAAG,MAAM;IAEpC,OAAO,MAAM,CAAC,UAAU,CACtB,IAAI,wBAAwB,CAAC,MAAM,EAAE,2BAA2B,CAAC,CAClE,CAAC;AACJ,CAAC","sourcesContent":["import {\n  type ColumnType,\n  type FunctionNode,\n  IdentifierNode,\n  type Kysely,\n  type KyselyPlugin,\n  OperationNodeTransformer,\n  type PluginTransformQueryArgs,\n  type PluginTransformResultArgs,\n  type QueryId,\n  type QueryResult,\n  type RootOperationNode,\n  type SchemableIdentifierNode,\n  type UnknownRow,\n} from 'kysely';\nimport type { TenantDataRegion } from '../regional';\n\nexport type Generated<T> =\n  T extends ColumnType<infer S, infer I, infer U>\n    ? ColumnType<S, I | undefined, U>\n    : ColumnType<T, T | undefined, T>;\n\nexport type Json = JsonValue;\n\nexport type JsonArray = JsonValue[];\n\nexport type JsonObject = {\n  [x: string]: JsonValue | undefined;\n};\n\nexport type JsonPrimitive = boolean | number | string | null;\n\nexport type JsonValue = JsonArray | JsonObject | JsonPrimitive;\n\nexport type Int8 = ColumnType<\n  string,\n  number | string | bigint,\n  number | string | bigint\n>;\n\nexport type Timestamp = ColumnType<Date | string, Date | string, Date | string>;\n\nexport type ControlPlaneOperationStatus =\n  | 'completed'\n  | 'failed'\n  | 'queued'\n  | 'running';\n\nexport type ControlPlaneOperationType =\n  | 'ach-verification'\n  | 'bank-rule-run'\n  | 'calendar-block'\n  | 'calendar-unblock'\n  | 'connection-archive'\n  | 'connection-connect'\n  | 'connection-extract'\n  | 'connection-sync'\n  | 'csv-import'\n  | 'csv-import-preview'\n  | 'export'\n  | 'flow-run'\n  | 'historical-gl-import'\n  | 'historical-statement-import'\n  | 'listing-journal-refresh'\n  | 'plaid-initial-sync'\n  | 'pms-cutover'\n  | 'pms-cutover-preview'\n  | 'provider-payment'\n  | 'provider-payment-recovery'\n  | 'recurring-transaction-run'\n  | 'reservation-journal-refresh'\n  | 'team-delete'\n  | 'team-demo-data'\n  | 'team-duplicate'\n  | 'team-initialize'\n  | 'transaction-journal-refresh'\n  | 'vri-statement-import'\n  | 'vri-team-migration'\n  | 'webhook-test';\n\nexport type ControlPlaneOperationResourceRole = 'result' | 'source' | 'target';\n\nexport type ControlPlaneOperationResourceType =\n  | 'bank-rule'\n  | 'connection'\n  | 'contact'\n  | 'file'\n  | 'flow'\n  | 'listing'\n  | 'recurring-fee'\n  | 'recurring-transaction-template'\n  | 'reservation'\n  | 'statement'\n  | 'team'\n  | 'transaction'\n  | 'webhook';\n\nexport type ControlPlaneOperationSourceType =\n  | 'destructive-job'\n  | 'regional-operation';\n\nexport interface ControlPlaneApp {\n  authentication: Generated<Json>;\n  category: string | null;\n  color: string | null;\n  createdAt: Generated<Timestamp>;\n  icon: string | null;\n  iconRound: string | null;\n  id: Generated<string>;\n  info: Json | null;\n  name: string;\n  type: string;\n  version: Generated<number>;\n}\n\nexport interface ControlPlaneAccountingIntegrityEvaluation {\n  caseCount: Generated<Int8>;\n  completedAt: Timestamp | null;\n  completedRunCount: Generated<number>;\n  createdAt: Generated<Timestamp>;\n  environment: string;\n  error: Json | null;\n  expectedRunCount: number;\n  failedRunCount: Generated<number>;\n  findingRowCount: Generated<Int8>;\n  gitRevision: string;\n  id: string;\n  results: Generated<Json>;\n  rowsScanned: Generated<Int8>;\n  skippedRunCount: Generated<number>;\n  source: string;\n  startedAt: Timestamp;\n  status: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneAccountingIntegrityDefinitionDraft {\n  artifact: Json;\n  artifactHash: string;\n  code: string;\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  ownerId: string;\n  previewResult: Json | null;\n  rationale: string;\n  revision: Generated<number>;\n  status: string;\n  templateId: string;\n  templateVersion: number;\n  title: string;\n  updatedAt: Generated<Timestamp>;\n  validationResult: Json | null;\n}\n\nexport interface ControlPlaneAccountingIntegrityDefinitionVersion {\n  artifact: Json;\n  artifactHash: string;\n  code: string;\n  draftId: string;\n  id: Generated<string>;\n  publishedAt: Generated<Timestamp>;\n  publishedBy: string;\n  rollout: string;\n  templateId: string;\n  templateVersion: number;\n  version: number;\n}\n\nexport interface ControlPlaneBookingChannel {\n  channelRef: string | null;\n  color: string | null;\n  createdAt: Generated<Timestamp>;\n  icon: string | null;\n  iconProvider: string | null;\n  id: Generated<string>;\n  lastGenerateRun: Timestamp | null;\n  logo: string | null;\n  logoProvider: string | null;\n  selectedBookingChannelIconCandidateId: string | null;\n  selectedBookingChannelLogoCandidateId: string | null;\n  uniqueRef: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneBookingChannelIconCandidate {\n  bookingChannelId: string;\n  cloudflareImageId: string | null;\n  color: string | null;\n  comment: string | null;\n  createdAt: Generated<Timestamp>;\n  externalIcon: string | null;\n  height: number | null;\n  id: Generated<string>;\n  provider: Generated<string>;\n  source: string | null;\n  updatedAt: Generated<Timestamp>;\n  width: number | null;\n}\n\nexport interface ControlPlaneCurrency {\n  name: string;\n}\n\nexport type DestructiveJobStep =\n  | 'completed'\n  | 'completion_effects'\n  | 'control_plane_deleted'\n  | 'control_plane_deleting'\n  | 'queued'\n  | 'regional_deleted'\n  | 'regional_deleting'\n  | 'validated';\n\nexport interface ControlPlaneDestructiveJob {\n  attempts: Generated<number>;\n  authorizedByTenantId: string;\n  availableAt: Generated<Timestamp>;\n  claimedAt: Timestamp | null;\n  claimedBy: string | null;\n  completedAt: Timestamp | null;\n  completionData: Json | null;\n  completionWarningCode: string | null;\n  completionWarningDetails: Json | null;\n  createdAt: Generated<Timestamp>;\n  dataRegion: TenantDataRegion;\n  failureCode: string | null;\n  failureDetails: Json | null;\n  force: Generated<boolean>;\n  id: Generated<string>;\n  jobType: 'tenant.delete';\n  maxAttempts: Generated<number>;\n  requestId: string | null;\n  requestedByActorType: 'api_key' | 'system' | 'user';\n  requestedByTenantId: string;\n  requestedByUserId: string | null;\n  startedAt: Timestamp | null;\n  status: Generated<'canceled' | 'completed' | 'failed' | 'queued' | 'running'>;\n  step: Generated<DestructiveJobStep>;\n  targetId: string;\n  triggerRunId: string | null;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneDestructiveJobAttempt {\n  attemptNumber: number;\n  failureCode: string | null;\n  failureDetails: Json | null;\n  finishedAt: Timestamp | null;\n  id: Generated<string>;\n  jobId: string;\n  startedAt: Generated<Timestamp>;\n  status: Generated<'completed' | 'failed' | 'running' | 'superseded'>;\n  step: DestructiveJobStep;\n  triggerRunId: string | null;\n}\n\nexport interface ControlPlaneDestructiveJobOperation {\n  action: 'abandon' | 'cancel' | 'retry';\n  actor: string;\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  jobId: string;\n  previousStatus: string;\n  previousStep: string;\n  reasonCode: string;\n}\n\nexport interface ControlPlaneOperation {\n  createdAt: Generated<Timestamp>;\n  failureCode: string | null;\n  failureMessage: string | null;\n  finishedAt: Timestamp | null;\n  id: Generated<string>;\n  initiatedByUserId: string | null;\n  requesterTenantId: string;\n  retainUntil: Timestamp | null;\n  sourceRegion: TenantDataRegion | null;\n  sourceVersion: Generated<number>;\n  startedAt: Timestamp | null;\n  status: Generated<ControlPlaneOperationStatus>;\n  type: ControlPlaneOperationType;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneOperationResource {\n  createdAt: Generated<Timestamp>;\n  isPrimary: Generated<boolean>;\n  operationId: string;\n  resourceId: string;\n  resourceType: ControlPlaneOperationResourceType;\n  role: ControlPlaneOperationResourceRole;\n}\n\nexport interface ControlPlaneOperationSource {\n  finishedAt: Timestamp | null;\n  operationId: string;\n  required: Generated<boolean>;\n  sourceId: string;\n  sourceType: ControlPlaneOperationSourceType;\n  sourceVersion: Generated<number>;\n  startedAt: Timestamp | null;\n  status: Generated<ControlPlaneOperationStatus>;\n}\n\nexport interface ControlPlaneEvent {\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  name: string;\n  packageName: string;\n  props: Json;\n  tenantId: string | null;\n  timestamp: Generated<Timestamp>;\n  trackedAt: Generated<Timestamp | null>;\n  userId: string | null;\n}\n\nexport interface ControlPlaneTrackingQueueFailure {\n  bodyFingerprint: string | null;\n  capturedAt: Generated<Timestamp>;\n  dispositionAt: Timestamp | null;\n  dispositionHistory: Generated<Json[]>;\n  dispositionReasonCode: string | null;\n  dispositionActor: string | null;\n  environment: 'production' | 'staging';\n  event: Json | null;\n  eventId: string | null;\n  expiresAt: Timestamp;\n  failureCode: string;\n  id: Generated<string>;\n  lastReplayedAt: Timestamp | null;\n  queueMessageId: string;\n  replayCount: Generated<number>;\n  recurrenceCount: Generated<number>;\n  status: Generated<'captured' | 'discarded' | 'replayed'>;\n  updatedAt: Generated<Timestamp>;\n  validationIssues: Json | null;\n}\n\nexport interface ControlPlaneConnectionRoute {\n  connectionId: string;\n  createdAt: Generated<Timestamp>;\n  tenantId: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneFeature {\n  createdAt: Generated<Timestamp>;\n  description: string | null;\n  id: Generated<string>;\n  issueUrl: string | null;\n  requiredApprovals: Generated<Json | null>;\n  status: Generated<string | null>;\n  tenantId: string | null;\n  title: string;\n  updatedAt: Generated<Timestamp>;\n  url: string | null;\n}\n\nexport interface ControlPlaneFeatureEnabledTeam {\n  createdAt: Generated<Timestamp>;\n  featureId: string;\n  id: Generated<string>;\n  status: Generated<string | null>;\n  tenantId: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneFeatureApproval {\n  createdAt: Generated<Timestamp>;\n  featureId: string;\n  id: Generated<string>;\n  status: Generated<string | null>;\n  updatedAt: Generated<Timestamp>;\n  userId: string;\n}\n\nexport interface ControlPlaneFlow {\n  appId: string;\n  createdAt: Generated<Timestamp>;\n  description: string | null;\n  eventListening: Generated<Json>;\n  id: Generated<string>;\n  isPublic: Generated<boolean>;\n  leftType: string | null;\n  mappingSchema: Generated<Json>;\n  revision: Generated<number>;\n  rightType: string | null;\n  runSchema: Generated<Json>;\n  settingSchema: Generated<Json>;\n  title: string;\n  type: 'pull' | 'push' | null;\n  uniqueRef: string;\n  updatedAt: Generated<Timestamp>;\n  useGlobalMapping: Generated<boolean>;\n}\n\nexport interface ControlPlaneIssueMessageOverwrite {\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  message: string;\n  pattern: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneLineClassification {\n  appConflict: Generated<boolean>;\n  appId: string | null;\n  createdAt: Generated<Timestamp>;\n  firstSeenAt: Timestamp;\n  id: Generated<string>;\n  kind: string;\n  lastSeenAt: Timestamp;\n  name: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneLineClassificationTeamUsage {\n  classificationId: string;\n  createdAt: Generated<Timestamp>;\n  dataRegion: string;\n  firstSeenAt: Timestamp;\n  lastSeenAt: Timestamp;\n  mappingStatus: string | null;\n  revision: number;\n  tenantId: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneReservationLineDefault {\n  accountKey: string | null;\n  bookingChannel: string | null;\n  bookingChannelKey: Generated<string>;\n  classificationId: string;\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  partnerTenantId: string | null;\n  revenueRecognition: string | null;\n  revision: Generated<number>;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneRolloutMonitorSnapshot {\n  monitorName: string;\n  result: Json;\n  snapshotDate: Timestamp;\n}\n\nexport interface ControlPlaneTransactionLineMapping {\n  accountAssignmentType: string | null;\n  createdAt: Generated<Timestamp>;\n  deletedAt: Timestamp | null;\n  id: Generated<string>;\n  lineType: string;\n  revision: Generated<number>;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneGeneralLedgerDefaultSet {\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  partnerTenantId: string | null;\n  publishedVersion: number | null;\n  sourceDataRegion: string;\n  sourceDigest: string | null;\n  sourceTenantId: string;\n  templateKey: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneGeneralLedgerDefaultVersion {\n  dataJson: Json;\n  defaultSetId: string;\n  id: Generated<string>;\n  publishedAt: Generated<Timestamp>;\n  sourceDigest: string;\n  sourceTenantId: string;\n  version: number;\n}\n\nexport interface ControlPlaneOutbox {\n  aggregateId: string;\n  aggregateType: string;\n  attempts: Generated<number>;\n  createdAt: Generated<Timestamp>;\n  dataRegion: string | null;\n  eventType: string;\n  eventVersion: number;\n  id: Generated<string>;\n  lastError: string | null;\n  payload: Generated<Json>;\n  publishedAt: Timestamp | null;\n}\n\nexport interface ControlPlanePartnerManagedTeamAssignment {\n  createdAt: Generated<Timestamp>;\n  managedTeamId: string;\n  partnerTenantId: string;\n  userId: string;\n}\n\nexport interface ControlPlaneOwnerPortal {\n  hostname: string;\n  tenantId: string;\n  status: Generated<'pending' | 'active' | 'disabled'>;\n  name: string;\n  logo: string | null;\n  colorPrimary: string;\n  showPlatformBranding: Generated<boolean>;\n  linksEnabled: Generated<boolean>;\n  sender: { from: string; replyTo: string } | null;\n  revision: Generated<number>;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneTenant {\n  billingPartnerId: string | null;\n  billingPlan: string | null;\n  createdAt: Generated<Timestamp>;\n  dataRegion: Generated<string>;\n  id: Generated<string>;\n  migrationStatus: Generated<string>;\n  name: string;\n  partnerId: string | null;\n  partnerDeniedPermissions: Generated<string[]>;\n  slug: string;\n  status: Generated<string | null>;\n  storageRealm: Generated<string>;\n  type: Generated<string | null>;\n  uniqueRef: string | null;\n  updatedAt: Generated<Timestamp | null>;\n}\n\nexport interface ControlPlaneTenantRegionMigration {\n  actor: string;\n  completedAt: Timestamp | null;\n  error: Json | null;\n  id: string;\n  reason: string;\n  sourceManifest: Json | null;\n  sourceRegion: string;\n  startedAt: Generated<Timestamp>;\n  status: Generated<string>;\n  targetManifest: Json | null;\n  targetRegion: string;\n  tenantId: string;\n  updatedAt: Generated<Timestamp>;\n}\n\nexport interface ControlPlaneTenantUser {\n  createdAt: Generated<Timestamp | null>;\n  id: Generated<string>;\n  lastInvitedAt: Timestamp | null;\n  managedTeamAccess: Generated<'all' | 'assigned'>;\n  permissionBundles: Generated<string[]>;\n  role: Generated<string | null>;\n  status: Generated<string | null>;\n  tenantId: string;\n  updatedAt: Generated<Timestamp | null>;\n  userId: string;\n}\n\nexport interface ControlPlaneToken {\n  createdAt: Generated<Timestamp>;\n  expiresAt: Timestamp | null;\n  id: Generated<string>;\n  nanoId: Generated<string>;\n  payload: Generated<Json>;\n  tenantId: string | null;\n  type: string | null;\n  userId: string | null;\n}\n\nexport interface ControlPlaneUser {\n  clerkId: string | null;\n  createdAt: Generated<Timestamp | null>;\n  email: string | null;\n  firstName: string | null;\n  id: Generated<string>;\n  internalOperationGrants: Generated<string[]>;\n  isAdmin: Generated<boolean>;\n  lastName: string | null;\n  lastSeen: Timestamp | null;\n  name: string | null;\n  partnerId: string | null;\n  phone: string | null;\n  secondaryEmails: Generated<Json>;\n  status: Generated<string | null>;\n  sub: string | null;\n  type: Generated<string | null>;\n  updatedAt: Generated<Timestamp | null>;\n}\n\nexport interface ControlPlaneDB {\n  'controlPlane.accountingIntegrityDefinitionDraft': ControlPlaneAccountingIntegrityDefinitionDraft;\n  'controlPlane.accountingIntegrityDefinitionVersion': ControlPlaneAccountingIntegrityDefinitionVersion;\n  'controlPlane.accountingIntegrityEvaluation': ControlPlaneAccountingIntegrityEvaluation;\n  'controlPlane.app': ControlPlaneApp;\n  'controlPlane.bookingChannel': ControlPlaneBookingChannel;\n  'controlPlane.bookingChannelIconCandidate': ControlPlaneBookingChannelIconCandidate;\n  'controlPlane.connectionRoute': ControlPlaneConnectionRoute;\n  'controlPlane.controlPlaneOutbox': ControlPlaneOutbox;\n  'controlPlane.currency': ControlPlaneCurrency;\n  'controlPlane.destructiveJob': ControlPlaneDestructiveJob;\n  'controlPlane.destructiveJobAttempt': ControlPlaneDestructiveJobAttempt;\n  'controlPlane.destructiveJobOperation': ControlPlaneDestructiveJobOperation;\n  'controlPlane.events': ControlPlaneEvent;\n  'controlPlane.feature': ControlPlaneFeature;\n  'controlPlane.featureApproval': ControlPlaneFeatureApproval;\n  'controlPlane.featureEnabledTeam': ControlPlaneFeatureEnabledTeam;\n  'controlPlane.generalLedgerDefaultSet': ControlPlaneGeneralLedgerDefaultSet;\n  'controlPlane.generalLedgerDefaultVersion': ControlPlaneGeneralLedgerDefaultVersion;\n  'controlPlane.lineClassification': ControlPlaneLineClassification;\n  'controlPlane.lineClassificationTeamUsage': ControlPlaneLineClassificationTeamUsage;\n  'controlPlane.operation': ControlPlaneOperation;\n  'controlPlane.operationResource': ControlPlaneOperationResource;\n  'controlPlane.operationSource': ControlPlaneOperationSource;\n  'controlPlane.partnerManagedTeamAssignment': ControlPlanePartnerManagedTeamAssignment;\n  'controlPlane.reservationLineDefault': ControlPlaneReservationLineDefault;\n  'controlPlane.rolloutMonitorSnapshot': ControlPlaneRolloutMonitorSnapshot;\n  'controlPlane.ownerPortal': ControlPlaneOwnerPortal;\n  'controlPlane.tenant': ControlPlaneTenant;\n  'controlPlane.tenantRegionMigration': ControlPlaneTenantRegionMigration;\n  'controlPlane.tenantUser': ControlPlaneTenantUser;\n  'controlPlane.trackingQueueFailure': ControlPlaneTrackingQueueFailure;\n  'controlPlane.token': ControlPlaneToken;\n  'controlPlane.transactionLineMapping': ControlPlaneTransactionLineMapping;\n  'controlPlane.user': ControlPlaneUser;\n  'controlPlaneCore.flow': ControlPlaneFlow;\n  'controlPlaneCore.issueMessageOverwrite': ControlPlaneIssueMessageOverwrite;\n}\n\nclass ControlPlaneSchemaTransformer extends OperationNodeTransformer {\n  constructor(\n    readonly schema: string,\n    readonly issueMessageOverwriteSchema: string\n  ) {\n    super();\n  }\n\n  protected override transformSchemableIdentifier(\n    node: SchemableIdentifierNode,\n    queryId: QueryId\n  ): SchemableIdentifierNode {\n    const transformed = super.transformSchemableIdentifier(node, queryId);\n    if (\n      transformed.schema?.name === 'controlPlane' ||\n      transformed.schema?.name === 'control_plane'\n    ) {\n      return { ...transformed, schema: IdentifierNode.create(this.schema) };\n    }\n    if (\n      transformed.schema?.name === 'controlPlaneCore' ||\n      transformed.schema?.name === 'control_plane_core'\n    ) {\n      return {\n        ...transformed,\n        schema: IdentifierNode.create(this.issueMessageOverwriteSchema),\n      };\n    }\n    return transformed;\n  }\n\n  protected override transformFunction(\n    node: FunctionNode,\n    queryId: QueryId\n  ): FunctionNode {\n    const transformed = super.transformFunction(node, queryId);\n    return node.func.startsWith('controlPlane.')\n      ? {\n          ...transformed,\n          func: `${this.schema}.${transformed.func.slice('controlPlane.'.length)}`,\n        }\n      : transformed;\n  }\n}\n\nclass ControlPlaneSchemaPlugin implements KyselyPlugin {\n  readonly #transformer: ControlPlaneSchemaTransformer;\n\n  constructor(schema: string, issueMessageOverwriteSchema: string) {\n    this.#transformer = new ControlPlaneSchemaTransformer(\n      schema,\n      issueMessageOverwriteSchema\n    );\n  }\n\n  transformQuery(args: PluginTransformQueryArgs): RootOperationNode {\n    return this.#transformer.transformNode(args.node, args.queryId);\n  }\n\n  async transformResult(\n    args: PluginTransformResultArgs\n  ): Promise<QueryResult<UnknownRow>> {\n    return args.result;\n  }\n}\n\nexport function withControlPlaneSchema(\n  kysely: Kysely<ControlPlaneDB>,\n  schema: string,\n  issueMessageOverwriteSchema = 'core'\n) {\n  return kysely.withPlugin(\n    new ControlPlaneSchemaPlugin(schema, issueMessageOverwriteSchema)\n  );\n}\n"]}

package/build/module/index.d.ts

@@ -8,12 +8,22 @@ export * from './postgres-error';
 export * from './query';
 export * from './regional';
 export * from './transaction';
+import type { TenantDataRegion } from './regional';
 import type { DB } from './v1.generated';
 type KyselyOpts = {
+    dataRegion?: TenantDataRegion;
     hyperdrive?: true;
+    onRetryDiagnostic?: (diagnostic: PostgresRetryDiagnostic) => void;
     repositoryName?: string;
     log?: true | LogConfig;
 };
+type PostgresRetryDiagnostic = {
+    attemptCount: number;
+    dataRegion?: TenantDataRegion;
+    operationKind: 'acquisition' | 'mutation' | 'select' | 'transaction';
+    outcome: 'exhausted' | 'notRetryable' | 'retrying';
+    repositoryName?: string;
+};
 type PgOpt = {
     forceDisableSSL?: boolean;
     probe?: true;
@@ -25,5 +35,6 @@ export type ControlPlaneTransaction = T<ControlPlaneDB>;
 export declare function useKysely<Database = DB>(postgres: Sql, options?: KyselyOpts): K<Database>;
 export declare function useAsyncPostgres(connectionString: string, opt?: PgOpt): Promise<postgres.Sql<{}>>;
 export declare function useAsyncKysely(connectionString: string, opt?: PgOpt & KyselyOpts): Promise<K<DB>>;
+export declare function useAsyncNeonWebSocketKysely(connectionString: string, opt?: Pick<PgOpt, 'probe'> & KyselyOpts): Promise<K<DB>>;
 export declare function useAsyncControlPlaneKysely(connectionString: string, opt?: PgOpt & KyselyOpts): Promise<K<ControlPlaneDB>>;
 export declare function forceCloseAllPostgres(timeoutMs?: number): Promise<void>;

package/build/module/index.js

@@ -1,4 +1,6 @@
 import { backOff } from 'exponential-backoff';
+import { Pool } from '@neondatabase/serverless';
+import { PostgresDialect } from 'kysely';
 import { PostgresJSDialect, PostgresJSDriver, } from 'kysely-postgres-js';
 import postgres from 'postgres';
 import { withControlPlaneSchema } from './control-plane';
@@ -22,19 +24,74 @@ function isRetryableConnectionError(error) {
     const message = error instanceof Error ? error.message : String(error);
     return retryableMessages.some((retryable) => message.includes(retryable));
 }
-function retryPostgres(operation, retry) {
-    return backOff(operation, {
-        numOfAttempts: postgresRetryAttempts,
-        startingDelay: parseDuration('1s'),
-        maxDelay: parseDuration('5s'),
-        retry: (error, attemptNumber) => attemptNumber < postgresRetryAttempts && retry(error, attemptNumber),
-    });
+function isConnectionReset(error) {
+    return (error instanceof Error && 'code' in error && error.code === 'ECONNRESET');
+}
+function isRetryableConnectionAcquisitionError(error) {
+    return (isRetryableConnectionError(error) ||
+        isConnectionReset(error) ||
+        (error instanceof Error &&
+            (error.message === 'connection attempt failed' ||
+                ('code' in error && error.code === 'CONNECTION_DESTROYED'))));
+}
+function isQueryRetryDiagnosticError(error) {
+    const message = error instanceof Error ? error.message : String(error);
+    return (isRetryableConnectionError(error) ||
+        isConnectionReset(error) ||
+        message === 'Egress connection severed outside of Hyperdrive' ||
+        message === 'Network connection lost.' ||
+        message === 'Node shutting down' ||
+        message === 'Node shutting down.');
+}
+async function retryPostgres(operation, retry, diagnostic) {
+    let attemptCount = 0;
+    let retried = false;
+    const emitDiagnostic = (outcome) => {
+        if (!diagnostic)
+            return;
+        diagnostic.emit({
+            attemptCount,
+            dataRegion: diagnostic.dataRegion,
+            operationKind: diagnostic.operationKind,
+            outcome,
+            repositoryName: diagnostic.repositoryName,
+        });
+    };
+    try {
+        return await backOff(async () => {
+            attemptCount++;
+            return await operation();
+        }, {
+            numOfAttempts: postgresRetryAttempts,
+            startingDelay: parseDuration('1s'),
+            maxDelay: parseDuration('5s'),
+            async retry(error, attemptNumber) {
+                const shouldRetry = attemptNumber < postgresRetryAttempts &&
+                    (await retry(error, attemptNumber));
+                if (shouldRetry && diagnostic) {
+                    retried = true;
+                    emitDiagnostic('retrying');
+                }
+                return shouldRetry;
+            },
+        });
+    }
+    catch (error) {
+        if (diagnostic?.shouldEmit(error)) {
+            emitDiagnostic(retried && attemptCount === postgresRetryAttempts
+                ? 'exhausted'
+                : 'notRetryable');
+        }
+        throw error;
+    }
 }
 class RetryingPostgresJSDialect extends PostgresJSDialect {
     config;
-    constructor(config) {
+    retryDiagnostic;
+    constructor(config, retryDiagnostic) {
         super(config);
         this.config = config;
+        this.retryDiagnostic = retryDiagnostic;
     }
     createDriver() {
         const driver = new PostgresJSDriver(this.config);
@@ -53,21 +110,37 @@ class RetryingPostgresJSDialect extends PostgresJSDialect {
                     inTransaction: false,
                 };
                 const connection = {
-                    executeQuery: (compiledQuery) => retryPostgres(() => state.connection.executeQuery(compiledQuery), async (error) => {
-                        const message = error instanceof Error ? error.message : String(error);
-                        if (state.inTransaction ||
-                            (!isRetryableConnectionError(error) &&
-                                (compiledQuery.query.kind !== 'SelectQueryNode' ||
-                                    (message !==
-                                        'Egress connection severed outside of Hyperdrive' &&
-                                        message !== 'Network connection lost.')))) {
-                            return false;
-                        }
-                        // Postgres.js already ejects a severed reserved connection.
-                        // Acquire a new reservation before replaying the safe read.
-                        state.connection = await driver.acquireConnection();
-                        return true;
-                    }),
+                    executeQuery: (compiledQuery) => {
+                        const operationKind = state.inTransaction
+                            ? 'transaction'
+                            : compiledQuery.query.kind === 'SelectQueryNode'
+                                ? 'select'
+                                : 'mutation';
+                        return retryPostgres(() => state.connection.executeQuery(compiledQuery), async (error) => {
+                            const message = error instanceof Error ? error.message : String(error);
+                            if (state.inTransaction ||
+                                (!isRetryableConnectionError(error) &&
+                                    (compiledQuery.query.kind !== 'SelectQueryNode' ||
+                                        (!isConnectionReset(error) &&
+                                            message !==
+                                                'Egress connection severed outside of Hyperdrive' &&
+                                            message !== 'Network connection lost.' &&
+                                            message !== 'Node shutting down' &&
+                                            message !== 'Node shutting down.')))) {
+                                return false;
+                            }
+                            // Postgres.js already ejects a severed reserved connection.
+                            // Acquire a new reservation before replaying the safe read.
+                            state.connection = await driver.acquireConnection();
+                            return true;
+                        }, this.retryDiagnostic
+                            ? {
+                                ...this.retryDiagnostic,
+                                operationKind,
+                                shouldEmit: isQueryRetryDiagnosticError,
+                            }
+                            : undefined);
+                    },
                     streamQuery: (compiledQuery, chunkSize) => {
                         if (chunkSize === undefined) {
                             throw new Error('Postgres stream chunk size is required');
@@ -120,16 +193,35 @@ class RetryingPostgresJSDialect extends PostgresJSDialect {
 }
 export function useKysely(postgres, options) {
     console.debug('Creating Kysely instance');
+    const emitRetryDiagnostic = options?.onRetryDiagnostic ??
+        (options?.repositoryName
+            ? (diagnostic) => console.warn('Postgres operation retry diagnostic', diagnostic)
+            : undefined);
     const reserveConnection = postgres.reserve.bind(postgres);
     postgres.reserve = async () => {
-        const connection = await retryPostgres(reserveConnection, isRetryableConnectionError);
+        const connection = await retryPostgres(reserveConnection, isRetryableConnectionAcquisitionError, emitRetryDiagnostic
+            ? {
+                dataRegion: options?.dataRegion,
+                emit: emitRetryDiagnostic,
+                operationKind: 'acquisition',
+                repositoryName: options?.repositoryName,
+                shouldEmit: isRetryableConnectionAcquisitionError,
+            }
+            : undefined);
         if (options?.hyperdrive) {
             const unsafe = connection.unsafe;
             connection.unsafe = (query, parameters, queryOptions) => unsafe(query, parameters, { ...queryOptions, prepare: true });
         }
         return connection;
     };
-    const kysely = getKysely(new RetryingPostgresJSDialect({ postgres }), {
+    const kysely = getKysely(new RetryingPostgresJSDialect({ postgres }, emitRetryDiagnostic
+        ? {
+            dataRegion: options?.dataRegion,
+            emit: emitRetryDiagnostic,
+            repositoryName: options?.repositoryName,
+            shouldEmit: isQueryRetryDiagnosticError,
+        }
+        : undefined), {
         repositoryName: options?.repositoryName,
         log: options?.log,
     });
@@ -153,8 +245,9 @@ export async function useAsyncPostgres(connectionString, opt) {
     if (_connections.length > 0)
         console.warn(`We already have ${_connections.length} Postgres connections`);
     for (let attempt = 1; attempt <= retries; attempt++) {
+        let sql;
         try {
-            const sql = postgres(connectionString, {
+            sql = postgres(connectionString, {
                 ssl: ssl ? 'require' : false,
                 target_session_attrs: 'read-write',
                 // enable prepare statements, pgbouncer on crunchy allows for prepared statements
@@ -180,6 +273,7 @@ export async function useAsyncPostgres(connectionString, opt) {
             return sql;
         }
         catch (err) {
+            await sql?.end({ timeout: 0 }).catch(() => undefined);
             if (attempt === retries)
                 throw err;
             console.warn(`Postgres connection failed (attempt ${attempt}), retrying...`);
@@ -193,6 +287,17 @@ export async function useAsyncKysely(connectionString, opt) {
     const sql = await useAsyncPostgres(connectionString, opt);
     return useKysely(sql, opt);
 }
+export async function useAsyncNeonWebSocketKysely(connectionString, opt) {
+    const pool = new Pool({ connectionString, max: 5 });
+    const kysely = getKysely(new PostgresDialect({ pool }), opt ?? {});
+    kysely.destroy = async () => {
+        console.debug('Closing Neon WebSocket Kysely connection');
+        await pool.end();
+    };
+    if (opt?.probe)
+        await pool.query('select 1');
+    return kysely;
+}
 export async function useAsyncControlPlaneKysely(connectionString, opt) {
     const sql = await useAsyncPostgres(connectionString, opt);
     return withControlPlaneSchema(useKysely(sql, opt), 'public');

package/build/module/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAQ9C,OAAO,EACL,iBAAiB,EAEjB,gBAAgB,GACjB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,QAAsB,MAAM,UAAU,CAAC;AAC9C,OAAO,EAAuB,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAC9E,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACtC,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAGtC,cAAc,SAAS,CAAC;AACxB,cAAc,cAAc,CAAC;AAC7B,cAAc,kBAAkB,CAAC;AACjC,cAAc,SAAS,CAAC;AACxB,cAAc,YAAY,CAAC;AAC3B,cAAc,eAAe,CAAC;AAE9B,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAelD,MAAM,iBAAiB,GAAG;IACxB,mBAAmB;IACnB,iBAAiB;IACjB,uCAAuC;IACvC,oBAAoB;CACrB,CAAC;AACF,MAAM,qBAAqB,GAAG,CAAC,CAAC;AAEhC,SAAS,0BAA0B,CAAC,KAAc;IAChD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,OAAO,iBAAiB,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;AAC5E,CAAC;AAED,SAAS,aAAa,CACpB,SAA2B,EAC3B,KAA4E;IAE5E,OAAO,OAAO,CAAC,SAAS,EAAE;QACxB,aAAa,EAAE,qBAAqB;QACpC,aAAa,EAAE,aAAa,CAAC,IAAI,CAAC;QAClC,QAAQ,EAAE,aAAa,CAAC,IAAI,CAAC;QAC7B,KAAK,EAAE,CAAC,KAAK,EAAE,aAAa,EAAE,EAAE,CAC9B,aAAa,GAAG,qBAAqB,IAAI,KAAK,CAAC,KAAK,EAAE,aAAa,CAAC;KACvE,CAAC,CAAC;AACL,CAAC;AAED,MAAM,yBAA0B,SAAQ,iBAAiB;IAC1B;IAA7B,YAA6B,MAA+B;QAC1D,KAAK,CAAC,MAAM,CAAC,CAAC;QADa,WAAM,GAAN,MAAM,CAAyB;IAE5D,CAAC;IAEQ,YAAY;QACnB,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,IAAI,OAAO,EAG5B,CAAC;QACJ,MAAM,aAAa,GAAG,CAAC,UAA8B,EAAE,EAAE;YACvD,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAC1C,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;YACxE,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;QAEF,OAAO;YACL,IAAI,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE;YACzB,iBAAiB,EAAE,KAAK,IAAI,EAAE;gBAC5B,MAAM,KAAK,GAAG;oBACZ,UAAU,EAAE,MAAM,MAAM,CAAC,iBAAiB,EAAE;oBAC5C,aAAa,EAAE,KAAK;iBACrB,CAAC;gBACF,MAAM,UAAU,GAAuB;oBACrC,YAAY,EAAE,CAAC,aAAa,EAAE,EAAE,CAC9B,aAAa,CACX,GAAG,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,YAAY,CAAC,aAAa,CAAC,EAClD,KAAK,EAAE,KAAK,EAAE,EAAE;wBACd,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;wBACzD,IACE,KAAK,CAAC,aAAa;4BACnB,CAAC,CAAC,0BAA0B,CAAC,KAAK,CAAC;gCACjC,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB;oCAC7C,CAAC,OAAO;wCACN,iDAAiD;wCACjD,OAAO,KAAK,0BAA0B,CAAC,CAAC,CAAC,EAC/C,CAAC;4BACD,OAAO,KAAK,CAAC;wBACf,CAAC;wBAED,4DAA4D;wBAC5D,4DAA4D;wBAC5D,KAAK,CAAC,UAAU,GAAG,MAAM,MAAM,CAAC,iBAAiB,EAAE,CAAC;wBACpD,OAAO,IAAI,CAAC;oBACd,CAAC,CACF;oBACH,WAAW,EAAE,CAAC,aAAa,EAAE,SAAS,EAAE,EAAE;wBACxC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;4BAC5B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;wBAC5D,CAAC;wBACD,OAAO,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;oBAChE,CAAC;iBACF,CAAC;gBACF,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBACnC,OAAO,UAAU,CAAC;YACpB,CAAC;YACD,gBAAgB,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE;gBAC/C,MAAM,KAAK,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxC,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;gBAC3B,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;oBAC5B,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;YACD,iBAAiB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;gBACtC,MAAM,KAAK,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxC,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBACnD,CAAC;wBAAS,CAAC;oBACT,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,mBAAmB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;gBACxC,MAAM,KAAK,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxC,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBACrD,CAAC;wBAAS,CAAC;oBACT,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,SAAS,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,EAAE,CACrD,MAAM,CAAC,SAAS,CACd,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,EACpC,aAAa,EACb,YAAY,CACb;YACH,mBAAmB,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,EAAE,CAC/D,MAAM,CAAC,mBAAmB,CACxB,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,EACpC,aAAa,EACb,YAAY,CACb;YACH,gBAAgB,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,EAAE,CAC5D,MAAM,CAAC,gBAAgB,CACrB,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,EACpC,aAAa,EACb,YAAY,CACb;YACH,iBAAiB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;gBACtC,MAAM,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,CAAC,CAAC;gBACrE,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACjC,CAAC;YACD,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE;SAChC,CAAC;IACJ,CAAC;CACF;AAKD,MAAM,UAAU,SAAS,CAAgB,QAAa,EAAE,OAAoB;IAC1E,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC1C,MAAM,iBAAiB,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1D,QAAQ,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC5B,MAAM,UAAU,GAAG,MAAM,aAAa,CACpC,iBAAiB,EACjB,0BAA0B,CAC3B,CAAC;QACF,IAAI,OAAO,EAAE,UAAU,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;YACjC,UAAU,CAAC,MAAM,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,CACtD,MAAM,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,GAAG,YAAY,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAClE,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;IACF,MAAM,MAAM,GAAG,SAAS,CACtB,IAAI,yBAAyB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAC3C;QACE,cAAc,EAAE,OAAO,EAAE,cAAc;QACvC,GAAG,EAAE,OAAO,EAAE,GAAG;KAClB,CACF,CAAC;IACF,MAAM,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC3C,MAAM,QAAQ,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC1D,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,KAAK,KAAK,CAAC,CAAC;YAAE,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,IAAI,YAAY,GAAmB,EAAE,CAAC;AACtC,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,gBAAwB,EAAE,GAAW;IAC1E,MAAM,OAAO,GAAG,IAAI,CAAC;IACrB,MAAM,KAAK,GAAG,GAAG,EAAE,KAAK,KAAK,IAAI,CAAC;IAClC,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7E,MAAM,GAAG,GAAG,GAAG,EAAE,eAAe,KAAK,IAAI,IAAI,GAAG,EAAE,UAAU,KAAK,IAAI,CAAC;IACtE,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAEtC,OAAO,CAAC,KAAK,CACX,2CAA2C,KAAK,KAAK,OAAO,kBAAkB,GAAG,cAAc,GAAG,CAAC,IAAI,EAAE,CAC1G,CAAC;IACF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC;QACzB,OAAO,CAAC,IAAI,CAAC,mBAAmB,YAAY,CAAC,MAAM,uBAAuB,CAAC,CAAC;IAE9E,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;QACpD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,QAAQ,CAAC,gBAAgB,EAAE;gBACrC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;gBAC5B,oBAAoB,EAAE,YAAY;gBAClC,iFAAiF;gBACjF,OAAO,EAAE,IAAI;gBACb,GAAG,CAAC,GAAG,EAAE,UAAU;oBACjB,CAAC,CAAC;wBACE,GAAG,EAAE,CAAC;wBACN,uEAAuE;wBACvE,YAAY,EAAE,EAAE;qBACjB;oBACH,CAAC,CAAC;wBACE,eAAe,EAAE,EAAE;wBACnB,YAAY,EAAE,EAAE;qBACjB,CAAC;aACP,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,GAAG,CAAA,UAAU,CAAC;YACtB,CAAC;YACD,wEAAwE;YACxE,8DAA8D;YAC9D,qEAAqE;YAErE,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvB,OAAO,GAAG,CAAC;QACb,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,OAAO,KAAK,OAAO;gBAAE,MAAM,GAAG,CAAC;YACnC,OAAO,CAAC,IAAI,CACV,uCAAuC,OAAO,gBAAgB,CAC/D,CAAC;YACF,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAC5B,MAAM,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IACD,MAAM,IAAI,WAAW,CAAC,+BAA+B,CAAC,CAAC;AACzD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,gBAAwB,EACxB,GAAwB;IAExB,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,SAAS,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAC7B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,gBAAwB,EACxB,GAAwB;IAExB,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,sBAAsB,CAAC,SAAS,CAAiB,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,CAAC,CAAC;AAC/E,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,SAAS,GAAG,IAAI;IAC1D,MAAM,OAAO,CAAC,IAAI,CAAC;QACjB,OAAO,CAAC,GAAG,CACT,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CACtB,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,SAAS,GAAG,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAC7D,CACF;QACD,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,SAAS,GAAG,IAAI,CAAC,CAAC;KACxD,CAAC,CAAC;IACH,YAAY,GAAG,EAAE,CAAC;AACpB,CAAC","sourcesContent":["import { backOff } from 'exponential-backoff';\nimport type {\n  DatabaseConnection,\n  Driver,\n  Kysely as K,\n  LogConfig,\n  Transaction as T,\n} from 'kysely';\nimport {\n  PostgresJSDialect,\n  type PostgresJSDialectConfig,\n  PostgresJSDriver,\n} from 'kysely-postgres-js';\nimport postgres, { type Sql } from 'postgres';\nimport { type ControlPlaneDB, withControlPlaneSchema } from './control-plane';\nimport { KyselyError } from './error';\nimport { getKysely } from './plugins';\n\nexport type { Insertable, LogEvent } from 'kysely';\nexport * from './error';\nexport * from './isomorphic';\nexport * from './postgres-error';\nexport * from './query';\nexport * from './regional';\nexport * from './transaction';\n\nimport { parseDuration } from '@vrplatform/utils';\nimport type { DB } from './v1.generated';\n\ntype KyselyOpts = {\n  hyperdrive?: true;\n  repositoryName?: string;\n  log?: true | LogConfig;\n};\n\ntype PgOpt = {\n  forceDisableSSL?: boolean;\n  probe?: true;\n  retries?: false | number;\n} & Pick<KyselyOpts, 'hyperdrive'>;\n\nconst retryableMessages = [\n  'CONNECTION_CLOSED',\n  'CONNECT_TIMEOUT',\n  'Idle connection closed by Hyperdrive.',\n  'server_login_retry',\n];\nconst postgresRetryAttempts = 3;\n\nfunction isRetryableConnectionError(error: unknown) {\n  const message = error instanceof Error ? error.message : String(error);\n  return retryableMessages.some((retryable) => message.includes(retryable));\n}\n\nfunction retryPostgres<T>(\n  operation: () => Promise<T>,\n  retry: (error: unknown, attemptNumber: number) => boolean | Promise<boolean>\n) {\n  return backOff(operation, {\n    numOfAttempts: postgresRetryAttempts,\n    startingDelay: parseDuration('1s'),\n    maxDelay: parseDuration('5s'),\n    retry: (error, attemptNumber) =>\n      attemptNumber < postgresRetryAttempts && retry(error, attemptNumber),\n  });\n}\n\nclass RetryingPostgresJSDialect extends PostgresJSDialect {\n  constructor(private readonly config: PostgresJSDialectConfig) {\n    super(config);\n  }\n\n  override createDriver(): Driver {\n    const driver = new PostgresJSDriver(this.config);\n    const connections = new WeakMap<\n      DatabaseConnection,\n      { connection: DatabaseConnection; inTransaction: boolean }\n    >();\n    const getConnection = (connection: DatabaseConnection) => {\n      const state = connections.get(connection);\n      if (!state) throw new Error('Postgres connection state is unavailable');\n      return state;\n    };\n\n    return {\n      init: () => driver.init(),\n      acquireConnection: async () => {\n        const state = {\n          connection: await driver.acquireConnection(),\n          inTransaction: false,\n        };\n        const connection: DatabaseConnection = {\n          executeQuery: (compiledQuery) =>\n            retryPostgres(\n              () => state.connection.executeQuery(compiledQuery),\n              async (error) => {\n                const message =\n                  error instanceof Error ? error.message : String(error);\n                if (\n                  state.inTransaction ||\n                  (!isRetryableConnectionError(error) &&\n                    (compiledQuery.query.kind !== 'SelectQueryNode' ||\n                      (message !==\n                        'Egress connection severed outside of Hyperdrive' &&\n                        message !== 'Network connection lost.')))\n                ) {\n                  return false;\n                }\n\n                // Postgres.js already ejects a severed reserved connection.\n                // Acquire a new reservation before replaying the safe read.\n                state.connection = await driver.acquireConnection();\n                return true;\n              }\n            ),\n          streamQuery: (compiledQuery, chunkSize) => {\n            if (chunkSize === undefined) {\n              throw new Error('Postgres stream chunk size is required');\n            }\n            return state.connection.streamQuery(compiledQuery, chunkSize);\n          },\n        };\n        connections.set(connection, state);\n        return connection;\n      },\n      beginTransaction: async (connection, settings) => {\n        const state = getConnection(connection);\n        state.inTransaction = true;\n        try {\n          await driver.beginTransaction(state.connection, settings);\n        } catch (error) {\n          state.inTransaction = false;\n          throw error;\n        }\n      },\n      commitTransaction: async (connection) => {\n        const state = getConnection(connection);\n        try {\n          await driver.commitTransaction(state.connection);\n        } finally {\n          state.inTransaction = false;\n        }\n      },\n      rollbackTransaction: async (connection) => {\n        const state = getConnection(connection);\n        try {\n          await driver.rollbackTransaction(state.connection);\n        } finally {\n          state.inTransaction = false;\n        }\n      },\n      savepoint: (connection, savepointName, compileQuery) =>\n        driver.savepoint(\n          getConnection(connection).connection,\n          savepointName,\n          compileQuery\n        ),\n      rollbackToSavepoint: (connection, savepointName, compileQuery) =>\n        driver.rollbackToSavepoint(\n          getConnection(connection).connection,\n          savepointName,\n          compileQuery\n        ),\n      releaseSavepoint: (connection, savepointName, compileQuery) =>\n        driver.releaseSavepoint(\n          getConnection(connection).connection,\n          savepointName,\n          compileQuery\n        ),\n      releaseConnection: async (connection) => {\n        await driver.releaseConnection(getConnection(connection).connection);\n        connections.delete(connection);\n      },\n      destroy: () => driver.destroy(),\n    };\n  }\n}\n\nexport type Kysely = K<DB>;\nexport type ControlPlaneKysely = K<ControlPlaneDB>;\nexport type ControlPlaneTransaction = T<ControlPlaneDB>;\nexport function useKysely<Database = DB>(postgres: Sql, options?: KyselyOpts) {\n  console.debug('Creating Kysely instance');\n  const reserveConnection = postgres.reserve.bind(postgres);\n  postgres.reserve = async () => {\n    const connection = await retryPostgres(\n      reserveConnection,\n      isRetryableConnectionError\n    );\n    if (options?.hyperdrive) {\n      const unsafe = connection.unsafe;\n      connection.unsafe = (query, parameters, queryOptions) =>\n        unsafe(query, parameters, { ...queryOptions, prepare: true });\n    }\n    return connection;\n  };\n  const kysely = getKysely<Database>(\n    new RetryingPostgresJSDialect({ postgres }),\n    {\n      repositoryName: options?.repositoryName,\n      log: options?.log,\n    }\n  );\n  kysely.destroy = async () => {\n    console.debug('Closing Kysely connection');\n    await postgres.end({ timeout: 5 }).catch(() => undefined);\n    const index = _connections.indexOf(postgres);\n    if (index !== -1) _connections.splice(index, 1);\n  };\n\n  return kysely;\n}\n\nlet _connections: postgres.Sql[] = [];\nexport async function useAsyncPostgres(connectionString: string, opt?: PgOpt) {\n  const delayMs = 2500;\n  const probe = opt?.probe === true;\n  const retries = probe ? (opt?.retries === false ? 0 : opt?.retries || 3) : 1;\n  const ssl = opt?.forceDisableSSL !== true && opt?.hyperdrive !== true;\n  const url = new URL(connectionString);\n\n  console.debug(\n    `Creating Postgres connection with probe=${probe}, ${retries} attempts, ssl=${ssl}, and port=${url.port}`\n  );\n  if (_connections.length > 0)\n    console.warn(`We already have ${_connections.length} Postgres connections`);\n\n  for (let attempt = 1; attempt <= retries; attempt++) {\n    try {\n      const sql = postgres(connectionString, {\n        ssl: ssl ? 'require' : false,\n        target_session_attrs: 'read-write',\n        // enable prepare statements, pgbouncer on crunchy allows for prepared statements\n        prepare: true,\n        ...(opt?.hyperdrive\n          ? {\n              max: 5,\n              // Retire idle clients before Hyperdrive can close them asynchronously.\n              idle_timeout: 10,\n            }\n          : {\n              connect_timeout: 10,\n              idle_timeout: 60,\n            }),\n      });\n\n      if (probe) {\n        await sql`select 1`;\n      }\n      // statement_timeout/lock_timeout are role-level settings on the cluster\n      // (ALTER ROLE application SET ...) so they cover every pooled\n      // connection; per-session SETs here only ever applied to one of them\n\n      _connections.push(sql);\n      return sql;\n    } catch (err) {\n      if (attempt === retries) throw err;\n      console.warn(\n        `Postgres connection failed (attempt ${attempt}), retrying...`\n      );\n      console.error('error', err);\n      await new Promise((res) => setTimeout(res, delayMs));\n    }\n  }\n  throw new KyselyError('Unable to connect to Postgres');\n}\n\nexport async function useAsyncKysely(\n  connectionString: string,\n  opt?: PgOpt & KyselyOpts\n) {\n  const sql = await useAsyncPostgres(connectionString, opt);\n  return useKysely(sql, opt);\n}\n\nexport async function useAsyncControlPlaneKysely(\n  connectionString: string,\n  opt?: PgOpt & KyselyOpts\n) {\n  const sql = await useAsyncPostgres(connectionString, opt);\n  return withControlPlaneSchema(useKysely<ControlPlaneDB>(sql, opt), 'public');\n}\n\nexport async function forceCloseAllPostgres(timeoutMs = 5000) {\n  await Promise.race([\n    Promise.all(\n      _connections.map((pg) =>\n        pg.end({ timeout: timeoutMs / 1000 }).catch(() => undefined)\n      )\n    ),\n    new Promise((res) => setTimeout(res, timeoutMs + 1000)),\n  ]);\n  _connections = [];\n}\n"]}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,0BAA0B,CAAC;AAQhD,OAAO,EAAE,eAAe,EAAE,MAAM,QAAQ,CAAC;AACzC,OAAO,EACL,iBAAiB,EAEjB,gBAAgB,GACjB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,QAAsB,MAAM,UAAU,CAAC;AAC9C,OAAO,EAAuB,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAC9E,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACtC,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAGtC,cAAc,SAAS,CAAC;AACxB,cAAc,cAAc,CAAC;AAC7B,cAAc,kBAAkB,CAAC;AACjC,cAAc,SAAS,CAAC;AACxB,cAAc,YAAY,CAAC;AAC3B,cAAc,eAAe,CAAC;AAE9B,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AA0BlD,MAAM,iBAAiB,GAAG;IACxB,mBAAmB;IACnB,iBAAiB;IACjB,uCAAuC;IACvC,oBAAoB;CACrB,CAAC;AACF,MAAM,qBAAqB,GAAG,CAAC,CAAC;AAEhC,SAAS,0BAA0B,CAAC,KAAc;IAChD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,OAAO,iBAAiB,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;AAC5E,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,OAAO,CACL,KAAK,YAAY,KAAK,IAAI,MAAM,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CACzE,CAAC;AACJ,CAAC;AAED,SAAS,qCAAqC,CAAC,KAAc;IAC3D,OAAO,CACL,0BAA0B,CAAC,KAAK,CAAC;QACjC,iBAAiB,CAAC,KAAK,CAAC;QACxB,CAAC,KAAK,YAAY,KAAK;YACrB,CAAC,KAAK,CAAC,OAAO,KAAK,2BAA2B;gBAC5C,CAAC,MAAM,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,sBAAsB,CAAC,CAAC,CAAC,CACjE,CAAC;AACJ,CAAC;AAED,SAAS,2BAA2B,CAAC,KAAc;IACjD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,OAAO,CACL,0BAA0B,CAAC,KAAK,CAAC;QACjC,iBAAiB,CAAC,KAAK,CAAC;QACxB,OAAO,KAAK,iDAAiD;QAC7D,OAAO,KAAK,0BAA0B;QACtC,OAAO,KAAK,oBAAoB;QAChC,OAAO,KAAK,qBAAqB,CAClC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,aAAa,CAC1B,SAA2B,EAC3B,KAA4E,EAC5E,UAGC;IAED,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,MAAM,cAAc,GAAG,CAAC,OAA2C,EAAE,EAAE;QACrE,IAAI,CAAC,UAAU;YAAE,OAAO;QACxB,UAAU,CAAC,IAAI,CAAC;YACd,YAAY;YACZ,UAAU,EAAE,UAAU,CAAC,UAAU;YACjC,aAAa,EAAE,UAAU,CAAC,aAAa;YACvC,OAAO;YACP,cAAc,EAAE,UAAU,CAAC,cAAc;SAC1C,CAAC,CAAC;IACL,CAAC,CAAC;IACF,IAAI,CAAC;QACH,OAAO,MAAM,OAAO,CAClB,KAAK,IAAI,EAAE;YACT,YAAY,EAAE,CAAC;YACf,OAAO,MAAM,SAAS,EAAE,CAAC;QAC3B,CAAC,EACD;YACE,aAAa,EAAE,qBAAqB;YACpC,aAAa,EAAE,aAAa,CAAC,IAAI,CAAC;YAClC,QAAQ,EAAE,aAAa,CAAC,IAAI,CAAC;YAC7B,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,aAAa;gBAC9B,MAAM,WAAW,GACf,aAAa,GAAG,qBAAqB;oBACrC,CAAC,MAAM,KAAK,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC,CAAC;gBACtC,IAAI,WAAW,IAAI,UAAU,EAAE,CAAC;oBAC9B,OAAO,GAAG,IAAI,CAAC;oBACf,cAAc,CAAC,UAAU,CAAC,CAAC;gBAC7B,CAAC;gBACD,OAAO,WAAW,CAAC;YACrB,CAAC;SACF,CACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,UAAU,EAAE,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAClC,cAAc,CACZ,OAAO,IAAI,YAAY,KAAK,qBAAqB;gBAC/C,CAAC,CAAC,WAAW;gBACb,CAAC,CAAC,cAAc,CACnB,CAAC;QACJ,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,yBAA0B,SAAQ,iBAAiB;IAEpC;IACA;IAFnB,YACmB,MAA+B,EAC/B,eAMhB;QAED,KAAK,CAAC,MAAM,CAAC,CAAC;QATG,WAAM,GAAN,MAAM,CAAyB;QAC/B,oBAAe,GAAf,eAAe,CAM/B;IAGH,CAAC;IAEQ,YAAY;QACnB,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,IAAI,OAAO,EAG5B,CAAC;QACJ,MAAM,aAAa,GAAG,CAAC,UAA8B,EAAE,EAAE;YACvD,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAC1C,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;YACxE,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;QAEF,OAAO;YACL,IAAI,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE;YACzB,iBAAiB,EAAE,KAAK,IAAI,EAAE;gBAC5B,MAAM,KAAK,GAAG;oBACZ,UAAU,EAAE,MAAM,MAAM,CAAC,iBAAiB,EAAE;oBAC5C,aAAa,EAAE,KAAK;iBACrB,CAAC;gBACF,MAAM,UAAU,GAAuB;oBACrC,YAAY,EAAE,CAAC,aAAa,EAAE,EAAE;wBAC9B,MAAM,aAAa,GAAG,KAAK,CAAC,aAAa;4BACvC,CAAC,CAAC,aAAa;4BACf,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB;gCAC9C,CAAC,CAAC,QAAQ;gCACV,CAAC,CAAC,UAAU,CAAC;wBACjB,OAAO,aAAa,CAClB,GAAG,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,YAAY,CAAC,aAAa,CAAC,EAClD,KAAK,EAAE,KAAK,EAAE,EAAE;4BACd,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;4BACzD,IACE,KAAK,CAAC,aAAa;gCACnB,CAAC,CAAC,0BAA0B,CAAC,KAAK,CAAC;oCACjC,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB;wCAC7C,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC;4CACxB,OAAO;gDACL,iDAAiD;4CACnD,OAAO,KAAK,0BAA0B;4CACtC,OAAO,KAAK,oBAAoB;4CAChC,OAAO,KAAK,qBAAqB,CAAC,CAAC,CAAC,EAC1C,CAAC;gCACD,OAAO,KAAK,CAAC;4BACf,CAAC;4BAED,4DAA4D;4BAC5D,4DAA4D;4BAC5D,KAAK,CAAC,UAAU,GAAG,MAAM,MAAM,CAAC,iBAAiB,EAAE,CAAC;4BACpD,OAAO,IAAI,CAAC;wBACd,CAAC,EACD,IAAI,CAAC,eAAe;4BAClB,CAAC,CAAC;gCACE,GAAG,IAAI,CAAC,eAAe;gCACvB,aAAa;gCACb,UAAU,EAAE,2BAA2B;6BACxC;4BACH,CAAC,CAAC,SAAS,CACd,CAAC;oBACJ,CAAC;oBACD,WAAW,EAAE,CAAC,aAAa,EAAE,SAAS,EAAE,EAAE;wBACxC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;4BAC5B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;wBAC5D,CAAC;wBACD,OAAO,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;oBAChE,CAAC;iBACF,CAAC;gBACF,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBACnC,OAAO,UAAU,CAAC;YACpB,CAAC;YACD,gBAAgB,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE;gBAC/C,MAAM,KAAK,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxC,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;gBAC3B,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;oBAC5B,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;YACD,iBAAiB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;gBACtC,MAAM,KAAK,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxC,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBACnD,CAAC;wBAAS,CAAC;oBACT,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,mBAAmB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;gBACxC,MAAM,KAAK,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxC,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBACrD,CAAC;wBAAS,CAAC;oBACT,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,SAAS,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,EAAE,CACrD,MAAM,CAAC,SAAS,CACd,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,EACpC,aAAa,EACb,YAAY,CACb;YACH,mBAAmB,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,EAAE,CAC/D,MAAM,CAAC,mBAAmB,CACxB,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,EACpC,aAAa,EACb,YAAY,CACb;YACH,gBAAgB,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,EAAE,CAC5D,MAAM,CAAC,gBAAgB,CACrB,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,EACpC,aAAa,EACb,YAAY,CACb;YACH,iBAAiB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;gBACtC,MAAM,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,CAAC,CAAC;gBACrE,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACjC,CAAC;YACD,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE;SAChC,CAAC;IACJ,CAAC;CACF;AAKD,MAAM,UAAU,SAAS,CAAgB,QAAa,EAAE,OAAoB;IAC1E,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC1C,MAAM,mBAAmB,GACvB,OAAO,EAAE,iBAAiB;QAC1B,CAAC,OAAO,EAAE,cAAc;YACtB,CAAC,CAAC,CAAC,UAAmC,EAAE,EAAE,CACtC,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE,UAAU,CAAC;YACnE,CAAC,CAAC,SAAS,CAAC,CAAC;IACjB,MAAM,iBAAiB,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1D,QAAQ,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC5B,MAAM,UAAU,GAAG,MAAM,aAAa,CACpC,iBAAiB,EACjB,qCAAqC,EACrC,mBAAmB;YACjB,CAAC,CAAC;gBACE,UAAU,EAAE,OAAO,EAAE,UAAU;gBAC/B,IAAI,EAAE,mBAAmB;gBACzB,aAAa,EAAE,aAAa;gBAC5B,cAAc,EAAE,OAAO,EAAE,cAAc;gBACvC,UAAU,EAAE,qCAAqC;aAClD;YACH,CAAC,CAAC,SAAS,CACd,CAAC;QACF,IAAI,OAAO,EAAE,UAAU,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;YACjC,UAAU,CAAC,MAAM,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,CACtD,MAAM,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,GAAG,YAAY,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAClE,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;IACF,MAAM,MAAM,GAAG,SAAS,CACtB,IAAI,yBAAyB,CAC3B,EAAE,QAAQ,EAAE,EACZ,mBAAmB;QACjB,CAAC,CAAC;YACE,UAAU,EAAE,OAAO,EAAE,UAAU;YAC/B,IAAI,EAAE,mBAAmB;YACzB,cAAc,EAAE,OAAO,EAAE,cAAc;YACvC,UAAU,EAAE,2BAA2B;SACxC;QACH,CAAC,CAAC,SAAS,CACd,EACD;QACE,cAAc,EAAE,OAAO,EAAE,cAAc;QACvC,GAAG,EAAE,OAAO,EAAE,GAAG;KAClB,CACF,CAAC;IACF,MAAM,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC3C,MAAM,QAAQ,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC1D,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,KAAK,KAAK,CAAC,CAAC;YAAE,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,IAAI,YAAY,GAAmB,EAAE,CAAC;AACtC,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,gBAAwB,EAAE,GAAW;IAC1E,MAAM,OAAO,GAAG,IAAI,CAAC;IACrB,MAAM,KAAK,GAAG,GAAG,EAAE,KAAK,KAAK,IAAI,CAAC;IAClC,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7E,MAAM,GAAG,GAAG,GAAG,EAAE,eAAe,KAAK,IAAI,IAAI,GAAG,EAAE,UAAU,KAAK,IAAI,CAAC;IACtE,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAEtC,OAAO,CAAC,KAAK,CACX,2CAA2C,KAAK,KAAK,OAAO,kBAAkB,GAAG,cAAc,GAAG,CAAC,IAAI,EAAE,CAC1G,CAAC;IACF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC;QACzB,OAAO,CAAC,IAAI,CAAC,mBAAmB,YAAY,CAAC,MAAM,uBAAuB,CAAC,CAAC;IAE9E,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;QACpD,IAAI,GAAoB,CAAC;QACzB,IAAI,CAAC;YACH,GAAG,GAAG,QAAQ,CAAC,gBAAgB,EAAE;gBAC/B,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;gBAC5B,oBAAoB,EAAE,YAAY;gBAClC,iFAAiF;gBACjF,OAAO,EAAE,IAAI;gBACb,GAAG,CAAC,GAAG,EAAE,UAAU;oBACjB,CAAC,CAAC;wBACE,GAAG,EAAE,CAAC;wBACN,uEAAuE;wBACvE,YAAY,EAAE,EAAE;qBACjB;oBACH,CAAC,CAAC;wBACE,eAAe,EAAE,EAAE;wBACnB,YAAY,EAAE,EAAE;qBACjB,CAAC;aACP,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,GAAG,CAAA,UAAU,CAAC;YACtB,CAAC;YACD,wEAAwE;YACxE,8DAA8D;YAC9D,qEAAqE;YAErE,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvB,OAAO,GAAG,CAAC;QACb,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACtD,IAAI,OAAO,KAAK,OAAO;gBAAE,MAAM,GAAG,CAAC;YACnC,OAAO,CAAC,IAAI,CACV,uCAAuC,OAAO,gBAAgB,CAC/D,CAAC;YACF,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAC5B,MAAM,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IACD,MAAM,IAAI,WAAW,CAAC,+BAA+B,CAAC,CAAC;AACzD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,gBAAwB,EACxB,GAAwB;IAExB,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,SAAS,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAC7B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAC/C,gBAAwB,EACxB,GAAuC;IAEvC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,EAAE,gBAAgB,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;IACpD,MAAM,MAAM,GAAG,SAAS,CAAK,IAAI,eAAe,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;IACvE,MAAM,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC1D,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC,CAAC;IACF,IAAI,GAAG,EAAE,KAAK;QAAE,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAC7C,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,gBAAwB,EACxB,GAAwB;IAExB,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,sBAAsB,CAAC,SAAS,CAAiB,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,CAAC,CAAC;AAC/E,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,SAAS,GAAG,IAAI;IAC1D,MAAM,OAAO,CAAC,IAAI,CAAC;QACjB,OAAO,CAAC,GAAG,CACT,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CACtB,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,SAAS,GAAG,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAC7D,CACF;QACD,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,SAAS,GAAG,IAAI,CAAC,CAAC;KACxD,CAAC,CAAC;IACH,YAAY,GAAG,EAAE,CAAC;AACpB,CAAC","sourcesContent":["import { backOff } from 'exponential-backoff';\nimport { Pool } from '@neondatabase/serverless';\nimport type {\n  DatabaseConnection,\n  Driver,\n  Kysely as K,\n  LogConfig,\n  Transaction as T,\n} from 'kysely';\nimport { PostgresDialect } from 'kysely';\nimport {\n  PostgresJSDialect,\n  type PostgresJSDialectConfig,\n  PostgresJSDriver,\n} from 'kysely-postgres-js';\nimport postgres, { type Sql } from 'postgres';\nimport { type ControlPlaneDB, withControlPlaneSchema } from './control-plane';\nimport { KyselyError } from './error';\nimport { getKysely } from './plugins';\n\nexport type { Insertable, LogEvent } from 'kysely';\nexport * from './error';\nexport * from './isomorphic';\nexport * from './postgres-error';\nexport * from './query';\nexport * from './regional';\nexport * from './transaction';\n\nimport { parseDuration } from '@vrplatform/utils';\nimport type { TenantDataRegion } from './regional';\nimport type { DB } from './v1.generated';\n\ntype KyselyOpts = {\n  dataRegion?: TenantDataRegion;\n  hyperdrive?: true;\n  onRetryDiagnostic?: (diagnostic: PostgresRetryDiagnostic) => void;\n  repositoryName?: string;\n  log?: true | LogConfig;\n};\n\ntype PostgresRetryDiagnostic = {\n  attemptCount: number;\n  dataRegion?: TenantDataRegion;\n  operationKind: 'acquisition' | 'mutation' | 'select' | 'transaction';\n  outcome: 'exhausted' | 'notRetryable' | 'retrying';\n  repositoryName?: string;\n};\n\ntype PgOpt = {\n  forceDisableSSL?: boolean;\n  probe?: true;\n  retries?: false | number;\n} & Pick<KyselyOpts, 'hyperdrive'>;\n\nconst retryableMessages = [\n  'CONNECTION_CLOSED',\n  'CONNECT_TIMEOUT',\n  'Idle connection closed by Hyperdrive.',\n  'server_login_retry',\n];\nconst postgresRetryAttempts = 3;\n\nfunction isRetryableConnectionError(error: unknown) {\n  const message = error instanceof Error ? error.message : String(error);\n  return retryableMessages.some((retryable) => message.includes(retryable));\n}\n\nfunction isConnectionReset(error: unknown) {\n  return (\n    error instanceof Error && 'code' in error && error.code === 'ECONNRESET'\n  );\n}\n\nfunction isRetryableConnectionAcquisitionError(error: unknown) {\n  return (\n    isRetryableConnectionError(error) ||\n    isConnectionReset(error) ||\n    (error instanceof Error &&\n      (error.message === 'connection attempt failed' ||\n        ('code' in error && error.code === 'CONNECTION_DESTROYED')))\n  );\n}\n\nfunction isQueryRetryDiagnosticError(error: unknown) {\n  const message = error instanceof Error ? error.message : String(error);\n  return (\n    isRetryableConnectionError(error) ||\n    isConnectionReset(error) ||\n    message === 'Egress connection severed outside of Hyperdrive' ||\n    message === 'Network connection lost.' ||\n    message === 'Node shutting down' ||\n    message === 'Node shutting down.'\n  );\n}\n\nasync function retryPostgres<T>(\n  operation: () => Promise<T>,\n  retry: (error: unknown, attemptNumber: number) => boolean | Promise<boolean>,\n  diagnostic?: Omit<PostgresRetryDiagnostic, 'attemptCount' | 'outcome'> & {\n    emit: (diagnostic: PostgresRetryDiagnostic) => void;\n    shouldEmit: (error: unknown) => boolean;\n  }\n) {\n  let attemptCount = 0;\n  let retried = false;\n  const emitDiagnostic = (outcome: PostgresRetryDiagnostic['outcome']) => {\n    if (!diagnostic) return;\n    diagnostic.emit({\n      attemptCount,\n      dataRegion: diagnostic.dataRegion,\n      operationKind: diagnostic.operationKind,\n      outcome,\n      repositoryName: diagnostic.repositoryName,\n    });\n  };\n  try {\n    return await backOff(\n      async () => {\n        attemptCount++;\n        return await operation();\n      },\n      {\n        numOfAttempts: postgresRetryAttempts,\n        startingDelay: parseDuration('1s'),\n        maxDelay: parseDuration('5s'),\n        async retry(error, attemptNumber) {\n          const shouldRetry =\n            attemptNumber < postgresRetryAttempts &&\n            (await retry(error, attemptNumber));\n          if (shouldRetry && diagnostic) {\n            retried = true;\n            emitDiagnostic('retrying');\n          }\n          return shouldRetry;\n        },\n      }\n    );\n  } catch (error) {\n    if (diagnostic?.shouldEmit(error)) {\n      emitDiagnostic(\n        retried && attemptCount === postgresRetryAttempts\n          ? 'exhausted'\n          : 'notRetryable'\n      );\n    }\n    throw error;\n  }\n}\n\nclass RetryingPostgresJSDialect extends PostgresJSDialect {\n  constructor(\n    private readonly config: PostgresJSDialectConfig,\n    private readonly retryDiagnostic?: Omit<\n      PostgresRetryDiagnostic,\n      'attemptCount' | 'operationKind' | 'outcome'\n    > & {\n      emit: (diagnostic: PostgresRetryDiagnostic) => void;\n      shouldEmit: (error: unknown) => boolean;\n    }\n  ) {\n    super(config);\n  }\n\n  override createDriver(): Driver {\n    const driver = new PostgresJSDriver(this.config);\n    const connections = new WeakMap<\n      DatabaseConnection,\n      { connection: DatabaseConnection; inTransaction: boolean }\n    >();\n    const getConnection = (connection: DatabaseConnection) => {\n      const state = connections.get(connection);\n      if (!state) throw new Error('Postgres connection state is unavailable');\n      return state;\n    };\n\n    return {\n      init: () => driver.init(),\n      acquireConnection: async () => {\n        const state = {\n          connection: await driver.acquireConnection(),\n          inTransaction: false,\n        };\n        const connection: DatabaseConnection = {\n          executeQuery: (compiledQuery) => {\n            const operationKind = state.inTransaction\n              ? 'transaction'\n              : compiledQuery.query.kind === 'SelectQueryNode'\n                ? 'select'\n                : 'mutation';\n            return retryPostgres(\n              () => state.connection.executeQuery(compiledQuery),\n              async (error) => {\n                const message =\n                  error instanceof Error ? error.message : String(error);\n                if (\n                  state.inTransaction ||\n                  (!isRetryableConnectionError(error) &&\n                    (compiledQuery.query.kind !== 'SelectQueryNode' ||\n                      (!isConnectionReset(error) &&\n                        message !==\n                          'Egress connection severed outside of Hyperdrive' &&\n                        message !== 'Network connection lost.' &&\n                        message !== 'Node shutting down' &&\n                        message !== 'Node shutting down.')))\n                ) {\n                  return false;\n                }\n\n                // Postgres.js already ejects a severed reserved connection.\n                // Acquire a new reservation before replaying the safe read.\n                state.connection = await driver.acquireConnection();\n                return true;\n              },\n              this.retryDiagnostic\n                ? {\n                    ...this.retryDiagnostic,\n                    operationKind,\n                    shouldEmit: isQueryRetryDiagnosticError,\n                  }\n                : undefined\n            );\n          },\n          streamQuery: (compiledQuery, chunkSize) => {\n            if (chunkSize === undefined) {\n              throw new Error('Postgres stream chunk size is required');\n            }\n            return state.connection.streamQuery(compiledQuery, chunkSize);\n          },\n        };\n        connections.set(connection, state);\n        return connection;\n      },\n      beginTransaction: async (connection, settings) => {\n        const state = getConnection(connection);\n        state.inTransaction = true;\n        try {\n          await driver.beginTransaction(state.connection, settings);\n        } catch (error) {\n          state.inTransaction = false;\n          throw error;\n        }\n      },\n      commitTransaction: async (connection) => {\n        const state = getConnection(connection);\n        try {\n          await driver.commitTransaction(state.connection);\n        } finally {\n          state.inTransaction = false;\n        }\n      },\n      rollbackTransaction: async (connection) => {\n        const state = getConnection(connection);\n        try {\n          await driver.rollbackTransaction(state.connection);\n        } finally {\n          state.inTransaction = false;\n        }\n      },\n      savepoint: (connection, savepointName, compileQuery) =>\n        driver.savepoint(\n          getConnection(connection).connection,\n          savepointName,\n          compileQuery\n        ),\n      rollbackToSavepoint: (connection, savepointName, compileQuery) =>\n        driver.rollbackToSavepoint(\n          getConnection(connection).connection,\n          savepointName,\n          compileQuery\n        ),\n      releaseSavepoint: (connection, savepointName, compileQuery) =>\n        driver.releaseSavepoint(\n          getConnection(connection).connection,\n          savepointName,\n          compileQuery\n        ),\n      releaseConnection: async (connection) => {\n        await driver.releaseConnection(getConnection(connection).connection);\n        connections.delete(connection);\n      },\n      destroy: () => driver.destroy(),\n    };\n  }\n}\n\nexport type Kysely = K<DB>;\nexport type ControlPlaneKysely = K<ControlPlaneDB>;\nexport type ControlPlaneTransaction = T<ControlPlaneDB>;\nexport function useKysely<Database = DB>(postgres: Sql, options?: KyselyOpts) {\n  console.debug('Creating Kysely instance');\n  const emitRetryDiagnostic =\n    options?.onRetryDiagnostic ??\n    (options?.repositoryName\n      ? (diagnostic: PostgresRetryDiagnostic) =>\n          console.warn('Postgres operation retry diagnostic', diagnostic)\n      : undefined);\n  const reserveConnection = postgres.reserve.bind(postgres);\n  postgres.reserve = async () => {\n    const connection = await retryPostgres(\n      reserveConnection,\n      isRetryableConnectionAcquisitionError,\n      emitRetryDiagnostic\n        ? {\n            dataRegion: options?.dataRegion,\n            emit: emitRetryDiagnostic,\n            operationKind: 'acquisition',\n            repositoryName: options?.repositoryName,\n            shouldEmit: isRetryableConnectionAcquisitionError,\n          }\n        : undefined\n    );\n    if (options?.hyperdrive) {\n      const unsafe = connection.unsafe;\n      connection.unsafe = (query, parameters, queryOptions) =>\n        unsafe(query, parameters, { ...queryOptions, prepare: true });\n    }\n    return connection;\n  };\n  const kysely = getKysely<Database>(\n    new RetryingPostgresJSDialect(\n      { postgres },\n      emitRetryDiagnostic\n        ? {\n            dataRegion: options?.dataRegion,\n            emit: emitRetryDiagnostic,\n            repositoryName: options?.repositoryName,\n            shouldEmit: isQueryRetryDiagnosticError,\n          }\n        : undefined\n    ),\n    {\n      repositoryName: options?.repositoryName,\n      log: options?.log,\n    }\n  );\n  kysely.destroy = async () => {\n    console.debug('Closing Kysely connection');\n    await postgres.end({ timeout: 5 }).catch(() => undefined);\n    const index = _connections.indexOf(postgres);\n    if (index !== -1) _connections.splice(index, 1);\n  };\n\n  return kysely;\n}\n\nlet _connections: postgres.Sql[] = [];\nexport async function useAsyncPostgres(connectionString: string, opt?: PgOpt) {\n  const delayMs = 2500;\n  const probe = opt?.probe === true;\n  const retries = probe ? (opt?.retries === false ? 0 : opt?.retries || 3) : 1;\n  const ssl = opt?.forceDisableSSL !== true && opt?.hyperdrive !== true;\n  const url = new URL(connectionString);\n\n  console.debug(\n    `Creating Postgres connection with probe=${probe}, ${retries} attempts, ssl=${ssl}, and port=${url.port}`\n  );\n  if (_connections.length > 0)\n    console.warn(`We already have ${_connections.length} Postgres connections`);\n\n  for (let attempt = 1; attempt <= retries; attempt++) {\n    let sql: Sql | undefined;\n    try {\n      sql = postgres(connectionString, {\n        ssl: ssl ? 'require' : false,\n        target_session_attrs: 'read-write',\n        // enable prepare statements, pgbouncer on crunchy allows for prepared statements\n        prepare: true,\n        ...(opt?.hyperdrive\n          ? {\n              max: 5,\n              // Retire idle clients before Hyperdrive can close them asynchronously.\n              idle_timeout: 10,\n            }\n          : {\n              connect_timeout: 10,\n              idle_timeout: 60,\n            }),\n      });\n\n      if (probe) {\n        await sql`select 1`;\n      }\n      // statement_timeout/lock_timeout are role-level settings on the cluster\n      // (ALTER ROLE application SET ...) so they cover every pooled\n      // connection; per-session SETs here only ever applied to one of them\n\n      _connections.push(sql);\n      return sql;\n    } catch (err) {\n      await sql?.end({ timeout: 0 }).catch(() => undefined);\n      if (attempt === retries) throw err;\n      console.warn(\n        `Postgres connection failed (attempt ${attempt}), retrying...`\n      );\n      console.error('error', err);\n      await new Promise((res) => setTimeout(res, delayMs));\n    }\n  }\n  throw new KyselyError('Unable to connect to Postgres');\n}\n\nexport async function useAsyncKysely(\n  connectionString: string,\n  opt?: PgOpt & KyselyOpts\n) {\n  const sql = await useAsyncPostgres(connectionString, opt);\n  return useKysely(sql, opt);\n}\n\nexport async function useAsyncNeonWebSocketKysely(\n  connectionString: string,\n  opt?: Pick<PgOpt, 'probe'> & KyselyOpts\n) {\n  const pool = new Pool({ connectionString, max: 5 });\n  const kysely = getKysely<DB>(new PostgresDialect({ pool }), opt ?? {});\n  kysely.destroy = async () => {\n    console.debug('Closing Neon WebSocket Kysely connection');\n    await pool.end();\n  };\n  if (opt?.probe) await pool.query('select 1');\n  return kysely;\n}\n\nexport async function useAsyncControlPlaneKysely(\n  connectionString: string,\n  opt?: PgOpt & KyselyOpts\n) {\n  const sql = await useAsyncPostgres(connectionString, opt);\n  return withControlPlaneSchema(useKysely<ControlPlaneDB>(sql, opt), 'public');\n}\n\nexport async function forceCloseAllPostgres(timeoutMs = 5000) {\n  await Promise.race([\n    Promise.all(\n      _connections.map((pg) =>\n        pg.end({ timeout: timeoutMs / 1000 }).catch(() => undefined)\n      )\n    ),\n    new Promise((res) => setTimeout(res, timeoutMs + 1000)),\n  ]);\n  _connections = [];\n}\n"]}