refactor: expose additional kysely postgres options

app/src/data/connection/postgres/PgPostgresConnection.ts

@@ -1,22 +1,26 @@
-import { Kysely, PostgresDialect } from "kysely";
+import { Kysely, PostgresDialect, type PostgresDialectConfig as KyselyPostgresDialectConfig } from "kysely";
 import { PostgresIntrospector } from "./PostgresIntrospector";
 import { PostgresConnection, plugins } from "./PostgresConnection";
 import { customIntrospector } from "../Connection";
 import type { Pool } from "pg";
 
+export type PostgresDialectConfig = Omit<KyselyPostgresDialectConfig, "pool"> & {
+   pool: Pool;
+};
+
 export class PgPostgresConnection extends PostgresConnection<Pool> {
    override name = "pg";
 
-   constructor(pool: Pool) {
+   constructor(config: PostgresDialectConfig) {
       const kysely = new Kysely({
          dialect: customIntrospector(PostgresDialect, PostgresIntrospector, {
             excludeTables: [],
-         }).create({ pool }),
+         }).create(config),
          plugins,
       });
 
       super(kysely);
-      this.client = pool;
+      this.client = config.pool;
    }
 
    override async close(): Promise<void> {
@@ -24,6 +28,6 @@ export class PgPostgresConnection extends PostgresConnection<Pool> {
    }
 }
 
-export function pg(pool: Pool): PgPostgresConnection {
-   return new PgPostgresConnection(pool);
+export function pg(config: PostgresDialectConfig): PgPostgresConnection {
+   return new PgPostgresConnection(config);
 }

app/src/data/connection/postgres/PostgresJsConnection.ts

@@ -2,22 +2,21 @@ import { Kysely } from "kysely";
 import { PostgresIntrospector } from "./PostgresIntrospector";
 import { PostgresConnection, plugins } from "./PostgresConnection";
 import { customIntrospector } from "../Connection";
-import { PostgresJSDialect } from "kysely-postgres-js";
-import type { Sql } from "postgres";
+import { PostgresJSDialect, type PostgresJSDialectConfig } from "kysely-postgres-js";
 
-export class PostgresJsConnection extends PostgresConnection<Sql> {
+export class PostgresJsConnection extends PostgresConnection<PostgresJSDialectConfig["postgres"]> {
    override name = "postgres-js";
 
-   constructor(opts: { postgres: Sql }) {
+   constructor(config: PostgresJSDialectConfig) {
       const kysely = new Kysely({
          dialect: customIntrospector(PostgresJSDialect, PostgresIntrospector, {
             excludeTables: [],
-         }).create({ postgres: opts.postgres }),
+         }).create(config),
          plugins,
       });
 
       super(kysely);
-      this.client = opts.postgres;
+      this.client = config.postgres;
    }
 
    override async close(): Promise<void> {
@@ -26,7 +25,7 @@ export class PostgresJsConnection extends PostgresConnection<Sql> {
 }
 
 export function postgresJs(
-   postgres: Sql,
+   config: PostgresJSDialectConfig,
 ): PostgresJsConnection {
-   return new PostgresJsConnection({ postgres });
+   return new PostgresJsConnection(config);
 }