@vrplatform/kysely 1.3.45-stage.4896 → 1.3.45-stage.490

package/build/main/error.js.map

@@ -1 +1 @@
-{"version":3,"file":"error.js","sourceRoot":"src/","sources":["error.ts"],"names":[],"mappings":";;;AAAA,MAAa,WAAY,SAAQ,KAAK;IACpC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,aAAa,CAAC;IAC5B,CAAC;CACF;AALD,kCAKC","sourcesContent":["export class KyselyError extends Error {\n  constructor(message: string) {\n    super(message);\n    this.name = 'KyselyError';\n  }\n}\n"]}
+{"version":3,"file":"error.js","sourceRoot":"src/","sources":["error.ts"],"names":[],"mappings":";;;AAAA,MAAa,WAAY,SAAQ,KAAK;IACpC,YACE,OAAe,EACC,OAAiB;QAEjC,KAAK,CAAC,OAAO,CAAC,CAAC;QAFC,YAAO,GAAP,OAAO,CAAU;QAGjC,IAAI,CAAC,IAAI,GAAG,aAAa,CAAC;IAC5B,CAAC;CACF;AARD,kCAQC","sourcesContent":["export class KyselyError extends Error {\n  constructor(\n    message: string,\n    public readonly details?: unknown\n  ) {\n    super(message);\n    this.name = 'KyselyError';\n  }\n}\n"]}

package/build/main/index.d.ts

@@ -1,5 +1,6 @@
-import type { Kysely as K, LogConfig } from 'kysely';
+import type { Kysely as K, LogConfig, Transaction as T } from 'kysely';
 import postgres, { type Sql } from 'postgres';
+import { type ControlPlaneDB } from './control-plane';
 export type { Insertable, LogEvent } from 'kysely';
 export * from './error';
 export * from './isomorphic';
@@ -7,29 +8,33 @@ export * from './postgres-error';
 export * from './query';
 export * from './regional';
 export * from './transaction';
+import type { TenantDataRegion } from './regional';
 import type { DB } from './v1.generated';
 type KyselyOpts = {
+    dataRegion?: TenantDataRegion;
+    hyperdrive?: true;
+    onRetryDiagnostic?: (diagnostic: PostgresRetryDiagnostic) => void;
     repositoryName?: string;
     log?: true | LogConfig;
 };
+type PostgresRetryDiagnostic = {
+    attemptCount: number;
+    dataRegion?: TenantDataRegion;
+    operationKind: 'acquisition' | 'mutation' | 'select' | 'transaction';
+    outcome: 'exhausted' | 'notRetryable' | 'retrying';
+    repositoryName?: string;
+};
 type PgOpt = {
     forceDisableSSL?: boolean;
     probe?: true;
     retries?: false | number;
-};
+} & Pick<KyselyOpts, 'hyperdrive'>;
 export type Kysely = K<DB>;
-export type Postgres = Sql;
-/**
- * @deprecated This method is deprecated in favor of the useAsyncKysely method.
- */
-export declare function useKysely(postgres: Sql, options?: KyselyOpts): K<DB>;
+export type ControlPlaneKysely = K<ControlPlaneDB>;
+export type ControlPlaneTransaction = T<ControlPlaneDB>;
+export declare function useKysely<Database = DB>(postgres: Sql, options?: KyselyOpts): K<Database>;
 export declare function useAsyncPostgres(connectionString: string, opt?: PgOpt): Promise<postgres.Sql<{}>>;
 export declare function useAsyncKysely(connectionString: string, opt?: PgOpt & KyselyOpts): Promise<K<DB>>;
-/**
- * @deprecated This method is deprecated in favor of the useAsyncKysely method.
- */
-export declare function usePostgres(connectionString?: string, opt?: Pick<PgOpt, 'forceDisableSSL'>): postgres.Sql<{}>;
+export declare function useAsyncNeonWebSocketKysely(connectionString: string, opt?: Pick<PgOpt, 'probe'> & KyselyOpts): Promise<K<DB>>;
+export declare function useAsyncControlPlaneKysely(connectionString: string, opt?: PgOpt & KyselyOpts): Promise<K<ControlPlaneDB>>;
 export declare function forceCloseAllPostgres(timeoutMs?: number): Promise<void>;
-export declare class DatabaseError extends Error {
-    constructor(message: string, cause: any);
-}

package/build/main/index.js

@@ -17,15 +17,18 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DatabaseError = void 0;
 exports.useKysely = useKysely;
 exports.useAsyncPostgres = useAsyncPostgres;
 exports.useAsyncKysely = useAsyncKysely;
-exports.usePostgres = usePostgres;
+exports.useAsyncNeonWebSocketKysely = useAsyncNeonWebSocketKysely;
+exports.useAsyncControlPlaneKysely = useAsyncControlPlaneKysely;
 exports.forceCloseAllPostgres = forceCloseAllPostgres;
 const exponential_backoff_1 = require("exponential-backoff");
+const serverless_1 = require("@neondatabase/serverless");
+const kysely_1 = require("kysely");
 const kysely_postgres_js_1 = require("kysely-postgres-js");
 const postgres_1 = __importDefault(require("postgres"));
+const control_plane_1 = require("./control-plane");
 const error_1 = require("./error");
 const plugins_1 = require("./plugins");
 __exportStar(require("./error"), exports);
@@ -39,43 +42,214 @@ const retryableMessages = [
     'CONNECTION_CLOSED',
     'CONNECT_TIMEOUT',
     'Idle connection closed by Hyperdrive.',
+    'server_login_retry',
 ];
-/**
- * @deprecated This method is deprecated in favor of the useAsyncKysely method.
- */
+const postgresRetryAttempts = 3;
+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 isQueryRetryDiagnosticError(error) {
+    const message = error instanceof Error ? error.message : String(error);
+    return (isRetryableConnectionError(error) ||
+        isConnectionReset(error) ||
+        message === 'Egress connection severed outside of Hyperdrive' ||
+        message === 'Network connection lost.' ||
+        message === 'Node shutting down' ||
+        message === 'Node shutting down.');
+}
+async function retryPostgres(operation, retry, diagnostic) {
+    let attemptCount = 0;
+    let retried = false;
+    const emitDiagnostic = (outcome) => {
+        if (!diagnostic)
+            return;
+        diagnostic.emit({
+            attemptCount,
+            dataRegion: diagnostic.dataRegion,
+            operationKind: diagnostic.operationKind,
+            outcome,
+            repositoryName: diagnostic.repositoryName,
+        });
+    };
+    try {
+        return await (0, exponential_backoff_1.backOff)(async () => {
+            attemptCount++;
+            return await operation();
+        }, {
+            numOfAttempts: postgresRetryAttempts,
+            startingDelay: (0, utils_1.parseDuration)('1s'),
+            maxDelay: (0, utils_1.parseDuration)('5s'),
+            async retry(error, attemptNumber) {
+                const shouldRetry = attemptNumber < postgresRetryAttempts &&
+                    (await retry(error, attemptNumber));
+                if (shouldRetry && diagnostic) {
+                    retried = true;
+                    emitDiagnostic('retrying');
+                }
+                return shouldRetry;
+            },
+        });
+    }
+    catch (error) {
+        if (diagnostic?.shouldEmit(error)) {
+            emitDiagnostic(retried && attemptCount === postgresRetryAttempts
+                ? 'exhausted'
+                : 'notRetryable');
+        }
+        throw error;
+    }
+}
+class RetryingPostgresJSDialect extends kysely_postgres_js_1.PostgresJSDialect {
+    constructor(config, retryDiagnostic) {
+        super(config);
+        this.config = config;
+        this.retryDiagnostic = retryDiagnostic;
+    }
+    createDriver() {
+        const driver = new kysely_postgres_js_1.PostgresJSDriver(this.config);
+        const connections = new WeakMap();
+        const getConnection = (connection) => {
+            const state = connections.get(connection);
+            if (!state)
+                throw new Error('Postgres connection state is unavailable');
+            return state;
+        };
+        return {
+            init: () => driver.init(),
+            acquireConnection: async () => {
+                const state = {
+                    connection: await driver.acquireConnection(),
+                    inTransaction: false,
+                };
+                const connection = {
+                    executeQuery: (compiledQuery) => {
+                        const operationKind = state.inTransaction
+                            ? 'transaction'
+                            : compiledQuery.query.kind === 'SelectQueryNode'
+                                ? 'select'
+                                : 'mutation';
+                        return retryPostgres(() => state.connection.executeQuery(compiledQuery), async (error) => {
+                            const message = error instanceof Error ? error.message : String(error);
+                            if (state.inTransaction ||
+                                (!isRetryableConnectionError(error) &&
+                                    (compiledQuery.query.kind !== 'SelectQueryNode' ||
+                                        (!isConnectionReset(error) &&
+                                            message !==
+                                                'Egress connection severed outside of Hyperdrive' &&
+                                            message !== 'Network connection lost.' &&
+                                            message !== 'Node shutting down' &&
+                                            message !== 'Node shutting down.')))) {
+                                return false;
+                            }
+                            // Postgres.js already ejects a severed reserved connection.
+                            // Acquire a new reservation before replaying the safe read.
+                            state.connection = await driver.acquireConnection();
+                            return true;
+                        }, this.retryDiagnostic
+                            ? {
+                                ...this.retryDiagnostic,
+                                operationKind,
+                                shouldEmit: isQueryRetryDiagnosticError,
+                            }
+                            : undefined);
+                    },
+                    streamQuery: (compiledQuery, chunkSize) => {
+                        if (chunkSize === undefined) {
+                            throw new Error('Postgres stream chunk size is required');
+                        }
+                        return state.connection.streamQuery(compiledQuery, chunkSize);
+                    },
+                };
+                connections.set(connection, state);
+                return connection;
+            },
+            beginTransaction: async (connection, settings) => {
+                const state = getConnection(connection);
+                state.inTransaction = true;
+                try {
+                    await driver.beginTransaction(state.connection, settings);
+                }
+                catch (error) {
+                    state.inTransaction = false;
+                    throw error;
+                }
+            },
+            commitTransaction: async (connection) => {
+                const state = getConnection(connection);
+                try {
+                    await driver.commitTransaction(state.connection);
+                }
+                finally {
+                    state.inTransaction = false;
+                }
+            },
+            rollbackTransaction: async (connection) => {
+                const state = getConnection(connection);
+                try {
+                    await driver.rollbackTransaction(state.connection);
+                }
+                finally {
+                    state.inTransaction = false;
+                }
+            },
+            savepoint: (connection, savepointName, compileQuery) => driver.savepoint(getConnection(connection).connection, savepointName, compileQuery),
+            rollbackToSavepoint: (connection, savepointName, compileQuery) => driver.rollbackToSavepoint(getConnection(connection).connection, savepointName, compileQuery),
+            releaseSavepoint: (connection, savepointName, compileQuery) => driver.releaseSavepoint(getConnection(connection).connection, savepointName, compileQuery),
+            releaseConnection: async (connection) => {
+                await driver.releaseConnection(getConnection(connection).connection);
+                connections.delete(connection);
+            },
+            destroy: () => driver.destroy(),
+        };
+    }
+}
 function useKysely(postgres, options) {
     console.debug('Creating Kysely instance');
-    const kysely = (0, plugins_1.getKysely)(new kysely_postgres_js_1.PostgresJSDialect({
-        postgres,
-        onReserveConnection: async (connection) => {
-            const executeQuery = connection.executeQuery.bind(connection);
-            connection.executeQuery = (arg) => (0, exponential_backoff_1.backOff)(async () => await executeQuery(arg), {
-                numOfAttempts: 3,
-                startingDelay: (0, utils_1.parseDuration)('1s'),
-                maxDelay: (0, utils_1.parseDuration)('5s'),
-                retry: (error) => {
-                    const message = error instanceof Error ? error.message : String(error);
-                    if (retryableMessages.some((m) => message?.includes(m)) ||
-                        (arg.query.kind === 'SelectQueryNode' &&
-                            (message ===
-                                'Egress connection severed outside of Hyperdrive' ||
-                                message === 'Network connection lost.'))) {
-                        return true;
-                    }
-                    return false;
-                },
-            });
-        },
-    }), {
+    const emitRetryDiagnostic = options?.onRetryDiagnostic ??
+        (options?.repositoryName
+            ? (diagnostic) => console.warn('Postgres operation retry diagnostic', diagnostic)
+            : undefined);
+    const reserveConnection = postgres.reserve.bind(postgres);
+    postgres.reserve = async () => {
+        const connection = await retryPostgres(reserveConnection, (error) => isRetryableConnectionError(error) ||
+            isConnectionReset(error) ||
+            (error instanceof Error &&
+                error.message === 'connection attempt failed'), emitRetryDiagnostic
+            ? {
+                dataRegion: options?.dataRegion,
+                emit: emitRetryDiagnostic,
+                operationKind: 'acquisition',
+                repositoryName: options?.repositoryName,
+                shouldEmit: (error) => isRetryableConnectionError(error) ||
+                    isConnectionReset(error) ||
+                    (error instanceof Error &&
+                        error.message === 'connection attempt failed'),
+            }
+            : undefined);
+        if (options?.hyperdrive) {
+            const unsafe = connection.unsafe;
+            connection.unsafe = (query, parameters, queryOptions) => unsafe(query, parameters, { ...queryOptions, prepare: true });
+        }
+        return connection;
+    };
+    const kysely = (0, plugins_1.getKysely)(new RetryingPostgresJSDialect({ postgres }, emitRetryDiagnostic
+        ? {
+            dataRegion: options?.dataRegion,
+            emit: emitRetryDiagnostic,
+            repositoryName: options?.repositoryName,
+            shouldEmit: isQueryRetryDiagnosticError,
+        }
+        : undefined), {
         repositoryName: options?.repositoryName,
         log: options?.log,
     });
     kysely.destroy = async () => {
         console.debug('Closing Kysely connection');
-        await Promise.race([
-            postgres.end({ timeout: 5000 }).catch(() => undefined),
-            new Promise((res) => setTimeout(res, 6000)),
-        ]);
+        await postgres.end({ timeout: 5 }).catch(() => undefined);
         const index = _connections.indexOf(postgres);
         if (index !== -1)
             _connections.splice(index, 1);
@@ -87,21 +261,29 @@ async function useAsyncPostgres(connectionString, opt) {
     const delayMs = 2500;
     const probe = opt?.probe === true;
     const retries = probe ? (opt?.retries === false ? 0 : opt?.retries || 3) : 1;
-    const ssl = opt?.forceDisableSSL !== true;
+    const ssl = opt?.forceDisableSSL !== true && opt?.hyperdrive !== true;
     const url = new URL(connectionString);
     console.debug(`Creating Postgres connection with probe=${probe}, ${retries} attempts, ssl=${ssl}, and port=${url.port}`);
     if (_connections.length > 0)
         console.warn(`We already have ${_connections.length} Postgres connections`);
     for (let attempt = 1; attempt <= retries; attempt++) {
+        let sql;
         try {
-            const sql = (0, postgres_1.default)(connectionString, {
+            sql = (0, postgres_1.default)(connectionString, {
                 ssl: ssl ? 'require' : false,
-                // connect_timeout: 10
-                // idle_timeout: 10
+                target_session_attrs: 'read-write',
                 // enable prepare statements, pgbouncer on crunchy allows for prepared statements
                 prepare: true,
-                connect_timeout: 10,
-                idle_timeout: 60,
+                ...(opt?.hyperdrive
+                    ? {
+                        max: 5,
+                        // Retire idle clients before Hyperdrive can close them asynchronously.
+                        idle_timeout: 10,
+                    }
+                    : {
+                        connect_timeout: 10,
+                        idle_timeout: 60,
+                    }),
             });
             if (probe) {
                 await sql `select 1`;
@@ -113,6 +295,7 @@ async function useAsyncPostgres(connectionString, opt) {
             return sql;
         }
         catch (err) {
+            await sql?.end({ timeout: 0 }).catch(() => undefined);
             if (attempt === retries)
                 throw err;
             console.warn(`Postgres connection failed (attempt ${attempt}), retrying...`);
@@ -126,33 +309,26 @@ async function useAsyncKysely(connectionString, opt) {
     const sql = await useAsyncPostgres(connectionString, opt);
     return useKysely(sql, opt);
 }
-/**
- * @deprecated This method is deprecated in favor of the useAsyncKysely method.
- */
-function usePostgres(connectionString, opt) {
-    if (!connectionString)
-        throw new error_1.KyselyError('No connection string');
-    const ssl = opt?.forceDisableSSL !== true;
-    const url = new URL(connectionString);
-    console.debug(`Creating sync Postgres connection with ssl=${ssl} and port=${url.port}`);
-    const pg = (0, postgres_1.default)(connectionString, {
-        ssl: ssl ? 'require' : false,
-    });
-    _connections.push(pg);
-    return pg;
+async function useAsyncNeonWebSocketKysely(connectionString, opt) {
+    const pool = new serverless_1.Pool({ connectionString, max: 5 });
+    const kysely = (0, plugins_1.getKysely)(new kysely_1.PostgresDialect({ pool }), opt ?? {});
+    kysely.destroy = async () => {
+        console.debug('Closing Neon WebSocket Kysely connection');
+        await pool.end();
+    };
+    if (opt?.probe)
+        await pool.query('select 1');
+    return kysely;
+}
+async function useAsyncControlPlaneKysely(connectionString, opt) {
+    const sql = await useAsyncPostgres(connectionString, opt);
+    return (0, control_plane_1.withControlPlaneSchema)(useKysely(sql, opt), 'public');
 }
 async function forceCloseAllPostgres(timeoutMs = 5000) {
     await Promise.race([
-        Promise.all(_connections.map((pg) => pg.end({ timeout: timeoutMs }).catch(() => undefined))),
+        Promise.all(_connections.map((pg) => pg.end({ timeout: timeoutMs / 1000 }).catch(() => undefined))),
         new Promise((res) => setTimeout(res, timeoutMs + 1000)),
     ]);
     _connections = [];
 }
-class DatabaseError extends Error {
-    constructor(message, cause) {
-        super(message);
-        this.cause = cause;
-    }
-}
-exports.DatabaseError = DatabaseError;
 //# sourceMappingURL=index.js.map

package/build/main/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;AAwCA,8BA6CC;AAGD,4CA4CC;AAED,wCAMC;AAKD,kCAeC;AAED,sDAUC;AA5KD,6DAA8C;AAE9C,2DAAuD;AACvD,wDAA8C;AAC9C,mCAAsC;AACtC,uCAAsC;AAGtC,0CAAwB;AACxB,+CAA6B;AAC7B,mDAAiC;AACjC,0CAAwB;AACxB,6CAA2B;AAC3B,gDAA8B;AAE9B,6CAAkD;AAclD,MAAM,iBAAiB,GAAG;IACxB,mBAAmB;IACnB,iBAAiB;IACjB,uCAAuC;CACxC,CAAC;AAIF;;GAEG;AACH,SAAgB,SAAS,CAAC,QAAa,EAAE,OAAoB;IAC3D,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAA,mBAAS,EACtB,IAAI,sCAAiB,CAAC;QACpB,QAAQ;QACR,mBAAmB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;YACxC,MAAM,YAAY,GAAG,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC9D,UAAU,CAAC,YAAY,GAAG,CAAC,GAAG,EAAE,EAAE,CAChC,IAAA,6BAAO,EAAC,KAAK,IAAI,EAAE,CAAC,MAAM,YAAY,CAAC,GAAG,CAAC,EAAE;gBAC3C,aAAa,EAAE,CAAC;gBAChB,aAAa,EAAE,IAAA,qBAAa,EAAC,IAAI,CAAC;gBAClC,QAAQ,EAAE,IAAA,qBAAa,EAAC,IAAI,CAAC;gBAC7B,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;oBACf,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACzD,IACE,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC;wBACnD,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB;4BACnC,CAAC,OAAO;gCACN,iDAAiD;gCACjD,OAAO,KAAK,0BAA0B,CAAC,CAAC,EAC5C,CAAC;wBACD,OAAO,IAAI,CAAC;oBACd,CAAC;oBACD,OAAO,KAAK,CAAC;gBACf,CAAC;aACF,CAAC,CAAC;QACP,CAAC;KACF,CAAC,EACF;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,OAAO,CAAC,IAAI,CAAC;YACjB,QAAQ,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;YACtD,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;SAC5C,CAAC,CAAC;QACH,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,CAAC;IAC1C,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,sBAAsB;gBACtB,mBAAmB;gBACnB,iFAAiF;gBACjF,OAAO,EAAE,IAAI;gBACb,eAAe,EAAE,EAAE;gBACnB,YAAY,EAAE,EAAE;aACjB,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;AAED;;GAEG;AACH,SAAgB,WAAW,CACzB,gBAAyB,EACzB,GAAoC;IAEpC,IAAI,CAAC,gBAAgB;QAAE,MAAM,IAAI,mBAAW,CAAC,sBAAsB,CAAC,CAAC;IACrE,MAAM,GAAG,GAAG,GAAG,EAAE,eAAe,KAAK,IAAI,CAAC;IAC1C,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,CAAC;IACtC,OAAO,CAAC,KAAK,CACX,8CAA8C,GAAG,aAAa,GAAG,CAAC,IAAI,EAAE,CACzE,CAAC;IACF,MAAM,EAAE,GAAG,IAAA,kBAAQ,EAAC,gBAAgB,EAAE;QACpC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;KAC7B,CAAC,CAAC;IACH,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACtB,OAAO,EAAE,CAAC;AACZ,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,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CACtD,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;AAED,MAAa,aAAc,SAAQ,KAAK;IACtC,YAAY,OAAe,EAAE,KAAU;QACrC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AALD,sCAKC","sourcesContent":["import { backOff } from 'exponential-backoff';\nimport type { Kysely as K, LogConfig } from 'kysely';\nimport { PostgresJSDialect } from 'kysely-postgres-js';\nimport postgres, { type Sql } from 'postgres';\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  repositoryName?: string;\n  log?: true | LogConfig;\n};\n\ntype PgOpt = {\n  forceDisableSSL?: boolean;\n  probe?: true;\n  retries?: false | number;\n};\n\nconst retryableMessages = [\n  'CONNECTION_CLOSED',\n  'CONNECT_TIMEOUT',\n  'Idle connection closed by Hyperdrive.',\n];\n\nexport type Kysely = K<DB>;\nexport type Postgres = Sql;\n/**\n * @deprecated This method is deprecated in favor of the useAsyncKysely method.\n */\nexport function useKysely(postgres: Sql, options?: KyselyOpts) {\n  console.debug('Creating Kysely instance');\n  const kysely = getKysely(\n    new PostgresJSDialect({\n      postgres,\n      onReserveConnection: async (connection) => {\n        const executeQuery = connection.executeQuery.bind(connection);\n        connection.executeQuery = (arg) =>\n          backOff(async () => await executeQuery(arg), {\n            numOfAttempts: 3,\n            startingDelay: parseDuration('1s'),\n            maxDelay: parseDuration('5s'),\n            retry: (error) => {\n              const message =\n                error instanceof Error ? error.message : String(error);\n              if (\n                retryableMessages.some((m) => message?.includes(m)) ||\n                (arg.query.kind === 'SelectQueryNode' &&\n                  (message ===\n                    'Egress connection severed outside of Hyperdrive' ||\n                    message === 'Network connection lost.'))\n              ) {\n                return true;\n              }\n              return false;\n            },\n          });\n      },\n    }),\n    {\n      repositoryName: options?.repositoryName,\n      log: options?.log,\n    }\n  );\n  kysely.destroy = async () => {\n    console.debug('Closing Kysely connection');\n    await Promise.race([\n      postgres.end({ timeout: 5000 }).catch(() => undefined),\n      new Promise((res) => setTimeout(res, 6000)),\n    ]);\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;\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        // connect_timeout: 10\n        // idle_timeout: 10\n        // enable prepare statements, pgbouncer on crunchy allows for prepared statements\n        prepare: true,\n        connect_timeout: 10,\n        idle_timeout: 60,\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\n/**\n * @deprecated This method is deprecated in favor of the useAsyncKysely method.\n */\nexport function usePostgres(\n  connectionString?: string,\n  opt?: Pick<PgOpt, 'forceDisableSSL'>\n) {\n  if (!connectionString) throw new KyselyError('No connection string');\n  const ssl = opt?.forceDisableSSL !== true;\n  const url = new URL(connectionString);\n  console.debug(\n    `Creating sync Postgres connection with ssl=${ssl} and port=${url.port}`\n  );\n  const pg = postgres(connectionString, {\n    ssl: ssl ? 'require' : false,\n  });\n  _connections.push(pg);\n  return pg;\n}\n\nexport async function forceCloseAllPostgres(timeoutMs = 5000) {\n  await Promise.race([\n    Promise.all(\n      _connections.map((pg) =>\n        pg.end({ timeout: timeoutMs }).catch(() => undefined)\n      )\n    ),\n    new Promise((res) => setTimeout(res, timeoutMs + 1000)),\n  ]);\n  _connections = [];\n}\n\nexport class DatabaseError extends Error {\n  constructor(message: string, cause: any) {\n    super(message);\n    this.cause = cause;\n  }\n}\n"]}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;AAsRA,8BA+DC;AAGD,4CAqDC;AAED,wCAMC;AAED,kEAYC;AAED,gEAMC;AAED,sDAUC;AAvbD,6DAA8C;AAC9C,yDAAgD;AAQhD,mCAAyC;AACzC,2DAI4B;AAC5B,wDAA8C;AAC9C,mDAA8E;AAC9E,mCAAsC;AACtC,uCAAsC;AAGtC,0CAAwB;AACxB,+CAA6B;AAC7B,mDAAiC;AACjC,0CAAwB;AACxB,6CAA2B;AAC3B,gDAA8B;AAE9B,6CAAkD;AA0BlD,MAAM,iBAAiB,GAAG;IACxB,mBAAmB;IACnB,iBAAiB;IACjB,uCAAuC;IACvC,oBAAoB;CACrB,CAAC;AACF,MAAM,qBAAqB,GAAG,CAAC,CAAC;AAEhC,SAAS,0BAA0B,CAAC,KAAc;IAChD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,OAAO,iBAAiB,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;AAC5E,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,OAAO,CACL,KAAK,YAAY,KAAK,IAAI,MAAM,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CACzE,CAAC;AACJ,CAAC;AAED,SAAS,2BAA2B,CAAC,KAAc;IACjD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,OAAO,CACL,0BAA0B,CAAC,KAAK,CAAC;QACjC,iBAAiB,CAAC,KAAK,CAAC;QACxB,OAAO,KAAK,iDAAiD;QAC7D,OAAO,KAAK,0BAA0B;QACtC,OAAO,KAAK,oBAAoB;QAChC,OAAO,KAAK,qBAAqB,CAClC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,aAAa,CAC1B,SAA2B,EAC3B,KAA4E,EAC5E,UAGC;IAED,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,MAAM,cAAc,GAAG,CAAC,OAA2C,EAAE,EAAE;QACrE,IAAI,CAAC,UAAU;YAAE,OAAO;QACxB,UAAU,CAAC,IAAI,CAAC;YACd,YAAY;YACZ,UAAU,EAAE,UAAU,CAAC,UAAU;YACjC,aAAa,EAAE,UAAU,CAAC,aAAa;YACvC,OAAO;YACP,cAAc,EAAE,UAAU,CAAC,cAAc;SAC1C,CAAC,CAAC;IACL,CAAC,CAAC;IACF,IAAI,CAAC;QACH,OAAO,MAAM,IAAA,6BAAO,EAClB,KAAK,IAAI,EAAE;YACT,YAAY,EAAE,CAAC;YACf,OAAO,MAAM,SAAS,EAAE,CAAC;QAC3B,CAAC,EACD;YACE,aAAa,EAAE,qBAAqB;YACpC,aAAa,EAAE,IAAA,qBAAa,EAAC,IAAI,CAAC;YAClC,QAAQ,EAAE,IAAA,qBAAa,EAAC,IAAI,CAAC;YAC7B,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,aAAa;gBAC9B,MAAM,WAAW,GACf,aAAa,GAAG,qBAAqB;oBACrC,CAAC,MAAM,KAAK,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC,CAAC;gBACtC,IAAI,WAAW,IAAI,UAAU,EAAE,CAAC;oBAC9B,OAAO,GAAG,IAAI,CAAC;oBACf,cAAc,CAAC,UAAU,CAAC,CAAC;gBAC7B,CAAC;gBACD,OAAO,WAAW,CAAC;YACrB,CAAC;SACF,CACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,UAAU,EAAE,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAClC,cAAc,CACZ,OAAO,IAAI,YAAY,KAAK,qBAAqB;gBAC/C,CAAC,CAAC,WAAW;gBACb,CAAC,CAAC,cAAc,CACnB,CAAC;QACJ,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,yBAA0B,SAAQ,sCAAiB;IACvD,YACmB,MAA+B,EAC/B,eAMhB;QAED,KAAK,CAAC,MAAM,CAAC,CAAC;QATG,WAAM,GAAN,MAAM,CAAyB;QAC/B,oBAAe,GAAf,eAAe,CAM/B;IAGH,CAAC;IAEQ,YAAY;QACnB,MAAM,MAAM,GAAG,IAAI,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;wBAC9B,MAAM,aAAa,GAAG,KAAK,CAAC,aAAa;4BACvC,CAAC,CAAC,aAAa;4BACf,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB;gCAC9C,CAAC,CAAC,QAAQ;gCACV,CAAC,CAAC,UAAU,CAAC;wBACjB,OAAO,aAAa,CAClB,GAAG,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,YAAY,CAAC,aAAa,CAAC,EAClD,KAAK,EAAE,KAAK,EAAE,EAAE;4BACd,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;4BACzD,IACE,KAAK,CAAC,aAAa;gCACnB,CAAC,CAAC,0BAA0B,CAAC,KAAK,CAAC;oCACjC,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB;wCAC7C,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC;4CACxB,OAAO;gDACL,iDAAiD;4CACnD,OAAO,KAAK,0BAA0B;4CACtC,OAAO,KAAK,oBAAoB;4CAChC,OAAO,KAAK,qBAAqB,CAAC,CAAC,CAAC,EAC1C,CAAC;gCACD,OAAO,KAAK,CAAC;4BACf,CAAC;4BAED,4DAA4D;4BAC5D,4DAA4D;4BAC5D,KAAK,CAAC,UAAU,GAAG,MAAM,MAAM,CAAC,iBAAiB,EAAE,CAAC;4BACpD,OAAO,IAAI,CAAC;wBACd,CAAC,EACD,IAAI,CAAC,eAAe;4BAClB,CAAC,CAAC;gCACE,GAAG,IAAI,CAAC,eAAe;gCACvB,aAAa;gCACb,UAAU,EAAE,2BAA2B;6BACxC;4BACH,CAAC,CAAC,SAAS,CACd,CAAC;oBACJ,CAAC;oBACD,WAAW,EAAE,CAAC,aAAa,EAAE,SAAS,EAAE,EAAE;wBACxC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;4BAC5B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;wBAC5D,CAAC;wBACD,OAAO,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;oBAChE,CAAC;iBACF,CAAC;gBACF,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBACnC,OAAO,UAAU,CAAC;YACpB,CAAC;YACD,gBAAgB,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE;gBAC/C,MAAM,KAAK,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxC,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC;gBAC3B,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;oBAC5B,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;YACD,iBAAiB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;gBACtC,MAAM,KAAK,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxC,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBACnD,CAAC;wBAAS,CAAC;oBACT,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,mBAAmB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;gBACxC,MAAM,KAAK,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxC,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBACrD,CAAC;wBAAS,CAAC;oBACT,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;gBAC9B,CAAC;YACH,CAAC;YACD,SAAS,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,EAAE,CACrD,MAAM,CAAC,SAAS,CACd,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,EACpC,aAAa,EACb,YAAY,CACb;YACH,mBAAmB,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,EAAE,CAC/D,MAAM,CAAC,mBAAmB,CACxB,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,EACpC,aAAa,EACb,YAAY,CACb;YACH,gBAAgB,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,EAAE,CAC5D,MAAM,CAAC,gBAAgB,CACrB,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,EACpC,aAAa,EACb,YAAY,CACb;YACH,iBAAiB,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;gBACtC,MAAM,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,UAAU,CAAC,CAAC;gBACrE,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACjC,CAAC;YACD,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE;SAChC,CAAC;IACJ,CAAC;CACF;AAKD,SAAgB,SAAS,CAAgB,QAAa,EAAE,OAAoB;IAC1E,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC1C,MAAM,mBAAmB,GACvB,OAAO,EAAE,iBAAiB;QAC1B,CAAC,OAAO,EAAE,cAAc;YACtB,CAAC,CAAC,CAAC,UAAmC,EAAE,EAAE,CACtC,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE,UAAU,CAAC;YACnE,CAAC,CAAC,SAAS,CAAC,CAAC;IACjB,MAAM,iBAAiB,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1D,QAAQ,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC5B,MAAM,UAAU,GAAG,MAAM,aAAa,CACpC,iBAAiB,EACjB,CAAC,KAAK,EAAE,EAAE,CACR,0BAA0B,CAAC,KAAK,CAAC;YACjC,iBAAiB,CAAC,KAAK,CAAC;YACxB,CAAC,KAAK,YAAY,KAAK;gBACrB,KAAK,CAAC,OAAO,KAAK,2BAA2B,CAAC,EAClD,mBAAmB;YACjB,CAAC,CAAC;gBACE,UAAU,EAAE,OAAO,EAAE,UAAU;gBAC/B,IAAI,EAAE,mBAAmB;gBACzB,aAAa,EAAE,aAAa;gBAC5B,cAAc,EAAE,OAAO,EAAE,cAAc;gBACvC,UAAU,EAAE,CAAC,KAAK,EAAE,EAAE,CACpB,0BAA0B,CAAC,KAAK,CAAC;oBACjC,iBAAiB,CAAC,KAAK,CAAC;oBACxB,CAAC,KAAK,YAAY,KAAK;wBACrB,KAAK,CAAC,OAAO,KAAK,2BAA2B,CAAC;aACnD;YACH,CAAC,CAAC,SAAS,CACd,CAAC;QACF,IAAI,OAAO,EAAE,UAAU,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;YACjC,UAAU,CAAC,MAAM,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,CACtD,MAAM,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,GAAG,YAAY,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAClE,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;IACF,MAAM,MAAM,GAAG,IAAA,mBAAS,EACtB,IAAI,yBAAyB,CAC3B,EAAE,QAAQ,EAAE,EACZ,mBAAmB;QACjB,CAAC,CAAC;YACE,UAAU,EAAE,OAAO,EAAE,UAAU;YAC/B,IAAI,EAAE,mBAAmB;YACzB,cAAc,EAAE,OAAO,EAAE,cAAc;YACvC,UAAU,EAAE,2BAA2B;SACxC;QACH,CAAC,CAAC,SAAS,CACd,EACD;QACE,cAAc,EAAE,OAAO,EAAE,cAAc;QACvC,GAAG,EAAE,OAAO,EAAE,GAAG;KAClB,CACF,CAAC;IACF,MAAM,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC3C,MAAM,QAAQ,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC1D,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,KAAK,KAAK,CAAC,CAAC;YAAE,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,IAAI,YAAY,GAAmB,EAAE,CAAC;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,GAAoB,CAAC;QACzB,IAAI,CAAC;YACH,GAAG,GAAG,IAAA,kBAAQ,EAAC,gBAAgB,EAAE;gBAC/B,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;gBAC5B,oBAAoB,EAAE,YAAY;gBAClC,iFAAiF;gBACjF,OAAO,EAAE,IAAI;gBACb,GAAG,CAAC,GAAG,EAAE,UAAU;oBACjB,CAAC,CAAC;wBACE,GAAG,EAAE,CAAC;wBACN,uEAAuE;wBACvE,YAAY,EAAE,EAAE;qBACjB;oBACH,CAAC,CAAC;wBACE,eAAe,EAAE,EAAE;wBACnB,YAAY,EAAE,EAAE;qBACjB,CAAC;aACP,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,GAAG,CAAA,UAAU,CAAC;YACtB,CAAC;YACD,wEAAwE;YACxE,8DAA8D;YAC9D,qEAAqE;YAErE,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvB,OAAO,GAAG,CAAC;QACb,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACtD,IAAI,OAAO,KAAK,OAAO;gBAAE,MAAM,GAAG,CAAC;YACnC,OAAO,CAAC,IAAI,CACV,uCAAuC,OAAO,gBAAgB,CAC/D,CAAC;YACF,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAC5B,MAAM,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IACD,MAAM,IAAI,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,2BAA2B,CAC/C,gBAAwB,EACxB,GAAuC;IAEvC,MAAM,IAAI,GAAG,IAAI,iBAAI,CAAC,EAAE,gBAAgB,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;IACpD,MAAM,MAAM,GAAG,IAAA,mBAAS,EAAK,IAAI,wBAAe,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;IACvE,MAAM,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC1D,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC,CAAC;IACF,IAAI,GAAG,EAAE,KAAK;QAAE,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAC7C,OAAO,MAAM,CAAC;AAChB,CAAC;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 { Pool } from '@neondatabase/serverless';\nimport type {\n  DatabaseConnection,\n  Driver,\n  Kysely as K,\n  LogConfig,\n  Transaction as T,\n} from 'kysely';\nimport { PostgresDialect } from 'kysely';\nimport {\n  PostgresJSDialect,\n  type PostgresJSDialectConfig,\n  PostgresJSDriver,\n} from 'kysely-postgres-js';\nimport postgres, { type Sql } from 'postgres';\nimport { type ControlPlaneDB, withControlPlaneSchema } from './control-plane';\nimport { KyselyError } from './error';\nimport { getKysely } from './plugins';\n\nexport type { Insertable, LogEvent } from 'kysely';\nexport * from './error';\nexport * from './isomorphic';\nexport * from './postgres-error';\nexport * from './query';\nexport * from './regional';\nexport * from './transaction';\n\nimport { parseDuration } from '@vrplatform/utils';\nimport type { TenantDataRegion } from './regional';\nimport type { DB } from './v1.generated';\n\ntype KyselyOpts = {\n  dataRegion?: TenantDataRegion;\n  hyperdrive?: true;\n  onRetryDiagnostic?: (diagnostic: PostgresRetryDiagnostic) => void;\n  repositoryName?: string;\n  log?: true | LogConfig;\n};\n\ntype PostgresRetryDiagnostic = {\n  attemptCount: number;\n  dataRegion?: TenantDataRegion;\n  operationKind: 'acquisition' | 'mutation' | 'select' | 'transaction';\n  outcome: 'exhausted' | 'notRetryable' | 'retrying';\n  repositoryName?: string;\n};\n\ntype PgOpt = {\n  forceDisableSSL?: boolean;\n  probe?: true;\n  retries?: false | number;\n} & Pick<KyselyOpts, 'hyperdrive'>;\n\nconst retryableMessages = [\n  'CONNECTION_CLOSED',\n  'CONNECT_TIMEOUT',\n  'Idle connection closed by Hyperdrive.',\n  'server_login_retry',\n];\nconst postgresRetryAttempts = 3;\n\nfunction isRetryableConnectionError(error: unknown) {\n  const message = error instanceof Error ? error.message : String(error);\n  return retryableMessages.some((retryable) => message.includes(retryable));\n}\n\nfunction isConnectionReset(error: unknown) {\n  return (\n    error instanceof Error && 'code' in error && error.code === 'ECONNRESET'\n  );\n}\n\nfunction isQueryRetryDiagnosticError(error: unknown) {\n  const message = error instanceof Error ? error.message : String(error);\n  return (\n    isRetryableConnectionError(error) ||\n    isConnectionReset(error) ||\n    message === 'Egress connection severed outside of Hyperdrive' ||\n    message === 'Network connection lost.' ||\n    message === 'Node shutting down' ||\n    message === 'Node shutting down.'\n  );\n}\n\nasync function retryPostgres<T>(\n  operation: () => Promise<T>,\n  retry: (error: unknown, attemptNumber: number) => boolean | Promise<boolean>,\n  diagnostic?: Omit<PostgresRetryDiagnostic, 'attemptCount' | 'outcome'> & {\n    emit: (diagnostic: PostgresRetryDiagnostic) => void;\n    shouldEmit: (error: unknown) => boolean;\n  }\n) {\n  let attemptCount = 0;\n  let retried = false;\n  const emitDiagnostic = (outcome: PostgresRetryDiagnostic['outcome']) => {\n    if (!diagnostic) return;\n    diagnostic.emit({\n      attemptCount,\n      dataRegion: diagnostic.dataRegion,\n      operationKind: diagnostic.operationKind,\n      outcome,\n      repositoryName: diagnostic.repositoryName,\n    });\n  };\n  try {\n    return await backOff(\n      async () => {\n        attemptCount++;\n        return await operation();\n      },\n      {\n        numOfAttempts: postgresRetryAttempts,\n        startingDelay: parseDuration('1s'),\n        maxDelay: parseDuration('5s'),\n        async retry(error, attemptNumber) {\n          const shouldRetry =\n            attemptNumber < postgresRetryAttempts &&\n            (await retry(error, attemptNumber));\n          if (shouldRetry && diagnostic) {\n            retried = true;\n            emitDiagnostic('retrying');\n          }\n          return shouldRetry;\n        },\n      }\n    );\n  } catch (error) {\n    if (diagnostic?.shouldEmit(error)) {\n      emitDiagnostic(\n        retried && attemptCount === postgresRetryAttempts\n          ? 'exhausted'\n          : 'notRetryable'\n      );\n    }\n    throw error;\n  }\n}\n\nclass RetryingPostgresJSDialect extends PostgresJSDialect {\n  constructor(\n    private readonly config: PostgresJSDialectConfig,\n    private readonly retryDiagnostic?: Omit<\n      PostgresRetryDiagnostic,\n      'attemptCount' | 'operationKind' | 'outcome'\n    > & {\n      emit: (diagnostic: PostgresRetryDiagnostic) => void;\n      shouldEmit: (error: unknown) => boolean;\n    }\n  ) {\n    super(config);\n  }\n\n  override createDriver(): Driver {\n    const driver = new PostgresJSDriver(this.config);\n    const connections = new WeakMap<\n      DatabaseConnection,\n      { connection: DatabaseConnection; inTransaction: boolean }\n    >();\n    const getConnection = (connection: DatabaseConnection) => {\n      const state = connections.get(connection);\n      if (!state) throw new Error('Postgres connection state is unavailable');\n      return state;\n    };\n\n    return {\n      init: () => driver.init(),\n      acquireConnection: async () => {\n        const state = {\n          connection: await driver.acquireConnection(),\n          inTransaction: false,\n        };\n        const connection: DatabaseConnection = {\n          executeQuery: (compiledQuery) => {\n            const operationKind = state.inTransaction\n              ? 'transaction'\n              : compiledQuery.query.kind === 'SelectQueryNode'\n                ? 'select'\n                : 'mutation';\n            return retryPostgres(\n              () => state.connection.executeQuery(compiledQuery),\n              async (error) => {\n                const message =\n                  error instanceof Error ? error.message : String(error);\n                if (\n                  state.inTransaction ||\n                  (!isRetryableConnectionError(error) &&\n                    (compiledQuery.query.kind !== 'SelectQueryNode' ||\n                      (!isConnectionReset(error) &&\n                        message !==\n                          'Egress connection severed outside of Hyperdrive' &&\n                        message !== 'Network connection lost.' &&\n                        message !== 'Node shutting down' &&\n                        message !== 'Node shutting down.')))\n                ) {\n                  return false;\n                }\n\n                // Postgres.js already ejects a severed reserved connection.\n                // Acquire a new reservation before replaying the safe read.\n                state.connection = await driver.acquireConnection();\n                return true;\n              },\n              this.retryDiagnostic\n                ? {\n                    ...this.retryDiagnostic,\n                    operationKind,\n                    shouldEmit: isQueryRetryDiagnosticError,\n                  }\n                : undefined\n            );\n          },\n          streamQuery: (compiledQuery, chunkSize) => {\n            if (chunkSize === undefined) {\n              throw new Error('Postgres stream chunk size is required');\n            }\n            return state.connection.streamQuery(compiledQuery, chunkSize);\n          },\n        };\n        connections.set(connection, state);\n        return connection;\n      },\n      beginTransaction: async (connection, settings) => {\n        const state = getConnection(connection);\n        state.inTransaction = true;\n        try {\n          await driver.beginTransaction(state.connection, settings);\n        } catch (error) {\n          state.inTransaction = false;\n          throw error;\n        }\n      },\n      commitTransaction: async (connection) => {\n        const state = getConnection(connection);\n        try {\n          await driver.commitTransaction(state.connection);\n        } finally {\n          state.inTransaction = false;\n        }\n      },\n      rollbackTransaction: async (connection) => {\n        const state = getConnection(connection);\n        try {\n          await driver.rollbackTransaction(state.connection);\n        } finally {\n          state.inTransaction = false;\n        }\n      },\n      savepoint: (connection, savepointName, compileQuery) =>\n        driver.savepoint(\n          getConnection(connection).connection,\n          savepointName,\n          compileQuery\n        ),\n      rollbackToSavepoint: (connection, savepointName, compileQuery) =>\n        driver.rollbackToSavepoint(\n          getConnection(connection).connection,\n          savepointName,\n          compileQuery\n        ),\n      releaseSavepoint: (connection, savepointName, compileQuery) =>\n        driver.releaseSavepoint(\n          getConnection(connection).connection,\n          savepointName,\n          compileQuery\n        ),\n      releaseConnection: async (connection) => {\n        await driver.releaseConnection(getConnection(connection).connection);\n        connections.delete(connection);\n      },\n      destroy: () => driver.destroy(),\n    };\n  }\n}\n\nexport type Kysely = K<DB>;\nexport type ControlPlaneKysely = K<ControlPlaneDB>;\nexport type ControlPlaneTransaction = T<ControlPlaneDB>;\nexport function useKysely<Database = DB>(postgres: Sql, options?: KyselyOpts) {\n  console.debug('Creating Kysely instance');\n  const emitRetryDiagnostic =\n    options?.onRetryDiagnostic ??\n    (options?.repositoryName\n      ? (diagnostic: PostgresRetryDiagnostic) =>\n          console.warn('Postgres operation retry diagnostic', diagnostic)\n      : undefined);\n  const reserveConnection = postgres.reserve.bind(postgres);\n  postgres.reserve = async () => {\n    const connection = await retryPostgres(\n      reserveConnection,\n      (error) =>\n        isRetryableConnectionError(error) ||\n        isConnectionReset(error) ||\n        (error instanceof Error &&\n          error.message === 'connection attempt failed'),\n      emitRetryDiagnostic\n        ? {\n            dataRegion: options?.dataRegion,\n            emit: emitRetryDiagnostic,\n            operationKind: 'acquisition',\n            repositoryName: options?.repositoryName,\n            shouldEmit: (error) =>\n              isRetryableConnectionError(error) ||\n              isConnectionReset(error) ||\n              (error instanceof Error &&\n                error.message === 'connection attempt failed'),\n          }\n        : undefined\n    );\n    if (options?.hyperdrive) {\n      const unsafe = connection.unsafe;\n      connection.unsafe = (query, parameters, queryOptions) =>\n        unsafe(query, parameters, { ...queryOptions, prepare: true });\n    }\n    return connection;\n  };\n  const kysely = getKysely<Database>(\n    new RetryingPostgresJSDialect(\n      { postgres },\n      emitRetryDiagnostic\n        ? {\n            dataRegion: options?.dataRegion,\n            emit: emitRetryDiagnostic,\n            repositoryName: options?.repositoryName,\n            shouldEmit: isQueryRetryDiagnosticError,\n          }\n        : undefined\n    ),\n    {\n      repositoryName: options?.repositoryName,\n      log: options?.log,\n    }\n  );\n  kysely.destroy = async () => {\n    console.debug('Closing Kysely connection');\n    await postgres.end({ timeout: 5 }).catch(() => undefined);\n    const index = _connections.indexOf(postgres);\n    if (index !== -1) _connections.splice(index, 1);\n  };\n\n  return kysely;\n}\n\nlet _connections: postgres.Sql[] = [];\nexport async function useAsyncPostgres(connectionString: string, opt?: PgOpt) {\n  const delayMs = 2500;\n  const probe = opt?.probe === true;\n  const retries = probe ? (opt?.retries === false ? 0 : opt?.retries || 3) : 1;\n  const ssl = opt?.forceDisableSSL !== true && opt?.hyperdrive !== true;\n  const url = new URL(connectionString);\n\n  console.debug(\n    `Creating Postgres connection with probe=${probe}, ${retries} attempts, ssl=${ssl}, and port=${url.port}`\n  );\n  if (_connections.length > 0)\n    console.warn(`We already have ${_connections.length} Postgres connections`);\n\n  for (let attempt = 1; attempt <= retries; attempt++) {\n    let sql: Sql | undefined;\n    try {\n      sql = postgres(connectionString, {\n        ssl: ssl ? 'require' : false,\n        target_session_attrs: 'read-write',\n        // enable prepare statements, pgbouncer on crunchy allows for prepared statements\n        prepare: true,\n        ...(opt?.hyperdrive\n          ? {\n              max: 5,\n              // Retire idle clients before Hyperdrive can close them asynchronously.\n              idle_timeout: 10,\n            }\n          : {\n              connect_timeout: 10,\n              idle_timeout: 60,\n            }),\n      });\n\n      if (probe) {\n        await sql`select 1`;\n      }\n      // statement_timeout/lock_timeout are role-level settings on the cluster\n      // (ALTER ROLE application SET ...) so they cover every pooled\n      // connection; per-session SETs here only ever applied to one of them\n\n      _connections.push(sql);\n      return sql;\n    } catch (err) {\n      await sql?.end({ timeout: 0 }).catch(() => undefined);\n      if (attempt === retries) throw err;\n      console.warn(\n        `Postgres connection failed (attempt ${attempt}), retrying...`\n      );\n      console.error('error', err);\n      await new Promise((res) => setTimeout(res, delayMs));\n    }\n  }\n  throw new KyselyError('Unable to connect to Postgres');\n}\n\nexport async function useAsyncKysely(\n  connectionString: string,\n  opt?: PgOpt & KyselyOpts\n) {\n  const sql = await useAsyncPostgres(connectionString, opt);\n  return useKysely(sql, opt);\n}\n\nexport async function useAsyncNeonWebSocketKysely(\n  connectionString: string,\n  opt?: Pick<PgOpt, 'probe'> & KyselyOpts\n) {\n  const pool = new Pool({ connectionString, max: 5 });\n  const kysely = getKysely<DB>(new PostgresDialect({ pool }), opt ?? {});\n  kysely.destroy = async () => {\n    console.debug('Closing Neon WebSocket Kysely connection');\n    await pool.end();\n  };\n  if (opt?.probe) await pool.query('select 1');\n  return kysely;\n}\n\nexport async function useAsyncControlPlaneKysely(\n  connectionString: string,\n  opt?: PgOpt & KyselyOpts\n) {\n  const sql = await useAsyncPostgres(connectionString, opt);\n  return withControlPlaneSchema(useKysely<ControlPlaneDB>(sql, opt), 'public');\n}\n\nexport async function forceCloseAllPostgres(timeoutMs = 5000) {\n  await Promise.race([\n    Promise.all(\n      _connections.map((pg) =>\n        pg.end({ timeout: timeoutMs / 1000 }).catch(() => undefined)\n      )\n    ),\n    new Promise((res) => setTimeout(res, timeoutMs + 1000)),\n  ]);\n  _connections = [];\n}\n"]}

package/build/main/isomorphic.d.ts

@@ -1,12 +1,12 @@
 import type { ControlledTransaction as CT, Kysely as K, Transaction as T } from 'kysely';
 export type { ColumnType, Expression, ExpressionBuilder, ExpressionOrFactory, Generated, Insertable, JSONColumnType, KyselyPlugin, Nullable, OperandExpression, PluginTransformQueryArgs, PluginTransformResultArgs, QueryResult, RootOperationNode, Selectable, SelectQueryBuilder, SqlBool, UnknownRow, Updateable, } from 'kysely';
 export { jsonArrayFrom } from 'kysely/helpers/postgres';
+export * from './domain';
 export * from './v1.generated';
 import type { DB } from './v1.generated';
-export type Transaction = T<DB>;
+export type DatabaseKysely<Database> = K<Database>;
+export type DatabaseTransaction<Database> = T<Database>;
+export type Transaction = DatabaseTransaction<DB>;
 export type ControlledTransaction<S extends string[] = string[]> = CT<DB, S>;
 export { sql } from 'kysely';
-export type Kysely = K<DB>;
-export declare class DatabaseError extends Error {
-    constructor(message: string, cause: any);
-}
+export type Kysely = DatabaseKysely<DB>;

package/build/main/isomorphic.js

@@ -14,17 +14,11 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DatabaseError = exports.sql = exports.jsonArrayFrom = void 0;
+exports.sql = exports.jsonArrayFrom = void 0;
 var postgres_1 = require("kysely/helpers/postgres");
 Object.defineProperty(exports, "jsonArrayFrom", { enumerable: true, get: function () { return postgres_1.jsonArrayFrom; } });
+__exportStar(require("./domain"), exports);
 __exportStar(require("./v1.generated"), exports);
 var kysely_1 = require("kysely");
 Object.defineProperty(exports, "sql", { enumerable: true, get: function () { return kysely_1.sql; } });
-class DatabaseError extends Error {
-    constructor(message, cause) {
-        super(message);
-        this.cause = cause;
-    }
-}
-exports.DatabaseError = DatabaseError;
 //# sourceMappingURL=isomorphic.js.map

package/build/main/isomorphic.js.map

@@ -1 +1 @@
-{"version":3,"file":"isomorphic.js","sourceRoot":"src/","sources":["isomorphic.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AA2BA,oDAAwD;AAA/C,yGAAA,aAAa,OAAA;AACtB,iDAA+B;AAM/B,iCAA6B;AAApB,6FAAA,GAAG,OAAA;AAIZ,MAAa,aAAc,SAAQ,KAAK;IACtC,YAAY,OAAe,EAAE,KAAU;QACrC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AALD,sCAKC","sourcesContent":["import type {\n  ControlledTransaction as CT,\n  Kysely as K,\n  Transaction as T,\n} from 'kysely';\n\nexport type {\n  ColumnType,\n  Expression,\n  ExpressionBuilder,\n  ExpressionOrFactory,\n  Generated,\n  Insertable,\n  JSONColumnType,\n  KyselyPlugin,\n  Nullable,\n  OperandExpression,\n  PluginTransformQueryArgs,\n  PluginTransformResultArgs,\n  QueryResult,\n  RootOperationNode,\n  Selectable,\n  SelectQueryBuilder,\n  SqlBool,\n  UnknownRow,\n  Updateable,\n} from 'kysely';\nexport { jsonArrayFrom } from 'kysely/helpers/postgres';\nexport * from './v1.generated';\n\nimport type { DB } from './v1.generated';\n\nexport type Transaction = T<DB>;\nexport type ControlledTransaction<S extends string[] = string[]> = CT<DB, S>;\nexport { sql } from 'kysely';\n\nexport type Kysely = K<DB>;\n\nexport class DatabaseError extends Error {\n  constructor(message: string, cause: any) {\n    super(message);\n    this.cause = cause;\n  }\n}\n"]}
+{"version":3,"file":"isomorphic.js","sourceRoot":"src/","sources":["isomorphic.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AA2BA,oDAAwD;AAA/C,yGAAA,aAAa,OAAA;AACtB,2CAAyB;AACzB,iDAA+B;AAQ/B,iCAA6B;AAApB,6FAAA,GAAG,OAAA","sourcesContent":["import type {\n  ControlledTransaction as CT,\n  Kysely as K,\n  Transaction as T,\n} from 'kysely';\n\nexport type {\n  ColumnType,\n  Expression,\n  ExpressionBuilder,\n  ExpressionOrFactory,\n  Generated,\n  Insertable,\n  JSONColumnType,\n  KyselyPlugin,\n  Nullable,\n  OperandExpression,\n  PluginTransformQueryArgs,\n  PluginTransformResultArgs,\n  QueryResult,\n  RootOperationNode,\n  Selectable,\n  SelectQueryBuilder,\n  SqlBool,\n  UnknownRow,\n  Updateable,\n} from 'kysely';\nexport { jsonArrayFrom } from 'kysely/helpers/postgres';\nexport * from './domain';\nexport * from './v1.generated';\n\nimport type { DB } from './v1.generated';\n\nexport type DatabaseKysely<Database> = K<Database>;\nexport type DatabaseTransaction<Database> = T<Database>;\nexport type Transaction = DatabaseTransaction<DB>;\nexport type ControlledTransaction<S extends string[] = string[]> = CT<DB, S>;\nexport { sql } from 'kysely';\n\nexport type Kysely = DatabaseKysely<DB>;\n"]}

package/build/main/local/digest.d.ts

@@ -1,3 +1,3 @@
-import { type Kysely } from '../isomorphic';
-export declare function ensureDigestFunction(kysely: Kysely): Promise<void>;
-export declare function ensureGenRandomBytesFunction(kysely: Kysely): Promise<void>;
+import { type DatabaseKysely } from '../isomorphic';
+export declare function ensureDigestFunction<Database>(kysely: DatabaseKysely<Database>): Promise<void>;
+export declare function ensureGenRandomBytesFunction<Database>(kysely: DatabaseKysely<Database>): Promise<void>;

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

@@ -1 +1 @@
-{"version":3,"file":"digest.js","sourceRoot":"src/","sources":["local/digest.ts"],"names":[],"mappings":";;AAEA,oDAeC;AAED,oEAwBC;AA3CD,8CAAiD;AAE1C,KAAK,UAAU,oBAAoB,CAAC,MAAc;IACvD,IAAI,CAAC;QACH,MAAM,IAAA,gBAAG,EAAA,6BAA6B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACzD,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAA,gBAAG,EAAA;;;;;;;;;KASR,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,4BAA4B,CAAC,MAAc;IAC/D,IAAI,CAAC;QACH,MAAM,IAAA,gBAAG,EAAA,4BAA4B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACxD,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAA,gBAAG,EAAA;;;;;;;;;;;;;;;;;;KAkBR,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC;AACH,CAAC","sourcesContent":["import { type Kysely, sql } from '../isomorphic';\n\nexport async function ensureDigestFunction(kysely: Kysely) {\n  try {\n    await sql`select digest('', 'sha256')`.execute(kysely);\n  } catch {\n    await sql`\n      create or replace function digest(input text, algo text)\n      returns bytea\n      language sql\n      immutable\n      strict\n      as $$\n        select '\\\\x0000000000000000000000000000000000000000000000000000000000000000'::bytea;\n      $$;\n    `.execute(kysely);\n  }\n}\n\nexport async function ensureGenRandomBytesFunction(kysely: Kysely) {\n  try {\n    await sql`select gen_random_bytes(1)`.execute(kysely);\n  } catch {\n    await sql`\n      create or replace function gen_random_bytes(p_len integer)\n      returns bytea\n      language sql\n      stable\n      strict\n      as $$\n        select substring(\n          decode(\n            repeat(\n              md5(random()::text || now()::text),\n              ((p_len + 15) / 16)\n            ),\n            'hex'\n          )\n          from 1 for p_len\n        );\n      $$;\n    `.execute(kysely);\n  }\n}\n"]}
+{"version":3,"file":"digest.js","sourceRoot":"src/","sources":["local/digest.ts"],"names":[],"mappings":";;AAEA,oDAiBC;AAED,oEA0BC;AA/CD,8CAAyD;AAElD,KAAK,UAAU,oBAAoB,CACxC,MAAgC;IAEhC,IAAI,CAAC;QACH,MAAM,IAAA,gBAAG,EAAA,6BAA6B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACzD,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAA,gBAAG,EAAA;;;;;;;;;KASR,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,4BAA4B,CAChD,MAAgC;IAEhC,IAAI,CAAC;QACH,MAAM,IAAA,gBAAG,EAAA,4BAA4B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACxD,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAA,gBAAG,EAAA;;;;;;;;;;;;;;;;;;KAkBR,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC;AACH,CAAC","sourcesContent":["import { type DatabaseKysely, sql } from '../isomorphic';\n\nexport async function ensureDigestFunction<Database>(\n  kysely: DatabaseKysely<Database>\n) {\n  try {\n    await sql`select digest('', 'sha256')`.execute(kysely);\n  } catch {\n    await sql`\n      create or replace function digest(input text, algo text)\n      returns bytea\n      language sql\n      immutable\n      strict\n      as $$\n        select '\\\\x0000000000000000000000000000000000000000000000000000000000000000'::bytea;\n      $$;\n    `.execute(kysely);\n  }\n}\n\nexport async function ensureGenRandomBytesFunction<Database>(\n  kysely: DatabaseKysely<Database>\n) {\n  try {\n    await sql`select gen_random_bytes(1)`.execute(kysely);\n  } catch {\n    await sql`\n      create or replace function gen_random_bytes(p_len integer)\n      returns bytea\n      language sql\n      stable\n      strict\n      as $$\n        select substring(\n          decode(\n            repeat(\n              md5(random()::text || now()::text),\n              ((p_len + 15) / 16)\n            ),\n            'hex'\n          )\n          from 1 for p_len\n        );\n      $$;\n    `.execute(kysely);\n  }\n}\n"]}

package/build/main/local/generate.js

@@ -45,11 +45,15 @@ const pglite_1 = require("@electric-sql/pglite");
 const kysely_1 = require("kysely");
 const kysely_codegen_1 = __importDefault(require("kysely-codegen"));
 const kysely_pglite_dialect_1 = require("kysely-pglite-dialect");
+const migration_1 = require("../migration");
 const plugins_1 = require("../plugins");
 async function normalizeGeneratedTypes(root) {
     const outFile = path.join(root, 'src/v1.generated.ts');
     const current = await Bun.file(outFile).text();
-    const next = current.replace('export type Timestamp = ColumnType<Date, Date | string, Date | string>;', 'export type Timestamp = ColumnType<Date | string, Date | string, Date | string>;');
+    const next = current
+        .replace('export type Timestamp = ColumnType<Date, Date | string, Date | string>;', 'export type Timestamp = ColumnType<Date | string, Date | string, Date | string>;')
+        .replace('  managedTeamAccess: Generated<string>;\n', '  managedTeamAccess: Generated<"all" | "assigned">;\n')
+        .replace('  slug: string;\n', '  slug: Generated<string>;\n');
     if (next !== current) {
         await Bun.write(outFile, next);
     }
@@ -68,15 +72,16 @@ async function migrate(root, client) {
 async function migrateKysely(root, kysely) {
     console.log('init migrating kysely ....');
     const migrator = new kysely_1.Migrator({
+        allowUnorderedMigrations: true,
         db: kysely,
-        migrationTableSchema: 'core',
-        migrationTableName: 'kysely_migration',
-        migrationLockTableName: 'kysely_migration_lock',
+        migrationTableSchema: migration_1.appMigrationEpoch.tableSchema,
+        migrationTableName: migration_1.appMigrationEpoch.tableName,
+        migrationLockTableName: migration_1.appMigrationEpoch.lockTableName,
         provider: new kysely_1.FileMigrationProvider({
             fs: node_fs_1.promises,
             path,
             // This needs to be an absolute path.
-            migrationFolder: path.join(root, 'migrations'),
+            migrationFolder: path.join(root, migration_1.appMigrationEpoch.folder),
         }),
     });
     console.log('migrating kysely ....');
@@ -123,9 +128,6 @@ async function generateDump(root, dumpPath) {
             outFile: path.join(root, 'src/v1.generated.ts'),
             defaultSchemas: ['xxx'],
             camelCase: true,
-            typeMapping: {
-                date: 'string',
-            },
             dialect: kysely_codegen_1.default.getDialect('postgres'),
         })
             .catch(() => {