@vrplatform/kysely 1.2.28 → 1.2.30

package/build/main/index.d.ts

@@ -6,15 +6,25 @@ export * from './v1.generated';
 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 declare function useKysely(postgres: Sql, options?: {
     log?: boolean;
 }): K<DB>;
-export declare function useAsyncPostgres(connectionString: string, ssl?: boolean, retries?: number, delayMs?: number): Promise<postgres.Sql<{}>>;
-export declare function useAsyncKysely(connectionString: string, ssl?: boolean): Promise<K<DB>>;
-export declare function usePostgres(connectionString?: string, ssl?: boolean): postgres.Sql<{}>;
-export declare function forceCloseAllPostgres(timeoutMs?: number): Promise<(void | undefined)[]>;
+export declare function useAsyncPostgres(connectionString: string, opt?: PgOpt): Promise<postgres.Sql<{}>>;
+export declare function useAsyncKysely(connectionString: string, opt?: PgOpt): 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 forceCloseAllPostgres(timeoutMs?: number): Promise<void>;
 export declare class DatabaseError extends Error {
     constructor(message: string, cause: any);
 }

package/build/main/index.js

@@ -34,27 +34,55 @@ __exportStar(require("./v1.generated"), exports);
 const node_path_1 = require("node:path");
 var kysely_2 = require("kysely");
 Object.defineProperty(exports, "sql", { enumerable: true, get: function () { return kysely_2.sql; } });
+/**
+ * @deprecated This method is deprecated in favor of the useAsyncKysely method.
+ */
 function useKysely(postgres, options = {}) {
-    return new kysely_1.Kysely({
+    console.debug('Creating Kysely instance');
+    const kysely = new kysely_1.Kysely({
         dialect: new kysely_postgres_js_1.PostgresJSDialect({
             postgres,
         }),
         plugins: [new kysely_1.CamelCasePlugin()],
         log: options?.log ? ['query', 'error'] : undefined,
     });
+    const destroy = kysely.destroy;
+    kysely.destroy = async () => {
+        console.debug('Closing Kysely connection');
+        return destroy();
+    };
+    return kysely;
 }
-async function useAsyncPostgres(connectionString, ssl, retries = 5, delayMs = 2500) {
+const _connections = [];
+async function useAsyncPostgres(connectionString, opt) {
+    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 = (0, postgres_1.default)(connectionString, {
                 ssl: ssl ? 'require' : false,
-                connection: {
-                    statement_timeout: 30000,
-                },
+                // 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`;
-            connections.push(sql);
+            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);
             return sql;
         }
         catch (err) {
@@ -67,22 +95,26 @@ async function useAsyncPostgres(connectionString, ssl, retries = 5, delayMs = 25
     }
     throw new Error('Unable to connect to Postgres');
 }
-async function useAsyncKysely(connectionString, ssl) {
-    const sql = await useAsyncPostgres(connectionString, ssl);
+async function useAsyncKysely(connectionString, opt) {
+    const sql = await useAsyncPostgres(connectionString, opt);
     return useKysely(sql);
 }
-const connections = [];
-function usePostgres(connectionString, ssl) {
+/**
+ * @deprecated This method is deprecated in favor of the useAsyncKysely method.
+ */
+function usePostgres(connectionString, opt) {
     if (!connectionString)
         throw new Error('No connection string');
+    const ssl = opt?.forceDisableSSL !== true;
     const pg = (0, postgres_1.default)(connectionString, {
         ssl: ssl ? 'require' : false,
     });
-    connections.push(pg);
+    _connections.push(pg);
     return pg;
 }
 async function forceCloseAllPostgres(timeoutMs = 5000) {
-    return await Promise.all(connections.map((pg) => pg.end({ timeout: timeoutMs }).catch(() => undefined)));
+    await Promise.all(_connections.map((pg) => pg.end({ timeout: timeoutMs }).catch(() => undefined)));
+    _connections.length = 0;
 }
 class DatabaseError extends Error {
     constructor(message, cause) {

package/build/main/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;AA6BA,8BAQC;AAED,4CA+BC;AAED,wCAGC;AAGD,kCAOC;AAED,sDAMC;AAYD,8BA+BC;AAxID,4CAAkD;AAClD,mCAA6E;AAC7E,2DAAuD;AACvD,wDAA8C;AAe9C,oDAAwD;AAA/C,yGAAA,aAAa,OAAA;AACtB,iDAA+B;AAE/B,yCAA2C;AAI3C,iCAA6B;AAApB,6FAAA,GAAG,OAAA;AAIZ,SAAgB,SAAS,CAAC,QAAa,EAAE,UAA6B,EAAE;IACtE,OAAO,IAAI,eAAC,CAAK;QACf,OAAO,EAAE,IAAI,sCAAiB,CAAC;YAC7B,QAAQ;SACT,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,wBAAe,EAAE,CAAC;QAChC,GAAG,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;KACnD,CAAC,CAAC;AACL,CAAC;AAEM,KAAK,UAAU,gBAAgB,CACpC,gBAAwB,EACxB,GAAa,EACb,OAAO,GAAG,CAAC,EACX,OAAO,GAAG,IAAI;IAEd,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,UAAU,EAAE;oBACV,iBAAiB,EAAE,KAAM;iBAC1B;gBACD,OAAO,EAAE,KAAK;aACf,CAAC,CAAC;YAEH,MAAM,GAAG,CAAA,UAAU,CAAC;YAEpB,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAEtB,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,KAAK,CAAC,+BAA+B,CAAC,CAAC;AACnD,CAAC;AAEM,KAAK,UAAU,cAAc,CAAC,gBAAwB,EAAE,GAAa;IAC1E,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;AACxB,CAAC;AAED,MAAM,WAAW,GAAwB,EAAE,CAAC;AAC5C,SAAgB,WAAW,CAAC,gBAAyB,EAAE,GAAa;IAClE,IAAI,CAAC,gBAAgB;QAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC/D,MAAM,EAAE,GAAG,IAAA,kBAAQ,EAAC,gBAAgB,EAAE;QACpC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;KAC7B,CAAC,CAAC;IACH,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACrB,OAAO,EAAE,CAAC;AACZ,CAAC;AAEM,KAAK,UAAU,qBAAqB,CAAC,SAAS,GAAG,IAAI;IAC1D,OAAO,MAAM,OAAO,CAAC,GAAG,CACtB,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CACrB,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CACtD,CACF,CAAC;AACJ,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;AAED,MAAM,OAAO,GACX,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAA,gBAAI,EAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AAE5D,KAAK,UAAU,SAAS,CAAI,CAAa;IAC9C,uDAAuD;IACvD,IAAI,CAAC,OAAO;QAAE,OAAO,MAAM,CAAC,CAAC;IAE7B,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;IAC/D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;IAC7D,MAAM,SAAS,GACb,KAAK,EAAE,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,OAAO;QACjD,CAAC,CAAC,IAAA,oBAAQ,EAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,SAAS,CAAC;IAChB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC;QACvB,OAAO,CAAC,GAAG,CACT,IAAI,SAAS,qBAAqB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,IAAI,CAC1E,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,IAAI,KAAK;YAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;;YAC5C,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;QAC3B,mBAAmB;QACnB,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;QAC1B,MAAM,UAAU,GAAG,IAAA,sBAAc,EAAC,CAAC,CAAC,CAAC;QACrC,OAAO,UAAU,CAAC,KAAK,CAAC;QACxB,KAAK,CAAC,KAAK,GAAG,UAAU,CAAC;QACzB,OAAO,CAAC,KAAK,CAAC,qBAAqB,SAAS,EAAE,EAAE,IAAA,sBAAc,EAAC,KAAK,CAAC,CAAC,CAAC;QACvE,OAAO,CAAC,GAAG,CACT,IAAI,SAAS,qBAAqB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,IAAI,CAC1E,CAAC;QACF,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC","sourcesContent":["import { serializeError } from '@finalytic/utils';\nimport { CamelCasePlugin, Kysely as K, type Transaction as T } from 'kysely';\nimport { PostgresJSDialect } from 'kysely-postgres-js';\nimport postgres, { type Sql } from 'postgres';\n\nexport type {\n  ColumnType,\n  Expression,\n  ExpressionBuilder,\n  ExpressionOrFactory,\n  Generated,\n  Insertable,\n  JSONColumnType,\n  OperandExpression,\n  Selectable,\n  SqlBool,\n  Updateable,\n} from 'kysely';\nexport { jsonArrayFrom } from 'kysely/helpers/postgres';\nexport * from './v1.generated';\n\nimport { join, relative } from 'node:path';\nimport type { DB } from './v1.generated';\n\nexport type Transaction = T<DB>;\nexport { sql } from 'kysely';\n\nexport type Kysely = K<DB>;\nexport type Postgres = Sql;\nexport function useKysely(postgres: Sql, options: { log?: boolean } = {}) {\n  return new K<DB>({\n    dialect: new PostgresJSDialect({\n      postgres,\n    }),\n    plugins: [new CamelCasePlugin()],\n    log: options?.log ? ['query', 'error'] : undefined,\n  });\n}\n\nexport async function useAsyncPostgres(\n  connectionString: string,\n  ssl?: boolean,\n  retries = 5,\n  delayMs = 2500\n) {\n  for (let attempt = 1; attempt <= retries; attempt++) {\n    try {\n      const sql = postgres(connectionString, {\n        ssl: ssl ? 'require' : false,\n        connection: {\n          statement_timeout: 30_000,\n        },\n        prepare: false,\n      });\n\n      await sql`select 1`;\n\n      connections.push(sql);\n\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 Error('Unable to connect to Postgres');\n}\n\nexport async function useAsyncKysely(connectionString: string, ssl?: boolean) {\n  const sql = await useAsyncPostgres(connectionString, ssl);\n  return useKysely(sql);\n}\n\nconst connections: postgres.Sql<any>[] = [];\nexport function usePostgres(connectionString?: string, ssl?: boolean) {\n  if (!connectionString) throw new Error('No connection string');\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  return await Promise.all(\n    connections.map((pg) =>\n      pg.end({ timeout: timeoutMs }).catch(() => undefined)\n    )\n  );\n}\n\nexport class DatabaseError extends Error {\n  constructor(message: string, cause: any) {\n    super(message);\n    this.cause = cause;\n  }\n}\n\nconst dirname =\n  typeof __dirname === 'string' ? join(__dirname, '../../..') : '';\n\nexport async function withStack<T>(a: Promise<T>) {\n  // skip if we can't get the stack, probably in a worker\n  if (!dirname) return await a;\n\n  const now = performance.now();\n  const error = new DatabaseError('Error in execute', undefined);\n  const stack = error.stack?.split('\\n').slice(3) || undefined;\n  const firstLine =\n    stack?.length && stack[0]?.includes('(') && dirname\n      ? relative(dirname, stack[0].split('(')[1].split(')')[0])\n      : undefined;\n  try {\n    const result = await a;\n    console.log(\n      `[${firstLine}] Execution time: ${Math.round(performance.now() - now)}ms`\n    );\n    return result;\n  } catch (e: any) {\n    if (stack) error.stack = stack.join('\\n').trim();\n    else error.stack = e.stack;\n    // error.cause = e;\n    error.message = e.message;\n    const serialized = serializeError(e);\n    delete serialized.stack;\n    error.cause = serialized;\n    console.error(`Error in execute: ${firstLine}`, serializeError(error));\n    console.log(\n      `[${firstLine}] Execution time: ${Math.round(performance.now() - now)}ms`\n    );\n    throw error;\n  }\n}\n"]}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;AAqCA,8BAiBC;AAGD,4CA6CC;AAED,wCAGC;AAKD,kCAWC;AAED,sDAOC;AAYD,8BA+BC;AA/KD,4CAAkD;AAClD,mCAA6E;AAC7E,2DAAuD;AACvD,wDAA8C;AAe9C,oDAAwD;AAA/C,yGAAA,aAAa,OAAA;AACtB,iDAA+B;AAE/B,yCAA2C;AAI3C,iCAA6B;AAApB,6FAAA,GAAG,OAAA;AASZ;;GAEG;AACH,SAAgB,SAAS,CAAC,QAAa,EAAE,UAA6B,EAAE;IACtE,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAI,eAAC,CAAK;QACvB,OAAO,EAAE,IAAI,sCAAiB,CAAC;YAC7B,QAAQ;SACT,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,wBAAe,EAAE,CAAC;QAChC,GAAG,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;KACnD,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;IAC/B,MAAM,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC3C,OAAO,OAAO,EAAE,CAAC;IACnB,CAAC,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,YAAY,GAAmB,EAAE,CAAC;AACjC,KAAK,UAAU,gBAAgB,CAAC,gBAAwB,EAAE,GAAW;IAC1E,MAAM,OAAO,GAAG,IAAI,CAAC;IACrB,MAAM,OAAO,GAAG,GAAG,EAAE,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,OAAO,IAAI,CAAC,CAAC;IAC/D,MAAM,GAAG,GAAG,GAAG,EAAE,eAAe,KAAK,IAAI,CAAC;IAE1C,OAAO,CAAC,KAAK,CACX,qCAAqC,OAAO,oBAAoB,GAAG,EAAE,CACtE,CAAC;IACF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,OAAO,CAAC,IAAI,CAAC,mBAAmB,YAAY,CAAC,MAAM,uBAAuB,CAAC,CAAC;IAC9E,CAAC;IAED,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,oDAAoD;gBACpD,OAAO,EAAE,KAAK;gBACd,eAAe,EAAE,EAAE;gBACnB,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,MAAM,GAAG,CAAA,UAAU,CAAC;YACpB,IAAI,CAAC;gBACH,+GAA+G;gBAC/G,MAAM,GAAG,CAAA,iCAAiC,CAAC;gBAC3C,MAAM,GAAG,CAAA,4BAA4B,CAAC;YACxC,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO,CAAC,IAAI,CAAC,6CAA6C,EAAE,CAAC,CAAC,CAAC;YACjE,CAAC;YAED,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,KAAK,CAAC,+BAA+B,CAAC,CAAC;AACnD,CAAC;AAEM,KAAK,UAAU,cAAc,CAAC,gBAAwB,EAAE,GAAW;IACxE,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;AACxB,CAAC;AAED;;GAEG;AACH,SAAgB,WAAW,CACzB,gBAAyB,EACzB,GAAoC;IAEpC,IAAI,CAAC,gBAAgB;QAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC/D,MAAM,GAAG,GAAG,GAAG,EAAE,eAAe,KAAK,IAAI,CAAC;IAC1C,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,GAAG,CACf,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,CAAC;IACF,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC;AAC1B,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;AAED,MAAM,OAAO,GACX,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAA,gBAAI,EAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AAE5D,KAAK,UAAU,SAAS,CAAI,CAAa;IAC9C,uDAAuD;IACvD,IAAI,CAAC,OAAO;QAAE,OAAO,MAAM,CAAC,CAAC;IAE7B,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;IAC/D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;IAC7D,MAAM,SAAS,GACb,KAAK,EAAE,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,OAAO;QACjD,CAAC,CAAC,IAAA,oBAAQ,EAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,SAAS,CAAC;IAChB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC;QACvB,OAAO,CAAC,GAAG,CACT,IAAI,SAAS,qBAAqB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,IAAI,CAC1E,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,IAAI,KAAK;YAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;;YAC5C,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;QAC3B,mBAAmB;QACnB,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;QAC1B,MAAM,UAAU,GAAG,IAAA,sBAAc,EAAC,CAAC,CAAC,CAAC;QACrC,OAAO,UAAU,CAAC,KAAK,CAAC;QACxB,KAAK,CAAC,KAAK,GAAG,UAAU,CAAC;QACzB,OAAO,CAAC,KAAK,CAAC,qBAAqB,SAAS,EAAE,EAAE,IAAA,sBAAc,EAAC,KAAK,CAAC,CAAC,CAAC;QACvE,OAAO,CAAC,GAAG,CACT,IAAI,SAAS,qBAAqB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,IAAI,CAC1E,CAAC;QACF,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC","sourcesContent":["import { serializeError } from '@finalytic/utils';\nimport { CamelCasePlugin, Kysely as K, type Transaction as T } from 'kysely';\nimport { PostgresJSDialect } from 'kysely-postgres-js';\nimport postgres, { type Sql } from 'postgres';\n\nexport type {\n  ColumnType,\n  Expression,\n  ExpressionBuilder,\n  ExpressionOrFactory,\n  Generated,\n  Insertable,\n  JSONColumnType,\n  OperandExpression,\n  Selectable,\n  SqlBool,\n  Updateable,\n} from 'kysely';\nexport { jsonArrayFrom } from 'kysely/helpers/postgres';\nexport * from './v1.generated';\n\nimport { join, relative } from 'node:path';\nimport type { DB } from './v1.generated';\n\nexport type Transaction = T<DB>;\nexport { sql } from 'kysely';\n\ntype PgOpt = {\n  forceDisableSSL?: boolean;\n  retries?: false | number;\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: { log?: boolean } = {}) {\n  console.debug('Creating Kysely instance');\n  const kysely = new K<DB>({\n    dialect: new PostgresJSDialect({\n      postgres,\n    }),\n    plugins: [new CamelCasePlugin()],\n    log: options?.log ? ['query', 'error'] : undefined,\n  });\n\n  const destroy = kysely.destroy;\n  kysely.destroy = async () => {\n    console.debug('Closing Kysely connection');\n    return destroy();\n  };\n\n  return kysely;\n}\n\nconst _connections: postgres.Sql[] = [];\nexport async function useAsyncPostgres(connectionString: string, opt?: PgOpt) {\n  const delayMs = 2500;\n  const retries = opt?.retries === false ? 0 : opt?.retries || 3;\n  const ssl = opt?.forceDisableSSL !== true;\n\n  console.debug(\n    `Creating Postgres connection with ${retries} retries and ssl=${ssl}`\n  );\n  if (_connections.length > 0) {\n    console.warn(`We already have ${_connections.length} Postgres connections`);\n  }\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        // disable prepare statements since we use pgbouncer\n        prepare: false,\n        connect_timeout: 10,\n        idle_timeout: 60,\n      });\n\n      await sql`select 1`;\n      try {\n        // connect 30s, idle 60s, max lifetime 45min, statement 60s, lock 10s, log_min_duration 20s, log_lock_waits off\n        await sql`set statement_timeout to '30s';`;\n        await sql`set lock_timeout to '10s';`;\n      } catch (e) {\n        console.warn('Failed to set statement timeout, continuing', e);\n      }\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 Error('Unable to connect to Postgres');\n}\n\nexport async function useAsyncKysely(connectionString: string, opt?: PgOpt) {\n  const sql = await useAsyncPostgres(connectionString, opt);\n  return useKysely(sql);\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 Error('No connection string');\n  const ssl = opt?.forceDisableSSL !== true;\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.all(\n    _connections.map((pg) =>\n      pg.end({ timeout: timeoutMs }).catch(() => undefined)\n    )\n  );\n  _connections.length = 0;\n}\n\nexport class DatabaseError extends Error {\n  constructor(message: string, cause: any) {\n    super(message);\n    this.cause = cause;\n  }\n}\n\nconst dirname =\n  typeof __dirname === 'string' ? join(__dirname, '../../..') : '';\n\nexport async function withStack<T>(a: Promise<T>) {\n  // skip if we can't get the stack, probably in a worker\n  if (!dirname) return await a;\n\n  const now = performance.now();\n  const error = new DatabaseError('Error in execute', undefined);\n  const stack = error.stack?.split('\\n').slice(3) || undefined;\n  const firstLine =\n    stack?.length && stack[0]?.includes('(') && dirname\n      ? relative(dirname, stack[0].split('(')[1].split(')')[0])\n      : undefined;\n  try {\n    const result = await a;\n    console.log(\n      `[${firstLine}] Execution time: ${Math.round(performance.now() - now)}ms`\n    );\n    return result;\n  } catch (e: any) {\n    if (stack) error.stack = stack.join('\\n').trim();\n    else error.stack = e.stack;\n    // error.cause = e;\n    error.message = e.message;\n    const serialized = serializeError(e);\n    delete serialized.stack;\n    error.cause = serialized;\n    console.error(`Error in execute: ${firstLine}`, serializeError(error));\n    console.log(\n      `[${firstLine}] Execution time: ${Math.round(performance.now() - now)}ms`\n    );\n    throw error;\n  }\n}\n"]}

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

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

package/build/main/{local.js → local/generate.js}

@@ -36,7 +36,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.useLocalKysely = useLocalKysely;
+exports.createLocalKysely = createLocalKysely;
 const node_fs_1 = require("node:fs");
 const promises_1 = require("node:fs/promises");
 const path = __importStar(require("node:path"));
@@ -45,15 +45,14 @@ const pglite_1 = require("@electric-sql/pglite");
 const kysely_1 = require("kysely");
 const kysely_codegen_1 = __importDefault(require("kysely-codegen"));
 const kysely_pglite_1 = require("kysely-pglite");
-const root = (0, node_path_1.join)(__dirname, '..');
-async function migrate(client) {
+async function migrate(root, client) {
     console.log('migrating pglite ....');
     const glob = new Bun.Glob('initial/*.sql');
     for await (const migration of glob.scan(root)) {
         await client.exec(await Bun.file((0, node_path_1.join)(root, migration)).text());
     }
 }
-async function migrateKysely(kysely) {
+async function migrateKysely(root, kysely) {
     console.log('init migrating kysely ....');
     const migrator = new kysely_1.Migrator({
         db: kysely,
@@ -86,13 +85,12 @@ async function migrateKysely(kysely) {
         process.exit(1);
     }
 }
-const dumpPath = (0, node_path_1.join)(root, '.data.tar');
-const dumpFile = (async () => {
+async function generateDump(root, dumpPath) {
     const dataExists = await (0, promises_1.exists)(dumpPath).catch(() => false);
     if (!dataExists) {
         console.log('Creating initial dump');
         const main = new pglite_1.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', JSON.stringify(err, null, 2));
             process.exit(1);
@@ -102,7 +100,7 @@ const dumpFile = (async () => {
             dialect,
             plugins: [new kysely_1.CamelCasePlugin()],
         });
-        await migrateKysely(kysely).catch(() => {
+        await migrateKysely(root, kysely).catch(() => {
             console.error('failed to migrate kysely');
             process.exit(1);
         });
@@ -129,11 +127,11 @@ const dumpFile = (async () => {
         return content;
     }
     return Bun.file(dumpPath);
-})();
-async function useLocalKysely() {
+}
+async function createLocalKysely(root, dumpPath) {
     let isDestroyed = false;
     const now = performance.now();
-    const dump = await dumpFile;
+    const dump = await generateDump(root, dumpPath);
     const pglite = new pglite_1.PGlite({ loadDataDir: dump });
     // warm up
     await pglite.sql `SELECT 1`;
@@ -155,4 +153,4 @@ async function useLocalKysely() {
     console.log(`Database initialized: ${Math.round(performance.now() - now)}ms`);
     return kysely;
 }
-//# sourceMappingURL=local.js.map
+//# sourceMappingURL=generate.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"generate.js","sourceRoot":"src/","sources":["local/generate.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA0GA,8CA8BC;AAxID,qCAAyC;AACzC,+CAA0C;AAC1C,gDAAkC;AAClC,yCAAiC;AACjC,iDAA8C;AAC9C,mCAKgB;AAChB,oEAAqC;AACrC,iDAA6C;AAG7C,KAAK,UAAU,OAAO,CAAC,IAAY,EAAE,MAAc;IACjD,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,IAAY,EAAE,MAAmB;IAC5D,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,KAAK,UAAU,YAAY,CAAC,IAAY,EAAE,QAAgB;IACxD,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,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YAC5C,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,IAAI,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YAC3C,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;AAEM,KAAK,UAAU,iBAAiB,CAAC,IAAY,EAAE,QAAgB;IACpE,IAAI,WAAW,GAAG,KAAK,CAAC;IACxB,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAE9B,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAEhD,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\nasync function migrate(root: string, 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(root: string, 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\nasync function generateDump(root: string, dumpPath: string) {\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(root, 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(root, 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 createLocalKysely(root: string, dumpPath: string) {\n  let isDestroyed = false;\n  const now = performance.now();\n\n  const dump = await generateDump(root, 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/main/local/index.d.ts

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

package/build/main/local/index.js

@@ -0,0 +1,36 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.dumpPath = exports.root = void 0;
+exports.useLocalKysely = useLocalKysely;
+const node_path_1 = require("node:path");
+const pglite_1 = require("@electric-sql/pglite");
+const kysely_1 = require("kysely");
+const kysely_pglite_1 = require("kysely-pglite");
+exports.root = (0, node_path_1.join)(__dirname, '../..');
+exports.dumpPath = (0, node_path_1.join)(exports.root, '.data.tar');
+async function useLocalKysely() {
+    let isDestroyed = false;
+    const now = performance.now();
+    const dump = Bun.file(exports.dumpPath);
+    const pglite = new pglite_1.PGlite({ loadDataDir: dump });
+    // warm up
+    await pglite.sql `SELECT 1`;
+    const { dialect } = new kysely_pglite_1.KyselyPGlite(pglite);
+    const kysely = new kysely_1.Kysely({
+        dialect,
+        plugins: [new kysely_1.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/main/local/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["local/index.ts"],"names":[],"mappings":";;;AASA,wCA8BC;AAvCD,yCAAiC;AACjC,iDAA8C;AAC9C,mCAAiD;AACjD,iDAA6C;AAGhC,QAAA,IAAI,GAAG,IAAA,gBAAI,EAAC,SAAS,EAAE,OAAO,CAAC,CAAC;AAChC,QAAA,QAAQ,GAAG,IAAA,gBAAI,EAAC,YAAI,EAAE,WAAW,CAAC,CAAC;AAEzC,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,gBAAQ,CAAC,CAAC;IAEhC,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 { 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/main/tsconfig.main.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/build/module/index.d.ts

@@ -6,15 +6,25 @@ export * from './v1.generated';
 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 declare function useKysely(postgres: Sql, options?: {
     log?: boolean;
 }): K<DB>;
-export declare function useAsyncPostgres(connectionString: string, ssl?: boolean, retries?: number, delayMs?: number): Promise<postgres.Sql<{}>>;
-export declare function useAsyncKysely(connectionString: string, ssl?: boolean): Promise<K<DB>>;
-export declare function usePostgres(connectionString?: string, ssl?: boolean): postgres.Sql<{}>;
-export declare function forceCloseAllPostgres(timeoutMs?: number): Promise<(void | undefined)[]>;
+export declare function useAsyncPostgres(connectionString: string, opt?: PgOpt): Promise<postgres.Sql<{}>>;
+export declare function useAsyncKysely(connectionString: string, opt?: PgOpt): 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 forceCloseAllPostgres(timeoutMs?: number): Promise<void>;
 export declare class DatabaseError extends Error {
     constructor(message: string, cause: any);
 }

package/build/module/index.js

@@ -6,27 +6,55 @@ export { jsonArrayFrom } from 'kysely/helpers/postgres';
 export * from './v1.generated';
 import { join, relative } from 'node:path';
 export { sql } from 'kysely';
+/**
+ * @deprecated This method is deprecated in favor of the useAsyncKysely method.
+ */
 export function useKysely(postgres, options = {}) {
-    return new K({
+    console.debug('Creating Kysely instance');
+    const kysely = new K({
         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, ssl, retries = 5, delayMs = 2500) {
+const _connections = [];
+export async function useAsyncPostgres(connectionString, opt) {
+    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`;
-            connections.push(sql);
+            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);
             return sql;
         }
         catch (err) {
@@ -39,22 +67,26 @@ export async function useAsyncPostgres(connectionString, ssl, retries = 5, delay
     }
     throw new Error('Unable to connect to Postgres');
 }
-export async function useAsyncKysely(connectionString, ssl) {
-    const sql = await useAsyncPostgres(connectionString, ssl);
+export async function useAsyncKysely(connectionString, opt) {
+    const sql = await useAsyncPostgres(connectionString, opt);
     return useKysely(sql);
 }
-const connections = [];
-export function usePostgres(connectionString, ssl) {
+/**
+ * @deprecated This method is deprecated in favor of the useAsyncKysely method.
+ */
+export function usePostgres(connectionString, opt) {
     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) => pg.end({ timeout: timeoutMs }).catch(() => undefined)));
+    await Promise.all(_connections.map((pg) => pg.end({ timeout: timeoutMs }).catch(() => undefined)));
+    _connections.length = 0;
 }
 export class DatabaseError extends Error {
     constructor(message, cause) {

package/build/module/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC,EAAyB,MAAM,QAAQ,CAAC;AAC7E,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,QAAsB,MAAM,UAAU,CAAC;AAe9C,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,cAAc,gBAAgB,CAAC;AAE/B,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAI3C,OAAO,EAAE,GAAG,EAAE,MAAM,QAAQ,CAAC;AAI7B,MAAM,UAAU,SAAS,CAAC,QAAa,EAAE,UAA6B,EAAE;IACtE,OAAO,IAAI,CAAC,CAAK;QACf,OAAO,EAAE,IAAI,iBAAiB,CAAC;YAC7B,QAAQ;SACT,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,eAAe,EAAE,CAAC;QAChC,GAAG,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;KACnD,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,gBAAwB,EACxB,GAAa,EACb,OAAO,GAAG,CAAC,EACX,OAAO,GAAG,IAAI;IAEd,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;QACpD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,QAAQ,CAAC,gBAAgB,EAAE;gBACrC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;gBAC5B,UAAU,EAAE;oBACV,iBAAiB,EAAE,MAAM;iBAC1B;gBACD,OAAO,EAAE,KAAK;aACf,CAAC,CAAC;YAEH,MAAM,GAAG,CAAA,UAAU,CAAC;YAEpB,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAEtB,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,KAAK,CAAC,+BAA+B,CAAC,CAAC;AACnD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,gBAAwB,EAAE,GAAa;IAC1E,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;AACxB,CAAC;AAED,MAAM,WAAW,GAAwB,EAAE,CAAC;AAC5C,MAAM,UAAU,WAAW,CAAC,gBAAyB,EAAE,GAAa;IAClE,IAAI,CAAC,gBAAgB;QAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC/D,MAAM,EAAE,GAAG,QAAQ,CAAC,gBAAgB,EAAE;QACpC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;KAC7B,CAAC,CAAC;IACH,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACrB,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,SAAS,GAAG,IAAI;IAC1D,OAAO,MAAM,OAAO,CAAC,GAAG,CACtB,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CACrB,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CACtD,CACF,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,aAAc,SAAQ,KAAK;IACtC,YAAY,OAAe,EAAE,KAAU;QACrC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AAED,MAAM,OAAO,GACX,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AAEnE,MAAM,CAAC,KAAK,UAAU,SAAS,CAAI,CAAa;IAC9C,uDAAuD;IACvD,IAAI,CAAC,OAAO;QAAE,OAAO,MAAM,CAAC,CAAC;IAE7B,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;IAC/D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;IAC7D,MAAM,SAAS,GACb,KAAK,EAAE,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,OAAO;QACjD,CAAC,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,SAAS,CAAC;IAChB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC;QACvB,OAAO,CAAC,GAAG,CACT,IAAI,SAAS,qBAAqB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,IAAI,CAC1E,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,IAAI,KAAK;YAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;;YAC5C,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;QAC3B,mBAAmB;QACnB,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;QAC1B,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;QACrC,OAAO,UAAU,CAAC,KAAK,CAAC;QACxB,KAAK,CAAC,KAAK,GAAG,UAAU,CAAC;QACzB,OAAO,CAAC,KAAK,CAAC,qBAAqB,SAAS,EAAE,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;QACvE,OAAO,CAAC,GAAG,CACT,IAAI,SAAS,qBAAqB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,IAAI,CAC1E,CAAC;QACF,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC","sourcesContent":["import { serializeError } from '@finalytic/utils';\nimport { CamelCasePlugin, Kysely as K, type Transaction as T } from 'kysely';\nimport { PostgresJSDialect } from 'kysely-postgres-js';\nimport postgres, { type Sql } from 'postgres';\n\nexport type {\n  ColumnType,\n  Expression,\n  ExpressionBuilder,\n  ExpressionOrFactory,\n  Generated,\n  Insertable,\n  JSONColumnType,\n  OperandExpression,\n  Selectable,\n  SqlBool,\n  Updateable,\n} from 'kysely';\nexport { jsonArrayFrom } from 'kysely/helpers/postgres';\nexport * from './v1.generated';\n\nimport { join, relative } from 'node:path';\nimport type { DB } from './v1.generated';\n\nexport type Transaction = T<DB>;\nexport { sql } from 'kysely';\n\nexport type Kysely = K<DB>;\nexport type Postgres = Sql;\nexport function useKysely(postgres: Sql, options: { log?: boolean } = {}) {\n  return new K<DB>({\n    dialect: new PostgresJSDialect({\n      postgres,\n    }),\n    plugins: [new CamelCasePlugin()],\n    log: options?.log ? ['query', 'error'] : undefined,\n  });\n}\n\nexport async function useAsyncPostgres(\n  connectionString: string,\n  ssl?: boolean,\n  retries = 5,\n  delayMs = 2500\n) {\n  for (let attempt = 1; attempt <= retries; attempt++) {\n    try {\n      const sql = postgres(connectionString, {\n        ssl: ssl ? 'require' : false,\n        connection: {\n          statement_timeout: 30_000,\n        },\n        prepare: false,\n      });\n\n      await sql`select 1`;\n\n      connections.push(sql);\n\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 Error('Unable to connect to Postgres');\n}\n\nexport async function useAsyncKysely(connectionString: string, ssl?: boolean) {\n  const sql = await useAsyncPostgres(connectionString, ssl);\n  return useKysely(sql);\n}\n\nconst connections: postgres.Sql<any>[] = [];\nexport function usePostgres(connectionString?: string, ssl?: boolean) {\n  if (!connectionString) throw new Error('No connection string');\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  return await Promise.all(\n    connections.map((pg) =>\n      pg.end({ timeout: timeoutMs }).catch(() => undefined)\n    )\n  );\n}\n\nexport class DatabaseError extends Error {\n  constructor(message: string, cause: any) {\n    super(message);\n    this.cause = cause;\n  }\n}\n\nconst dirname =\n  typeof __dirname === 'string' ? join(__dirname, '../../..') : '';\n\nexport async function withStack<T>(a: Promise<T>) {\n  // skip if we can't get the stack, probably in a worker\n  if (!dirname) return await a;\n\n  const now = performance.now();\n  const error = new DatabaseError('Error in execute', undefined);\n  const stack = error.stack?.split('\\n').slice(3) || undefined;\n  const firstLine =\n    stack?.length && stack[0]?.includes('(') && dirname\n      ? relative(dirname, stack[0].split('(')[1].split(')')[0])\n      : undefined;\n  try {\n    const result = await a;\n    console.log(\n      `[${firstLine}] Execution time: ${Math.round(performance.now() - now)}ms`\n    );\n    return result;\n  } catch (e: any) {\n    if (stack) error.stack = stack.join('\\n').trim();\n    else error.stack = e.stack;\n    // error.cause = e;\n    error.message = e.message;\n    const serialized = serializeError(e);\n    delete serialized.stack;\n    error.cause = serialized;\n    console.error(`Error in execute: ${firstLine}`, serializeError(error));\n    console.log(\n      `[${firstLine}] Execution time: ${Math.round(performance.now() - now)}ms`\n    );\n    throw error;\n  }\n}\n"]}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC,EAAyB,MAAM,QAAQ,CAAC;AAC7E,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,QAAsB,MAAM,UAAU,CAAC;AAe9C,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,cAAc,gBAAgB,CAAC;AAE/B,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAI3C,OAAO,EAAE,GAAG,EAAE,MAAM,QAAQ,CAAC;AAS7B;;GAEG;AACH,MAAM,UAAU,SAAS,CAAC,QAAa,EAAE,UAA6B,EAAE;IACtE,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,CAAK;QACvB,OAAO,EAAE,IAAI,iBAAiB,CAAC;YAC7B,QAAQ;SACT,CAAC;QACF,OAAO,EAAE,CAAC,IAAI,eAAe,EAAE,CAAC;QAChC,GAAG,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;KACnD,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;IAC/B,MAAM,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC3C,OAAO,OAAO,EAAE,CAAC;IACnB,CAAC,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,YAAY,GAAmB,EAAE,CAAC;AACxC,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,gBAAwB,EAAE,GAAW;IAC1E,MAAM,OAAO,GAAG,IAAI,CAAC;IACrB,MAAM,OAAO,GAAG,GAAG,EAAE,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,OAAO,IAAI,CAAC,CAAC;IAC/D,MAAM,GAAG,GAAG,GAAG,EAAE,eAAe,KAAK,IAAI,CAAC;IAE1C,OAAO,CAAC,KAAK,CACX,qCAAqC,OAAO,oBAAoB,GAAG,EAAE,CACtE,CAAC;IACF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,OAAO,CAAC,IAAI,CAAC,mBAAmB,YAAY,CAAC,MAAM,uBAAuB,CAAC,CAAC;IAC9E,CAAC;IAED,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;QACpD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,QAAQ,CAAC,gBAAgB,EAAE;gBACrC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;gBAC5B,sBAAsB;gBACtB,mBAAmB;gBACnB,oDAAoD;gBACpD,OAAO,EAAE,KAAK;gBACd,eAAe,EAAE,EAAE;gBACnB,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,MAAM,GAAG,CAAA,UAAU,CAAC;YACpB,IAAI,CAAC;gBACH,+GAA+G;gBAC/G,MAAM,GAAG,CAAA,iCAAiC,CAAC;gBAC3C,MAAM,GAAG,CAAA,4BAA4B,CAAC;YACxC,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO,CAAC,IAAI,CAAC,6CAA6C,EAAE,CAAC,CAAC,CAAC;YACjE,CAAC;YAED,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,KAAK,CAAC,+BAA+B,CAAC,CAAC;AACnD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,gBAAwB,EAAE,GAAW;IACxE,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC;AACxB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,WAAW,CACzB,gBAAyB,EACzB,GAAoC;IAEpC,IAAI,CAAC,gBAAgB;QAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC/D,MAAM,GAAG,GAAG,GAAG,EAAE,eAAe,KAAK,IAAI,CAAC;IAC1C,MAAM,EAAE,GAAG,QAAQ,CAAC,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;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,SAAS,GAAG,IAAI;IAC1D,MAAM,OAAO,CAAC,GAAG,CACf,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,CAAC;IACF,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,OAAO,aAAc,SAAQ,KAAK;IACtC,YAAY,OAAe,EAAE,KAAU;QACrC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AAED,MAAM,OAAO,GACX,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AAEnE,MAAM,CAAC,KAAK,UAAU,SAAS,CAAI,CAAa;IAC9C,uDAAuD;IACvD,IAAI,CAAC,OAAO;QAAE,OAAO,MAAM,CAAC,CAAC;IAE7B,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;IAC/D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;IAC7D,MAAM,SAAS,GACb,KAAK,EAAE,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,OAAO;QACjD,CAAC,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,SAAS,CAAC;IAChB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC;QACvB,OAAO,CAAC,GAAG,CACT,IAAI,SAAS,qBAAqB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,IAAI,CAC1E,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,IAAI,KAAK;YAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;;YAC5C,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;QAC3B,mBAAmB;QACnB,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;QAC1B,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;QACrC,OAAO,UAAU,CAAC,KAAK,CAAC;QACxB,KAAK,CAAC,KAAK,GAAG,UAAU,CAAC;QACzB,OAAO,CAAC,KAAK,CAAC,qBAAqB,SAAS,EAAE,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;QACvE,OAAO,CAAC,GAAG,CACT,IAAI,SAAS,qBAAqB,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,IAAI,CAC1E,CAAC;QACF,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC","sourcesContent":["import { serializeError } from '@finalytic/utils';\nimport { CamelCasePlugin, Kysely as K, type Transaction as T } from 'kysely';\nimport { PostgresJSDialect } from 'kysely-postgres-js';\nimport postgres, { type Sql } from 'postgres';\n\nexport type {\n  ColumnType,\n  Expression,\n  ExpressionBuilder,\n  ExpressionOrFactory,\n  Generated,\n  Insertable,\n  JSONColumnType,\n  OperandExpression,\n  Selectable,\n  SqlBool,\n  Updateable,\n} from 'kysely';\nexport { jsonArrayFrom } from 'kysely/helpers/postgres';\nexport * from './v1.generated';\n\nimport { join, relative } from 'node:path';\nimport type { DB } from './v1.generated';\n\nexport type Transaction = T<DB>;\nexport { sql } from 'kysely';\n\ntype PgOpt = {\n  forceDisableSSL?: boolean;\n  retries?: false | number;\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: { log?: boolean } = {}) {\n  console.debug('Creating Kysely instance');\n  const kysely = new K<DB>({\n    dialect: new PostgresJSDialect({\n      postgres,\n    }),\n    plugins: [new CamelCasePlugin()],\n    log: options?.log ? ['query', 'error'] : undefined,\n  });\n\n  const destroy = kysely.destroy;\n  kysely.destroy = async () => {\n    console.debug('Closing Kysely connection');\n    return destroy();\n  };\n\n  return kysely;\n}\n\nconst _connections: postgres.Sql[] = [];\nexport async function useAsyncPostgres(connectionString: string, opt?: PgOpt) {\n  const delayMs = 2500;\n  const retries = opt?.retries === false ? 0 : opt?.retries || 3;\n  const ssl = opt?.forceDisableSSL !== true;\n\n  console.debug(\n    `Creating Postgres connection with ${retries} retries and ssl=${ssl}`\n  );\n  if (_connections.length > 0) {\n    console.warn(`We already have ${_connections.length} Postgres connections`);\n  }\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        // disable prepare statements since we use pgbouncer\n        prepare: false,\n        connect_timeout: 10,\n        idle_timeout: 60,\n      });\n\n      await sql`select 1`;\n      try {\n        // connect 30s, idle 60s, max lifetime 45min, statement 60s, lock 10s, log_min_duration 20s, log_lock_waits off\n        await sql`set statement_timeout to '30s';`;\n        await sql`set lock_timeout to '10s';`;\n      } catch (e) {\n        console.warn('Failed to set statement timeout, continuing', e);\n      }\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 Error('Unable to connect to Postgres');\n}\n\nexport async function useAsyncKysely(connectionString: string, opt?: PgOpt) {\n  const sql = await useAsyncPostgres(connectionString, opt);\n  return useKysely(sql);\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 Error('No connection string');\n  const ssl = opt?.forceDisableSSL !== true;\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.all(\n    _connections.map((pg) =>\n      pg.end({ timeout: timeoutMs }).catch(() => undefined)\n    )\n  );\n  _connections.length = 0;\n}\n\nexport class DatabaseError extends Error {\n  constructor(message: string, cause: any) {\n    super(message);\n    this.cause = cause;\n  }\n}\n\nconst dirname =\n  typeof __dirname === 'string' ? join(__dirname, '../../..') : '';\n\nexport async function withStack<T>(a: Promise<T>) {\n  // skip if we can't get the stack, probably in a worker\n  if (!dirname) return await a;\n\n  const now = performance.now();\n  const error = new DatabaseError('Error in execute', undefined);\n  const stack = error.stack?.split('\\n').slice(3) || undefined;\n  const firstLine =\n    stack?.length && stack[0]?.includes('(') && dirname\n      ? relative(dirname, stack[0].split('(')[1].split(')')[0])\n      : undefined;\n  try {\n    const result = await a;\n    console.log(\n      `[${firstLine}] Execution time: ${Math.round(performance.now() - now)}ms`\n    );\n    return result;\n  } catch (e: any) {\n    if (stack) error.stack = stack.join('\\n').trim();\n    else error.stack = e.stack;\n    // error.cause = e;\n    error.message = e.message;\n    const serialized = serializeError(e);\n    delete serialized.stack;\n    error.cause = serialized;\n    console.error(`Error in execute: ${firstLine}`, serializeError(error));\n    console.log(\n      `[${firstLine}] Execution time: ${Math.round(performance.now() - now)}ms`\n    );\n    throw error;\n  }\n}\n"]}

package/build/module/local/generate.d.ts

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

package/build/module/{local.js → local/generate.js}

@@ -6,15 +6,14 @@ import { PGlite } from '@electric-sql/pglite';
 import { CamelCasePlugin, FileMigrationProvider, Kysely, Migrator, } from 'kysely';
 import codegen from 'kysely-codegen';
 import { KyselyPGlite } from 'kysely-pglite';
-const root = join(__dirname, '..');
-async function migrate(client) {
+async function migrate(root, client) {
     console.log('migrating pglite ....');
     const glob = new Bun.Glob('initial/*.sql');
     for await (const migration of glob.scan(root)) {
         await client.exec(await Bun.file(join(root, migration)).text());
     }
 }
-async function migrateKysely(kysely) {
+async function migrateKysely(root, kysely) {
     console.log('init migrating kysely ....');
     const migrator = new Migrator({
         db: kysely,
@@ -47,13 +46,12 @@ async function migrateKysely(kysely) {
         process.exit(1);
     }
 }
-const dumpPath = join(root, '.data.tar');
-const dumpFile = (async () => {
+async function generateDump(root, dumpPath) {
     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', JSON.stringify(err, null, 2));
             process.exit(1);
@@ -63,7 +61,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);
         });
@@ -90,11 +88,11 @@ const dumpFile = (async () => {
         return content;
     }
     return Bun.file(dumpPath);
-})();
-export async function useLocalKysely() {
+}
+export async function createLocalKysely(root, dumpPath) {
     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
     await pglite.sql `SELECT 1`;
@@ -116,4 +114,4 @@ export async function useLocalKysely() {
     console.log(`Database initialized: ${Math.round(performance.now() - now)}ms`);
     return kysely;
 }
-//# sourceMappingURL=local.js.map
+//# sourceMappingURL=generate.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"generate.js","sourceRoot":"src/","sources":["local/generate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAC9C,OAAO,EACL,eAAe,EACf,qBAAqB,EACrB,MAAM,EACN,QAAQ,GACT,MAAM,QAAQ,CAAC;AAChB,OAAO,OAAO,MAAM,gBAAgB,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAG7C,KAAK,UAAU,OAAO,CAAC,IAAY,EAAE,MAAc;IACjD,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,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAClE,CAAC;AACH,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,IAAY,EAAE,MAAmB;IAC5D,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;IAC1C,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;QAC5B,EAAE,EAAE,MAAM;QACV,oBAAoB,EAAE,MAAM;QAC5B,kBAAkB,EAAE,kBAAkB;QACtC,sBAAsB,EAAE,uBAAuB;QAC/C,QAAQ,EAAE,IAAI,qBAAqB,CAAC;YAClC,EAAE;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,KAAK,UAAU,YAAY,CAAC,IAAY,EAAE,QAAgB;IACxD,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,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,MAAM,EAAE,CAAC;QAC1B,MAAM,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YAC5C,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,YAAY,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,MAAM,CAAK;YAC5B,OAAO;YACP,OAAO,EAAE,CAAC,IAAI,eAAe,EAAE,CAAC;SACjC,CAAC,CAAC;QACH,MAAM,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YAC3C,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;YAC1C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QACH,MAAM,OAAO;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,OAAO,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;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,IAAY,EAAE,QAAgB;IACpE,IAAI,WAAW,GAAG,KAAK,CAAC;IACxB,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAE9B,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAEhD,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 { 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\nasync function migrate(root: string, 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(root: string, 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\nasync function generateDump(root: string, dumpPath: string) {\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(root, 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(root, 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 createLocalKysely(root: string, dumpPath: string) {\n  let isDestroyed = false;\n  const now = performance.now();\n\n  const dump = await generateDump(root, 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/local/index.d.ts

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