@vrplatform/kysely 1.3.1 → 1.3.3

package/build/main/index.d.ts

@@ -1,4 +1,4 @@
-import { Kysely as K, type Transaction as T } from 'kysely';
+import { Kysely as K, type LogConfig, type Transaction as T } from 'kysely';
 import postgres, { type Sql } from 'postgres';
 export type { ColumnType, Expression, ExpressionBuilder, ExpressionOrFactory, Generated, Insertable, JSONColumnType, OperandExpression, Selectable, SelectQueryBuilder, SqlBool, Updateable, } from 'kysely';
 export { jsonArrayFrom } from 'kysely/helpers/postgres';
@@ -6,6 +6,10 @@ export * from './v1.generated';
 import type { DB } from './v1.generated';
 export type Transaction = T<DB>;
 export { sql } from 'kysely';
+type KyselyOpts = {
+    repositoryName?: string;
+    log?: true | LogConfig;
+};
 type PgOpt = {
     forceDisableSSL?: boolean;
     retries?: false | number;
@@ -15,11 +19,9 @@ 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 useKysely(postgres: Sql, options?: KyselyOpts): K<DB>;
 export declare function useAsyncPostgres(connectionString: string, opt?: PgOpt): Promise<postgres.Sql<{}>>;
-export declare function useAsyncKysely(connectionString: string, opt?: PgOpt): Promise<K<DB>>;
+export declare function useAsyncKysely(connectionString: string, opt?: PgOpt & KyselyOpts): Promise<K<DB>>;
 /**
  * @deprecated This method is deprecated in favor of the useAsyncKysely method.
  */

package/build/main/index.js

@@ -32,19 +32,26 @@ var postgres_2 = require("kysely/helpers/postgres");
 Object.defineProperty(exports, "jsonArrayFrom", { enumerable: true, get: function () { return postgres_2.jsonArrayFrom; } });
 __exportStar(require("./v1.generated"), exports);
 const node_path_1 = require("node:path");
+const pganalyze_comment_plugin_1 = require("./pganalyze-comment-plugin");
 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 = {}) {
+function useKysely(postgres, options) {
     console.debug('Creating Kysely instance');
+    const plugins = [new kysely_1.CamelCasePlugin()];
+    if (options?.repositoryName) {
+        plugins.push(new pganalyze_comment_plugin_1.KyselyPganalyzeCommentPlugin({
+            repository: options.repositoryName,
+        }));
+    }
     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,
+        plugins,
+        log: options?.log === true ? ['query', 'error'] : options?.log,
     });
     kysely.destroy = async () => {
         console.debug('Closing Kysely connection');
@@ -73,8 +80,8 @@ async function useAsyncPostgres(connectionString, opt) {
                 ssl: ssl ? 'require' : false,
                 // connect_timeout: 10
                 // idle_timeout: 10
-                // disable prepare statements since we use pgbouncer
-                prepare: false,
+                // enable prepare statements, pgbouncer on crunchy allows for prepared statements
+                prepare: true,
                 connect_timeout: 10,
                 idle_timeout: 60,
             });
@@ -102,7 +109,7 @@ async function useAsyncPostgres(connectionString, opt) {
 }
 async function useAsyncKysely(connectionString, opt) {
     const sql = await useAsyncPostgres(connectionString, opt);
-    return useKysely(sql);
+    return useKysely(sql, opt);
 }
 /**
  * @deprecated This method is deprecated in favor of the useAsyncKysely method.

package/build/main/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"src/","sources":["index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;AAsCA,8BAqBC;AAGD,4CA6CC;AAED,wCAGC;AAKD,kCAeC;AAED,sDAUC;AAYD,8BA+BC;AA3LD,4CAAkD;AAClD,mCAA6E;AAC7E,2DAAuD;AACvD,wDAA8C;AAgB9C,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,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC3C,MAAM,OAAO,CAAC,IAAI,CAAC;YACjB,QAAQ,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;YACtD,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;SAC5C,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,KAAK,KAAK,CAAC,CAAC;YAAE,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,IAAI,YAAY,GAAmB,EAAE,CAAC;AAC/B,KAAK,UAAU,gBAAgB,CAAC,gBAAwB,EAAE,GAAW;IAC1E,MAAM,OAAO,GAAG,IAAI,CAAC;IACrB,MAAM,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;IAC1C,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAEtC,OAAO,CAAC,KAAK,CACX,qCAAqC,OAAO,oBAAoB,GAAG,aAAa,GAAG,CAAC,IAAI,EAAE,CAC3F,CAAC;IACF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC;QACzB,OAAO,CAAC,IAAI,CAAC,mBAAmB,YAAY,CAAC,MAAM,uBAAuB,CAAC,CAAC;IAE9E,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;QACpD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAA,kBAAQ,EAAC,gBAAgB,EAAE;gBACrC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;gBAC5B,sBAAsB;gBACtB,mBAAmB;gBACnB,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,GAAG,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,CAAC;IACtC,OAAO,CAAC,KAAK,CACX,8CAA8C,GAAG,aAAa,GAAG,CAAC,IAAI,EAAE,CACzE,CAAC;IACF,MAAM,EAAE,GAAG,IAAA,kBAAQ,EAAC,gBAAgB,EAAE;QACpC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;KAC7B,CAAC,CAAC;IACH,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACtB,OAAO,EAAE,CAAC;AACZ,CAAC;AAEM,KAAK,UAAU,qBAAqB,CAAC,SAAS,GAAG,IAAI;IAC1D,MAAM,OAAO,CAAC,IAAI,CAAC;QACjB,OAAO,CAAC,GAAG,CACT,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CACtB,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CACtD,CACF;QACD,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,SAAS,GAAG,IAAI,CAAC,CAAC;KACxD,CAAC,CAAC;IACH,YAAY,GAAG,EAAE,CAAC;AACpB,CAAC;AAED,MAAa,aAAc,SAAQ,KAAK;IACtC,YAAY,OAAe,EAAE,KAAU;QACrC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AALD,sCAKC;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  SelectQueryBuilder,\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  kysely.destroy = async () => {\n    console.debug('Closing Kysely connection');\n    await Promise.race([\n      postgres.end({ timeout: 5000 }).catch(() => undefined),\n      new Promise((res) => setTimeout(res, 6000)),\n    ]);\n    const index = _connections.indexOf(postgres);\n    if (index !== -1) _connections.splice(index, 1);\n  };\n\n  return kysely;\n}\n\nlet _connections: postgres.Sql[] = [];\nexport async function useAsyncPostgres(connectionString: string, opt?: PgOpt) {\n  const delayMs = 2500;\n  const retries = opt?.retries === false ? 0 : opt?.retries || 3;\n  const ssl = opt?.forceDisableSSL !== true;\n  const url = new URL(connectionString);\n\n  console.debug(\n    `Creating Postgres connection with ${retries} retries and ssl=${ssl} and port=${url.port}`\n  );\n  if (_connections.length > 0)\n    console.warn(`We already have ${_connections.length} Postgres connections`);\n\n  for (let attempt = 1; attempt <= retries; attempt++) {\n    try {\n      const sql = postgres(connectionString, {\n        ssl: ssl ? 'require' : false,\n        // connect_timeout: 10\n        // idle_timeout: 10\n        // 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 url = new URL(connectionString);\n  console.debug(\n    `Creating sync Postgres connection with ssl=${ssl} and port=${url.port}`\n  );\n  const pg = postgres(connectionString, {\n    ssl: ssl ? 'require' : false,\n  });\n  _connections.push(pg);\n  return pg;\n}\n\nexport async function forceCloseAllPostgres(timeoutMs = 5000) {\n  await Promise.race([\n    Promise.all(\n      _connections.map((pg) =>\n        pg.end({ timeout: timeoutMs }).catch(() => undefined)\n      )\n    ),\n    new Promise((res) => setTimeout(res, timeoutMs + 1000)),\n  ]);\n  _connections = [];\n}\n\nexport class DatabaseError extends Error {\n  constructor(message: string, cause: any) {\n    super(message);\n    this.cause = cause;\n  }\n}\n\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":";;;;;;;;;;;;;;;;;;;;AAkDA,8BA6BC;AAGD,4CA6CC;AAED,wCAMC;AAKD,kCAeC;AAED,sDAUC;AAYD,8BA+BC;AAlND,4CAAkD;AAClD,mCAMgB;AAChB,2DAAuD;AACvD,wDAA8C;AAgB9C,oDAAwD;AAA/C,yGAAA,aAAa,OAAA;AACtB,iDAA+B;AAE/B,yCAA2C;AAC3C,yEAA0E;AAI1E,iCAA6B;AAApB,6FAAA,GAAG,OAAA;AAcZ;;GAEG;AACH,SAAgB,SAAS,CAAC,QAAa,EAAE,OAAoB;IAC3D,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC1C,MAAM,OAAO,GAAmB,CAAC,IAAI,wBAAe,EAAE,CAAC,CAAC;IACxD,IAAI,OAAO,EAAE,cAAc,EAAE,CAAC;QAC5B,OAAO,CAAC,IAAI,CACV,IAAI,uDAA4B,CAAC;YAC/B,UAAU,EAAE,OAAO,CAAC,cAAc;SACnC,CAAC,CACH,CAAC;IACJ,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,eAAC,CAAK;QACvB,OAAO,EAAE,IAAI,sCAAiB,CAAC;YAC7B,QAAQ;SACT,CAAC;QACF,OAAO;QACP,GAAG,EAAE,OAAO,EAAE,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,GAAG;KAC/D,CAAC,CAAC;IAEH,MAAM,CAAC,OAAO,GAAG,KAAK,IAAI,EAAE;QAC1B,OAAO,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC3C,MAAM,OAAO,CAAC,IAAI,CAAC;YACjB,QAAQ,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;YACtD,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;SAC5C,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,KAAK,KAAK,CAAC,CAAC;YAAE,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,IAAI,YAAY,GAAmB,EAAE,CAAC;AAC/B,KAAK,UAAU,gBAAgB,CAAC,gBAAwB,EAAE,GAAW;IAC1E,MAAM,OAAO,GAAG,IAAI,CAAC;IACrB,MAAM,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;IAC1C,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAEtC,OAAO,CAAC,KAAK,CACX,qCAAqC,OAAO,oBAAoB,GAAG,aAAa,GAAG,CAAC,IAAI,EAAE,CAC3F,CAAC;IACF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC;QACzB,OAAO,CAAC,IAAI,CAAC,mBAAmB,YAAY,CAAC,MAAM,uBAAuB,CAAC,CAAC;IAE9E,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;QACpD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAA,kBAAQ,EAAC,gBAAgB,EAAE;gBACrC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;gBAC5B,sBAAsB;gBACtB,mBAAmB;gBACnB,iFAAiF;gBACjF,OAAO,EAAE,IAAI;gBACb,eAAe,EAAE,EAAE;gBACnB,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,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,CAClC,gBAAwB,EACxB,GAAwB;IAExB,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC1D,OAAO,SAAS,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,SAAgB,WAAW,CACzB,gBAAyB,EACzB,GAAoC;IAEpC,IAAI,CAAC,gBAAgB;QAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC/D,MAAM,GAAG,GAAG,GAAG,EAAE,eAAe,KAAK,IAAI,CAAC;IAC1C,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,CAAC;IACtC,OAAO,CAAC,KAAK,CACX,8CAA8C,GAAG,aAAa,GAAG,CAAC,IAAI,EAAE,CACzE,CAAC;IACF,MAAM,EAAE,GAAG,IAAA,kBAAQ,EAAC,gBAAgB,EAAE;QACpC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;KAC7B,CAAC,CAAC;IACH,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACtB,OAAO,EAAE,CAAC;AACZ,CAAC;AAEM,KAAK,UAAU,qBAAqB,CAAC,SAAS,GAAG,IAAI;IAC1D,MAAM,OAAO,CAAC,IAAI,CAAC;QACjB,OAAO,CAAC,GAAG,CACT,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CACtB,EAAE,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CACtD,CACF;QACD,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,SAAS,GAAG,IAAI,CAAC,CAAC;KACxD,CAAC,CAAC;IACH,YAAY,GAAG,EAAE,CAAC;AACpB,CAAC;AAED,MAAa,aAAc,SAAQ,KAAK;IACtC,YAAY,OAAe,EAAE,KAAU;QACrC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AALD,sCAKC;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 {\n  CamelCasePlugin,\n  Kysely as K,\n  type KyselyPlugin,\n  type LogConfig,\n  type Transaction as T,\n} 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  SelectQueryBuilder,\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 { KyselyPganalyzeCommentPlugin } from './pganalyze-comment-plugin';\nimport type { DB } from './v1.generated';\n\nexport type Transaction = T<DB>;\nexport { sql } from 'kysely';\n\ntype KyselyOpts = {\n  repositoryName?: string;\n  log?: true | LogConfig;\n};\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?: KyselyOpts) {\n  console.debug('Creating Kysely instance');\n  const plugins: KyselyPlugin[] = [new CamelCasePlugin()];\n  if (options?.repositoryName) {\n    plugins.push(\n      new KyselyPganalyzeCommentPlugin({\n        repository: options.repositoryName,\n      })\n    );\n  }\n  const kysely = new K<DB>({\n    dialect: new PostgresJSDialect({\n      postgres,\n    }),\n    plugins,\n    log: options?.log === true ? ['query', 'error'] : options?.log,\n  });\n\n  kysely.destroy = async () => {\n    console.debug('Closing Kysely connection');\n    await Promise.race([\n      postgres.end({ timeout: 5000 }).catch(() => undefined),\n      new Promise((res) => setTimeout(res, 6000)),\n    ]);\n    const index = _connections.indexOf(postgres);\n    if (index !== -1) _connections.splice(index, 1);\n  };\n\n  return kysely;\n}\n\nlet _connections: postgres.Sql[] = [];\nexport async function useAsyncPostgres(connectionString: string, opt?: PgOpt) {\n  const delayMs = 2500;\n  const retries = opt?.retries === false ? 0 : opt?.retries || 3;\n  const ssl = opt?.forceDisableSSL !== true;\n  const url = new URL(connectionString);\n\n  console.debug(\n    `Creating Postgres connection with ${retries} retries and ssl=${ssl} and port=${url.port}`\n  );\n  if (_connections.length > 0)\n    console.warn(`We already have ${_connections.length} Postgres connections`);\n\n  for (let attempt = 1; attempt <= retries; attempt++) {\n    try {\n      const sql = postgres(connectionString, {\n        ssl: ssl ? 'require' : false,\n        // connect_timeout: 10\n        // idle_timeout: 10\n        // enable prepare statements, pgbouncer on crunchy allows for prepared statements\n        prepare: true,\n        connect_timeout: 10,\n        idle_timeout: 60,\n      });\n\n      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(\n  connectionString: string,\n  opt?: PgOpt & KyselyOpts\n) {\n  const sql = await useAsyncPostgres(connectionString, opt);\n  return useKysely(sql, opt);\n}\n\n/**\n * @deprecated This method is deprecated in favor of the useAsyncKysely method.\n */\nexport function usePostgres(\n  connectionString?: string,\n  opt?: Pick<PgOpt, 'forceDisableSSL'>\n) {\n  if (!connectionString) throw new Error('No connection string');\n  const ssl = opt?.forceDisableSSL !== true;\n  const url = new URL(connectionString);\n  console.debug(\n    `Creating sync Postgres connection with ssl=${ssl} and port=${url.port}`\n  );\n  const pg = postgres(connectionString, {\n    ssl: ssl ? 'require' : false,\n  });\n  _connections.push(pg);\n  return pg;\n}\n\nexport async function forceCloseAllPostgres(timeoutMs = 5000) {\n  await Promise.race([\n    Promise.all(\n      _connections.map((pg) =>\n        pg.end({ timeout: timeoutMs }).catch(() => undefined)\n      )\n    ),\n    new Promise((res) => setTimeout(res, timeoutMs + 1000)),\n  ]);\n  _connections = [];\n}\n\nexport class DatabaseError extends Error {\n  constructor(message: string, cause: any) {\n    super(message);\n    this.cause = cause;\n  }\n}\n\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/index.d.ts

@@ -1,5 +1,9 @@
-import { Kysely } from 'kysely';
+import { Kysely, type LogConfig } 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>>;
+export type UseLocalKyselyOptions = {
+    repositoryName?: string;
+    log?: true | LogConfig;
+};
+export declare function useLocalKysely(args?: UseLocalKyselyOptions): Promise<Kysely<DB>>;

package/build/main/local/index.js

@@ -6,9 +6,10 @@ 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");
+const pganalyze_comment_plugin_1 = require("../pganalyze-comment-plugin");
 exports.root = (0, node_path_1.join)(__dirname, '../..');
 exports.dumpPath = (0, node_path_1.join)(exports.root, '.data.tar');
-async function useLocalKysely() {
+async function useLocalKysely(args = {}) {
     let isDestroyed = false;
     const now = performance.now();
     const dump = Bun.file(exports.dumpPath);
@@ -18,7 +19,13 @@ async function useLocalKysely() {
     const { dialect } = new kysely_pglite_1.KyselyPGlite(pglite);
     const kysely = new kysely_1.Kysely({
         dialect,
-        plugins: [new kysely_1.CamelCasePlugin()],
+        log: args.log === true ? ['query', 'error'] : args.log,
+        plugins: [
+            new kysely_1.CamelCasePlugin(),
+            new pganalyze_comment_plugin_1.KyselyPganalyzeCommentPlugin({
+                repository: args.repositoryName ?? 'localTesting',
+            }),
+        ],
     });
     const originalDestroy = kysely.destroy.bind(kysely);
     kysely.destroy = async () => {

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

@@ -1 +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"]}
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["local/index.ts"],"names":[],"mappings":";;;AAcA,wCAoCC;AAlDD,yCAAiC;AACjC,iDAA8C;AAC9C,mCAAiE;AACjE,iDAA6C;AAC7C,0EAA2E;AAG9D,QAAA,IAAI,GAAG,IAAA,gBAAI,EAAC,SAAS,EAAE,OAAO,CAAC,CAAC;AAChC,QAAA,QAAQ,GAAG,IAAA,gBAAI,EAAC,YAAI,EAAE,WAAW,CAAC,CAAC;AAMzC,KAAK,UAAU,cAAc,CAAC,OAA8B,EAAE;IACnE,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,GAAG,EAAE,IAAI,CAAC,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG;QACtD,OAAO,EAAE;YACP,IAAI,wBAAe,EAAE;YACrB,IAAI,uDAA4B,CAAC;gBAC/B,UAAU,EAAE,IAAI,CAAC,cAAc,IAAI,cAAc;aAClD,CAAC;SACH;KACF,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, type LogConfig } from 'kysely';\nimport { KyselyPGlite } from 'kysely-pglite';\nimport { KyselyPganalyzeCommentPlugin } from '../pganalyze-comment-plugin';\nimport type { DB } from '../v1.generated';\n\nexport const root = join(__dirname, '../..');\nexport const dumpPath = join(root, '.data.tar');\n\nexport type UseLocalKyselyOptions = {\n  repositoryName?: string;\n  log?: true | LogConfig;\n};\nexport async function useLocalKysely(args: UseLocalKyselyOptions = {}) {\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    log: args.log === true ? ['query', 'error'] : args.log,\n    plugins: [\n      new CamelCasePlugin(),\n      new KyselyPganalyzeCommentPlugin({\n        repository: args.repositoryName ?? 'localTesting',\n      }),\n    ],\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/pganalyze-comment-plugin/comment-transformer.d.ts

@@ -0,0 +1,21 @@
+import { type DeleteQueryNode, type InsertQueryNode, OperationNodeTransformer, type RootOperationNode, type SelectQueryNode, type UpdateQueryNode } from 'kysely';
+/**
+ * Transformer that adds a SQL comment to queries for pganalyze tracking.
+ * The comment format is: repository: REPOSITORY_NAME
+ *
+ * This works by appending the comment to queries using the endModifiers property,
+ * similar to how modifyEnd works.
+ */
+export declare class CommentTransformer extends OperationNodeTransformer {
+    readonly _repositoryName: string;
+    constructor(_repositoryName: string);
+    transformRootNode(node: RootOperationNode): RootOperationNode;
+    protected transformSelectQuery(node: SelectQueryNode): SelectQueryNode;
+    protected transformInsertQuery(node: InsertQueryNode): InsertQueryNode;
+    protected transformUpdateQuery(node: UpdateQueryNode): UpdateQueryNode;
+    protected transformDeleteQuery(node: DeleteQueryNode): DeleteQueryNode;
+    private addCommentToSelectQuery;
+    private addCommentToInsertQuery;
+    private addCommentToUpdateQuery;
+    private addCommentToDeleteQuery;
+}

package/build/main/pganalyze-comment-plugin/comment-transformer.js

@@ -0,0 +1,90 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CommentTransformer = void 0;
+const kysely_1 = require("kysely");
+function createCommentNode(repositoryName) {
+    return kysely_1.SelectModifierNode.createWithExpression(kysely_1.sql.raw(`/* repository: ${repositoryName} */`).toOperationNode());
+}
+/**
+ * Transformer that adds a SQL comment to queries for pganalyze tracking.
+ * The comment format is: repository: REPOSITORY_NAME
+ *
+ * This works by appending the comment to queries using the endModifiers property,
+ * similar to how modifyEnd works.
+ */
+class CommentTransformer extends kysely_1.OperationNodeTransformer {
+    constructor(_repositoryName) {
+        super();
+        this._repositoryName = _repositoryName;
+    }
+    transformRootNode(node) {
+        if (node.kind === 'SelectQueryNode') {
+            return this.transformSelectQuery(node);
+        }
+        if (node.kind === 'InsertQueryNode') {
+            return this.transformInsertQuery(node);
+        }
+        if (node.kind === 'UpdateQueryNode') {
+            return this.transformUpdateQuery(node);
+        }
+        if (node.kind === 'DeleteQueryNode') {
+            return this.transformDeleteQuery(node);
+        }
+        return node;
+    }
+    transformSelectQuery(node) {
+        const transformed = super.transformSelectQuery(node);
+        return this.addCommentToSelectQuery(transformed);
+    }
+    transformInsertQuery(node) {
+        const transformed = super.transformInsertQuery(node);
+        return this.addCommentToInsertQuery(transformed);
+    }
+    transformUpdateQuery(node) {
+        const transformed = super.transformUpdateQuery(node);
+        return this.addCommentToUpdateQuery(transformed);
+    }
+    transformDeleteQuery(node) {
+        const transformed = super.transformDeleteQuery(node);
+        return this.addCommentToDeleteQuery(transformed);
+    }
+    addCommentToSelectQuery(node) {
+        // Append comment using endModifiers property, similar to modifyEnd
+        const commentNode = createCommentNode(this._repositoryName);
+        return {
+            ...node,
+            endModifiers: node.endModifiers
+                ? [...node.endModifiers, commentNode]
+                : [commentNode],
+        };
+    }
+    addCommentToInsertQuery(node) {
+        const commentNode = createCommentNode(this._repositoryName);
+        return {
+            ...node,
+            endModifiers: node.endModifiers
+                ? [...node.endModifiers, commentNode]
+                : [commentNode],
+        };
+    }
+    addCommentToUpdateQuery(node) {
+        const commentNode = createCommentNode(this._repositoryName);
+        return {
+            ...node,
+            endModifiers: node.endModifiers
+                ? [...node.endModifiers, commentNode]
+                : [commentNode],
+        };
+    }
+    addCommentToDeleteQuery(node) {
+        const commentNode = createCommentNode(this._repositoryName);
+        return {
+            ...node,
+            endModifiers: node.endModifiers
+                ? [...node.endModifiers, commentNode]
+                : [commentNode],
+        };
+    }
+}
+exports.CommentTransformer = CommentTransformer;
+//# sourceMappingURL=comment-transformer.js.map

package/build/main/pganalyze-comment-plugin/comment-transformer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"comment-transformer.js","sourceRoot":"src/","sources":["pganalyze-comment-plugin/comment-transformer.ts"],"names":[],"mappings":";;;AAAA,mCASgB;AAEhB,SAAS,iBAAiB,CAAC,cAAsB;IAC/C,OAAO,2BAAkB,CAAC,oBAAoB,CAC5C,YAAG,CAAC,GAAG,CAAC,kBAAkB,cAAc,KAAK,CAAC,CAAC,eAAe,EAAE,CACjE,CAAC;AACJ,CAAC;AACD;;;;;;GAMG;AACH,MAAa,kBAAmB,SAAQ,iCAAwB;IAC9D,YAAqB,eAAuB;QAC1C,KAAK,EAAE,CAAC;QADW,oBAAe,GAAf,eAAe,CAAQ;IAE5C,CAAC;IAED,iBAAiB,CAAC,IAAuB;QACvC,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACpC,OAAO,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACpC,OAAO,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACpC,OAAO,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACpC,OAAO,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEkB,oBAAoB,CACrC,IAAqB;QAErB,MAAM,WAAW,GAAG,KAAK,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACrD,OAAO,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC;IAEkB,oBAAoB,CACrC,IAAqB;QAErB,MAAM,WAAW,GAAG,KAAK,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACrD,OAAO,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC;IAEkB,oBAAoB,CACrC,IAAqB;QAErB,MAAM,WAAW,GAAG,KAAK,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACrD,OAAO,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC;IAEkB,oBAAoB,CACrC,IAAqB;QAErB,MAAM,WAAW,GAAG,KAAK,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACrD,OAAO,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC;IAEO,uBAAuB,CAAC,IAAqB;QACnD,mEAAmE;QACnE,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAE5D,OAAO;YACL,GAAG,IAAI;YACP,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC7B,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC;gBACrC,CAAC,CAAC,CAAC,WAAW,CAAC;SAClB,CAAC;IACJ,CAAC;IAEO,uBAAuB,CAAC,IAAqB;QACnD,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAE5D,OAAO;YACL,GAAG,IAAI;YACP,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC7B,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC;gBACrC,CAAC,CAAC,CAAC,WAAW,CAAC;SAClB,CAAC;IACJ,CAAC;IAEO,uBAAuB,CAAC,IAAqB;QACnD,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAE5D,OAAO;YACL,GAAG,IAAI;YACP,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC7B,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC;gBACrC,CAAC,CAAC,CAAC,WAAW,CAAC;SAClB,CAAC;IACJ,CAAC;IAEO,uBAAuB,CAAC,IAAqB;QACnD,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAE5D,OAAO;YACL,GAAG,IAAI;YACP,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC7B,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC;gBACrC,CAAC,CAAC,CAAC,WAAW,CAAC;SAClB,CAAC;IACJ,CAAC;CACF;AA7FD,gDA6FC","sourcesContent":["import {\n  type DeleteQueryNode,\n  type InsertQueryNode,\n  OperationNodeTransformer,\n  type RootOperationNode,\n  SelectModifierNode,\n  type SelectQueryNode,\n  sql,\n  type UpdateQueryNode,\n} from 'kysely';\n\nfunction createCommentNode(repositoryName: string): SelectModifierNode {\n  return SelectModifierNode.createWithExpression(\n    sql.raw(`/* repository: ${repositoryName} */`).toOperationNode()\n  );\n}\n/**\n * Transformer that adds a SQL comment to queries for pganalyze tracking.\n * The comment format is: repository: REPOSITORY_NAME\n *\n * This works by appending the comment to queries using the endModifiers property,\n * similar to how modifyEnd works.\n */\nexport class CommentTransformer extends OperationNodeTransformer {\n  constructor(readonly _repositoryName: string) {\n    super();\n  }\n\n  transformRootNode(node: RootOperationNode): RootOperationNode {\n    if (node.kind === 'SelectQueryNode') {\n      return this.transformSelectQuery(node);\n    }\n    if (node.kind === 'InsertQueryNode') {\n      return this.transformInsertQuery(node);\n    }\n    if (node.kind === 'UpdateQueryNode') {\n      return this.transformUpdateQuery(node);\n    }\n    if (node.kind === 'DeleteQueryNode') {\n      return this.transformDeleteQuery(node);\n    }\n    return node;\n  }\n\n  protected override transformSelectQuery(\n    node: SelectQueryNode\n  ): SelectQueryNode {\n    const transformed = super.transformSelectQuery(node);\n    return this.addCommentToSelectQuery(transformed);\n  }\n\n  protected override transformInsertQuery(\n    node: InsertQueryNode\n  ): InsertQueryNode {\n    const transformed = super.transformInsertQuery(node);\n    return this.addCommentToInsertQuery(transformed);\n  }\n\n  protected override transformUpdateQuery(\n    node: UpdateQueryNode\n  ): UpdateQueryNode {\n    const transformed = super.transformUpdateQuery(node);\n    return this.addCommentToUpdateQuery(transformed);\n  }\n\n  protected override transformDeleteQuery(\n    node: DeleteQueryNode\n  ): DeleteQueryNode {\n    const transformed = super.transformDeleteQuery(node);\n    return this.addCommentToDeleteQuery(transformed);\n  }\n\n  private addCommentToSelectQuery(node: SelectQueryNode): SelectQueryNode {\n    // Append comment using endModifiers property, similar to modifyEnd\n    const commentNode = createCommentNode(this._repositoryName);\n\n    return {\n      ...node,\n      endModifiers: node.endModifiers\n        ? [...node.endModifiers, commentNode]\n        : [commentNode],\n    };\n  }\n\n  private addCommentToInsertQuery(node: InsertQueryNode): InsertQueryNode {\n    const commentNode = createCommentNode(this._repositoryName);\n\n    return {\n      ...node,\n      endModifiers: node.endModifiers\n        ? [...node.endModifiers, commentNode]\n        : [commentNode],\n    };\n  }\n\n  private addCommentToUpdateQuery(node: UpdateQueryNode): UpdateQueryNode {\n    const commentNode = createCommentNode(this._repositoryName);\n\n    return {\n      ...node,\n      endModifiers: node.endModifiers\n        ? [...node.endModifiers, commentNode]\n        : [commentNode],\n    };\n  }\n\n  private addCommentToDeleteQuery(node: DeleteQueryNode): DeleteQueryNode {\n    const commentNode = createCommentNode(this._repositoryName);\n\n    return {\n      ...node,\n      endModifiers: node.endModifiers\n        ? [...node.endModifiers, commentNode]\n        : [commentNode],\n    };\n  }\n}\n"]}

package/build/main/pganalyze-comment-plugin/index.d.ts

@@ -0,0 +1,9 @@
+import type { KyselyPlugin, PluginTransformQueryArgs, PluginTransformResultArgs, QueryResult, RootOperationNode, UnknownRow } from 'kysely';
+export declare class KyselyPganalyzeCommentPlugin implements KyselyPlugin {
+    #private;
+    constructor(options: {
+        repository: string;
+    });
+    transformQuery(args: PluginTransformQueryArgs): RootOperationNode;
+    transformResult(args: PluginTransformResultArgs): Promise<QueryResult<UnknownRow>>;
+}

package/build/main/pganalyze-comment-plugin/index.js

@@ -0,0 +1,31 @@
+"use strict";
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var _KyselyPganalyzeCommentPlugin_transformer;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KyselyPganalyzeCommentPlugin = void 0;
+const comment_transformer_1 = require("./comment-transformer");
+class KyselyPganalyzeCommentPlugin {
+    constructor(options) {
+        _KyselyPganalyzeCommentPlugin_transformer.set(this, void 0);
+        __classPrivateFieldSet(this, _KyselyPganalyzeCommentPlugin_transformer, new comment_transformer_1.CommentTransformer(options.repository), "f");
+    }
+    transformQuery(args) {
+        return __classPrivateFieldGet(this, _KyselyPganalyzeCommentPlugin_transformer, "f").transformRootNode(args.node);
+    }
+    async transformResult(args) {
+        return args.result;
+    }
+}
+exports.KyselyPganalyzeCommentPlugin = KyselyPganalyzeCommentPlugin;
+_KyselyPganalyzeCommentPlugin_transformer = new WeakMap();
+//# sourceMappingURL=index.js.map

package/build/main/pganalyze-comment-plugin/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["pganalyze-comment-plugin/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAQA,+DAA2D;AAE3D,MAAa,4BAA4B;IAGvC,YAAY,OAA+B;QAFlC,4DAAiC;QAGxC,uBAAA,IAAI,6CAAgB,IAAI,wCAAkB,CAAC,OAAO,CAAC,UAAU,CAAC,MAAA,CAAC;IACjE,CAAC;IAED,cAAc,CAAC,IAA8B;QAC3C,OAAO,uBAAA,IAAI,iDAAa,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,IAA+B;QAE/B,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;CACF;AAhBD,oEAgBC","sourcesContent":["import type {\n  KyselyPlugin,\n  PluginTransformQueryArgs,\n  PluginTransformResultArgs,\n  QueryResult,\n  RootOperationNode,\n  UnknownRow,\n} from 'kysely';\nimport { CommentTransformer } from './comment-transformer';\n\nexport class KyselyPganalyzeCommentPlugin implements KyselyPlugin {\n  readonly #transformer: CommentTransformer;\n\n  constructor(options: { repository: string }) {\n    this.#transformer = new CommentTransformer(options.repository);\n  }\n\n  transformQuery(args: PluginTransformQueryArgs): RootOperationNode {\n    return this.#transformer.transformRootNode(args.node);\n  }\n\n  async transformResult(\n    args: PluginTransformResultArgs\n  ): Promise<QueryResult<UnknownRow>> {\n    return args.result;\n  }\n}\n"]}

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

@@ -0,0 +1,15 @@
+import type { Kysely } from 'kysely';
+import { type UseLocalKyselyOptions } from '../local';
+import type { DB } from '../v1.generated';
+/**
+ * @deprecated Use `useLocalKyselyResource` instead.
+ */
+export declare function useTestKysely(cleanupStrategy: 'afterAll', args?: UseLocalKyselyOptions): Promise<{
+    kysely: Kysely<DB>;
+}>;
+export declare function useTestAutoCleanupKysely(args?: UseLocalKyselyOptions): Promise<Kysely<DB>>;
+type ReturnType = Kysely<DB> & {
+    [Symbol.dispose]: () => Promise<void>;
+};
+export declare function useLocalKyselyResource(args?: UseLocalKyselyOptions): Promise<ReturnType>;
+export {};

package/build/main/test/index.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useTestKysely = useTestKysely;
+exports.useTestAutoCleanupKysely = useTestAutoCleanupKysely;
+exports.useLocalKyselyResource = useLocalKyselyResource;
+const bun_test_1 = require("bun:test");
+const local_1 = require("../local");
+/**
+ * @deprecated Use `useLocalKyselyResource` instead.
+ */
+async function useTestKysely(cleanupStrategy, args) {
+    const kysely = await (0, local_1.useLocalKysely)(args);
+    if (cleanupStrategy === 'afterAll')
+        (0, bun_test_1.afterAll)(async () => {
+            await kysely.destroy();
+        });
+    return {
+        kysely,
+    };
+}
+async function useTestAutoCleanupKysely(args) {
+    const kysely = await (0, local_1.useLocalKysely)(args);
+    (0, bun_test_1.afterAll)(async () => {
+        await kysely.destroy();
+    });
+    return kysely;
+}
+async function useLocalKyselyResource(args) {
+    const kysely = await (0, local_1.useLocalKysely)(args);
+    return new Proxy({}, {
+        get(_target, prop) {
+            if (prop === Symbol.dispose) {
+                return async () => {
+                    await kysely.destroy();
+                };
+            }
+            if (prop === 'kysely') {
+                return kysely;
+            }
+            if (typeof kysely[prop] === 'function') {
+                return kysely[prop].bind(kysely);
+            }
+            return kysely[prop];
+        },
+    });
+}
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"src/","sources":["test/index.ts"],"names":[],"mappings":";;AAQA,sCAcC;AAED,4DAQC;AAMD,wDAwBC;AA9DD,uCAAoC;AAEpC,oCAAsE;AAGtE;;GAEG;AACI,KAAK,UAAU,aAAa,CACjC,eAA2B,EAC3B,IAA4B;IAE5B,MAAM,MAAM,GAAG,MAAM,IAAA,sBAAc,EAAC,IAAI,CAAC,CAAC;IAE1C,IAAI,eAAe,KAAK,UAAU;QAChC,IAAA,mBAAQ,EAAC,KAAK,IAAI,EAAE;YAClB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QACzB,CAAC,CAAC,CAAC;IAEL,OAAO;QACL,MAAM;KACP,CAAC;AACJ,CAAC;AAEM,KAAK,UAAU,wBAAwB,CAAC,IAA4B;IACzE,MAAM,MAAM,GAAG,MAAM,IAAA,sBAAc,EAAC,IAAI,CAAC,CAAC;IAE1C,IAAA,mBAAQ,EAAC,KAAK,IAAI,EAAE;QAClB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC;AAMM,KAAK,UAAU,sBAAsB,CAC1C,IAA4B;IAE5B,MAAM,MAAM,GAAG,MAAM,IAAA,sBAAc,EAAC,IAAI,CAAC,CAAC;IAE1C,OAAO,IAAI,KAAK,CACd,EAAE,EACF;QACE,GAAG,CAAC,OAAO,EAAE,IAAI;YACf,IAAI,IAAI,KAAK,MAAM,CAAC,OAAO,EAAE,CAAC;gBAC5B,OAAO,KAAK,IAAI,EAAE;oBAChB,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;gBACzB,CAAC,CAAC;YACJ,CAAC;YACD,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,IAAI,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACvC,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;KACF,CACY,CAAC;AAClB,CAAC","sourcesContent":["import { afterAll } from 'bun:test';\nimport type { Kysely } from 'kysely';\nimport { type UseLocalKyselyOptions, useLocalKysely } from '../local';\nimport type { DB } from '../v1.generated';\n\n/**\n * @deprecated Use `useLocalKyselyResource` instead.\n */\nexport async function useTestKysely(\n  cleanupStrategy: 'afterAll',\n  args?: UseLocalKyselyOptions\n) {\n  const kysely = await useLocalKysely(args);\n\n  if (cleanupStrategy === 'afterAll')\n    afterAll(async () => {\n      await kysely.destroy();\n    });\n\n  return {\n    kysely,\n  };\n}\n\nexport async function useTestAutoCleanupKysely(args?: UseLocalKyselyOptions) {\n  const kysely = await useLocalKysely(args);\n\n  afterAll(async () => {\n    await kysely.destroy();\n  });\n\n  return kysely;\n}\n\ntype ReturnType = Kysely<DB> & {\n  [Symbol.dispose]: () => Promise<void>;\n};\n\nexport async function useLocalKyselyResource(\n  args?: UseLocalKyselyOptions\n): Promise<ReturnType> {\n  const kysely = await useLocalKysely(args);\n\n  return new Proxy(\n    {},\n    {\n      get(_target, prop) {\n        if (prop === Symbol.dispose) {\n          return async () => {\n            await kysely.destroy();\n          };\n        }\n        if (prop === 'kysely') {\n          return kysely;\n        }\n        if (typeof kysely[prop] === 'function') {\n          return kysely[prop].bind(kysely);\n        }\n        return kysely[prop];\n      },\n    }\n  ) as ReturnType;\n}\n"]}

package/build/main/tsconfig.main.tsbuildinfo

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

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

@@ -1214,6 +1214,7 @@ export interface PublicReservation {
     lineMap: Json | null;
     listingConnectionId: string | null;
     listingId: string | null;
+    matchPaymentCustomerRef: string | null;
     metadata: Json | null;
     nights: number | null;
     otaReservationId: string | null;