immich/server/src/sql-tools/schema-from-database.ts

import { Kysely } from 'kysely';
import { PostgresJSDialect } from 'kysely-postgres-js';
import { Sql } from 'postgres';
import { ReaderContext } from 'src/sql-tools/contexts/reader-context';
import { readers } from 'src/sql-tools/readers';
import { DatabaseSchema, PostgresDB, SchemaFromDatabaseOptions } from 'src/sql-tools/types';

export type DatabaseLike = Sql | Kysely<any>;

const isKysely = (db: DatabaseLike): db is Kysely<any> => db instanceof Kysely;

/**
 * Load schema from a database url
 */
export const schemaFromDatabase = async (
  database: DatabaseLike,
  options: SchemaFromDatabaseOptions = {},
): Promise<DatabaseSchema> => {
  const db = isKysely(database)
    ? (database as Kysely<PostgresDB>)
    : new Kysely<PostgresDB>({ dialect: new PostgresJSDialect({ postgres: database }) });
  const ctx = new ReaderContext(options);

  try {
    for (const reader of readers) {
      await reader(ctx, db);
    }

    return ctx.build();
  } finally {
    // only close the connection it we created it
    if (!isKysely(database)) {
      await db.destroy();
    }
  }
};