@vrplatform/kysely 1.3.45-stage.6299 → 1.3.45-stage.63

package/build/migrations-v2/0000000000032-tenant-delete-ownership-cascades.ts

@@ -0,0 +1,44 @@
+import type { Kysely } from 'kysely';
+import { sql } from 'kysely';
+
+export async function up(db: Kysely<unknown>): Promise<void> {
+  await sql`
+    alter table public.connection
+      drop constraint connection_tenant_id_fkey,
+      add constraint connection_tenant_id_fkey
+        foreign key (tenant_id)
+        references public.tenant(id)
+        on update restrict
+        on delete cascade
+  `.execute(db);
+  await sql`
+    alter table accounting.account_reservation_line_type
+      drop constraint account_reservation_line_type_account_id_fkey,
+      add constraint account_reservation_line_type_account_id_fkey
+        foreign key (account_id)
+        references accounting.account(id)
+        on update restrict
+        on delete cascade
+  `.execute(db);
+}
+
+export async function down(db: Kysely<unknown>): Promise<void> {
+  await sql`
+    alter table accounting.account_reservation_line_type
+      drop constraint account_reservation_line_type_account_id_fkey,
+      add constraint account_reservation_line_type_account_id_fkey
+        foreign key (account_id)
+        references accounting.account(id)
+        on update restrict
+        on delete restrict
+  `.execute(db);
+  await sql`
+    alter table public.connection
+      drop constraint connection_tenant_id_fkey,
+      add constraint connection_tenant_id_fkey
+        foreign key (tenant_id)
+        references public.tenant(id)
+        on update restrict
+        on delete restrict
+  `.execute(db);
+}

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[];
@@ -42,6 +43,36 @@ export interface ControlPlaneAccountingIntegrityEvaluation {
     status: string;
     updatedAt: Generated<Timestamp>;
 }
+export interface ControlPlaneAccountingIntegrityDefinitionDraft {
+    artifact: Json;
+    artifactHash: string;
+    code: string;
+    createdAt: Generated<Timestamp>;
+    id: Generated<string>;
+    ownerId: string;
+    previewResult: Json | null;
+    rationale: string;
+    revision: Generated<number>;
+    status: string;
+    templateId: string;
+    templateVersion: number;
+    title: string;
+    updatedAt: Generated<Timestamp>;
+    validationResult: Json | null;
+}
+export interface ControlPlaneAccountingIntegrityDefinitionVersion {
+    artifact: Json;
+    artifactHash: string;
+    code: string;
+    draftId: string;
+    id: Generated<string>;
+    publishedAt: Generated<Timestamp>;
+    publishedBy: string;
+    rollout: string;
+    templateId: string;
+    templateVersion: number;
+    version: number;
+}
 export interface ControlPlaneBookingChannel {
     channelRef: string | null;
     color: string | null;
@@ -74,6 +105,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<'completed' | 'failed' | 'queued' | 'running'>;
+    type: string;
+    updatedAt: Generated<Timestamp>;
+}
+export interface ControlPlaneOperationResource {
+    createdAt: Generated<Timestamp>;
+    isPrimary: Generated<boolean>;
+    operationId: string;
+    resourceId: string;
+    resourceType: string;
+    role: 'result' | 'source' | 'target';
+}
+export interface ControlPlaneOperationSource {
+    finishedAt: Timestamp | null;
+    operationId: string;
+    required: Generated<boolean>;
+    sourceId: string;
+    sourceType: 'destructive-job' | 'regional-operation';
+    sourceVersion: Generated<number>;
+    startedAt: Timestamp | null;
+    status: Generated<'completed' | 'failed' | 'queued' | 'running'>;
+}
 export interface ControlPlaneEvent {
     createdAt: Generated<Timestamp>;
     id: Generated<string>;
@@ -85,6 +202,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>;
@@ -119,6 +257,25 @@ export interface ControlPlaneFeatureApproval {
     updatedAt: Generated<Timestamp>;
     userId: string;
 }
+export interface ControlPlaneFlow {
+    appId: string;
+    createdAt: Generated<Timestamp>;
+    description: string | null;
+    eventListening: Generated<Json>;
+    id: Generated<string>;
+    isPublic: Generated<boolean>;
+    leftType: string | null;
+    mappingSchema: Generated<Json>;
+    revision: Generated<number>;
+    rightType: string | null;
+    runSchema: Generated<Json>;
+    settingSchema: Generated<Json>;
+    title: string;
+    type: 'pull' | 'push' | null;
+    uniqueRef: string;
+    updatedAt: Generated<Timestamp>;
+    useGlobalMapping: Generated<boolean>;
+}
 export interface ControlPlaneIssueMessageOverwrite {
     createdAt: Generated<Timestamp>;
     id: Generated<string>;
@@ -126,6 +283,74 @@ export interface ControlPlaneIssueMessageOverwrite {
     pattern: string;
     updatedAt: Generated<Timestamp>;
 }
+export interface ControlPlaneLineClassification {
+    appConflict: Generated<boolean>;
+    appId: string | null;
+    createdAt: Generated<Timestamp>;
+    firstSeenAt: Timestamp;
+    id: Generated<string>;
+    kind: string;
+    lastSeenAt: Timestamp;
+    name: string;
+    updatedAt: Generated<Timestamp>;
+}
+export interface ControlPlaneLineClassificationTeamUsage {
+    classificationId: string;
+    createdAt: Generated<Timestamp>;
+    dataRegion: string;
+    firstSeenAt: Timestamp;
+    lastSeenAt: Timestamp;
+    mappingStatus: string | null;
+    revision: number;
+    tenantId: string;
+    updatedAt: Generated<Timestamp>;
+}
+export interface ControlPlaneReservationLineDefault {
+    accountKey: string | null;
+    bookingChannel: string | null;
+    bookingChannelKey: Generated<string>;
+    classificationId: string;
+    createdAt: Generated<Timestamp>;
+    id: Generated<string>;
+    partnerTenantId: string | null;
+    revenueRecognition: string | null;
+    revision: Generated<number>;
+    updatedAt: Generated<Timestamp>;
+}
+export interface ControlPlaneRolloutMonitorSnapshot {
+    monitorName: string;
+    result: Json;
+    snapshotDate: Timestamp;
+}
+export interface ControlPlaneTransactionLineMapping {
+    accountAssignmentType: string | null;
+    createdAt: Generated<Timestamp>;
+    deletedAt: Timestamp | null;
+    id: Generated<string>;
+    lineType: string;
+    revision: Generated<number>;
+    updatedAt: Generated<Timestamp>;
+}
+export interface ControlPlaneGeneralLedgerDefaultSet {
+    createdAt: Generated<Timestamp>;
+    id: Generated<string>;
+    partnerTenantId: string | null;
+    publishedVersion: number | null;
+    sourceDataRegion: string;
+    sourceDigest: string | null;
+    sourceTenantId: string;
+    templateKey: string;
+    updatedAt: Generated<Timestamp>;
+}
+export interface ControlPlaneGeneralLedgerDefaultVersion {
+    dataJson: Json;
+    defaultSetId: string;
+    id: Generated<string>;
+    publishedAt: Generated<Timestamp>;
+    sourceDigest: string;
+    sourceTenantId: string;
+    version: number;
+}
 export interface ControlPlaneOutbox {
     aggregateId: string;
     aggregateType: string;
@@ -151,6 +376,7 @@ export interface ControlPlaneTenant {
     createdAt: Generated<Timestamp>;
     dataRegion: Generated<string>;
     id: Generated<string>;
+    migrationStatus: Generated<string>;
     name: string;
     partnerId: string | null;
     partnerDeniedPermissions: Generated<string[]>;
@@ -161,6 +387,21 @@ export interface ControlPlaneTenant {
     uniqueRef: string | null;
     updatedAt: Generated<Timestamp | null>;
 }
+export interface ControlPlaneTenantRegionMigration {
+    actor: string;
+    completedAt: Timestamp | null;
+    error: Json | null;
+    id: string;
+    reason: string;
+    sourceManifest: Json | null;
+    sourceRegion: string;
+    startedAt: Generated<Timestamp>;
+    status: Generated<string>;
+    targetManifest: Json | null;
+    targetRegion: string;
+    tenantId: string;
+    updatedAt: Generated<Timestamp>;
+}
 export interface ControlPlaneTenantUser {
     createdAt: Generated<Timestamp | null>;
     id: Generated<string>;
@@ -189,6 +430,7 @@ export interface ControlPlaneUser {
     email: string | null;
     firstName: string | null;
     id: Generated<string>;
+    internalOperationGrants: Generated<string[]>;
     isAdmin: Generated<boolean>;
     lastName: string | null;
     lastSeen: Timestamp | null;
@@ -202,6 +444,8 @@ export interface ControlPlaneUser {
     updatedAt: Generated<Timestamp | null>;
 }
 export interface ControlPlaneDB {
+    'controlPlane.accountingIntegrityDefinitionDraft': ControlPlaneAccountingIntegrityDefinitionDraft;
+    'controlPlane.accountingIntegrityDefinitionVersion': ControlPlaneAccountingIntegrityDefinitionVersion;
     'controlPlane.accountingIntegrityEvaluation': ControlPlaneAccountingIntegrityEvaluation;
     'controlPlane.app': ControlPlaneApp;
     'controlPlane.bookingChannel': ControlPlaneBookingChannel;
@@ -209,15 +453,31 @@ 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;
     'controlPlane.featureEnabledTeam': ControlPlaneFeatureEnabledTeam;
+    'controlPlane.generalLedgerDefaultSet': ControlPlaneGeneralLedgerDefaultSet;
+    '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.tenant': ControlPlaneTenant;
+    'controlPlane.tenantRegionMigration': ControlPlaneTenantRegionMigration;
     'controlPlane.tenantUser': ControlPlaneTenantUser;
+    'controlPlane.trackingQueueFailure': ControlPlaneTrackingQueueFailure;
     'controlPlane.token': ControlPlaneToken;
+    'controlPlane.transactionLineMapping': ControlPlaneTransactionLineMapping;
     'controlPlane.user': ControlPlaneUser;
+    'controlPlaneCore.flow': ControlPlaneFlow;
     'controlPlaneCore.issueMessageOverwrite': ControlPlaneIssueMessageOverwrite;
 }
 export declare function withControlPlaneSchema(kysely: Kysely<ControlPlaneDB>, schema: string, issueMessageOverwriteSchema?: string): Kysely<ControlPlaneDB>;

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;AAgQhB,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 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 ControlPlaneIssueMessageOverwrite {\n  createdAt: Generated<Timestamp>;\n  id: Generated<string>;\n  message: string;\n  pattern: string;\n  updatedAt: Generated<Timestamp>;\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  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 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  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.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.partnerManagedTeamAssignment': ControlPlanePartnerManagedTeamAssignment;\n  'controlPlane.tenant': ControlPlaneTenant;\n  'controlPlane.tenantUser': ControlPlaneTenantUser;\n  'controlPlane.token': ControlPlaneToken;\n  'controlPlane.user': ControlPlaneUser;\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;AA+hBhB,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 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<'completed' | 'failed' | 'queued' | 'running'>;\n  type: string;\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: string;\n  role: 'result' | 'source' | 'target';\n}\n\nexport interface ControlPlaneOperationSource {\n  finishedAt: Timestamp | null;\n  operationId: string;\n  required: Generated<boolean>;\n  sourceId: string;\n  sourceType: 'destructive-job' | 'regional-operation';\n  sourceVersion: Generated<number>;\n  startedAt: Timestamp | null;\n  status: Generated<'completed' | 'failed' | 'queued' | 'running'>;\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 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.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

@@ -10,15 +10,15 @@ export * from './regional';
 export * from './transaction';
 import type { DB } from './v1.generated';
 type KyselyOpts = {
+    hyperdrive?: true;
     repositoryName?: string;
     log?: true | LogConfig;
 };
 type PgOpt = {
     forceDisableSSL?: boolean;
-    hyperdrive?: true;
     probe?: true;
     retries?: false | number;
-};
+} & Pick<KyselyOpts, 'hyperdrive'>;
 export type Kysely = K<DB>;
 export type ControlPlaneKysely = K<ControlPlaneDB>;
 export type ControlPlaneTransaction = T<ControlPlaneDB>;

package/build/module/index.js

@@ -1,5 +1,5 @@
 import { backOff } from 'exponential-backoff';
-import { PostgresJSDialect } from 'kysely-postgres-js';
+import { PostgresJSDialect, PostgresJSDriver, } from 'kysely-postgres-js';
 import postgres from 'postgres';
 import { withControlPlaneSchema } from './control-plane';
 import { KyselyError } from './error';
@@ -15,40 +15,121 @@ const retryableMessages = [
     'CONNECTION_CLOSED',
     'CONNECT_TIMEOUT',
     'Idle connection closed by Hyperdrive.',
+    'server_login_retry',
 ];
+const postgresRetryAttempts = 3;
 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: 3,
+        numOfAttempts: postgresRetryAttempts,
         startingDelay: parseDuration('1s'),
         maxDelay: parseDuration('5s'),
-        retry,
+        retry: (error, attemptNumber) => attemptNumber < postgresRetryAttempts && retry(error, attemptNumber),
     });
 }
+class RetryingPostgresJSDialect extends PostgresJSDialect {
+    config;
+    constructor(config) {
+        super(config);
+        this.config = config;
+    }
+    createDriver() {
+        const driver = new PostgresJSDriver(this.config);
+        const connections = new WeakMap();
+        const getConnection = (connection) => {
+            const state = connections.get(connection);
+            if (!state)
+                throw new Error('Postgres connection state is unavailable');
+            return state;
+        };
+        return {
+            init: () => driver.init(),
+            acquireConnection: async () => {
+                const state = {
+                    connection: await driver.acquireConnection(),
+                    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;
+                    }),
+                    streamQuery: (compiledQuery, chunkSize) => {
+                        if (chunkSize === undefined) {
+                            throw new Error('Postgres stream chunk size is required');
+                        }
+                        return state.connection.streamQuery(compiledQuery, chunkSize);
+                    },
+                };
+                connections.set(connection, state);
+                return connection;
+            },
+            beginTransaction: async (connection, settings) => {
+                const state = getConnection(connection);
+                state.inTransaction = true;
+                try {
+                    await driver.beginTransaction(state.connection, settings);
+                }
+                catch (error) {
+                    state.inTransaction = false;
+                    throw error;
+                }
+            },
+            commitTransaction: async (connection) => {
+                const state = getConnection(connection);
+                try {
+                    await driver.commitTransaction(state.connection);
+                }
+                finally {
+                    state.inTransaction = false;
+                }
+            },
+            rollbackTransaction: async (connection) => {
+                const state = getConnection(connection);
+                try {
+                    await driver.rollbackTransaction(state.connection);
+                }
+                finally {
+                    state.inTransaction = false;
+                }
+            },
+            savepoint: (connection, savepointName, compileQuery) => driver.savepoint(getConnection(connection).connection, savepointName, compileQuery),
+            rollbackToSavepoint: (connection, savepointName, compileQuery) => driver.rollbackToSavepoint(getConnection(connection).connection, savepointName, compileQuery),
+            releaseSavepoint: (connection, savepointName, compileQuery) => driver.releaseSavepoint(getConnection(connection).connection, savepointName, compileQuery),
+            releaseConnection: async (connection) => {
+                await driver.releaseConnection(getConnection(connection).connection);
+                connections.delete(connection);
+            },
+            destroy: () => driver.destroy(),
+        };
+    }
+}
 export function useKysely(postgres, options) {
     console.debug('Creating Kysely instance');
     const reserveConnection = postgres.reserve.bind(postgres);
-    postgres.reserve = () => retryPostgres(reserveConnection, isRetryableConnectionError);
-    const kysely = getKysely(new PostgresJSDialect({
-        postgres,
-        onReserveConnection: async (connection) => {
-            const executeQuery = connection.executeQuery.bind(connection);
-            connection.executeQuery = (arg) => retryPostgres(async () => await executeQuery(arg), (error) => {
-                const message = error instanceof Error ? error.message : String(error);
-                if (isRetryableConnectionError(error) ||
-                    (arg.query.kind === 'SelectQueryNode' &&
-                        (message ===
-                            'Egress connection severed outside of Hyperdrive' ||
-                            message === 'Network connection lost.'))) {
-                    return true;
-                }
-                return false;
-            });
-        },
-    }), {
+    postgres.reserve = async () => {
+        const connection = await retryPostgres(reserveConnection, isRetryableConnectionError);
+        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 }), {
         repositoryName: options?.repositoryName,
         log: options?.log,
     });