@vrplatform/kysely 1.3.45-stage.31 → 1.3.45-stage.313

package/build/main/index.js

@@ -46,6 +46,9 @@ function isRetryableConnectionError(error) {
     const message = error instanceof Error ? error.message : String(error);
     return retryableMessages.some((retryable) => message.includes(retryable));
 }
+function isConnectionReset(error) {
+    return (error instanceof Error && 'code' in error && error.code === 'ECONNRESET');
+}
 function retryPostgres(operation, retry) {
     return (0, exponential_backoff_1.backOff)(operation, {
         numOfAttempts: postgresRetryAttempts,
@@ -81,9 +84,11 @@ class RetryingPostgresJSDialect extends kysely_postgres_js_1.PostgresJSDialect {
                         if (state.inTransaction ||
                             (!isRetryableConnectionError(error) &&
                                 (compiledQuery.query.kind !== 'SelectQueryNode' ||
-                                    (message !==
-                                        'Egress connection severed outside of Hyperdrive' &&
-                                        message !== 'Network connection lost.')))) {
+                                    (!isConnectionReset(error) &&
+                                        message !==
+                                            'Egress connection severed outside of Hyperdrive' &&
+                                        message !== 'Network connection lost.' &&
+                                        message !== 'Node shutting down')))) {
                             return false;
                         }
                         // Postgres.js already ejects a severed reserved connection.
@@ -145,7 +150,7 @@ function useKysely(postgres, options) {
     console.debug('Creating Kysely instance');
     const reserveConnection = postgres.reserve.bind(postgres);
     postgres.reserve = async () => {
-        const connection = await retryPostgres(reserveConnection, isRetryableConnectionError);
+        const connection = await retryPostgres(reserveConnection, (error) => isRetryableConnectionError(error) || isConnectionReset(error));
         if (options?.hyperdrive) {
             const unsafe = connection.unsafe;
             connection.unsafe = (query, parameters, queryOptions) => unsafe(query, parameters, { ...queryOptions, prepare: true });

package/build/main/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;AAqLA,8BA8BC;AAGD,4CAmDC;AAED,wCAMC;AAED,gEAMC;AAED,sDAUC;AArSD,6DAA8C;AAQ9C,2DAI4B;AAC5B,wDAA8C;AAC9C,mDAA8E;AAC9E,mCAAsC;AACtC,uCAAsC;AAGtC,0CAAwB;AACxB,+CAA6B;AAC7B,mDAAiC;AACjC,0CAAwB;AACxB,6CAA2B;AAC3B,gDAA8B;AAE9B,6CAAkD;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,IAAA,6BAAO,EAAC,SAAS,EAAE;QACxB,aAAa,EAAE,qBAAqB;QACpC,aAAa,EAAE,IAAA,qBAAa,EAAC,IAAI,CAAC;QAClC,QAAQ,EAAE,IAAA,qBAAa,EAAC,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,sCAAiB;IACvD,YAA6B,MAA+B;QAC1D,KAAK,CAAC,MAAM,CAAC,CAAC;QADa,WAAM,GAAN,MAAM,CAAyB;IAE5D,CAAC;IAEQ,YAAY;QACnB,MAAM,MAAM,GAAG,IAAI,qCAAgB,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,SAAgB,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,IAAA,mBAAS,EACtB,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;AAC/B,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,IAAA,kBAAQ,EAAC,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,mBAAW,CAAC,+BAA+B,CAAC,CAAC;AACzD,CAAC;AAEM,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;AAEM,KAAK,UAAU,0BAA0B,CAC9C,gBAAwB,EACxB,GAAwB;IAExB,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,IAAA,sCAAsB,EAAC,SAAS,CAAiB,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,CAAC,CAAC;AAC/E,CAAC;AAEM,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":";;;;;;;;;;;;;;;;;;;AA6LA,8BA8BC;AAGD,4CAmDC;AAED,wCAMC;AAED,gEAMC;AAED,sDAUC;AA7SD,6DAA8C;AAQ9C,2DAI4B;AAC5B,wDAA8C;AAC9C,mDAA8E;AAC9E,mCAAsC;AACtC,uCAAsC;AAGtC,0CAAwB;AACxB,+CAA6B;AAC7B,mDAAiC;AACjC,0CAAwB;AACxB,6CAA2B;AAC3B,gDAA8B;AAE9B,6CAAkD;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,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,aAAa,CACpB,SAA2B,EAC3B,KAA4E;IAE5E,OAAO,IAAA,6BAAO,EAAC,SAAS,EAAE;QACxB,aAAa,EAAE,qBAAqB;QACpC,aAAa,EAAE,IAAA,qBAAa,EAAC,IAAI,CAAC;QAClC,QAAQ,EAAE,IAAA,qBAAa,EAAC,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,sCAAiB;IACvD,YAA6B,MAA+B;QAC1D,KAAK,CAAC,MAAM,CAAC,CAAC;QADa,WAAM,GAAN,MAAM,CAAyB;IAE5D,CAAC;IAEQ,YAAY;QACnB,MAAM,MAAM,GAAG,IAAI,qCAAgB,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,CAAC,iBAAiB,CAAC,KAAK,CAAC;wCACxB,OAAO;4CACL,iDAAiD;wCACnD,OAAO,KAAK,0BAA0B;wCACtC,OAAO,KAAK,oBAAoB,CAAC,CAAC,CAAC,EACzC,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,SAAgB,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,CAAC,KAAK,EAAE,EAAE,CAAC,0BAA0B,CAAC,KAAK,CAAC,IAAI,iBAAiB,CAAC,KAAK,CAAC,CACzE,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,IAAA,mBAAS,EACtB,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;AAC/B,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,IAAA,kBAAQ,EAAC,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,mBAAW,CAAC,+BAA+B,CAAC,CAAC;AACzD,CAAC;AAEM,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;AAEM,KAAK,UAAU,0BAA0B,CAC9C,gBAAwB,EACxB,GAAwB;IAExB,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,IAAA,sCAAsB,EAAC,SAAS,CAAiB,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,CAAC,CAAC;AAC/E,CAAC;AAEM,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 isConnectionReset(error: unknown) {\n  return (\n    error instanceof Error && 'code' in error && error.code === 'ECONNRESET'\n  );\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                      (!isConnectionReset(error) &&\n                        message !==\n                          'Egress connection severed outside of Hyperdrive' &&\n                        message !== 'Network connection lost.' &&\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            ),\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      (error) => isRetryableConnectionError(error) || isConnectionReset(error)\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"]}

package/build/main/local/index.js

@@ -64,17 +64,68 @@ async function migrateToLatest(kysely) {
         throw error;
     }
 }
-async function useLocalKysely(args = {}) {
-    let isDestroyed = false;
-    // const now = performance.now();
-    const dump = Bun.file(exports.dumpPath);
+// Every spec file used to replay this identical bootstrap - migrations, the
+// schema surgery, and the control-plane seed copies - against a freshly loaded
+// snapshot, which measured ~345ms per file against ~185ms for the snapshot load
+// alone. None of it depends on the caller, so it runs once and is cached as a
+// prepared snapshot keyed by everything that determines its content.
+const preparedDumpPrefix = '.data.prepared.';
+let preparedDumpPromise;
+async function preparedDumpKey() {
+    const [snapshot, appMigrations, controlPlaneMigrations] = await Promise.all([
+        node_fs_1.promises.stat(exports.dumpPath),
+        node_fs_1.promises.readdir(path.join(exports.root, migration_1.appMigrationEpoch.folder)),
+        node_fs_1.promises.readdir(path.join(exports.root, 'control-plane-migrations')),
+    ]);
+    const hasher = new Bun.CryptoHasher('sha256');
+    hasher.update(`${snapshot.size}:${snapshot.mtimeMs}`);
+    hasher.update(appMigrations.sort().join(','));
+    hasher.update(controlPlaneMigrations.sort().join(','));
+    // Source of the bootstrap itself, so editing the surgery rebuilds the
+    // snapshot. Read through toString() rather than the file: this module is also
+    // published as CommonJS, where import.meta is a compile error.
+    for (const step of [
+        buildPreparedDump,
+        migrateToLatest,
+        digest_1.ensureDigestFunction,
+        digest_1.ensureGenRandomBytesFunction,
+    ]) {
+        hasher.update(step.toString());
+    }
+    return hasher.digest('hex').slice(0, 16);
+}
+async function resolvePreparedDump() {
+    const key = await preparedDumpKey();
+    const target = path.join(exports.root, `${preparedDumpPrefix}${key}.tar`);
+    const cached = Bun.file(target);
+    if (await cached.exists()) {
+        return cached;
+    }
+    const content = await buildPreparedDump();
+    // Sibling workers build this concurrently on a cold cache; rename is atomic,
+    // so a reader never observes a half-written snapshot.
+    const staging = `${target}.${process.pid}.tmp`;
+    await Bun.write(staging, content);
+    await node_fs_1.promises.rename(staging, target);
+    // Only finished snapshots: a sibling worker's staging file also carries the
+    // prefix, and removing it would break its rename.
+    const current = path.basename(target);
+    for (const entry of await node_fs_1.promises.readdir(exports.root)) {
+        if (entry.startsWith(preparedDumpPrefix) &&
+            entry.endsWith('.tar') &&
+            entry !== current) {
+            await node_fs_1.promises.rm(path.join(exports.root, entry), { force: true });
+        }
+    }
+    return content;
+}
+async function buildPreparedDump() {
     const pglite = new pglite_1.PGlite({
-        loadDataDir: dump,
+        loadDataDir: Bun.file(exports.dumpPath),
     });
     const dialect = new kysely_pglite_dialect_1.PGliteDialect(pglite);
     const kysely = (0, plugins_1.getKysely)(dialect, {
-        repositoryName: args?.repositoryName ?? 'localTesting',
-        log: args?.log,
+        repositoryName: 'localTestingPrepare',
     });
     await (0, digest_1.ensureDigestFunction)(kysely);
     await (0, digest_1.ensureGenRandomBytesFunction)(kysely);
@@ -84,7 +135,7 @@ async function useLocalKysely(args = {}) {
     await (0, kysely_1.sql) `create schema public`.execute(kysely);
     await (0, kysely_1.sql) `create schema core`.execute(kysely);
     const controlPlaneBootstrapKysely = (0, plugins_1.getKysely)(dialect, {
-        repositoryName: `${args.repositoryName ?? 'localTesting'}/control-plane-bootstrap`,
+        repositoryName: 'localTestingPrepare/control-plane-bootstrap',
     });
     const controlPlaneMigrator = new kysely_1.Migrator({
         db: controlPlaneBootstrapKysely,
@@ -134,6 +185,26 @@ async function useLocalKysely(args = {}) {
     end;
     $$;
   `.execute(controlPlaneBootstrapKysely);
+    // The canonical VRPlatform tenant. Production code falls back to this exact id
+    // in getCalculatedBillingPartnerId, so specs need it to exist without each of
+    // them creating - and mutating - their own copy of a shared singleton.
+    await kysely
+        .insertInto('public.tenant')
+        .values({
+        id: '7bfec41f-aa3f-4602-95d2-f7996e4f0e59',
+        name: 'VRPlatform',
+        type: 'partner',
+        status: 'active',
+        isActive: true,
+        isGeneralLedger: false,
+        isOnboarding: false,
+        partnerId: null,
+        billingPartnerId: null,
+        defaultCurrency: 'usd',
+        statementStartAt: new Date('2020-01-01T00:00:00Z'),
+    })
+        .onConflict((oc) => oc.column('id').doNothing())
+        .execute();
     await (0, kysely_1.sql) `
     insert into control_plane.app (
       authentication,
@@ -289,6 +360,25 @@ async function useLocalKysely(args = {}) {
     from public.token
     on conflict (id) do nothing;
   `.execute(controlPlaneBootstrapKysely);
+    const content = await pglite.dumpDataDir('none');
+    await kysely.destroy();
+    return content;
+}
+async function useLocalKysely(args = {}) {
+    let isDestroyed = false;
+    preparedDumpPromise ?? (preparedDumpPromise = resolvePreparedDump());
+    const pglite = new pglite_1.PGlite({
+        loadDataDir: await preparedDumpPromise,
+    });
+    // The constructor is lazy. The bootstrap this replaced forced the snapshot to
+    // load before returning, so keep that contract: a returned handle is usable,
+    // and a bad snapshot fails here instead of inside an unrelated test.
+    await pglite.waitReady;
+    const dialect = new kysely_pglite_dialect_1.PGliteDialect(pglite);
+    const kysely = (0, plugins_1.getKysely)(dialect, {
+        repositoryName: args?.repositoryName ?? 'localTesting',
+        log: args?.log,
+    });
     const controlPlaneKysely = (0, control_plane_1.withControlPlaneSchema)((0, plugins_1.getKysely)(dialect, {
         repositoryName: `${args.repositoryName ?? 'localTesting'}/control-plane`,
         log: args.log,

package/build/main/local/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["local/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8CA,wCA+QC;AA7TD,qCAAyC;AACzC,gDAAkC;AAClC,iDAA8C;AAC9C,mCAA8E;AAC9E,iEAAsD;AACtD,oDAA+E;AAE/E,4CAAiD;AACjD,wCAAuC;AAEvC,qCAA8E;AAEjE,QAAA,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;AACrC,QAAA,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,YAAI,EAAE,WAAW,CAAC,CAAC;AAWrD,KAAK,UAAU,eAAe,CAC5B,MAAyC;IAEzC,MAAM,QAAQ,GAAG,IAAI,iBAAQ,CAAC;QAC5B,wBAAwB,EAAE,IAAI;QAC9B,EAAE,EAAE,MAAM;QACV,oBAAoB,EAAE,6BAAiB,CAAC,WAAW;QACnD,kBAAkB,EAAE,6BAAiB,CAAC,SAAS;QAC/C,sBAAsB,EAAE,6BAAiB,CAAC,aAAa;QACvD,QAAQ,EAAE,IAAI,8BAAqB,CAAC;YAClC,EAAE,EAAF,kBAAE;YACF,IAAI;YACJ,eAAe,EAAE,IAAI,CAAC,IAAI,CAAC,YAAI,EAAE,6BAAiB,CAAC,MAAM,CAAC;SAC3D,CAAC;KACH,CAAC,CAAC;IAEH,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,eAAe,EAAE,CAAC;IACnD,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,cAAc,CAClC,OAA8B,EAAE;IAEhC,IAAI,WAAW,GAAG,KAAK,CAAC;IACxB,iCAAiC;IAEjC,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,gBAAQ,CAAC,CAAC;IAEhC,MAAM,MAAM,GAAG,IAAI,eAAM,CAAC;QACxB,WAAW,EAAE,IAAI;KAClB,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,IAAI,qCAAa,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAA,mBAAS,EAAK,OAAO,EAAE;QACpC,cAAc,EAAE,IAAI,EAAE,cAAc,IAAI,cAAc;QACtD,GAAG,EAAE,IAAI,EAAE,GAAG;KACf,CAAC,CAAC;IACH,MAAM,IAAA,6BAAoB,EAAC,MAAM,CAAC,CAAC;IACnC,MAAM,IAAA,qCAA4B,EAAC,MAAM,CAAC,CAAC;IAC3C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAE9B,MAAM,IAAA,YAAG,EAAA,+CAA+C,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACzE,MAAM,IAAA,YAAG,EAAA,2CAA2C,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACrE,MAAM,IAAA,YAAG,EAAA,sBAAsB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAChD,MAAM,IAAA,YAAG,EAAA,oBAAoB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAE9C,MAAM,2BAA2B,GAAG,IAAA,mBAAS,EAAiB,OAAO,EAAE;QACrE,cAAc,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,cAAc,0BAA0B;KACnF,CAAC,CAAC;IACH,MAAM,oBAAoB,GAAG,IAAI,iBAAQ,CAAC;QACxC,EAAE,EAAE,2BAA2B;QAC/B,oBAAoB,EAAE,QAAQ;QAC9B,kBAAkB,EAAE,gCAAgC;QACpD,sBAAsB,EAAE,qCAAqC;QAC7D,QAAQ,EAAE,IAAI,8BAAqB,CAAC;YAClC,EAAE,EAAF,kBAAE;YACF,IAAI;YACJ,eAAe,EAAE,IAAI,CAAC,IAAI,CAAC,YAAI,EAAE,0BAA0B,CAAC;SAC7D,CAAC;KACH,CAAC,CAAC;IACH,MAAM,EAAE,KAAK,EAAE,0BAA0B,EAAE,GACzC,MAAM,oBAAoB,CAAC,eAAe,EAAE,CAAC;IAC/C,IAAI,0BAA0B,EAAE,CAAC;QAC/B,MAAM,0BAA0B,CAAC;IACnC,CAAC;IAED,MAAM,IAAA,YAAG,EAAA,yCAAyC,CAAC,OAAO,CACxD,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,YAAG,EAAA,4DAA4D,CAAC,OAAO,CAC3E,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,YAAG,EAAA,kBAAkB,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACjE,MAAM,IAAA,YAAG,EAAA,6CAA6C,CAAC,OAAO,CAC5D,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,YAAG,EAAA,+CAA+C,CAAC,OAAO,CAC9D,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,YAAG,EAAA,2CAA2C,CAAC,OAAO,CAC1D,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;GAKR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;GAsBR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IAEvC,MAAM,kBAAkB,GAAG,IAAA,sCAAsB,EAC/C,IAAA,mBAAS,EAAiB,OAAO,EAAE;QACjC,cAAc,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,cAAc,gBAAgB;QACxE,GAAG,EAAE,IAAI,CAAC,GAAG;KACd,CAAC,EACF,eAAe,EACf,eAAe,CAChB,CAAC;IACF,kBAAkB,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;IAC5C,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,oBAAoB,EAAE;QAClD,KAAK,EAAE,kBAAkB;KAC1B,CAAC,CAAC;IAEH,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACpD,MAAM,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,IAAI,WAAW;YAAE,OAAO;QACxB,WAAW,GAAG,IAAI,CAAC;QACnB,wCAAwC;QACxC,MAAM,eAAe,EAAE,CAAC;QACxB,cAAc;QACd,0EAA0E;QAC1E,IAAI;IACN,CAAC,CAAC;IAEF,iFAAiF;IAEjF,OAAO,MAAqB,CAAC;AAC/B,CAAC","sourcesContent":["import { promises as fs } from 'node:fs';\nimport * as path from 'node:path';\nimport { PGlite } from '@electric-sql/pglite';\nimport { FileMigrationProvider, type LogConfig, Migrator, sql } from 'kysely';\nimport { PGliteDialect } from 'kysely-pglite-dialect';\nimport { type ControlPlaneDB, withControlPlaneSchema } from '../control-plane';\nimport type { ControlPlaneKysely } from '../index';\nimport { appMigrationEpoch } from '../migration';\nimport { getKysely } from '../plugins';\nimport type { DB } from '../v1.generated';\nimport { ensureDigestFunction, ensureGenRandomBytesFunction } from './digest';\n\nexport const root = path.join(__dirname, '../..');\nexport const dumpPath = path.join(root, '.data.tar');\n\nexport type UseLocalKyselyOptions = {\n  repositoryName?: string;\n  log?: true | LogConfig;\n};\n\nexport type LocalKysely = import('kysely').Kysely<DB> & {\n  controlPlaneKysely: ControlPlaneKysely;\n};\n\nasync function migrateToLatest<Database>(\n  kysely: import('kysely').Kysely<Database>\n) {\n  const migrator = new Migrator({\n    allowUnorderedMigrations: true,\n    db: kysely,\n    migrationTableSchema: appMigrationEpoch.tableSchema,\n    migrationTableName: appMigrationEpoch.tableName,\n    migrationLockTableName: appMigrationEpoch.lockTableName,\n    provider: new FileMigrationProvider({\n      fs,\n      path,\n      migrationFolder: path.join(root, appMigrationEpoch.folder),\n    }),\n  });\n\n  const { error } = await migrator.migrateToLatest();\n  if (error) {\n    throw error;\n  }\n}\n\nexport async function useLocalKysely(\n  args: UseLocalKyselyOptions = {}\n): Promise<LocalKysely> {\n  let isDestroyed = false;\n  // const now = performance.now();\n\n  const dump = Bun.file(dumpPath);\n\n  const pglite = new PGlite({\n    loadDataDir: dump,\n  });\n  const dialect = new PGliteDialect(pglite);\n  const kysely = getKysely<DB>(dialect, {\n    repositoryName: args?.repositoryName ?? 'localTesting',\n    log: args?.log,\n  });\n  await ensureDigestFunction(kysely);\n  await ensureGenRandomBytesFunction(kysely);\n  await migrateToLatest(kysely);\n\n  await sql`alter schema public rename to regional_public`.execute(kysely);\n  await sql`alter schema core rename to regional_core`.execute(kysely);\n  await sql`create schema public`.execute(kysely);\n  await sql`create schema core`.execute(kysely);\n\n  const controlPlaneBootstrapKysely = getKysely<ControlPlaneDB>(dialect, {\n    repositoryName: `${args.repositoryName ?? 'localTesting'}/control-plane-bootstrap`,\n  });\n  const controlPlaneMigrator = new Migrator({\n    db: controlPlaneBootstrapKysely,\n    migrationTableSchema: 'public',\n    migrationTableName: 'control_plane_kysely_migration',\n    migrationLockTableName: 'control_plane_kysely_migration_lock',\n    provider: new FileMigrationProvider({\n      fs,\n      path,\n      migrationFolder: path.join(root, 'control-plane-migrations'),\n    }),\n  });\n  const { error: controlPlaneMigrationError } =\n    await controlPlaneMigrator.migrateToLatest();\n  if (controlPlaneMigrationError) {\n    throw controlPlaneMigrationError;\n  }\n\n  await sql`alter table core.flow set schema public`.execute(\n    controlPlaneBootstrapKysely\n  );\n  await sql`alter table core.issue_message_overwrite set schema public`.execute(\n    controlPlaneBootstrapKysely\n  );\n  await sql`drop schema core`.execute(controlPlaneBootstrapKysely);\n  await sql`alter schema public rename to control_plane`.execute(\n    controlPlaneBootstrapKysely\n  );\n  await sql`alter schema regional_public rename to public`.execute(\n    controlPlaneBootstrapKysely\n  );\n  await sql`alter schema regional_core rename to core`.execute(\n    controlPlaneBootstrapKysely\n  );\n  await sql`\n    do $$\n    declare\n      definition text;\n    begin\n      select pg_get_functiondef(\n        'control_plane.normalize_tenant_slug(text)'::regprocedure\n      ) into definition;\n      execute replace(definition, 'public.', 'control_plane.');\n\n      select pg_get_functiondef(\n        'control_plane.build_unique_tenant_slug(text,uuid)'::regprocedure\n      ) into definition;\n      execute replace(definition, 'public.', 'control_plane.');\n\n      select pg_get_functiondef(\n        'control_plane.refresh_operation_state(uuid)'::regprocedure\n      ) into definition;\n      execute replace(definition, 'public.', 'control_plane.');\n\n      select pg_get_functiondef(\n        'control_plane.refresh_destructive_job_operation()'::regprocedure\n      ) into definition;\n      execute replace(definition, 'public.', 'control_plane.');\n    end;\n    $$;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.app (\n      authentication,\n      category,\n      color,\n      created_at,\n      icon,\n      icon_round,\n      id,\n      info,\n      name,\n      type,\n      version\n    )\n    select\n      authentication,\n      category,\n      color,\n      created_at,\n      icon,\n      icon_round,\n      id,\n      info,\n      name,\n      type,\n      version\n    from public.app\n    on conflict (id) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.currency (name)\n    select name\n    from public.currency\n    on conflict (name) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.booking_channel (\n      channel_ref,\n      color,\n      created_at,\n      icon,\n      icon_provider,\n      id,\n      last_generate_run,\n      logo,\n      logo_provider,\n      selected_booking_channel_icon_candidate_id,\n      selected_booking_channel_logo_candidate_id,\n      unique_ref,\n      updated_at\n    )\n    select\n      channel_ref,\n      color,\n      created_at,\n      icon,\n      icon_provider,\n      id,\n      last_generate_run,\n      logo,\n      logo_provider,\n      selected_booking_channel_icon_candidate_id,\n      selected_booking_channel_logo_candidate_id,\n      unique_ref,\n      updated_at\n    from public.booking_channel\n    on conflict (id) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.booking_channel_icon_candidate (\n      booking_channel_id,\n      cloudflare_image_id,\n      color,\n      comment,\n      created_at,\n      external_icon,\n      height,\n      id,\n      provider,\n      source,\n      updated_at,\n      width\n    )\n    select\n      booking_channel_id,\n      cloudflare_image_id,\n      color,\n      comment,\n      created_at,\n      external_icon,\n      height,\n      id,\n      provider,\n      source,\n      updated_at,\n      width\n    from public.booking_channel_icon_candidate\n    on conflict (id) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.tenant (\n      billing_partner_id,\n      billing_plan,\n      created_at,\n      data_region,\n      id,\n      name,\n      partner_id,\n      slug,\n      status,\n      storage_realm,\n      type,\n      unique_ref,\n      updated_at\n    )\n    select\n      billing_partner_id,\n      billing_plan,\n      created_at,\n      data_region,\n      id,\n      name,\n      partner_id,\n      slug,\n      status,\n      storage_realm,\n      type,\n      unique_ref,\n      updated_at\n    from public.tenant\n    on conflict (id) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.token (\n      created_at,\n      expires_at,\n      id,\n      nano_id,\n      payload,\n      tenant_id,\n      type,\n      user_id\n    )\n    select\n      created_at,\n      expires_at,\n      id,\n      nano_id,\n      payload,\n      tenant_id,\n      type,\n      user_id\n    from public.token\n    on conflict (id) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n\n  const controlPlaneKysely = withControlPlaneSchema(\n    getKysely<ControlPlaneDB>(dialect, {\n      repositoryName: `${args.repositoryName ?? 'localTesting'}/control-plane`,\n      log: args.log,\n    }),\n    'control_plane',\n    'control_plane'\n  );\n  controlPlaneKysely.destroy = async () => {};\n  Object.defineProperty(kysely, 'controlPlaneKysely', {\n    value: controlPlaneKysely,\n  });\n\n  const originalDestroy = kysely.destroy.bind(kysely);\n  kysely.destroy = async () => {\n    if (isDestroyed) return;\n    isDestroyed = true;\n    //const destroyTime = performance.now();\n    await originalDestroy();\n    //console.log(\n    //  `Database destroyed: ${Math.round(performance.now() - destroyTime)}ms`\n    //);\n  };\n\n  // console.log(`Database initialized: ${Math.round(performance.now() - now)}ms`);\n\n  return kysely as LocalKysely;\n}\n"]}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["local/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqXA,wCA8CC;AAnaD,qCAAyC;AACzC,gDAAkC;AAClC,iDAA8C;AAC9C,mCAA8E;AAC9E,iEAAsD;AACtD,oDAA+E;AAE/E,4CAAiD;AACjD,wCAAuC;AAEvC,qCAA8E;AAEjE,QAAA,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;AACrC,QAAA,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,YAAI,EAAE,WAAW,CAAC,CAAC;AAWrD,KAAK,UAAU,eAAe,CAC5B,MAAyC;IAEzC,MAAM,QAAQ,GAAG,IAAI,iBAAQ,CAAC;QAC5B,wBAAwB,EAAE,IAAI;QAC9B,EAAE,EAAE,MAAM;QACV,oBAAoB,EAAE,6BAAiB,CAAC,WAAW;QACnD,kBAAkB,EAAE,6BAAiB,CAAC,SAAS;QAC/C,sBAAsB,EAAE,6BAAiB,CAAC,aAAa;QACvD,QAAQ,EAAE,IAAI,8BAAqB,CAAC;YAClC,EAAE,EAAF,kBAAE;YACF,IAAI;YACJ,eAAe,EAAE,IAAI,CAAC,IAAI,CAAC,YAAI,EAAE,6BAAiB,CAAC,MAAM,CAAC;SAC3D,CAAC;KACH,CAAC,CAAC;IAEH,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,eAAe,EAAE,CAAC;IACnD,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,4EAA4E;AAC5E,+EAA+E;AAC/E,gFAAgF;AAChF,8EAA8E;AAC9E,qEAAqE;AACrE,MAAM,kBAAkB,GAAG,iBAAiB,CAAC;AAE7C,IAAI,mBAA8C,CAAC;AAEnD,KAAK,UAAU,eAAe;IAC5B,MAAM,CAAC,QAAQ,EAAE,aAAa,EAAE,sBAAsB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QAC1E,kBAAE,CAAC,IAAI,CAAC,gBAAQ,CAAC;QACjB,kBAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,YAAI,EAAE,6BAAiB,CAAC,MAAM,CAAC,CAAC;QACrD,kBAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,YAAI,EAAE,0BAA0B,CAAC,CAAC;KACxD,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC9C,MAAM,CAAC,MAAM,CAAC,GAAG,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC;IACtD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAC9C,MAAM,CAAC,MAAM,CAAC,sBAAsB,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IACvD,sEAAsE;IACtE,8EAA8E;IAC9E,+DAA+D;IAC/D,KAAK,MAAM,IAAI,IAAI;QACjB,iBAAiB;QACjB,eAAe;QACf,6BAAoB;QACpB,qCAA4B;KAC7B,EAAE,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IACjC,CAAC;IACD,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED,KAAK,UAAU,mBAAmB;IAChC,MAAM,GAAG,GAAG,MAAM,eAAe,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,YAAI,EAAE,GAAG,kBAAkB,GAAG,GAAG,MAAM,CAAC,CAAC;IAClE,MAAM,MAAM,GAAG,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAChC,IAAI,MAAM,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;QAC1B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,iBAAiB,EAAE,CAAC;IAC1C,6EAA6E;IAC7E,sDAAsD;IACtD,MAAM,OAAO,GAAG,GAAG,MAAM,IAAI,OAAO,CAAC,GAAG,MAAM,CAAC;IAC/C,MAAM,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAClC,MAAM,kBAAE,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAEjC,4EAA4E;IAC5E,kDAAkD;IAClD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IACtC,KAAK,MAAM,KAAK,IAAI,MAAM,kBAAE,CAAC,OAAO,CAAC,YAAI,CAAC,EAAE,CAAC;QAC3C,IACE,KAAK,CAAC,UAAU,CAAC,kBAAkB,CAAC;YACpC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;YACtB,KAAK,KAAK,OAAO,EACjB,CAAC;YACD,MAAM,kBAAE,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,YAAI,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,iBAAiB;IAC9B,MAAM,MAAM,GAAG,IAAI,eAAM,CAAC;QACxB,WAAW,EAAE,GAAG,CAAC,IAAI,CAAC,gBAAQ,CAAC;KAChC,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,IAAI,qCAAa,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAA,mBAAS,EAAK,OAAO,EAAE;QACpC,cAAc,EAAE,qBAAqB;KACtC,CAAC,CAAC;IACH,MAAM,IAAA,6BAAoB,EAAC,MAAM,CAAC,CAAC;IACnC,MAAM,IAAA,qCAA4B,EAAC,MAAM,CAAC,CAAC;IAC3C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAE9B,MAAM,IAAA,YAAG,EAAA,+CAA+C,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACzE,MAAM,IAAA,YAAG,EAAA,2CAA2C,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACrE,MAAM,IAAA,YAAG,EAAA,sBAAsB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAChD,MAAM,IAAA,YAAG,EAAA,oBAAoB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAE9C,MAAM,2BAA2B,GAAG,IAAA,mBAAS,EAAiB,OAAO,EAAE;QACrE,cAAc,EAAE,6CAA6C;KAC9D,CAAC,CAAC;IACH,MAAM,oBAAoB,GAAG,IAAI,iBAAQ,CAAC;QACxC,EAAE,EAAE,2BAA2B;QAC/B,oBAAoB,EAAE,QAAQ;QAC9B,kBAAkB,EAAE,gCAAgC;QACpD,sBAAsB,EAAE,qCAAqC;QAC7D,QAAQ,EAAE,IAAI,8BAAqB,CAAC;YAClC,EAAE,EAAF,kBAAE;YACF,IAAI;YACJ,eAAe,EAAE,IAAI,CAAC,IAAI,CAAC,YAAI,EAAE,0BAA0B,CAAC;SAC7D,CAAC;KACH,CAAC,CAAC;IACH,MAAM,EAAE,KAAK,EAAE,0BAA0B,EAAE,GACzC,MAAM,oBAAoB,CAAC,eAAe,EAAE,CAAC;IAC/C,IAAI,0BAA0B,EAAE,CAAC;QAC/B,MAAM,0BAA0B,CAAC;IACnC,CAAC;IAED,MAAM,IAAA,YAAG,EAAA,yCAAyC,CAAC,OAAO,CACxD,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,YAAG,EAAA,4DAA4D,CAAC,OAAO,CAC3E,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,YAAG,EAAA,kBAAkB,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACjE,MAAM,IAAA,YAAG,EAAA,6CAA6C,CAAC,OAAO,CAC5D,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,YAAG,EAAA,+CAA+C,CAAC,OAAO,CAC9D,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,YAAG,EAAA,2CAA2C,CAAC,OAAO,CAC1D,2BAA2B,CAC5B,CAAC;IACF,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,+EAA+E;IAC/E,8EAA8E;IAC9E,uEAAuE;IACvE,MAAM,MAAM;SACT,UAAU,CAAC,eAAe,CAAC;SAC3B,MAAM,CAAC;QACN,EAAE,EAAE,sCAAsC;QAC1C,IAAI,EAAE,YAAY;QAClB,IAAI,EAAE,SAAS;QACf,MAAM,EAAE,QAAQ;QAChB,QAAQ,EAAE,IAAI;QACd,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,KAAK;QACnB,SAAS,EAAE,IAAI;QACf,gBAAgB,EAAE,IAAI;QACtB,eAAe,EAAE,KAAK;QACtB,gBAAgB,EAAE,IAAI,IAAI,CAAC,sBAAsB,CAAC;KACnD,CAAC;SACD,UAAU,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,CAAC;SAC/C,OAAO,EAAE,CAAC;IAEb,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;GAKR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IACvC,MAAM,IAAA,YAAG,EAAA;;;;;;;;;;;;;;;;;;;;;;GAsBR,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;IAEvC,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IACjD,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IACvB,OAAO,OAAO,CAAC;AACjB,CAAC;AAEM,KAAK,UAAU,cAAc,CAClC,OAA8B,EAAE;IAEhC,IAAI,WAAW,GAAG,KAAK,CAAC;IAExB,mBAAmB,KAAnB,mBAAmB,GAAK,mBAAmB,EAAE,EAAC;IAC9C,MAAM,MAAM,GAAG,IAAI,eAAM,CAAC;QACxB,WAAW,EAAE,MAAM,mBAAmB;KACvC,CAAC,CAAC;IACH,8EAA8E;IAC9E,6EAA6E;IAC7E,qEAAqE;IACrE,MAAM,MAAM,CAAC,SAAS,CAAC;IACvB,MAAM,OAAO,GAAG,IAAI,qCAAa,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAA,mBAAS,EAAK,OAAO,EAAE;QACpC,cAAc,EAAE,IAAI,EAAE,cAAc,IAAI,cAAc;QACtD,GAAG,EAAE,IAAI,EAAE,GAAG;KACf,CAAC,CAAC;IAEH,MAAM,kBAAkB,GAAG,IAAA,sCAAsB,EAC/C,IAAA,mBAAS,EAAiB,OAAO,EAAE;QACjC,cAAc,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,cAAc,gBAAgB;QACxE,GAAG,EAAE,IAAI,CAAC,GAAG;KACd,CAAC,EACF,eAAe,EACf,eAAe,CAChB,CAAC;IACF,kBAAkB,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE,GAAE,CAAC,CAAC;IAC5C,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,oBAAoB,EAAE;QAClD,KAAK,EAAE,kBAAkB;KAC1B,CAAC,CAAC;IAEH,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACpD,MAAM,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,IAAI,WAAW;YAAE,OAAO;QACxB,WAAW,GAAG,IAAI,CAAC;QACnB,wCAAwC;QACxC,MAAM,eAAe,EAAE,CAAC;QACxB,cAAc;QACd,0EAA0E;QAC1E,IAAI;IACN,CAAC,CAAC;IAEF,iFAAiF;IAEjF,OAAO,MAAqB,CAAC;AAC/B,CAAC","sourcesContent":["import { promises as fs } from 'node:fs';\nimport * as path from 'node:path';\nimport { PGlite } from '@electric-sql/pglite';\nimport { FileMigrationProvider, type LogConfig, Migrator, sql } from 'kysely';\nimport { PGliteDialect } from 'kysely-pglite-dialect';\nimport { type ControlPlaneDB, withControlPlaneSchema } from '../control-plane';\nimport type { ControlPlaneKysely } from '../index';\nimport { appMigrationEpoch } from '../migration';\nimport { getKysely } from '../plugins';\nimport type { DB } from '../v1.generated';\nimport { ensureDigestFunction, ensureGenRandomBytesFunction } from './digest';\n\nexport const root = path.join(__dirname, '../..');\nexport const dumpPath = path.join(root, '.data.tar');\n\nexport type UseLocalKyselyOptions = {\n  repositoryName?: string;\n  log?: true | LogConfig;\n};\n\nexport type LocalKysely = import('kysely').Kysely<DB> & {\n  controlPlaneKysely: ControlPlaneKysely;\n};\n\nasync function migrateToLatest<Database>(\n  kysely: import('kysely').Kysely<Database>\n) {\n  const migrator = new Migrator({\n    allowUnorderedMigrations: true,\n    db: kysely,\n    migrationTableSchema: appMigrationEpoch.tableSchema,\n    migrationTableName: appMigrationEpoch.tableName,\n    migrationLockTableName: appMigrationEpoch.lockTableName,\n    provider: new FileMigrationProvider({\n      fs,\n      path,\n      migrationFolder: path.join(root, appMigrationEpoch.folder),\n    }),\n  });\n\n  const { error } = await migrator.migrateToLatest();\n  if (error) {\n    throw error;\n  }\n}\n\n// Every spec file used to replay this identical bootstrap - migrations, the\n// schema surgery, and the control-plane seed copies - against a freshly loaded\n// snapshot, which measured ~345ms per file against ~185ms for the snapshot load\n// alone. None of it depends on the caller, so it runs once and is cached as a\n// prepared snapshot keyed by everything that determines its content.\nconst preparedDumpPrefix = '.data.prepared.';\n\nlet preparedDumpPromise: Promise<Blob> | undefined;\n\nasync function preparedDumpKey(): Promise<string> {\n  const [snapshot, appMigrations, controlPlaneMigrations] = await Promise.all([\n    fs.stat(dumpPath),\n    fs.readdir(path.join(root, appMigrationEpoch.folder)),\n    fs.readdir(path.join(root, 'control-plane-migrations')),\n  ]);\n\n  const hasher = new Bun.CryptoHasher('sha256');\n  hasher.update(`${snapshot.size}:${snapshot.mtimeMs}`);\n  hasher.update(appMigrations.sort().join(','));\n  hasher.update(controlPlaneMigrations.sort().join(','));\n  // Source of the bootstrap itself, so editing the surgery rebuilds the\n  // snapshot. Read through toString() rather than the file: this module is also\n  // published as CommonJS, where import.meta is a compile error.\n  for (const step of [\n    buildPreparedDump,\n    migrateToLatest,\n    ensureDigestFunction,\n    ensureGenRandomBytesFunction,\n  ]) {\n    hasher.update(step.toString());\n  }\n  return hasher.digest('hex').slice(0, 16);\n}\n\nasync function resolvePreparedDump(): Promise<Blob> {\n  const key = await preparedDumpKey();\n  const target = path.join(root, `${preparedDumpPrefix}${key}.tar`);\n  const cached = Bun.file(target);\n  if (await cached.exists()) {\n    return cached;\n  }\n\n  const content = await buildPreparedDump();\n  // Sibling workers build this concurrently on a cold cache; rename is atomic,\n  // so a reader never observes a half-written snapshot.\n  const staging = `${target}.${process.pid}.tmp`;\n  await Bun.write(staging, content);\n  await fs.rename(staging, target);\n\n  // Only finished snapshots: a sibling worker's staging file also carries the\n  // prefix, and removing it would break its rename.\n  const current = path.basename(target);\n  for (const entry of await fs.readdir(root)) {\n    if (\n      entry.startsWith(preparedDumpPrefix) &&\n      entry.endsWith('.tar') &&\n      entry !== current\n    ) {\n      await fs.rm(path.join(root, entry), { force: true });\n    }\n  }\n\n  return content;\n}\n\nasync function buildPreparedDump(): Promise<Blob> {\n  const pglite = new PGlite({\n    loadDataDir: Bun.file(dumpPath),\n  });\n  const dialect = new PGliteDialect(pglite);\n  const kysely = getKysely<DB>(dialect, {\n    repositoryName: 'localTestingPrepare',\n  });\n  await ensureDigestFunction(kysely);\n  await ensureGenRandomBytesFunction(kysely);\n  await migrateToLatest(kysely);\n\n  await sql`alter schema public rename to regional_public`.execute(kysely);\n  await sql`alter schema core rename to regional_core`.execute(kysely);\n  await sql`create schema public`.execute(kysely);\n  await sql`create schema core`.execute(kysely);\n\n  const controlPlaneBootstrapKysely = getKysely<ControlPlaneDB>(dialect, {\n    repositoryName: 'localTestingPrepare/control-plane-bootstrap',\n  });\n  const controlPlaneMigrator = new Migrator({\n    db: controlPlaneBootstrapKysely,\n    migrationTableSchema: 'public',\n    migrationTableName: 'control_plane_kysely_migration',\n    migrationLockTableName: 'control_plane_kysely_migration_lock',\n    provider: new FileMigrationProvider({\n      fs,\n      path,\n      migrationFolder: path.join(root, 'control-plane-migrations'),\n    }),\n  });\n  const { error: controlPlaneMigrationError } =\n    await controlPlaneMigrator.migrateToLatest();\n  if (controlPlaneMigrationError) {\n    throw controlPlaneMigrationError;\n  }\n\n  await sql`alter table core.flow set schema public`.execute(\n    controlPlaneBootstrapKysely\n  );\n  await sql`alter table core.issue_message_overwrite set schema public`.execute(\n    controlPlaneBootstrapKysely\n  );\n  await sql`drop schema core`.execute(controlPlaneBootstrapKysely);\n  await sql`alter schema public rename to control_plane`.execute(\n    controlPlaneBootstrapKysely\n  );\n  await sql`alter schema regional_public rename to public`.execute(\n    controlPlaneBootstrapKysely\n  );\n  await sql`alter schema regional_core rename to core`.execute(\n    controlPlaneBootstrapKysely\n  );\n  await sql`\n    do $$\n    declare\n      definition text;\n    begin\n      select pg_get_functiondef(\n        'control_plane.normalize_tenant_slug(text)'::regprocedure\n      ) into definition;\n      execute replace(definition, 'public.', 'control_plane.');\n\n      select pg_get_functiondef(\n        'control_plane.build_unique_tenant_slug(text,uuid)'::regprocedure\n      ) into definition;\n      execute replace(definition, 'public.', 'control_plane.');\n\n      select pg_get_functiondef(\n        'control_plane.refresh_operation_state(uuid)'::regprocedure\n      ) into definition;\n      execute replace(definition, 'public.', 'control_plane.');\n\n      select pg_get_functiondef(\n        'control_plane.refresh_destructive_job_operation()'::regprocedure\n      ) into definition;\n      execute replace(definition, 'public.', 'control_plane.');\n    end;\n    $$;\n  `.execute(controlPlaneBootstrapKysely);\n  // The canonical VRPlatform tenant. Production code falls back to this exact id\n  // in getCalculatedBillingPartnerId, so specs need it to exist without each of\n  // them creating - and mutating - their own copy of a shared singleton.\n  await kysely\n    .insertInto('public.tenant')\n    .values({\n      id: '7bfec41f-aa3f-4602-95d2-f7996e4f0e59',\n      name: 'VRPlatform',\n      type: 'partner',\n      status: 'active',\n      isActive: true,\n      isGeneralLedger: false,\n      isOnboarding: false,\n      partnerId: null,\n      billingPartnerId: null,\n      defaultCurrency: 'usd',\n      statementStartAt: new Date('2020-01-01T00:00:00Z'),\n    })\n    .onConflict((oc) => oc.column('id').doNothing())\n    .execute();\n\n  await sql`\n    insert into control_plane.app (\n      authentication,\n      category,\n      color,\n      created_at,\n      icon,\n      icon_round,\n      id,\n      info,\n      name,\n      type,\n      version\n    )\n    select\n      authentication,\n      category,\n      color,\n      created_at,\n      icon,\n      icon_round,\n      id,\n      info,\n      name,\n      type,\n      version\n    from public.app\n    on conflict (id) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.currency (name)\n    select name\n    from public.currency\n    on conflict (name) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.booking_channel (\n      channel_ref,\n      color,\n      created_at,\n      icon,\n      icon_provider,\n      id,\n      last_generate_run,\n      logo,\n      logo_provider,\n      selected_booking_channel_icon_candidate_id,\n      selected_booking_channel_logo_candidate_id,\n      unique_ref,\n      updated_at\n    )\n    select\n      channel_ref,\n      color,\n      created_at,\n      icon,\n      icon_provider,\n      id,\n      last_generate_run,\n      logo,\n      logo_provider,\n      selected_booking_channel_icon_candidate_id,\n      selected_booking_channel_logo_candidate_id,\n      unique_ref,\n      updated_at\n    from public.booking_channel\n    on conflict (id) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.booking_channel_icon_candidate (\n      booking_channel_id,\n      cloudflare_image_id,\n      color,\n      comment,\n      created_at,\n      external_icon,\n      height,\n      id,\n      provider,\n      source,\n      updated_at,\n      width\n    )\n    select\n      booking_channel_id,\n      cloudflare_image_id,\n      color,\n      comment,\n      created_at,\n      external_icon,\n      height,\n      id,\n      provider,\n      source,\n      updated_at,\n      width\n    from public.booking_channel_icon_candidate\n    on conflict (id) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.tenant (\n      billing_partner_id,\n      billing_plan,\n      created_at,\n      data_region,\n      id,\n      name,\n      partner_id,\n      slug,\n      status,\n      storage_realm,\n      type,\n      unique_ref,\n      updated_at\n    )\n    select\n      billing_partner_id,\n      billing_plan,\n      created_at,\n      data_region,\n      id,\n      name,\n      partner_id,\n      slug,\n      status,\n      storage_realm,\n      type,\n      unique_ref,\n      updated_at\n    from public.tenant\n    on conflict (id) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n  await sql`\n    insert into control_plane.token (\n      created_at,\n      expires_at,\n      id,\n      nano_id,\n      payload,\n      tenant_id,\n      type,\n      user_id\n    )\n    select\n      created_at,\n      expires_at,\n      id,\n      nano_id,\n      payload,\n      tenant_id,\n      type,\n      user_id\n    from public.token\n    on conflict (id) do nothing;\n  `.execute(controlPlaneBootstrapKysely);\n\n  const content = await pglite.dumpDataDir('none');\n  await kysely.destroy();\n  return content;\n}\n\nexport async function useLocalKysely(\n  args: UseLocalKyselyOptions = {}\n): Promise<LocalKysely> {\n  let isDestroyed = false;\n\n  preparedDumpPromise ??= resolvePreparedDump();\n  const pglite = new PGlite({\n    loadDataDir: await preparedDumpPromise,\n  });\n  // The constructor is lazy. The bootstrap this replaced forced the snapshot to\n  // load before returning, so keep that contract: a returned handle is usable,\n  // and a bad snapshot fails here instead of inside an unrelated test.\n  await pglite.waitReady;\n  const dialect = new PGliteDialect(pglite);\n  const kysely = getKysely<DB>(dialect, {\n    repositoryName: args?.repositoryName ?? 'localTesting',\n    log: args?.log,\n  });\n\n  const controlPlaneKysely = withControlPlaneSchema(\n    getKysely<ControlPlaneDB>(dialect, {\n      repositoryName: `${args.repositoryName ?? 'localTesting'}/control-plane`,\n      log: args.log,\n    }),\n    'control_plane',\n    'control_plane'\n  );\n  controlPlaneKysely.destroy = async () => {};\n  Object.defineProperty(kysely, 'controlPlaneKysely', {\n    value: controlPlaneKysely,\n  });\n\n  const originalDestroy = kysely.destroy.bind(kysely);\n  kysely.destroy = async () => {\n    if (isDestroyed) return;\n    isDestroyed = true;\n    //const destroyTime = performance.now();\n    await originalDestroy();\n    //console.log(\n    //  `Database destroyed: ${Math.round(performance.now() - destroyTime)}ms`\n    //);\n  };\n\n  // console.log(`Database initialized: ${Math.round(performance.now() - now)}ms`);\n\n  return kysely as LocalKysely;\n}\n"]}

package/build/main/postgres-error.d.ts

@@ -1,8 +1,8 @@
 /**
- * Matches PostgreSQL errors from both production's postgres.js driver and
- * PGlite, which expose constraint names under different property names.
+ * Matches PostgreSQL errors by code and, when requested, constraint. Production
+ * postgres.js and PGlite expose constraint names under different properties.
  */
 export declare function isPostgresError(error: unknown, expected: {
     code: string;
-    constraint: string;
+    constraint?: string;
 }): boolean;

package/build/main/postgres-error.js

@@ -2,8 +2,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.isPostgresError = isPostgresError;
 /**
- * Matches PostgreSQL errors from both production's postgres.js driver and
- * PGlite, which expose constraint names under different property names.
+ * Matches PostgreSQL errors by code and, when requested, constraint. Production
+ * postgres.js and PGlite expose constraint names under different properties.
  */
 function isPostgresError(error, expected) {
     if (!(error instanceof Error) ||
@@ -16,6 +16,7 @@ function isPostgresError(error, expected) {
         : 'constraint_name' in error && typeof error.constraint_name === 'string'
             ? error.constraint_name
             : undefined;
-    return error.code === expected.code && constraint === expected.constraint;
+    return (error.code === expected.code &&
+        (expected.constraint === undefined || constraint === expected.constraint));
 }
 //# sourceMappingURL=postgres-error.js.map

package/build/main/postgres-error.js.map

@@ -1 +1 @@
-{"version":3,"file":"postgres-error.js","sourceRoot":"src/","sources":["postgres-error.ts"],"names":[],"mappings":";;AAIA,0CAoBC;AAxBD;;;GAGG;AACH,SAAgB,eAAe,CAC7B,KAAc,EACd,QAA8C;IAE9C,IACE,CAAC,CAAC,KAAK,YAAY,KAAK,CAAC;QACzB,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC;QAClB,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,EAC9B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,UAAU,GACd,YAAY,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ;QAC3D,CAAC,CAAC,KAAK,CAAC,UAAU;QAClB,CAAC,CAAC,iBAAiB,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,eAAe,KAAK,QAAQ;YACvE,CAAC,CAAC,KAAK,CAAC,eAAe;YACvB,CAAC,CAAC,SAAS,CAAC;IAElB,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,IAAI,UAAU,KAAK,QAAQ,CAAC,UAAU,CAAC;AAC5E,CAAC","sourcesContent":["/**\n * Matches PostgreSQL errors from both production's postgres.js driver and\n * PGlite, which expose constraint names under different property names.\n */\nexport function isPostgresError(\n  error: unknown,\n  expected: { code: string; constraint: string }\n): boolean {\n  if (\n    !(error instanceof Error) ||\n    !('code' in error) ||\n    typeof error.code !== 'string'\n  ) {\n    return false;\n  }\n\n  const constraint =\n    'constraint' in error && typeof error.constraint === 'string'\n      ? error.constraint\n      : 'constraint_name' in error && typeof error.constraint_name === 'string'\n        ? error.constraint_name\n        : undefined;\n\n  return error.code === expected.code && constraint === expected.constraint;\n}\n"]}
+{"version":3,"file":"postgres-error.js","sourceRoot":"src/","sources":["postgres-error.ts"],"names":[],"mappings":";;AAIA,0CAuBC;AA3BD;;;GAGG;AACH,SAAgB,eAAe,CAC7B,KAAc,EACd,QAA+C;IAE/C,IACE,CAAC,CAAC,KAAK,YAAY,KAAK,CAAC;QACzB,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC;QAClB,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,EAC9B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,UAAU,GACd,YAAY,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ;QAC3D,CAAC,CAAC,KAAK,CAAC,UAAU;QAClB,CAAC,CAAC,iBAAiB,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,eAAe,KAAK,QAAQ;YACvE,CAAC,CAAC,KAAK,CAAC,eAAe;YACvB,CAAC,CAAC,SAAS,CAAC;IAElB,OAAO,CACL,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI;QAC5B,CAAC,QAAQ,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,QAAQ,CAAC,UAAU,CAAC,CAC1E,CAAC;AACJ,CAAC","sourcesContent":["/**\n * Matches PostgreSQL errors by code and, when requested, constraint. Production\n * postgres.js and PGlite expose constraint names under different properties.\n */\nexport function isPostgresError(\n  error: unknown,\n  expected: { code: string; constraint?: string }\n): boolean {\n  if (\n    !(error instanceof Error) ||\n    !('code' in error) ||\n    typeof error.code !== 'string'\n  ) {\n    return false;\n  }\n\n  const constraint =\n    'constraint' in error && typeof error.constraint === 'string'\n      ? error.constraint\n      : 'constraint_name' in error && typeof error.constraint_name === 'string'\n        ? error.constraint_name\n        : undefined;\n\n  return (\n    error.code === expected.code &&\n    (expected.constraint === undefined || constraint === expected.constraint)\n  );\n}\n"]}

package/build/main/test/index.d.ts

@@ -1,7 +1,30 @@
 import { type LocalKysely, type UseLocalKyselyOptions } from '../local';
-export declare function useTestAutoCleanupKysely(args?: UseLocalKyselyOptions): Promise<LocalKysely>;
+type UseTestKyselyOptions = UseLocalKyselyOptions & {
+    /**
+     * Force a private database even when the worker shares one. Required by specs
+     * that model several regions at once, where two handles must be two distinct
+     * physical databases.
+     */
+    isolated?: boolean;
+};
+declare global {
+    var __vrplatformSharedKysely: Promise<LocalKysely> | undefined;
+}
+/**
+ * Locally each worker process shares one database, because specs isolate by
+ * tenant and the faster suite is what makes running it before a push realistic.
+ *
+ * CI keeps a database per spec file. It retries nothing, so a badly scoped spec
+ * would surface there as a red build in an unrelated file rather than as a
+ * local failure the author can reproduce.
+ *
+ * VR_TEST_SHARED_KYSELY overrides either way; `isolated: true` opts a single
+ * call out when a spec genuinely needs two physical databases at once.
+ */
+export declare function resolveSharedKysely(env: Record<string, string | undefined>): boolean;
+export declare function useTestAutoCleanupKysely(args?: UseTestKyselyOptions): Promise<LocalKysely>;
 type ReturnType = LocalKysely & {
     [Symbol.dispose]: () => Promise<void>;
 };
-export declare function useLocalKyselyResource(args?: UseLocalKyselyOptions): Promise<ReturnType>;
+export declare function useLocalKyselyResource(args?: UseTestKyselyOptions): Promise<ReturnType>;
 export {};

package/build/main/test/index.js

@@ -1,23 +1,54 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.resolveSharedKysely = resolveSharedKysely;
 exports.useTestAutoCleanupKysely = useTestAutoCleanupKysely;
 exports.useLocalKyselyResource = useLocalKyselyResource;
 const bun_test_1 = require("bun:test");
 const local_1 = require("../local");
+/**
+ * Locally each worker process shares one database, because specs isolate by
+ * tenant and the faster suite is what makes running it before a push realistic.
+ *
+ * CI keeps a database per spec file. It retries nothing, so a badly scoped spec
+ * would surface there as a red build in an unrelated file rather than as a
+ * local failure the author can reproduce.
+ *
+ * VR_TEST_SHARED_KYSELY overrides either way; `isolated: true` opts a single
+ * call out when a spec genuinely needs two physical databases at once.
+ */
+function resolveSharedKysely(env) {
+    if (env.VR_TEST_SHARED_KYSELY)
+        return env.VR_TEST_SHARED_KYSELY !== '0';
+    return !(env.CI || env.GITHUB_ACTIONS);
+}
+const sharedKysely = resolveSharedKysely(process.env);
+function acquireKysely(args) {
+    if (!sharedKysely || args?.isolated)
+        return (0, local_1.useLocalKysely)(args);
+    globalThis.__vrplatformSharedKysely ?? (globalThis.__vrplatformSharedKysely = (0, local_1.useLocalKysely)(args));
+    return globalThis.__vrplatformSharedKysely;
+}
+function ownsKysely(args) {
+    return !sharedKysely || Boolean(args?.isolated);
+}
 async function useTestAutoCleanupKysely(args) {
-    const kysely = await (0, local_1.useLocalKysely)(args);
-    (0, bun_test_1.afterAll)(async () => {
-        await kysely.destroy();
-    });
+    const kysely = await acquireKysely(args);
+    if (ownsKysely(args)) {
+        (0, bun_test_1.afterAll)(async () => {
+            await kysely.destroy();
+        });
+    }
     return kysely;
 }
 async function useLocalKyselyResource(args) {
-    const kysely = await (0, local_1.useLocalKysely)(args);
+    const kysely = await acquireKysely(args);
     return new Proxy({}, {
         get(_target, prop) {
             // somehow not working if using Symbol.dispose
             if (prop === Symbol.asyncDispose) {
                 return async () => {
+                    if (!ownsKysely(args))
+                        return;
                     await kysely.destroy();
                 };
             }

package/build/main/test/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["test/index.ts"],"names":[],"mappings":";;AAOA,4DAQC;AAMD,wDAyBC;AA9CD,uCAAoC;AACpC,oCAIkB;AAEX,KAAK,UAAU,wBAAwB,CAAC,IAA4B;IACzE,MAAM,MAAM,GAAG,MAAM,IAAA,sBAAc,EAAC,IAAI,CAAC,CAAC;IAE1C,IAAA,mBAAQ,EAAC,KAAK,IAAI,EAAE;QAClB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC;AAMM,KAAK,UAAU,sBAAsB,CAC1C,IAA4B;IAE5B,MAAM,MAAM,GAAG,MAAM,IAAA,sBAAc,EAAC,IAAI,CAAC,CAAC;IAE1C,OAAO,IAAI,KAAK,CACd,EAAE,EACF;QACE,GAAG,CAAC,OAAO,EAAE,IAAI;YACf,8CAA8C;YAC9C,IAAI,IAAI,KAAK,MAAM,CAAC,YAAY,EAAE,CAAC;gBACjC,OAAO,KAAK,IAAI,EAAE;oBAChB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;gBACzB,CAAC,CAAC;YACJ,CAAC;YACD,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,IAAI,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACvC,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;KACF,CACY,CAAC;AAClB,CAAC","sourcesContent":["import { afterAll } from 'bun:test';\nimport {\n  type LocalKysely,\n  type UseLocalKyselyOptions,\n  useLocalKysely,\n} from '../local';\n\nexport async function useTestAutoCleanupKysely(args?: UseLocalKyselyOptions) {\n  const kysely = await useLocalKysely(args);\n\n  afterAll(async () => {\n    await kysely.destroy();\n  });\n\n  return kysely;\n}\n\ntype ReturnType = LocalKysely & {\n  [Symbol.dispose]: () => Promise<void>;\n};\n\nexport async function useLocalKyselyResource(\n  args?: UseLocalKyselyOptions\n): Promise<ReturnType> {\n  const kysely = await useLocalKysely(args);\n\n  return new Proxy(\n    {},\n    {\n      get(_target, prop) {\n        // somehow not working if using Symbol.dispose\n        if (prop === Symbol.asyncDispose) {\n          return async () => {\n            await kysely.destroy();\n          };\n        }\n        if (prop === 'kysely') {\n          return kysely;\n        }\n        if (typeof kysely[prop] === 'function') {\n          return kysely[prop].bind(kysely);\n        }\n        return kysely[prop];\n      },\n    }\n  ) as ReturnType;\n}\n"]}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["test/index.ts"],"names":[],"mappings":";;AA+BA,kDAKC;AAcD,4DAUC;AAMD,wDA0BC;AA5FD,uCAAoC;AACpC,oCAIkB;AAelB;;;;;;;;;;GAUG;AACH,SAAgB,mBAAmB,CACjC,GAAuC;IAEvC,IAAI,GAAG,CAAC,qBAAqB;QAAE,OAAO,GAAG,CAAC,qBAAqB,KAAK,GAAG,CAAC;IACxE,OAAO,CAAC,CAAC,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,cAAc,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,YAAY,GAAG,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;AAEtD,SAAS,aAAa,CAAC,IAA2B;IAChD,IAAI,CAAC,YAAY,IAAI,IAAI,EAAE,QAAQ;QAAE,OAAO,IAAA,sBAAc,EAAC,IAAI,CAAC,CAAC;IACjE,UAAU,CAAC,wBAAwB,KAAnC,UAAU,CAAC,wBAAwB,GAAK,IAAA,sBAAc,EAAC,IAAI,CAAC,EAAC;IAC7D,OAAO,UAAU,CAAC,wBAAwB,CAAC;AAC7C,CAAC;AAED,SAAS,UAAU,CAAC,IAA2B;IAC7C,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;AAClD,CAAC;AAEM,KAAK,UAAU,wBAAwB,CAAC,IAA2B;IACxE,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,CAAC;IAEzC,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACrB,IAAA,mBAAQ,EAAC,KAAK,IAAI,EAAE;YAClB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QACzB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAMM,KAAK,UAAU,sBAAsB,CAC1C,IAA2B;IAE3B,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,CAAC;IAEzC,OAAO,IAAI,KAAK,CACd,EAAE,EACF;QACE,GAAG,CAAC,OAAO,EAAE,IAAI;YACf,8CAA8C;YAC9C,IAAI,IAAI,KAAK,MAAM,CAAC,YAAY,EAAE,CAAC;gBACjC,OAAO,KAAK,IAAI,EAAE;oBAChB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;wBAAE,OAAO;oBAC9B,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;gBACzB,CAAC,CAAC;YACJ,CAAC;YACD,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,IAAI,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACvC,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;KACF,CACY,CAAC;AAClB,CAAC","sourcesContent":["import { afterAll } from 'bun:test';\nimport {\n  type LocalKysely,\n  type UseLocalKyselyOptions,\n  useLocalKysely,\n} from '../local';\n\ntype UseTestKyselyOptions = UseLocalKyselyOptions & {\n  /**\n   * Force a private database even when the worker shares one. Required by specs\n   * that model several regions at once, where two handles must be two distinct\n   * physical databases.\n   */\n  isolated?: boolean;\n};\n\ndeclare global {\n  var __vrplatformSharedKysely: Promise<LocalKysely> | undefined;\n}\n\n/**\n * Locally each worker process shares one database, because specs isolate by\n * tenant and the faster suite is what makes running it before a push realistic.\n *\n * CI keeps a database per spec file. It retries nothing, so a badly scoped spec\n * would surface there as a red build in an unrelated file rather than as a\n * local failure the author can reproduce.\n *\n * VR_TEST_SHARED_KYSELY overrides either way; `isolated: true` opts a single\n * call out when a spec genuinely needs two physical databases at once.\n */\nexport function resolveSharedKysely(\n  env: Record<string, string | undefined>\n): boolean {\n  if (env.VR_TEST_SHARED_KYSELY) return env.VR_TEST_SHARED_KYSELY !== '0';\n  return !(env.CI || env.GITHUB_ACTIONS);\n}\n\nconst sharedKysely = resolveSharedKysely(process.env);\n\nfunction acquireKysely(args?: UseTestKyselyOptions): Promise<LocalKysely> {\n  if (!sharedKysely || args?.isolated) return useLocalKysely(args);\n  globalThis.__vrplatformSharedKysely ??= useLocalKysely(args);\n  return globalThis.__vrplatformSharedKysely;\n}\n\nfunction ownsKysely(args?: UseTestKyselyOptions): boolean {\n  return !sharedKysely || Boolean(args?.isolated);\n}\n\nexport async function useTestAutoCleanupKysely(args?: UseTestKyselyOptions) {\n  const kysely = await acquireKysely(args);\n\n  if (ownsKysely(args)) {\n    afterAll(async () => {\n      await kysely.destroy();\n    });\n  }\n\n  return kysely;\n}\n\ntype ReturnType = LocalKysely & {\n  [Symbol.dispose]: () => Promise<void>;\n};\n\nexport async function useLocalKyselyResource(\n  args?: UseTestKyselyOptions\n): Promise<ReturnType> {\n  const kysely = await acquireKysely(args);\n\n  return new Proxy(\n    {},\n    {\n      get(_target, prop) {\n        // somehow not working if using Symbol.dispose\n        if (prop === Symbol.asyncDispose) {\n          return async () => {\n            if (!ownsKysely(args)) return;\n            await kysely.destroy();\n          };\n        }\n        if (prop === 'kysely') {\n          return kysely;\n        }\n        if (typeof kysely[prop] === 'function') {\n          return kysely[prop].bind(kysely);\n        }\n        return kysely[prop];\n      },\n    }\n  ) as ReturnType;\n}\n"]}

package/build/main/v1.generated.d.ts

@@ -3,7 +3,7 @@
  * Please do not edit it manually.
  */
 import type { ColumnType } from "kysely";
-export type AccountingJournalEntryInactiveReason = "cancelledReservationLine" | "currencyMismatch" | "generalLedgerStartAt" | "inactiveAccount" | "listingInactive" | "nonPostingAccount" | "reservationGeneralLedgerStatusInactive" | "transactionInactive" | "unknownInactiveReason";
+export type AccountingJournalEntryInactiveReason = "cancelledReservationLine" | "currencyMismatch" | "generalLedgerStartAt" | "historicalLedger" | "inactiveAccount" | "listingInactive" | "listingOwnershipPeriodNotFound" | "nonPostingAccount" | "reservationGeneralLedgerStatusInactive" | "transactionInactive" | "unbalancedGroupInactiveLeg" | "unknownInactiveReason";
 export type AccountingJournalEntryLedger = "historical" | "operating" | "trust";
 export type AccountingTransactionAttachmentStatus = "attached" | "deleting" | "ready" | "uploading";
 export type ArrayType<T> = ArrayTypeImpl<T> extends (infer U)[] ? U[] : ArrayTypeImpl<T>;
@@ -417,6 +417,8 @@ export interface AccountingTransaction {
     recurringFeeId: string | null;
     recurringPattern: Json | null;
     recurringTemplateId: string | null;
+    returnedFromTransactionId: string | null;
+    returnEffectiveAt: Timestamp | null;
     /**
      * Human-readable system ID (EXP/DEP/PAY/TRF-XXX-XXXX-XXX depending on transaction.type).
      */
@@ -2008,6 +2010,7 @@ export interface PublicRecurringFeeType {
     name: string;
 }
 export interface PublicReservation {
+    adults: number | null;
     bookedAt: Generated<Timestamp | null>;
     bookerName: string | null;
     bookingPlatform: string | null;
@@ -2026,6 +2029,7 @@ export interface PublicReservation {
     channelId: string | null;
     checkIn: Timestamp | null;
     checkOut: Timestamp | null;
+    children: number | null;
     confirmationCode: string | null;
     connectionId: string | null;
     createdAt: Generated<Timestamp | null>;