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

package/dist/generation/generate.js

@@ -30,24 +30,27 @@ 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 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 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: ""
 };
@@ -62,159 +65,178 @@ async function generate(input) {
     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 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 ctx = {
+        runId: node_crypto_1.default.randomUUID(),
+        dialect: input.dialect,
+        query: undefined
     };
-    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);
+    return ctxStorage.run(ctx, async () => {
+        init(input);
+        let tables = await getTableNames();
+        if (tables.length === 0) {
+            throw new Error("No tables found");
         }
-        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.tables != null) {
+            tables = tables.filter((x) => input.tables?.includes(x));
         }
-        if (x.kind === "postOne") {
-            fs.writeFileSync(path.join(typesDirPath, x.typeDataName + ".ts"), x.typeData);
+        if (input.excludeTables != null) {
+            tables = tables.filter((x) => !input.excludeTables?.includes(x));
         }
-        if (x.kind === "patchOne") {
-            fs.writeFileSync(path.join(typesDirPath, x.typeDataName + ".ts"), x.typeData);
+        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 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, "_"));
+        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) {
+        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 (getCtx().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;
+                }
+                d = d.replace(/`enum`/g, enumMarker);
+                d += ";";
+                acc.push(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);
-        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);
+            }, [])
+                .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);
+    });
 }
 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 mysql = new MySQL_1.MySQL({
             user,
             password,
             host,
             port,
             database
         });
-        query = mysql.query.bind(mysql);
+        ctx.query = mysql.query.bind(mysql);
+    }
+    if (ctx.dialect === "postgresql") {
+        const pool = new pg_1.Pool({
+            host: host ?? "localhost",
+            port: port ?? 5432,
+            user,
+            password,
+            database
+        });
+        ctx.query = (q, values) => pool.query(q, values ?? []).then((r) => r.rows);
     }
 }
 // It's a bit awkward to put __whereNeedsProcessing, __prepareWhere on the class,
@@ -258,7 +280,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 +293,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) {
@@ -1068,7 +1090,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;
@@ -1489,7 +1511,7 @@ async function getArtifacts(tables, includeMappedFields, specialCaseUuidColumn)
             }
             return {
                 kind: "scalar",
-                type: getBaseJSONType(t.Type),
+                type: getBaseJSONType(t.Type, getCtx().dialect),
                 name: t.Field,
                 nullable,
                 hasDefaultValue: !!t.Default
@@ -1534,7 +1556,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 = [];
@@ -1604,7 +1626,7 @@ const getRelationInfo = _.memoize(async function getRelationInfo(table) {
     out = out.concat(relationsManyToMany);
     out = _.sortBy((x) => x.table, out);
     return out;
-});
+}, (table) => getCtx().runId + ":" + table);
 function getRelationManyToOneFieldName(x) {
     return changeCase.camelCase(x.foreignKey.replace(new RegExp(x.referencedKey + "$", "i"), ""));
 }
@@ -1627,11 +1649,12 @@ async function getJunctionTables() {
 }
 // `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 +1669,18 @@ 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`, [table]);
+    }
+    else {
+        throw new Error("Unsupported dialect: " + dialect);
+    }
     const xs = await Promise.all(_.uniqWith(_.isEqual, rs.map(async (v) => {
         return {
             table: table,
@@ -1658,10 +1691,13 @@ const getRelationsManyToOne = _.memoize(async function getRelationsManyToOne(tab
         };
     })));
     return _.sortBy((x) => x.referencedTable, 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 +1712,29 @@ 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`, [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));
-});
+}, (table) => getCtx().runId + ":" + table);
 async function getPrimaryColumn(table) {
     const tableMeta = await getTableMeta(table);
     const columns = tableMeta.filter((x) => x.Key === "PRI");
@@ -1698,7 +1744,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 +1756,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 +1772,116 @@ async function getUuidColumn(table) {
         nullable: column.Null === "YES"
     };
 }
-const getTableMeta = _.memoize(async function getTableMeta(table) {
+const getPgEnumDefinition = (0, memoize_1.default)(async function getPgEnumDefinition(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 = $1 AND n.nspname = 'public'
+     ORDER BY e.enumsortorder`, [udtName]);
+    if (rows.length === 0)
+        return null;
+    const labels = rows.map((r) => String(r.enumlabel).replace(/'/g, "''"));
+    return "enum('" + labels.join("', '") + "')";
+}, (udtName) => getCtx().runId + ":" + udtName);
+const getTableMeta = (0, memoize_1.default)(async function getTableMeta(table) {
+    const { dialect, query } = getCtx();
     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_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 udtNames = [
+            ...new Set(columns
+                .filter((c) => c.data_type === "USER-DEFINED" && c.udt_name != null)
+                .map((c) => c.udt_name))
+        ];
+        const enumDefs = await Promise.all(udtNames.map((udt) => getPgEnumDefinition(udt)));
+        const enumMap = new Map(udtNames.map((udt, i) => [udt, enumDefs[i] ?? "varchar(255)"]));
+        return columns.map((c) => {
+            let type;
+            if (c.data_type === "USER-DEFINED" && c.udt_name != null) {
+                type = enumMap.get(c.udt_name) ?? "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) ?? "",
+                Default: c.Default ?? ""
+            };
+        });
+    }
     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) => {
+            let def = `"${c.Field.replace(/"/g, '""')}" ${c.Type} ${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 +1917,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 +2024,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";
     }
@@ -1882,9 +2046,15 @@ function getPropertyFormat(sqlType) {
     return undefined;
 }
 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);
 }
 function getMysql2sqliteSrc() {