@technicity/data-service-generator 0.23.0-next.0 → 0.23.0-next.10

package/dist/generation/generate.js

@@ -30,24 +30,29 @@ exports.generate = generate;
 const path = __importStar(require("node:path"));
 const fs = __importStar(require("node:fs"));
 const os = __importStar(require("node:os"));
+const node_async_hooks_1 = require("node:async_hooks");
 const child_process = __importStar(require("node:child_process"));
 const node_crypto_1 = __importDefault(require("node:crypto"));
+const pino_1 = require("pino");
 const prettier = __importStar(require("prettier"));
 const changeCase = __importStar(require("change-case"));
 const fse = __importStar(require("fs-extra"));
 const _ = __importStar(require("lodash/fp"));
+const memoize_1 = __importDefault(require("lodash/memoize"));
 const json_schema_to_typescript_1 = require("json-schema-to-typescript");
 const getDuplicates_1 = require("../lib/getDuplicates");
 const isNotNullOrUndefined_1 = require("../lib/isNotNullOrUndefined");
+const pg_1 = require("pg");
 const MySQL_1 = require("../runtime/lib/MySQL");
 const capitalizeFirstLetter_1 = require("../lib/capitalizeFirstLetter");
-// json-schema-to-typescript inlines everything. We don't want that,
-// so use `tsType`, and add imports manually.
-// https://github.com/bcherny/json-schema-to-typescript#custom-schema-properties
-// https://github.com/bcherny/json-schema-to-typescript/issues/258
-// Meh
-let query = undefined;
-let dialect = "mysql";
+const pg2sqliteSchema_1 = require("./pg2sqliteSchema");
+const ctxStorage = new node_async_hooks_1.AsyncLocalStorage();
+function getCtx() {
+    const c = ctxStorage.getStore();
+    if (!c)
+        throw new Error("generate() context missing");
+    return c;
+}
 const json2TsOpts = {
     bannerComment: ""
 };
@@ -59,162 +64,237 @@ async function generate(input) {
     if (input.dialect == null) {
         throw new Error("Must specify `dialect`");
     }
+    const log = (0, pino_1.pino)({
+        name: "generate",
+        level: input.logLevel ?? process.env.LOG_LEVEL ?? "info",
+        transport: {
+            target: "pino-pretty",
+            options: {
+                colorize: true
+            }
+        }
+    });
     const specialCaseUuidColumn = input.specialCaseUuidColumn ?? true;
     const includeMappedFields = input.includeMappedFields ?? true;
     const supplementClientOpts = input.supplementClientOpts ?? true;
-    dialect = input.dialect;
-    init(input);
-    let tables = await getTableNames();
-    if (tables.length === 0) {
-        throw new Error("No tables found");
-    }
-    if (input.tables != null) {
-        tables = tables.filter((x) => input.tables?.includes(x));
-    }
-    if (input.excludeTables != null) {
-        tables = tables.filter((x) => !input.excludeTables?.includes(x));
-    }
-    const data = await Promise.all(tables.flatMap((x) => [
-        getGetOneData(x, includeMappedFields),
-        getGetListData(x),
-        getGetListPaginatedData(x),
-        getPostOneData(x, specialCaseUuidColumn, includeMappedFields),
-        getPatchOneData(x, specialCaseUuidColumn, includeMappedFields),
-        getPatchListData(x),
-        getDeleteOneData(x),
-        getDeleteListData(x)
-    ]));
-    const artifacts = await getArtifacts(tables, includeMappedFields, specialCaseUuidColumn);
-    const artifactsSource = getArtifactsSource(artifacts);
-    const sdkSource = await getSDKSource(data, specialCaseUuidColumn, supplementClientOpts, artifacts, input.outputSqliteSchema);
-    const sdkFilename = "index.ts";
-    const sourceIRuntimeFilePath = fs.existsSync(path.join(__dirname, "../runtime", "IRuntime.ts"))
-        ? path.join(__dirname, "../runtime", "IRuntime.ts")
-        : path.join(__dirname, "../runtime", "IRuntime.js");
-    const IRuntimeFilename = path.basename(sourceIRuntimeFilePath);
-    const artifactsFilename = "artifacts.ts";
-    const tsConfigJSON = {
-        compilerOptions: {
-            module: "commonjs",
-            moduleResolution: "node",
-            target: "es2020",
-            declaration: true,
-            outDir: "./sdk-ts"
-        },
-        include: [sdkFilename, artifactsFilename, IRuntimeFilename]
+    const runId = node_crypto_1.default.randomUUID();
+    const ctx = {
+        runId,
+        log: log.child({}),
+        dialect: input.dialect,
+        query: undefined,
+        pool: undefined
     };
-    const packageJSON = {
-        name: "temp",
-        version: "1.0.0",
-        // Deps need to be included so that they're inlined by ncc
-        dependencies: require("../../package.json").dependencies,
-        devDependencies: {
-            "@types/node": require("../../package.json").devDependencies["@types/node"],
-            typescript: require("../../package.json").devDependencies.typescript
-        },
-        // Not `resolutions` because npm used for install
-        overrides: {
-            // Fix for: `Cannot find type definition file for 'glob'`
-            glob: ">9.0.0"
+    log.debug({
+        runId: ctx.runId,
+        dialect: input.dialect,
+        database: input.database,
+        outdir: input.outdir
+    }, "generate() started");
+    return ctxStorage.run(ctx, async () => {
+        init(input);
+        ctx.log.debug("init() completed");
+        let tables = await getTableNames();
+        ctx.log.debug({ tableCount: tables.length, tables }, "getTableNames() completed");
+        if (tables.length === 0) {
+            throw new Error("No tables found");
         }
-    };
-    const tmpDirPath = path.join(os.tmpdir(), 
-    // _ because - in filename is not supported by mysql2sqlite
-    `dsg_${node_crypto_1.default.randomUUID()}`.replace(/-/g, "_"));
-    fse.mkdirpSync(tmpDirPath);
-    fs.writeFileSync(path.join(tmpDirPath, sdkFilename), sdkSource);
-    fs.writeFileSync(path.join(tmpDirPath, artifactsFilename), artifactsSource);
-    fse.copyFileSync(sourceIRuntimeFilePath, path.join(tmpDirPath, IRuntimeFilename));
-    const typesDirPath = path.join(tmpDirPath, "types");
-    fse.mkdirpSync(typesDirPath);
-    fs.writeFileSync(path.join(typesDirPath, "_shared.ts"), getTypeShared());
-    for (let x of data) {
-        if (x.kind === "getOne") {
-            fs.writeFileSync(path.join(typesDirPath, x.typeFieldsName + ".ts"), x.typeFields);
-            fs.writeFileSync(path.join(typesDirPath, x.typeReturnBaseName + ".ts"), x.typeReturnBase);
+        if (input.tables != null) {
+            tables = tables.filter((x) => input.tables?.includes(x));
         }
-        if (x.kind === "getList") {
-            fs.writeFileSync(path.join(typesDirPath, x.typeWhereName + ".ts"), x.typeWhere);
-            fs.writeFileSync(path.join(typesDirPath, x.typeOrderByName + ".ts"), x.typeOrderBy);
+        if (input.excludeTables != null) {
+            tables = tables.filter((x) => !input.excludeTables?.includes(x));
         }
-        if (x.kind === "postOne") {
-            fs.writeFileSync(path.join(typesDirPath, x.typeDataName + ".ts"), x.typeData);
-        }
-        if (x.kind === "patchOne") {
-            fs.writeFileSync(path.join(typesDirPath, x.typeDataName + ".ts"), x.typeData);
+        ctx.log.debug({ tableCount: tables.length, tables }, "tables after filter");
+        const data = await Promise.all(tables.flatMap((x) => [
+            getGetOneData(x, includeMappedFields),
+            getGetListData(x),
+            getGetListPaginatedData(x),
+            getPostOneData(x, specialCaseUuidColumn, includeMappedFields),
+            getPatchOneData(x, specialCaseUuidColumn, includeMappedFields),
+            getPatchListData(x),
+            getDeleteOneData(x),
+            getDeleteListData(x)
+        ]));
+        ctx.log.debug({ inputLength: data.length }, "SDK input data collected");
+        const artifacts = await getArtifacts(tables, includeMappedFields, specialCaseUuidColumn);
+        ctx.log.debug("getArtifacts() completed");
+        const artifactsSource = getArtifactsSource(artifacts);
+        const sdkSource = await getSDKSource(data, specialCaseUuidColumn, supplementClientOpts, artifacts, input.outputSqliteSchema);
+        ctx.log.debug("getSDKSource() completed");
+        const sdkFilename = "index.ts";
+        const sourceIRuntimeFilePath = fs.existsSync(path.join(__dirname, "../runtime", "IRuntime.ts"))
+            ? path.join(__dirname, "../runtime", "IRuntime.ts")
+            : path.join(__dirname, "../runtime", "IRuntime.js");
+        const IRuntimeFilename = path.basename(sourceIRuntimeFilePath);
+        const artifactsFilename = "artifacts.ts";
+        const tsConfigJSON = {
+            compilerOptions: {
+                module: "commonjs",
+                moduleResolution: "node",
+                target: "es2020",
+                declaration: true,
+                outDir: "./sdk-ts"
+            },
+            include: [sdkFilename, artifactsFilename, IRuntimeFilename]
+        };
+        const packageJSON = {
+            name: "temp",
+            version: "1.0.0",
+            // Deps need to be included so that they're inlined by ncc
+            dependencies: require("../../package.json").dependencies,
+            devDependencies: {
+                "@types/node": require("../../package.json").devDependencies["@types/node"],
+                typescript: require("../../package.json").devDependencies.typescript
+            },
+            // Not `resolutions` because npm used for install
+            overrides: {
+                // Fix for: `Cannot find type definition file for 'glob'`
+                glob: ">9.0.0"
+            }
+        };
+        const tmpDirPath = path.join(os.tmpdir(), 
+        // _ because - in filename is not supported by mysql2sqlite
+        `dsg_${node_crypto_1.default.randomUUID()}`.replace(/-/g, "_"));
+        ctx.log.debug({ tmpDirPath }, "writing SDK and artifacts to tmp dir");
+        fse.mkdirpSync(tmpDirPath);
+        fs.writeFileSync(path.join(tmpDirPath, sdkFilename), sdkSource);
+        fs.writeFileSync(path.join(tmpDirPath, artifactsFilename), artifactsSource);
+        fse.copyFileSync(sourceIRuntimeFilePath, path.join(tmpDirPath, IRuntimeFilename));
+        const typesDirPath = path.join(tmpDirPath, "types");
+        fse.mkdirpSync(typesDirPath);
+        fs.writeFileSync(path.join(typesDirPath, "_shared.ts"), getTypeShared());
+        for (let x of data) {
+            if (x.kind === "getOne") {
+                fs.writeFileSync(path.join(typesDirPath, x.typeFieldsName + ".ts"), x.typeFields);
+                fs.writeFileSync(path.join(typesDirPath, x.typeReturnBaseName + ".ts"), x.typeReturnBase);
+            }
+            if (x.kind === "getList") {
+                fs.writeFileSync(path.join(typesDirPath, x.typeWhereName + ".ts"), x.typeWhere);
+                fs.writeFileSync(path.join(typesDirPath, x.typeOrderByName + ".ts"), x.typeOrderBy);
+            }
+            if (x.kind === "postOne") {
+                fs.writeFileSync(path.join(typesDirPath, x.typeDataName + ".ts"), x.typeData);
+            }
+            if (x.kind === "patchOne") {
+                fs.writeFileSync(path.join(typesDirPath, x.typeDataName + ".ts"), x.typeData);
+            }
         }
-    }
-    fs.writeFileSync(path.join(typesDirPath, "index.ts"), getTypeTypesIndex(data));
-    fs.writeFileSync(path.join(tmpDirPath, "package.json"), JSON.stringify(packageJSON, null, 2));
-    fs.writeFileSync(path.join(tmpDirPath, "tsconfig.json"), JSON.stringify(tsConfigJSON, null, 2));
-    fse.copySync(__dirname, path.join(tmpDirPath, "src"));
-    const tmpBuildOutputPath = path.join(tmpDirPath, "sdk-ts");
-    const outdir = path.resolve(input.outdir);
-    const sdkOutputPath = path.join(outdir, "sdk-ts");
-    const nccVersion = "^0.33.0";
-    child_process.execSync("npm i", { cwd: tmpDirPath, stdio: "inherit" });
-    child_process.execSync(`npm_config_yes=true npx -p @vercel/ncc@${nccVersion} ncc build ./${sdkFilename} -o ${tmpBuildOutputPath} -e ./artifacts`, { cwd: tmpDirPath, stdio: "inherit" });
-    // TODO: workaround for artifacts.js not being output by ncc
-    fs.writeFileSync(path.join(tmpBuildOutputPath, "artifacts.js"), artifactsSource
-        .replace("export const artifacts: IArtifacts = ", "module.exports.artifacts = ")
-        .split("\n")
-        // Remove import
-        .slice(2)
-        .join("\n"));
-    // TODO: workaround for IRuntime.d.ts not being included
-    // copyFileSync hangs for some reason, so use writeFileSync + readFileSync instead
-    fs.writeFileSync(path.join(tmpBuildOutputPath, "IRuntime.d.ts"), fs.existsSync(path.join(__dirname, "../runtime", "IRuntime.d.ts"))
-        ? fs.readFileSync(path.join(__dirname, "../runtime", "IRuntime.d.ts"), "utf-8")
-        : fs.readFileSync(sourceIRuntimeFilePath, "utf-8"));
-    if (dialect === "mysql" && input.outputSqliteSchema) {
-        // Since mysql2sqlite outputs a malformed string if a column
-        // has the name `enum`, temporarily change the name to something else,
-        // then change it back.
-        const enumMarker = "`" + node_crypto_1.default.randomUUID() + "`";
-        const schemaMySql = Object.values(artifacts)
-            .reduce((acc, x) => {
-            let d = x.dump?.schema;
-            if (!d) {
-                return acc;
+        fs.writeFileSync(path.join(typesDirPath, "index.ts"), getTypeTypesIndex(data));
+        fs.writeFileSync(path.join(tmpDirPath, "package.json"), JSON.stringify(packageJSON, null, 2));
+        fs.writeFileSync(path.join(tmpDirPath, "tsconfig.json"), JSON.stringify(tsConfigJSON, null, 2));
+        fse.copySync(__dirname, path.join(tmpDirPath, "src"));
+        const tmpBuildOutputPath = path.join(tmpDirPath, "sdk-ts");
+        const outdir = path.resolve(input.outdir);
+        const sdkOutputPath = path.join(outdir, "sdk-ts");
+        const nccVersion = "^0.33.0";
+        child_process.execSync("npm i", { cwd: tmpDirPath, stdio: "inherit" });
+        child_process.execSync(`npm_config_yes=true npx -p @vercel/ncc@${nccVersion} ncc build ./${sdkFilename} -o ${tmpBuildOutputPath} -e ./artifacts`, { cwd: tmpDirPath, stdio: "inherit" });
+        // TODO: workaround for artifacts.js not being output by ncc
+        fs.writeFileSync(path.join(tmpBuildOutputPath, "artifacts.js"), artifactsSource
+            .replace("export const artifacts: IArtifacts = ", "module.exports.artifacts = ")
+            .split("\n")
+            // Remove import
+            .slice(2)
+            .join("\n"));
+        // TODO: workaround for IRuntime.d.ts not being included
+        // copyFileSync hangs for some reason, so use writeFileSync + readFileSync instead
+        fs.writeFileSync(path.join(tmpBuildOutputPath, "IRuntime.d.ts"), fs.existsSync(path.join(__dirname, "../runtime", "IRuntime.d.ts"))
+            ? fs.readFileSync(path.join(__dirname, "../runtime", "IRuntime.d.ts"), "utf-8")
+            : fs.readFileSync(sourceIRuntimeFilePath, "utf-8"));
+        if (input.outputSqliteSchema) {
+            let schemaSqlite = null;
+            if (ctx.dialect === "mysql") {
+                // Since mysql2sqlite outputs a malformed string if a column
+                // has the name `enum`, temporarily change the name to something else,
+                // then change it back.
+                const enumMarker = "`" + node_crypto_1.default.randomUUID() + "`";
+                const schemaMySql = Object.values(artifacts)
+                    .reduce((acc, x) => {
+                    let d = x.dump?.schema;
+                    if (!d) {
+                        return acc;
+                    }
+                    d = d.replace(/`enum`/g, enumMarker);
+                    d += ";";
+                    acc.push(d);
+                    return acc;
+                }, [])
+                    .join("\n\n");
+                const mysql2SqliteSrc = getMysql2sqliteSrc();
+                const mysql2SqlitePath = path.join(tmpDirPath, "mysql2sqlite");
+                fs.writeFileSync(mysql2SqlitePath, mysql2SqliteSrc);
+                fs.chmodSync(mysql2SqlitePath, 0o755);
+                const tmpMySqlSchemaFilename = "tmp.sql";
+                const tmpMySqlSchemaPath = path.join(tmpDirPath, tmpMySqlSchemaFilename);
+                fs.writeFileSync(tmpMySqlSchemaPath, schemaMySql);
+                schemaSqlite = child_process
+                    .execFileSync(mysql2SqlitePath, [tmpMySqlSchemaFilename], {
+                    cwd: tmpDirPath
+                })
+                    .toString();
+                schemaSqlite = schemaSqlite.replace(new RegExp(enumMarker, "g"), "`enum`");
             }
-            d = d.replace(/`enum`/g, enumMarker);
-            d += ";";
-            acc.push(d);
-            return acc;
-        }, [])
-            .join("\n\n");
-        const mysql2SqliteSrc = getMysql2sqliteSrc();
-        const mysql2SqlitePath = path.join(tmpDirPath, "mysql2sqlite");
-        fs.writeFileSync(mysql2SqlitePath, mysql2SqliteSrc);
-        fs.chmodSync(mysql2SqlitePath, 0o755);
-        const tmpMySqlSchemaFilename = "tmp.sql";
-        const tmpMySqlSchemaPath = path.join(tmpDirPath, tmpMySqlSchemaFilename);
-        fs.writeFileSync(tmpMySqlSchemaPath, schemaMySql);
-        let schemaSqlite = child_process
-            .execFileSync(mysql2SqlitePath, [tmpMySqlSchemaFilename], { cwd: tmpDirPath })
-            .toString();
-        schemaSqlite = schemaSqlite.replace(new RegExp(enumMarker, "g"), "`enum`");
-        const src = prettier.format(`module.exports = { schema: \`${schemaSqlite.replace(/`/g, "\\`")}\` }`, { parser: "babel" });
-        fs.writeFileSync(path.join(tmpBuildOutputPath, "artifacts.sqlite.js"), src);
-    }
-    if (!fs.existsSync(outdir)) {
-        fse.mkdirpSync(outdir);
-    }
-    fse.emptyDirSync(sdkOutputPath);
-    fse.copySync(tmpBuildOutputPath, sdkOutputPath);
-    fse.removeSync(tmpDirPath);
+            else if (ctx.dialect === "postgresql") {
+                schemaSqlite = await (0, pg2sqliteSchema_1.convertPgSchemaToSqlite)(ctx.pool);
+            }
+            if (schemaSqlite) {
+                const src = prettier.format(`module.exports = { schema: \`${schemaSqlite.replace(/`/g, "\\`")}\` }`, { parser: "babel" });
+                fs.writeFileSync(path.join(tmpBuildOutputPath, "artifacts.sqlite.js"), src);
+            }
+        }
+        if (!fs.existsSync(outdir)) {
+            fse.mkdirpSync(outdir);
+        }
+        fse.emptyDirSync(sdkOutputPath);
+        fse.copySync(tmpBuildOutputPath, sdkOutputPath);
+        ctx.log.debug({ outdir, sdkOutputPath }, "copy to output dir completed");
+        fse.removeSync(tmpDirPath);
+        ctx.log.debug("generate() completed");
+    });
 }
 function init(input) {
+    const ctx = getCtx();
     const { database, user, password, host, port, server } = input;
-    if (dialect === "mysql") {
-        let mysql = new MySQL_1.MySQL({
+    if (ctx.dialect === "mysql") {
+        const connectionOpts = {
             user,
             password,
-            host,
-            port,
-            database
-        });
-        query = mysql.query.bind(mysql);
+            host: host ?? "localhost",
+            port: port ?? 3306,
+            database,
+            ...input.clientOpts
+        };
+        ctx.log.debug({
+            dialect: "mysql",
+            host: connectionOpts.host,
+            port: connectionOpts.port,
+            database,
+            user
+        }, "connecting to MySQL");
+        const mysql = new MySQL_1.MySQL(connectionOpts);
+        ctx.query = mysql.query.bind(mysql);
+        ctx.pool = mysql;
+    }
+    if (ctx.dialect === "postgresql") {
+        const connectionOpts = {
+            host: host ?? "localhost",
+            port: port ?? 5432,
+            user,
+            password,
+            database,
+            ...input.clientOpts
+        };
+        ctx.log.debug({
+            dialect: "postgresql",
+            host: connectionOpts.host,
+            port: connectionOpts.port,
+            database,
+            user
+        }, "connecting to PostgreSQL");
+        const pool = new pg_1.Pool(connectionOpts);
+        ctx.query = (q, values) => pool.query(q, values ?? []).then((r) => r.rows);
+        ctx.pool = pool;
     }
 }
 // It's a bit awkward to put __whereNeedsProcessing, __prepareWhere on the class,
@@ -258,7 +338,7 @@ async function getSDKSource(input, specialCaseUuidColumn, supplementClientOpts,
       runtime: any;
       clientOpts: { [k: string]: any; },
       otherOpts?: { [k: string]: any; },
-      passBeforeValueToAfterCallback: boolean,
+      passBeforeValueToAfterCallback?: boolean,
      }) {
       let otherOpts = opts.otherOpts ?? {};
       if (opts.clientOpts.filename === ":memory:") {
@@ -271,7 +351,7 @@ async function getSDKSource(input, specialCaseUuidColumn, supplementClientOpts,
         : "otherOpts"}, artifacts);
       this.onHandlerMap = new Map();
       this.eventTarget = new EventTarget();
-      this.passBeforeValueToAfterCallback = opts.passBeforeValueToAfterCallback;
+      this.passBeforeValueToAfterCallback = opts.passBeforeValueToAfterCallback ?? false;
     }
 
     $use(middleware: TMiddleware) {
@@ -601,6 +681,7 @@ function getMethodSourceOnHandlerPostOne(x) {
     return `on${(0, capitalizeFirstLetter_1.capitalizeFirstLetter)(x.methodName)}(handler:
     (sdk: InstanceType<typeof SDK>, input: { data: ${x.typeDataName} },
     output: Partial<${getTypeReturnName(x.table)}>,
+    before: null,
     context: TContext,
   ) => Promise<void>
 ): void {
@@ -715,6 +796,7 @@ function getMethodSourceOnHandlerDeleteOne(x, findOnes) {
         .map((findOne) => `{ ${findOne.name}: ${findOne.type}${findOne.nullable ? " | null" : ""} }`)
         .join(" | ")}, },
     output: void,
+    before: ${getTypeReturnName(x.table)} | null,
     context: TContext,
   ) => Promise<void>
 ): void {
@@ -1068,7 +1150,7 @@ async function getMappedFields(table) {
             name: "uuid",
             // Replace `Id` with `Uuid`
             as: x.foreignKey.slice(0, -2) + "Uuid",
-            type: getBaseJSONType(uuidColumn.Type)
+            type: getBaseJSONType(uuidColumn.Type, getCtx().dialect)
         });
     }
     return out;
@@ -1424,6 +1506,7 @@ function getArtifactsSource(artifacts) {
     return prettier.format(src, { parser: "typescript" });
 }
 async function getArtifacts(tables, includeMappedFields, specialCaseUuidColumn) {
+    const ctx = getCtx();
     const tableMetaList = await Promise.all(tables.map(async (table) => {
         const [tableMeta, primaryKey, dumpSchema] = await Promise.all([
             getTableMeta(table),
@@ -1489,7 +1572,7 @@ async function getArtifacts(tables, includeMappedFields, specialCaseUuidColumn)
             }
             return {
                 kind: "scalar",
-                type: getBaseJSONType(t.Type),
+                type: getBaseJSONType(t.Type, ctx.dialect),
                 name: t.Field,
                 nullable,
                 hasDefaultValue: !!t.Default
@@ -1534,7 +1617,7 @@ async function getArtifacts(tables, includeMappedFields, specialCaseUuidColumn)
     }, {});
     return artifacts;
 }
-const getRelationInfo = _.memoize(async function getRelationInfo(table) {
+const getRelationInfo = (0, memoize_1.default)(async function getRelationInfo(table) {
     const relationsManyToOne = await getRelationsManyToOne(table);
     const relationsOneToMany = await getRelationsOneToMany(table);
     let out = [];
@@ -1602,14 +1685,14 @@ const getRelationInfo = _.memoize(async function getRelationInfo(table) {
         return acc;
     }, []);
     out = out.concat(relationsManyToMany);
-    out = _.sortBy((x) => x.table, out);
+    out = _.sortBy([(x) => x.table, (x) => x.name], out);
     return out;
-});
+}, (table) => getCtx().runId + ":" + table);
 function getRelationManyToOneFieldName(x) {
     return changeCase.camelCase(x.foreignKey.replace(new RegExp(x.referencedKey + "$", "i"), ""));
 }
 // TODO: not sure if this logic is correct
-async function getJunctionTables() {
+const getJunctionTables = (0, memoize_1.default)(async function getJunctionTables() {
     const tables = await getTableNames();
     return (await Promise.all(tables.map(async (table) => {
         const relations = await getRelationsManyToOne(table);
@@ -1624,14 +1707,15 @@ async function getJunctionTables() {
         }
         return null;
     }))).filter(isNotNullOrUndefined_1.isNotNullOrUndefined);
-}
+}, () => getCtx().runId);
 // `from` relations
 // https://stackoverflow.com/a/54732547
-// Note: this function
-// isn't pure (because of `dialect`), but that's fine because its value
-// isn't expected to change.
-const getRelationsManyToOne = _.memoize(async function getRelationsManyToOne(table) {
-    const sql = `
+const getRelationsManyToOne = (0, memoize_1.default)(async function getRelationsManyToOne(table) {
+    const { dialect, query } = getCtx();
+    const tableMeta = await getTableMeta(table);
+    let rs;
+    if (dialect === "mysql") {
+        const sql = `
   SELECT
     TABLE_SCHEMA as db,
     TABLE_NAME as t1,
@@ -1646,8 +1730,19 @@ const getRelationsManyToOne = _.memoize(async function getRelationsManyToOne(tab
     AND REFERENCED_TABLE_NAME IS NOT NULL
     AND (TABLE_NAME = ?);
   `;
-    const tableMeta = await getTableMeta(table);
-    const rs = await query(sql, [table]);
+        rs = await query(sql, [table]);
+    }
+    else if (dialect === "postgresql") {
+        rs = await query(`SELECT kcu.column_name AS "t1Field", ccu.table_name AS t2, ccu.column_name AS "t2Field"
+       FROM information_schema.key_column_usage kcu
+       JOIN information_schema.referential_constraints rc ON kcu.constraint_name = rc.constraint_name AND kcu.table_schema = rc.constraint_schema
+       JOIN information_schema.constraint_column_usage ccu ON rc.unique_constraint_name = ccu.constraint_name AND rc.unique_constraint_schema = ccu.table_schema
+       WHERE kcu.table_schema = 'public' AND kcu.table_name = $1
+       ORDER BY ccu.table_name, ccu.column_name`, [table]);
+    }
+    else {
+        throw new Error("Unsupported dialect: " + dialect);
+    }
     const xs = await Promise.all(_.uniqWith(_.isEqual, rs.map(async (v) => {
         return {
             table: table,
@@ -1657,11 +1752,14 @@ const getRelationsManyToOne = _.memoize(async function getRelationsManyToOne(tab
             nullable: tableMeta.find((m) => m.Field === v.t1Field)?.Null === "YES"
         };
     })));
-    return _.sortBy((x) => x.referencedTable, xs);
-});
+    return _.sortBy([(x) => x.referencedTable, (x) => x.referencedKey, (x) => x.foreignKey], xs);
+}, (table) => getCtx().runId + ":" + table);
 // `to` relations
-const getRelationsOneToMany = _.memoize(async function getRelationsOneToMany(table) {
-    const sql = `
+const getRelationsOneToMany = (0, memoize_1.default)(async function getRelationsOneToMany(table) {
+    const { dialect, query } = getCtx();
+    let rs;
+    if (dialect === "mysql") {
+        const sql = `
   SELECT
     TABLE_SCHEMA as db,
     TABLE_NAME as t1,
@@ -1676,19 +1774,30 @@ const getRelationsOneToMany = _.memoize(async function getRelationsOneToMany(tab
     AND REFERENCED_TABLE_NAME IS NOT NULL
     AND (REFERENCED_TABLE_NAME = ?);
   `;
-    const rs = await query(sql, [table]);
+        rs = await query(sql, [table]);
+    }
+    else if (dialect === "postgresql") {
+        rs = await query(`SELECT kcu.table_name AS t1, kcu.column_name AS "t1Field", ccu.column_name AS "t2Field"
+       FROM information_schema.key_column_usage kcu
+       JOIN information_schema.referential_constraints rc ON kcu.constraint_name = rc.constraint_name AND kcu.table_schema = rc.constraint_schema
+       JOIN information_schema.constraint_column_usage ccu ON rc.unique_constraint_name = ccu.constraint_name AND rc.unique_constraint_schema = ccu.table_schema
+       WHERE kcu.table_schema = 'public' AND ccu.table_name = $1
+       ORDER BY kcu.table_name, kcu.column_name`, [table]);
+    }
+    else {
+        throw new Error("Unsupported dialect: " + dialect);
+    }
     const xs = await Promise.all(_.uniqWith(_.isEqual, rs.map(async (v) => {
         return {
             table: table,
             foreignKey: v.t2Field,
             referencedTable: v.t1,
             referencedKey: v.t1Field,
-            // TODO? I think this is right, since it's one-to-many, so a list
             nullable: false
         };
     })));
-    return _.sortBy((x) => x.referencedKey, _.sortBy((x) => x.referencedTable, xs));
-});
+    return _.sortBy([(x) => x.referencedTable, (x) => x.referencedKey, (x) => x.foreignKey], xs);
+}, (table) => getCtx().runId + ":" + table);
 async function getPrimaryColumn(table) {
     const tableMeta = await getTableMeta(table);
     const columns = tableMeta.filter((x) => x.Key === "PRI");
@@ -1698,7 +1807,7 @@ async function getPrimaryColumn(table) {
     const column = columns[0];
     return {
         name: column.Field,
-        type: getBaseJSONType(column.Type),
+        type: getBaseJSONType(column.Type, getCtx().dialect),
         nullable: column.Null === "YES"
     };
 }
@@ -1710,7 +1819,7 @@ async function getUniqueColumns(table, specialCaseUuidColumn) {
         (specialCaseUuidColumn && x.Field === "uuid"))
         .map((x) => ({
         name: x.Field,
-        type: getBaseJSONType(x.Type),
+        type: getBaseJSONType(x.Type, getCtx().dialect),
         nullable: x.Null === "YES"
     }));
 }
@@ -1726,23 +1835,150 @@ async function getUuidColumn(table) {
         nullable: column.Null === "YES"
     };
 }
-const getTableMeta = _.memoize(async function getTableMeta(table) {
+const getPgEnumDefinition = (0, memoize_1.default)(async function getPgEnumDefinition(udtSchema, udtName) {
+    const { dialect, query } = getCtx();
+    if (dialect !== "postgresql")
+        return null;
+    const rows = await query(`SELECT e.enumlabel FROM pg_enum e
+     JOIN pg_type t ON e.enumtypid = t.oid
+     JOIN pg_catalog.pg_namespace n ON t.typnamespace = n.oid
+     WHERE t.typname = $2 AND n.nspname = $1
+     ORDER BY e.enumsortorder`, [udtSchema, udtName]);
+    if (rows.length === 0)
+        return null;
+    const labels = rows.map((r) => String(r.enumlabel).replace(/'/g, "''"));
+    return "enum('" + labels.join("', '") + "')";
+}, (udtSchema, udtName) => getCtx().runId + ":" + udtSchema + ":" + udtName);
+const getTableMeta = (0, memoize_1.default)(async function getTableMeta(table) {
+    const ctx = getCtx();
+    const { dialect, query } = ctx;
+    ctx.log.debug({ table }, "getTableMeta() fetching");
     if (dialect === "mysql") {
         return query("DESCRIBE ??", [table]).then((xs) => _.sortBy((x) => x.Field, xs));
     }
+    if (dialect === "postgresql") {
+        const columns = await query(`SELECT column_name AS "Field", data_type, udt_schema, udt_name, character_maximum_length AS char_max, is_nullable, column_default AS "Default"
+       FROM information_schema.columns
+       WHERE table_schema = 'public' AND table_name = $1
+       ORDER BY ordinal_position`, [table]);
+        const keyInfo = await query(`SELECT a.attname AS col, 'PRI' AS key_type
+       FROM pg_index i
+       JOIN pg_attribute a ON a.attrelid = i.indrelid AND a.attnum = ANY(i.indkey) AND NOT a.attisdropped AND a.attnum > 0
+       JOIN pg_class c ON c.oid = i.indrelid
+       JOIN pg_namespace n ON n.oid = c.relnamespace
+       WHERE n.nspname = 'public' AND c.relname = $1 AND i.indisprimary
+       UNION ALL
+       SELECT kcu.column_name AS col, 'UNI' AS key_type
+       FROM information_schema.table_constraints tc
+       JOIN information_schema.key_column_usage kcu ON tc.constraint_name = kcu.constraint_name AND tc.table_schema = kcu.table_schema
+       WHERE tc.table_schema = 'public' AND tc.table_name = $1 AND tc.constraint_type = 'UNIQUE'
+       UNION ALL
+       SELECT kcu.column_name AS col, 'MUL' AS key_type
+       FROM information_schema.table_constraints tc
+       JOIN information_schema.key_column_usage kcu ON tc.constraint_name = kcu.constraint_name AND tc.table_schema = kcu.table_schema
+       WHERE tc.table_schema = 'public' AND tc.table_name = $1 AND tc.constraint_type = 'FOREIGN KEY'`, [table]);
+        const keyMap = new Map();
+        for (const k of keyInfo) {
+            if (!keyMap.has(k.col) || k.key_type === "PRI")
+                keyMap.set(k.col, k.key_type);
+        }
+        const udtKeys = [
+            ...new Map(columns
+                .filter((c) => c.data_type === "USER-DEFINED" &&
+                c.udt_schema != null &&
+                c.udt_name != null)
+                .map((c) => [`${c.udt_schema}.${c.udt_name}`, c])).keys()
+        ];
+        const udtPairs = udtKeys.map((k) => {
+            const [s, n] = k.split(".", 2);
+            return [s, n];
+        });
+        const enumDefs = await Promise.all(udtPairs.map(([schema, name]) => getPgEnumDefinition(schema, name)));
+        const enumMap = new Map(udtKeys.map((k, i) => [k, enumDefs[i] ?? "varchar(255)"]));
+        const cols = columns.map((c) => {
+            let type;
+            if (c.data_type === "USER-DEFINED" &&
+                c.udt_schema != null &&
+                c.udt_name != null) {
+                const enumKey = `${c.udt_schema}.${c.udt_name}`;
+                type = enumMap.get(enumKey) ?? "character varying(255)";
+            }
+            else {
+                type = c.data_type;
+                if ((c.data_type === "character varying" || c.data_type === "character") &&
+                    c.char_max != null) {
+                    type += "(" + c.char_max + ")";
+                }
+            }
+            return {
+                Field: c.Field,
+                Type: type,
+                Null: c.is_nullable === "YES" ? "YES" : "NO",
+                Key: keyMap.get(c.Field) ?? "",
+                // Preserve `null` when there is no default so that
+                // required-field detection (via `hasDefault`) works
+                // consistently with the MySQL `DESCRIBE` output.
+                Default: c.Default,
+                ...(c.data_type === "USER-DEFINED" &&
+                    c.udt_schema != null &&
+                    c.udt_name != null
+                    ? { PgType: c.udt_name }
+                    : {})
+            };
+        });
+        return _.sortBy((c) => c.Field, cols);
+    }
     throw new Error("Unsupported dialect: " + dialect);
-});
-function getShowCreateTable(table) {
+}, (table) => getCtx().runId + ":" + table);
+async function getShowCreateTable(table) {
+    const { dialect, query } = getCtx();
     if (dialect === "mysql") {
         return query("SHOW CREATE TABLE ??", [table]).then((xs) => xs[0]["Create Table"]
             // https://github.com/bradzacher/mysqldump/blob/66839a57e572a07c046b0ba98753f30a7026cbd8/src/getSchemaDump.ts#L65
             .replace(/AUTO_INCREMENT\s*=\s*\d+ /g, ""));
     }
+    if (dialect === "postgresql") {
+        const [tableMeta, relations] = await Promise.all([
+            getTableMeta(table),
+            getRelationsManyToOne(table)
+        ]);
+        const refByFk = new Map(relations.map((r) => [r.foreignKey, r]));
+        const columnDefs = tableMeta.map((c) => {
+            const isSerialPk = c.Key === "PRI" &&
+                c.Default != null &&
+                c.Default !== "" &&
+                /nextval\s*\(/i.test(c.Default);
+            if (isSerialPk) {
+                const baseType = (c.PgType ?? c.Type).toLowerCase();
+                const serialType = baseType === "bigint" || baseType === "int8"
+                    ? "BIGSERIAL"
+                    : baseType === "smallint" || baseType === "int2"
+                        ? "SMALLSERIAL"
+                        : "SERIAL";
+                return `"${c.Field.replace(/"/g, '""')}" ${serialType} PRIMARY KEY`;
+            }
+            const pgType = c.PgType ?? c.Type;
+            let def = `"${c.Field.replace(/"/g, '""')}" ${pgType} ${c.Null === "YES" ? "NULL" : "NOT NULL"}`;
+            if (c.Default != null && c.Default !== "") {
+                def += ` DEFAULT ${c.Default}`;
+            }
+            if (c.Key === "PRI")
+                def += " PRIMARY KEY";
+            if (c.Key === "UNI")
+                def += " UNIQUE";
+            const ref = refByFk.get(c.Field);
+            if (ref != null) {
+                def += ` REFERENCES "${ref.referencedTable.replace(/"/g, '""')}" ("${ref.referencedKey.replace(/"/g, '""')}")`;
+            }
+            return def;
+        });
+        return `CREATE TABLE "${table.replace(/"/g, '""')}" (\n  ${columnDefs.join(",\n  ")}\n);`;
+    }
     return Promise.resolve(null);
 }
 function getJSONSchemaObjProperties(tableMeta) {
     return tableMeta.reduce((acc, m) => {
-        const baseType = getBaseJSONType(m.Type);
+        const baseType = getBaseJSONType(m.Type, getCtx().dialect);
         const format = getPropertyFormat(m.Type);
         const nullable = m.Null === "YES";
         const isEnum = m.Type.startsWith("enum");
@@ -1778,7 +2014,29 @@ function getJSONTypes(baseType, nullable) {
     return baseType;
 }
 // https://github.com/mysqljs/mysql#type-casting
-function getBaseJSONType(sqlType) {
+function getBaseJSONType(sqlType, dialect) {
+    if (dialect === "postgresql") {
+        if (sqlType === "boolean" || sqlType === "bool")
+            return "boolean";
+        if (["smallint", "int2", "integer", "int4", "bigint", "int8"].includes(sqlType)) {
+            return "integer";
+        }
+        if (["real", "float4", "double precision", "float8"].includes(sqlType) ||
+            sqlType.startsWith("numeric") ||
+            sqlType.startsWith("decimal")) {
+            return "number";
+        }
+        if (["text", "uuid", "json", "jsonb"].includes(sqlType) ||
+            sqlType === "date" ||
+            sqlType === "time" ||
+            sqlType.startsWith("timestamp") ||
+            sqlType.startsWith("character varying") ||
+            sqlType.startsWith("character") ||
+            sqlType.startsWith("enum")) {
+            return "string";
+        }
+        throw new Error("Unable to map to JSON type: " + sqlType);
+    }
     if (
     // TODO?
     sqlType === "tinyint(1)" ||
@@ -1863,7 +2121,10 @@ function getPropertyEnum(sqlType) {
     return c;
 }
 function getPropertyFormat(sqlType) {
-    if (sqlType === "datetime" || sqlType === "datetime2" || sqlType === "timestamp") {
+    if (sqlType === "datetime" ||
+        sqlType === "datetime2" ||
+        sqlType === "timestamp" ||
+        sqlType.startsWith("timestamp")) {
         // TODO: not sure this is correct for `timestamp`
         return "date-time";
     }
@@ -1881,12 +2142,18 @@ function getPropertyFormat(sqlType) {
     }
     return undefined;
 }
-async function getTableNames() {
+const getTableNames = (0, memoize_1.default)(async function getTableNames() {
+    const { dialect, query } = getCtx();
     if (dialect === "mysql") {
         return query("SHOW TABLES").then((xs) => xs.flatMap((x) => Object.values(x)).sort());
     }
+    if (dialect === "postgresql") {
+        return query(`SELECT table_name FROM information_schema.tables
+         WHERE table_schema = 'public' AND table_type = 'BASE TABLE'
+         ORDER BY table_name`).then((rows) => rows.map((r) => r.table_name));
+    }
     throw new Error("Unsupported dialect: " + dialect);
-}
+}, () => getCtx().runId);
 function getMysql2sqliteSrc() {
     return `#!/usr/bin/awk -f