@vrplatform/kysely 1.2.28 → 1.2.30

package/build/module/local/index.js

@@ -0,0 +1,32 @@
+import { join } from 'node:path';
+import { PGlite } from '@electric-sql/pglite';
+import { CamelCasePlugin, Kysely } from 'kysely';
+import { KyselyPGlite } from 'kysely-pglite';
+export const root = join(__dirname, '../..');
+export const dumpPath = join(root, '.data.tar');
+export async function useLocalKysely() {
+    let isDestroyed = false;
+    const now = performance.now();
+    const dump = Bun.file(dumpPath);
+    const pglite = new PGlite({ loadDataDir: dump });
+    // warm up
+    await pglite.sql `SELECT 1`;
+    const { dialect } = new KyselyPGlite(pglite);
+    const kysely = new Kysely({
+        dialect,
+        plugins: [new CamelCasePlugin()],
+    });
+    const originalDestroy = kysely.destroy.bind(kysely);
+    kysely.destroy = async () => {
+        if (isDestroyed)
+            return;
+        isDestroyed = true;
+        const destroyTime = performance.now();
+        await originalDestroy().catch(() => undefined);
+        await pglite.close().catch(() => undefined);
+        console.log(`Database destroyed: ${Math.round(performance.now() - destroyTime)}ms`);
+    };
+    console.log(`Database initialized: ${Math.round(performance.now() - now)}ms`);
+    return kysely;
+}
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["local/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AACjD,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAG7C,MAAM,CAAC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;AAC7C,MAAM,CAAC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;AAEhD,MAAM,CAAC,KAAK,UAAU,cAAc;IAClC,IAAI,WAAW,GAAG,KAAK,CAAC;IACxB,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAE9B,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAEhC,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;IACjD,UAAU;IACV,MAAM,MAAM,CAAC,GAAG,CAAA,UAAU,CAAC;IAC3B,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IAC7C,MAAM,MAAM,GAAG,IAAI,MAAM,CAAK;QAC5B,OAAO;QACP,OAAO,EAAE,CAAC,IAAI,eAAe,EAAE,CAAC;KACjC,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,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACtC,MAAM,eAAe,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC/C,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC5C,OAAO,CAAC,GAAG,CACT,uBAAuB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,IAAI,CACvE,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC;IAE9E,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["import { join } from 'node:path';\nimport { PGlite } from '@electric-sql/pglite';\nimport { CamelCasePlugin, Kysely } from 'kysely';\nimport { KyselyPGlite } from 'kysely-pglite';\nimport type { DB } from '../v1.generated';\n\nexport const root = join(__dirname, '../..');\nexport const dumpPath = join(root, '.data.tar');\n\nexport async function useLocalKysely() {\n  let isDestroyed = false;\n  const now = performance.now();\n\n  const dump = Bun.file(dumpPath);\n\n  const pglite = new PGlite({ loadDataDir: dump });\n  // warm up\n  await pglite.sql`SELECT 1`;\n  const { dialect } = new KyselyPGlite(pglite);\n  const kysely = new Kysely<DB>({\n    dialect,\n    plugins: [new CamelCasePlugin()],\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().catch(() => undefined);\n    await pglite.close().catch(() => undefined);\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;\n}\n"]}

package/build/module/tsconfig.esm.tsbuildinfo

@@ -1 +1 @@
-{"root":["../../src/index.ts","../../src/local.ts","../../src/test.ts","../../src/v1.generated.ts"],"version":"5.9.3"}
+{"root":["../../src/index.ts","../../src/test.ts","../../src/v1.generated.ts","../../src/local/generate.ts","../../src/local/index.ts"],"version":"5.9.3"}

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.2.28",
+  "version": "1.2.30",
   "description": "",
   "main": "build/main/index.js",
   "module": "build/module/index.js",
@@ -13,8 +13,8 @@
       "require": "./build/main/index.js"
     },
     "./local": {
-      "import": "./build/module/local.js",
-      "require": "./build/main/local.js"
+      "import": "./build/module/local/index.js",
+      "require": "./build/main/local/index.js"
     },
     "./test": {
       "import": "./build/module/test.js",

package/src/index.ts

@@ -25,38 +25,70 @@ import type { DB } from './v1.generated';
 export type Transaction = T<DB>;
 export { sql } from 'kysely';
 
+type PgOpt = {
+  forceDisableSSL?: boolean;
+  retries?: false | number;
+};
+
 export type Kysely = K<DB>;
 export type Postgres = Sql;
+/**
+ * @deprecated This method is deprecated in favor of the useAsyncKysely method.
+ */
 export function useKysely(postgres: Sql, options: { log?: boolean } = {}) {
-  return new K<DB>({
+  console.debug('Creating Kysely instance');
+  const kysely = new K<DB>({
     dialect: new PostgresJSDialect({
       postgres,
     }),
     plugins: [new CamelCasePlugin()],
     log: options?.log ? ['query', 'error'] : undefined,
   });
+
+  const destroy = kysely.destroy;
+  kysely.destroy = async () => {
+    console.debug('Closing Kysely connection');
+    return destroy();
+  };
+
+  return kysely;
 }
 
-export async function useAsyncPostgres(
-  connectionString: string,
-  ssl?: boolean,
-  retries = 5,
-  delayMs = 2500
-) {
+const _connections: postgres.Sql[] = [];
+export async function useAsyncPostgres(connectionString: string, opt?: PgOpt) {
+  const delayMs = 2500;
+  const retries = opt?.retries === false ? 0 : opt?.retries || 3;
+  const ssl = opt?.forceDisableSSL !== true;
+
+  console.debug(
+    `Creating Postgres connection with ${retries} retries and ssl=${ssl}`
+  );
+  if (_connections.length > 0) {
+    console.warn(`We already have ${_connections.length} Postgres connections`);
+  }
+
   for (let attempt = 1; attempt <= retries; attempt++) {
     try {
       const sql = postgres(connectionString, {
         ssl: ssl ? 'require' : false,
-        connection: {
-          statement_timeout: 30_000,
-        },
+        // connect_timeout: 10
+        // idle_timeout: 10
+        // disable prepare statements since we use pgbouncer
         prepare: false,
+        connect_timeout: 10,
+        idle_timeout: 60,
       });
 
       await sql`select 1`;
+      try {
+        // connect 30s, idle 60s, max lifetime 45min, statement 60s, lock 10s, log_min_duration 20s, log_lock_waits off
+        await sql`set statement_timeout to '30s';`;
+        await sql`set lock_timeout to '10s';`;
+      } catch (e) {
+        console.warn('Failed to set statement timeout, continuing', e);
+      }
 
-      connections.push(sql);
-
+      _connections.push(sql);
       return sql;
     } catch (err) {
       if (attempt === retries) throw err;
@@ -70,27 +102,34 @@ export async function useAsyncPostgres(
   throw new Error('Unable to connect to Postgres');
 }
 
-export async function useAsyncKysely(connectionString: string, ssl?: boolean) {
-  const sql = await useAsyncPostgres(connectionString, ssl);
+export async function useAsyncKysely(connectionString: string, opt?: PgOpt) {
+  const sql = await useAsyncPostgres(connectionString, opt);
   return useKysely(sql);
 }
 
-const connections: postgres.Sql<any>[] = [];
-export function usePostgres(connectionString?: string, ssl?: boolean) {
+/**
+ * @deprecated This method is deprecated in favor of the useAsyncKysely method.
+ */
+export function usePostgres(
+  connectionString?: string,
+  opt?: Pick<PgOpt, 'forceDisableSSL'>
+) {
   if (!connectionString) throw new Error('No connection string');
+  const ssl = opt?.forceDisableSSL !== true;
   const pg = postgres(connectionString, {
     ssl: ssl ? 'require' : false,
   });
-  connections.push(pg);
+  _connections.push(pg);
   return pg;
 }
 
 export async function forceCloseAllPostgres(timeoutMs = 5000) {
-  return await Promise.all(
-    connections.map((pg) =>
+  await Promise.all(
+    _connections.map((pg) =>
       pg.end({ timeout: timeoutMs }).catch(() => undefined)
     )
   );
+  _connections.length = 0;
 }
 
 export class DatabaseError extends Error {

package/src/{local.ts → local/generate.ts}

@@ -11,10 +11,9 @@ import {
 } from 'kysely';
 import codegen from 'kysely-codegen';
 import { KyselyPGlite } from 'kysely-pglite';
-import type { DB } from './v1.generated';
+import type { DB } from '../v1.generated';
 
-const root = join(__dirname, '..');
-async function migrate(client: PGlite) {
+async function migrate(root: string, client: PGlite) {
   console.log('migrating pglite ....');
   const glob = new Bun.Glob('initial/*.sql');
   for await (const migration of glob.scan(root)) {
@@ -22,7 +21,7 @@ async function migrate(client: PGlite) {
   }
 }
 
-async function migrateKysely(kysely: Kysely<any>) {
+async function migrateKysely(root: string, kysely: Kysely<any>) {
   console.log('init migrating kysely ....');
   const migrator = new Migrator({
     db: kysely,
@@ -58,13 +57,12 @@ async function migrateKysely(kysely: Kysely<any>) {
   }
 }
 
-const dumpPath = join(root, '.data.tar');
-const dumpFile = (async () => {
+async function generateDump(root: string, dumpPath: string) {
   const dataExists = await exists(dumpPath).catch(() => false);
   if (!dataExists) {
     console.log('Creating initial dump');
     const main = new PGlite();
-    await migrate(main).catch(async (err) => {
+    await migrate(root, main).catch(async (err) => {
       console.error('failed to migrate initial dump');
       await Bun.write(
         '../logs/migrate-initial-dump.json',
@@ -77,7 +75,7 @@ const dumpFile = (async () => {
       dialect,
       plugins: [new CamelCasePlugin()],
     });
-    await migrateKysely(kysely).catch(() => {
+    await migrateKysely(root, kysely).catch(() => {
       console.error('failed to migrate kysely');
       process.exit(1);
     });
@@ -104,13 +102,13 @@ const dumpFile = (async () => {
     return content;
   }
   return Bun.file(dumpPath);
-})();
+}
 
-export async function useLocalKysely() {
+export async function createLocalKysely(root: string, dumpPath: string) {
   let isDestroyed = false;
   const now = performance.now();
 
-  const dump = await dumpFile;
+  const dump = await generateDump(root, dumpPath);
 
   const pglite = new PGlite({ loadDataDir: dump });
   // warm up

package/src/local/index.ts

@@ -0,0 +1,40 @@
+import { join } from 'node:path';
+import { PGlite } from '@electric-sql/pglite';
+import { CamelCasePlugin, Kysely } from 'kysely';
+import { KyselyPGlite } from 'kysely-pglite';
+import type { DB } from '../v1.generated';
+
+export const root = join(__dirname, '../..');
+export const dumpPath = join(root, '.data.tar');
+
+export async function useLocalKysely() {
+  let isDestroyed = false;
+  const now = performance.now();
+
+  const dump = Bun.file(dumpPath);
+
+  const pglite = new PGlite({ loadDataDir: dump });
+  // warm up
+  await pglite.sql`SELECT 1`;
+  const { dialect } = new KyselyPGlite(pglite);
+  const kysely = new Kysely<DB>({
+    dialect,
+    plugins: [new CamelCasePlugin()],
+  });
+
+  const originalDestroy = kysely.destroy.bind(kysely);
+  kysely.destroy = async () => {
+    if (isDestroyed) return;
+    isDestroyed = true;
+    const destroyTime = performance.now();
+    await originalDestroy().catch(() => undefined);
+    await pglite.close().catch(() => undefined);
+    console.log(
+      `Database destroyed: ${Math.round(performance.now() - destroyTime)}ms`
+    );
+  };
+
+  console.log(`Database initialized: ${Math.round(performance.now() - now)}ms`);
+
+  return kysely;
+}

package/build/main/local.d.ts

@@ -1,3 +0,0 @@
-import { Kysely } from 'kysely';
-import type { DB } from './v1.generated';
-export declare function useLocalKysely(): Promise<Kysely<DB>>;

package/build/main/local.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"local.js","sourceRoot":"src/","sources":["local.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4GA,wCA8BC;AA1ID,qCAAyC;AACzC,+CAA0C;AAC1C,gDAAkC;AAClC,yCAAiC;AACjC,iDAA8C;AAC9C,mCAKgB;AAChB,oEAAqC;AACrC,iDAA6C;AAG7C,MAAM,IAAI,GAAG,IAAA,gBAAI,EAAC,SAAS,EAAE,IAAI,CAAC,CAAC;AACnC,KAAK,UAAU,OAAO,CAAC,MAAc;IACnC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC3C,IAAI,KAAK,EAAE,MAAM,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC9C,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,IAAA,gBAAI,EAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAClE,CAAC;AACH,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,MAAmB;IAC9C,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;IAC1C,MAAM,QAAQ,GAAG,IAAI,iBAAQ,CAAC;QAC5B,EAAE,EAAE,MAAM;QACV,oBAAoB,EAAE,MAAM;QAC5B,kBAAkB,EAAE,kBAAkB;QACtC,sBAAsB,EAAE,uBAAuB;QAC/C,QAAQ,EAAE,IAAI,8BAAqB,CAAC;YAClC,EAAE,EAAF,kBAAE;YACF,IAAI;YACJ,qCAAqC;YACrC,eAAe,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC;SAC/C,CAAC;KACH,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,MAAM,QAAQ,CAAC,eAAe,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;QACrE,OAAO,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACnC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,KAAK,MAAM,EAAE,IAAI,OAAO,IAAI,EAAE,EAAE,CAAC;QAC/B,IAAI,EAAE,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,aAAa,6BAA6B,CAAC,CAAC;QAC3E,CAAC;aAAM,IAAI,EAAE,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;YACjC,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,CAAC,aAAa,GAAG,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IAED,IAAI,KAAK,EAAE,CAAC;QACV,OAAO,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACnC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,MAAM,QAAQ,GAAG,IAAA,gBAAI,EAAC,IAAI,EAAE,WAAW,CAAC,CAAC;AACzC,MAAM,QAAQ,GAAG,CAAC,KAAK,IAAI,EAAE;IAC3B,MAAM,UAAU,GAAG,MAAM,IAAA,iBAAM,EAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;IAC7D,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;QACrC,MAAM,IAAI,GAAG,IAAI,eAAM,EAAE,CAAC;QAC1B,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YACtC,OAAO,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;YAChD,MAAM,GAAG,CAAC,KAAK,CACb,mCAAmC,EACnC,IAAI,CAAC,SAAS,CAAC,GAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CACpC,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QACH,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,4BAAY,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,eAAM,CAAK;YAC5B,OAAO;YACP,OAAO,EAAE,CAAC,IAAI,wBAAe,EAAE,CAAC;SACjC,CAAC,CAAC;QACH,MAAM,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YACrC,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;YAC1C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QACH,MAAM,wBAAO;aACV,QAAQ,CAAC;YACR,EAAE,EAAE,MAAM;YACV,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,qBAAqB,CAAC;YAC/C,cAAc,EAAE,CAAC,KAAK,CAAC;YACvB,SAAS,EAAE,IAAI;YACf,OAAO,EAAE,wBAAO,CAAC,UAAU,CAAC,UAAU,CAAC;SACxC,CAAC;aACD,KAAK,CAAC,GAAG,EAAE;YACV,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC3C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QACL,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YACxD,OAAO,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACrC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QACH,MAAM,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YAC5C,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;YACtC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QACH,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,OAAO,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAC5B,CAAC,CAAC,EAAE,CAAC;AAEE,KAAK,UAAU,cAAc;IAClC,IAAI,WAAW,GAAG,KAAK,CAAC;IACxB,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAE9B,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC;IAE5B,MAAM,MAAM,GAAG,IAAI,eAAM,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;IACjD,UAAU;IACV,MAAM,MAAM,CAAC,GAAG,CAAA,UAAU,CAAC;IAC3B,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,4BAAY,CAAC,MAAM,CAAC,CAAC;IAC7C,MAAM,MAAM,GAAG,IAAI,eAAM,CAAK;QAC5B,OAAO;QACP,OAAO,EAAE,CAAC,IAAI,wBAAe,EAAE,CAAC;KACjC,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,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACtC,MAAM,eAAe,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC/C,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC5C,OAAO,CAAC,GAAG,CACT,uBAAuB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,IAAI,CACvE,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC;IAE9E,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["import { promises as fs } from 'node:fs';\nimport { exists } from 'node:fs/promises';\nimport * as path from 'node:path';\nimport { join } from 'node:path';\nimport { PGlite } from '@electric-sql/pglite';\nimport {\n  CamelCasePlugin,\n  FileMigrationProvider,\n  Kysely,\n  Migrator,\n} from 'kysely';\nimport codegen from 'kysely-codegen';\nimport { KyselyPGlite } from 'kysely-pglite';\nimport type { DB } from './v1.generated';\n\nconst root = join(__dirname, '..');\nasync function migrate(client: PGlite) {\n  console.log('migrating pglite ....');\n  const glob = new Bun.Glob('initial/*.sql');\n  for await (const migration of glob.scan(root)) {\n    await client.exec(await Bun.file(join(root, migration)).text());\n  }\n}\n\nasync function migrateKysely(kysely: Kysely<any>) {\n  console.log('init migrating kysely ....');\n  const migrator = new Migrator({\n    db: kysely,\n    migrationTableSchema: 'core',\n    migrationTableName: 'kysely_migration',\n    migrationLockTableName: 'kysely_migration_lock',\n    provider: new FileMigrationProvider({\n      fs,\n      path,\n      // This needs to be an absolute path.\n      migrationFolder: path.join(root, 'migrations'),\n    }),\n  });\n\n  console.log('migrating kysely ....');\n  const { error, results } = await migrator.migrateToLatest().catch(() => {\n    console.error('failed to migrate');\n    process.exit(1);\n  });\n\n  for (const it of results ?? []) {\n    if (it.status === 'Success') {\n      console.log(`migration \"${it.migrationName}\" was executed successfully`);\n    } else if (it.status === 'Error') {\n      console.error(`failed to execute migration \"${it.migrationName}\"`);\n    }\n  }\n\n  if (error) {\n    console.error('failed to migrate');\n    console.error(error);\n    process.exit(1);\n  }\n}\n\nconst dumpPath = join(root, '.data.tar');\nconst dumpFile = (async () => {\n  const dataExists = await exists(dumpPath).catch(() => false);\n  if (!dataExists) {\n    console.log('Creating initial dump');\n    const main = new PGlite();\n    await migrate(main).catch(async (err) => {\n      console.error('failed to migrate initial dump');\n      await Bun.write(\n        '../logs/migrate-initial-dump.json',\n        JSON.stringify(err as any, null, 2)\n      );\n      process.exit(1);\n    });\n    const { dialect } = new KyselyPGlite(main);\n    const kysely = new Kysely<DB>({\n      dialect,\n      plugins: [new CamelCasePlugin()],\n    });\n    await migrateKysely(kysely).catch(() => {\n      console.error('failed to migrate kysely');\n      process.exit(1);\n    });\n    await codegen\n      .generate({\n        db: kysely,\n        outFile: path.join(root, 'src/v1.generated.ts'),\n        defaultSchemas: ['xxx'],\n        camelCase: true,\n        dialect: codegen.getDialect('postgres'),\n      })\n      .catch(() => {\n        console.error('failed to generate kysely');\n        process.exit(1);\n      });\n    const content = await main.dumpDataDir('none').catch(() => {\n      console.error('failed to dump data');\n      process.exit(1);\n    });\n    await Bun.write(dumpPath, content).catch(() => {\n      console.error('failed to write dump');\n      process.exit(1);\n    });\n    return content;\n  }\n  return Bun.file(dumpPath);\n})();\n\nexport async function useLocalKysely() {\n  let isDestroyed = false;\n  const now = performance.now();\n\n  const dump = await dumpFile;\n\n  const pglite = new PGlite({ loadDataDir: dump });\n  // warm up\n  await pglite.sql`SELECT 1`;\n  const { dialect } = new KyselyPGlite(pglite);\n  const kysely = new Kysely<DB>({\n    dialect,\n    plugins: [new CamelCasePlugin()],\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().catch(() => undefined);\n    await pglite.close().catch(() => undefined);\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;\n}\n"]}