schema-seed-adapter-postgres 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 schema-seed contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,5 @@
+# @schema-seed/adapter-postgres
+
+Part of the [schema-seed](https://github.com/AliNazar-111/schema-seed) project.
+
+For full documentation and usage, please visit the [main repository](https://github.com/AliNazar-111/schema-seed).

package/dist/index.cjs

@@ -0,0 +1,178 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  PostgresAdapter: () => PostgresAdapter,
+  createPostgresAdapter: () => createPostgresAdapter
+});
+module.exports = __toCommonJS(index_exports);
+var import_pg = __toESM(require("pg"), 1);
+var import_schema_seed_core = require("schema-seed-core");
+var { Client } = import_pg.default;
+var PostgresAdapter = class {
+  client;
+  constructor(connectionString) {
+    this.client = new Client({ connectionString });
+  }
+  async connect() {
+    await this.client.connect();
+  }
+  async disconnect() {
+    await this.client.end();
+  }
+  async begin() {
+    await this.client.query("BEGIN");
+  }
+  async commit() {
+    await this.client.query("COMMIT");
+  }
+  async rollback() {
+    await this.client.query("ROLLBACK");
+  }
+  async introspectSchema() {
+    const schema = { tables: {} };
+    const tablesRes = await this.client.query(`
+      SELECT table_name, table_schema
+      FROM information_schema.tables
+      WHERE table_schema NOT IN ('information_schema', 'pg_catalog')
+      AND table_type = 'BASE TABLE'
+    `);
+    for (const row of tablesRes.rows) {
+      const tableName = row.table_name;
+      schema.tables[tableName] = {
+        name: tableName,
+        schema: row.table_schema,
+        columns: {},
+        foreignKeys: [],
+        uniqueConstraints: []
+      };
+    }
+    const columnsRes = await this.client.query(`
+      SELECT table_name, column_name, data_type, is_nullable, column_default, udt_name
+      FROM information_schema.columns
+      WHERE table_schema NOT IN ('information_schema', 'pg_catalog')
+    `);
+    for (const row of columnsRes.rows) {
+      if (!schema.tables[row.table_name]) continue;
+      const col = {
+        name: row.column_name,
+        type: this.mapPostgresType(row.data_type, row.udt_name),
+        rawType: row.data_type,
+        nullable: row.is_nullable === "YES",
+        defaultValue: row.column_default,
+        isAutoIncrement: row.column_default?.includes("nextval") || false
+      };
+      schema.tables[row.table_name].columns[row.column_name] = col;
+    }
+    const fkRes = await this.client.query(`
+      SELECT
+          tc.table_name, kcu.column_name, 
+          ccu.table_name AS foreign_table_name,
+          ccu.column_name AS foreign_column_name 
+      FROM 
+          information_schema.table_constraints AS tc 
+          JOIN information_schema.key_column_usage AS kcu
+            ON tc.constraint_name = kcu.constraint_name
+          JOIN information_schema.constraint_column_usage AS ccu
+            ON ccu.constraint_name = tc.constraint_name
+      WHERE tc.constraint_type = 'FOREIGN KEY'
+    `);
+    for (const row of fkRes.rows) {
+      if (!schema.tables[row.table_name]) continue;
+      schema.tables[row.table_name].foreignKeys.push({
+        columns: [row.column_name],
+        referencedTable: row.foreign_table_name,
+        referencedColumns: [row.foreign_column_name]
+      });
+    }
+    const pkRes = await this.client.query(`
+      SELECT tc.table_name, kcu.column_name
+      FROM information_schema.table_constraints tc
+      JOIN information_schema.key_column_usage kcu ON tc.constraint_name = kcu.constraint_name
+      WHERE tc.constraint_type = 'PRIMARY KEY'
+    `);
+    for (const row of pkRes.rows) {
+      if (!schema.tables[row.table_name]) continue;
+      if (!schema.tables[row.table_name].primaryKey) {
+        schema.tables[row.table_name].primaryKey = { columns: [] };
+      }
+      schema.tables[row.table_name].primaryKey.columns.push(row.column_name);
+    }
+    return schema;
+  }
+  async insertBatch(batch) {
+    const { tableName, rows } = batch;
+    if (rows.length === 0) return;
+    const columns = Object.keys(rows[0]);
+    const placeholders = rows.map(
+      (_, rowIndex) => `(${columns.map((_2, colIndex) => `$${rowIndex * columns.length + colIndex + 1}`).join(", ")})`
+    ).join(", ");
+    const values = rows.flatMap((row) => columns.map((col) => row[col]));
+    const query = `INSERT INTO "${tableName}" (${columns.map((c) => `"${c}"`).join(", ")}) VALUES ${placeholders}`;
+    await this.client.query(query, values);
+  }
+  async truncateTables(tableNames) {
+    if (tableNames.length === 0) return;
+    const quotedNames = tableNames.map((t) => `"${t}"`).join(", ");
+    await this.client.query(`TRUNCATE ${quotedNames} RESTART IDENTITY CASCADE`);
+  }
+  capabilities = {
+    enums: true,
+    deferrableConstraints: true,
+    returning: true,
+    identityInsert: false
+  };
+  mapPostgresType(pgType, udtName) {
+    const type = pgType.toLowerCase();
+    const udt = udtName.toLowerCase();
+    if (type.includes("int") || type.includes("serial")) {
+      if (type.includes("big") || udt.includes("int8")) return import_schema_seed_core.NormalizedSqlType.BIGINT;
+      return import_schema_seed_core.NormalizedSqlType.INT;
+    }
+    if (type.includes("char") || type.includes("text") || type === "character varying") return import_schema_seed_core.NormalizedSqlType.STRING;
+    if (type.includes("bool")) return import_schema_seed_core.NormalizedSqlType.BOOLEAN;
+    if (type.includes("timestamp") || type.includes("date")) return import_schema_seed_core.NormalizedSqlType.DATETIME;
+    if (type.includes("json")) return import_schema_seed_core.NormalizedSqlType.JSON;
+    if (type.includes("numeric") || type.includes("decimal")) return import_schema_seed_core.NormalizedSqlType.DECIMAL;
+    if (type.includes("float") || type.includes("real")) return import_schema_seed_core.NormalizedSqlType.FLOAT;
+    if (udt === "uuid") return import_schema_seed_core.NormalizedSqlType.UUID;
+    return import_schema_seed_core.NormalizedSqlType.STRING;
+  }
+};
+function createPostgresAdapter(connectionString) {
+  return new PostgresAdapter(connectionString);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  PostgresAdapter,
+  createPostgresAdapter
+});
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import pg from 'pg'\nconst { Client } = pg\nimport {\n    SqlAdapter,\n    SchemaGraph,\n    SeedBatch,\n    NormalizedSqlType,\n    TableSchema,\n    ColumnSchema\n} from 'schema-seed-core'\n\nexport class PostgresAdapter implements SqlAdapter {\n    private client: pg.Client\n\n    constructor(connectionString: string) {\n        this.client = new Client({ connectionString })\n    }\n\n    async connect(): Promise<void> {\n        await this.client.connect()\n    }\n\n    async disconnect(): Promise<void> {\n        await this.client.end()\n    }\n\n    async begin(): Promise<void> {\n        await this.client.query('BEGIN')\n    }\n\n    async commit(): Promise<void> {\n        await this.client.query('COMMIT')\n    }\n\n    async rollback(): Promise<void> {\n        await this.client.query('ROLLBACK')\n    }\n\n    async introspectSchema(): Promise<SchemaGraph> {\n        const schema: SchemaGraph = { tables: {} }\n\n        // Get tables\n        const tablesRes = await this.client.query(`\n      SELECT table_name, table_schema\n      FROM information_schema.tables\n      WHERE table_schema NOT IN ('information_schema', 'pg_catalog')\n      AND table_type = 'BASE TABLE'\n    `)\n\n        for (const row of tablesRes.rows) {\n            const tableName = row.table_name\n            schema.tables[tableName] = {\n                name: tableName,\n                schema: row.table_schema,\n                columns: {},\n                foreignKeys: [],\n                uniqueConstraints: []\n            }\n        }\n\n        // Get columns\n        const columnsRes = await this.client.query(`\n      SELECT table_name, column_name, data_type, is_nullable, column_default, udt_name\n      FROM information_schema.columns\n      WHERE table_schema NOT IN ('information_schema', 'pg_catalog')\n    `)\n\n        for (const row of columnsRes.rows) {\n            if (!schema.tables[row.table_name]) continue\n\n            const col: ColumnSchema = {\n                name: row.column_name,\n                type: this.mapPostgresType(row.data_type, row.udt_name),\n                rawType: row.data_type,\n                nullable: row.is_nullable === 'YES',\n                defaultValue: row.column_default,\n                isAutoIncrement: row.column_default?.includes('nextval') || false\n            }\n            schema.tables[row.table_name].columns[row.column_name] = col\n        }\n\n        // Get Foreign Keys\n        const fkRes = await this.client.query(`\n      SELECT\n          tc.table_name, kcu.column_name, \n          ccu.table_name AS foreign_table_name,\n          ccu.column_name AS foreign_column_name \n      FROM \n          information_schema.table_constraints AS tc \n          JOIN information_schema.key_column_usage AS kcu\n            ON tc.constraint_name = kcu.constraint_name\n          JOIN information_schema.constraint_column_usage AS ccu\n            ON ccu.constraint_name = tc.constraint_name\n      WHERE tc.constraint_type = 'FOREIGN KEY'\n    `)\n\n        for (const row of fkRes.rows) {\n            if (!schema.tables[row.table_name]) continue\n            schema.tables[row.table_name].foreignKeys.push({\n                columns: [row.column_name],\n                referencedTable: row.foreign_table_name,\n                referencedColumns: [row.foreign_column_name]\n            })\n        }\n\n        // Get Primary Keys\n        const pkRes = await this.client.query(`\n      SELECT tc.table_name, kcu.column_name\n      FROM information_schema.table_constraints tc\n      JOIN information_schema.key_column_usage kcu ON tc.constraint_name = kcu.constraint_name\n      WHERE tc.constraint_type = 'PRIMARY KEY'\n    `)\n\n        for (const row of pkRes.rows) {\n            if (!schema.tables[row.table_name]) continue\n            if (!schema.tables[row.table_name].primaryKey) {\n                schema.tables[row.table_name].primaryKey = { columns: [] }\n            }\n            schema.tables[row.table_name].primaryKey!.columns.push(row.column_name)\n        }\n\n        return schema\n    }\n\n    async insertBatch(batch: SeedBatch): Promise<void> {\n        const { tableName, rows } = batch\n        if (rows.length === 0) return\n\n        const columns = Object.keys(rows[0])\n        const placeholders = rows.map((_, rowIndex) =>\n            `(${columns.map((_, colIndex) => `$${rowIndex * columns.length + colIndex + 1}`).join(', ')})`\n        ).join(', ')\n\n        const values = rows.flatMap(row => columns.map(col => row[col]))\n        const query = `INSERT INTO \"${tableName}\" (${columns.map(c => `\"${c}\"`).join(', ')}) VALUES ${placeholders}`\n\n        await this.client.query(query, values)\n    }\n\n    async truncateTables(tableNames: string[]): Promise<void> {\n        if (tableNames.length === 0) return\n        const quotedNames = tableNames.map(t => `\"${t}\"`).join(', ')\n        await this.client.query(`TRUNCATE ${quotedNames} RESTART IDENTITY CASCADE`)\n    }\n\n    readonly capabilities = {\n        enums: true,\n        deferrableConstraints: true,\n        returning: true,\n        identityInsert: false\n    }\n\n    private mapPostgresType(pgType: string, udtName: string): NormalizedSqlType {\n        const type = pgType.toLowerCase()\n        const udt = udtName.toLowerCase()\n\n        if (type.includes('int') || type.includes('serial')) {\n            if (type.includes('big') || udt.includes('int8')) return NormalizedSqlType.BIGINT\n            return NormalizedSqlType.INT\n        }\n        if (type.includes('char') || type.includes('text') || type === 'character varying') return NormalizedSqlType.STRING\n        if (type.includes('bool')) return NormalizedSqlType.BOOLEAN\n        if (type.includes('timestamp') || type.includes('date')) return NormalizedSqlType.DATETIME\n        if (type.includes('json')) return NormalizedSqlType.JSON\n        if (type.includes('numeric') || type.includes('decimal')) return NormalizedSqlType.DECIMAL\n        if (type.includes('float') || type.includes('real')) return NormalizedSqlType.FLOAT\n        if (udt === 'uuid') return NormalizedSqlType.UUID\n\n        return NormalizedSqlType.STRING\n    }\n}\n\nexport function createPostgresAdapter(connectionString: string): PostgresAdapter {\n    return new PostgresAdapter(connectionString)\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,gBAAe;AAEf,8BAOO;AARP,IAAM,EAAE,OAAO,IAAI,UAAAA;AAUZ,IAAM,kBAAN,MAA4C;AAAA,EACvC;AAAA,EAER,YAAY,kBAA0B;AAClC,SAAK,SAAS,IAAI,OAAO,EAAE,iBAAiB,CAAC;AAAA,EACjD;AAAA,EAEA,MAAM,UAAyB;AAC3B,UAAM,KAAK,OAAO,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAM,aAA4B;AAC9B,UAAM,KAAK,OAAO,IAAI;AAAA,EAC1B;AAAA,EAEA,MAAM,QAAuB;AACzB,UAAM,KAAK,OAAO,MAAM,OAAO;AAAA,EACnC;AAAA,EAEA,MAAM,SAAwB;AAC1B,UAAM,KAAK,OAAO,MAAM,QAAQ;AAAA,EACpC;AAAA,EAEA,MAAM,WAA0B;AAC5B,UAAM,KAAK,OAAO,MAAM,UAAU;AAAA,EACtC;AAAA,EAEA,MAAM,mBAAyC;AAC3C,UAAM,SAAsB,EAAE,QAAQ,CAAC,EAAE;AAGzC,UAAM,YAAY,MAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA,KAK7C;AAEG,eAAW,OAAO,UAAU,MAAM;AAC9B,YAAM,YAAY,IAAI;AACtB,aAAO,OAAO,SAAS,IAAI;AAAA,QACvB,MAAM;AAAA,QACN,QAAQ,IAAI;AAAA,QACZ,SAAS,CAAC;AAAA,QACV,aAAa,CAAC;AAAA,QACd,mBAAmB,CAAC;AAAA,MACxB;AAAA,IACJ;AAGA,UAAM,aAAa,MAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA,KAI9C;AAEG,eAAW,OAAO,WAAW,MAAM;AAC/B,UAAI,CAAC,OAAO,OAAO,IAAI,UAAU,EAAG;AAEpC,YAAM,MAAoB;AAAA,QACtB,MAAM,IAAI;AAAA,QACV,MAAM,KAAK,gBAAgB,IAAI,WAAW,IAAI,QAAQ;AAAA,QACtD,SAAS,IAAI;AAAA,QACb,UAAU,IAAI,gBAAgB;AAAA,QAC9B,cAAc,IAAI;AAAA,QAClB,iBAAiB,IAAI,gBAAgB,SAAS,SAAS,KAAK;AAAA,MAChE;AACA,aAAO,OAAO,IAAI,UAAU,EAAE,QAAQ,IAAI,WAAW,IAAI;AAAA,IAC7D;AAGA,UAAM,QAAQ,MAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAYzC;AAEG,eAAW,OAAO,MAAM,MAAM;AAC1B,UAAI,CAAC,OAAO,OAAO,IAAI,UAAU,EAAG;AACpC,aAAO,OAAO,IAAI,UAAU,EAAE,YAAY,KAAK;AAAA,QAC3C,SAAS,CAAC,IAAI,WAAW;AAAA,QACzB,iBAAiB,IAAI;AAAA,QACrB,mBAAmB,CAAC,IAAI,mBAAmB;AAAA,MAC/C,CAAC;AAAA,IACL;AAGA,UAAM,QAAQ,MAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA,KAKzC;AAEG,eAAW,OAAO,MAAM,MAAM;AAC1B,UAAI,CAAC,OAAO,OAAO,IAAI,UAAU,EAAG;AACpC,UAAI,CAAC,OAAO,OAAO,IAAI,UAAU,EAAE,YAAY;AAC3C,eAAO,OAAO,IAAI,UAAU,EAAE,aAAa,EAAE,SAAS,CAAC,EAAE;AAAA,MAC7D;AACA,aAAO,OAAO,IAAI,UAAU,EAAE,WAAY,QAAQ,KAAK,IAAI,WAAW;AAAA,IAC1E;AAEA,WAAO;AAAA,EACX;AAAA,EAEA,MAAM,YAAY,OAAiC;AAC/C,UAAM,EAAE,WAAW,KAAK,IAAI;AAC5B,QAAI,KAAK,WAAW,EAAG;AAEvB,UAAM,UAAU,OAAO,KAAK,KAAK,CAAC,CAAC;AACnC,UAAM,eAAe,KAAK;AAAA,MAAI,CAAC,GAAG,aAC9B,IAAI,QAAQ,IAAI,CAACC,IAAG,aAAa,IAAI,WAAW,QAAQ,SAAS,WAAW,CAAC,EAAE,EAAE,KAAK,IAAI,CAAC;AAAA,IAC/F,EAAE,KAAK,IAAI;AAEX,UAAM,SAAS,KAAK,QAAQ,SAAO,QAAQ,IAAI,SAAO,IAAI,GAAG,CAAC,CAAC;AAC/D,UAAM,QAAQ,gBAAgB,SAAS,MAAM,QAAQ,IAAI,OAAK,IAAI,CAAC,GAAG,EAAE,KAAK,IAAI,CAAC,YAAY,YAAY;AAE1G,UAAM,KAAK,OAAO,MAAM,OAAO,MAAM;AAAA,EACzC;AAAA,EAEA,MAAM,eAAe,YAAqC;AACtD,QAAI,WAAW,WAAW,EAAG;AAC7B,UAAM,cAAc,WAAW,IAAI,OAAK,IAAI,CAAC,GAAG,EAAE,KAAK,IAAI;AAC3D,UAAM,KAAK,OAAO,MAAM,YAAY,WAAW,2BAA2B;AAAA,EAC9E;AAAA,EAES,eAAe;AAAA,IACpB,OAAO;AAAA,IACP,uBAAuB;AAAA,IACvB,WAAW;AAAA,IACX,gBAAgB;AAAA,EACpB;AAAA,EAEQ,gBAAgB,QAAgB,SAAoC;AACxE,UAAM,OAAO,OAAO,YAAY;AAChC,UAAM,MAAM,QAAQ,YAAY;AAEhC,QAAI,KAAK,SAAS,KAAK,KAAK,KAAK,SAAS,QAAQ,GAAG;AACjD,UAAI,KAAK,SAAS,KAAK,KAAK,IAAI,SAAS,MAAM,EAAG,QAAO,0CAAkB;AAC3E,aAAO,0CAAkB;AAAA,IAC7B;AACA,QAAI,KAAK,SAAS,MAAM,KAAK,KAAK,SAAS,MAAM,KAAK,SAAS,oBAAqB,QAAO,0CAAkB;AAC7G,QAAI,KAAK,SAAS,MAAM,EAAG,QAAO,0CAAkB;AACpD,QAAI,KAAK,SAAS,WAAW,KAAK,KAAK,SAAS,MAAM,EAAG,QAAO,0CAAkB;AAClF,QAAI,KAAK,SAAS,MAAM,EAAG,QAAO,0CAAkB;AACpD,QAAI,KAAK,SAAS,SAAS,KAAK,KAAK,SAAS,SAAS,EAAG,QAAO,0CAAkB;AACnF,QAAI,KAAK,SAAS,OAAO,KAAK,KAAK,SAAS,MAAM,EAAG,QAAO,0CAAkB;AAC9E,QAAI,QAAQ,OAAQ,QAAO,0CAAkB;AAE7C,WAAO,0CAAkB;AAAA,EAC7B;AACJ;AAEO,SAAS,sBAAsB,kBAA2C;AAC7E,SAAO,IAAI,gBAAgB,gBAAgB;AAC/C;","names":["pg","_"]}

package/dist/index.d.cts

@@ -0,0 +1,24 @@
+import { SqlAdapter, SchemaGraph, SeedBatch } from 'schema-seed-core';
+
+declare class PostgresAdapter implements SqlAdapter {
+    private client;
+    constructor(connectionString: string);
+    connect(): Promise<void>;
+    disconnect(): Promise<void>;
+    begin(): Promise<void>;
+    commit(): Promise<void>;
+    rollback(): Promise<void>;
+    introspectSchema(): Promise<SchemaGraph>;
+    insertBatch(batch: SeedBatch): Promise<void>;
+    truncateTables(tableNames: string[]): Promise<void>;
+    readonly capabilities: {
+        enums: boolean;
+        deferrableConstraints: boolean;
+        returning: boolean;
+        identityInsert: boolean;
+    };
+    private mapPostgresType;
+}
+declare function createPostgresAdapter(connectionString: string): PostgresAdapter;
+
+export { PostgresAdapter, createPostgresAdapter };

package/dist/index.d.ts

@@ -0,0 +1,24 @@
+import { SqlAdapter, SchemaGraph, SeedBatch } from 'schema-seed-core';
+
+declare class PostgresAdapter implements SqlAdapter {
+    private client;
+    constructor(connectionString: string);
+    connect(): Promise<void>;
+    disconnect(): Promise<void>;
+    begin(): Promise<void>;
+    commit(): Promise<void>;
+    rollback(): Promise<void>;
+    introspectSchema(): Promise<SchemaGraph>;
+    insertBatch(batch: SeedBatch): Promise<void>;
+    truncateTables(tableNames: string[]): Promise<void>;
+    readonly capabilities: {
+        enums: boolean;
+        deferrableConstraints: boolean;
+        returning: boolean;
+        identityInsert: boolean;
+    };
+    private mapPostgresType;
+}
+declare function createPostgresAdapter(connectionString: string): PostgresAdapter;
+
+export { PostgresAdapter, createPostgresAdapter };

package/dist/index.js

@@ -0,0 +1,144 @@
+// src/index.ts
+import pg from "pg";
+import {
+  NormalizedSqlType
+} from "schema-seed-core";
+var { Client } = pg;
+var PostgresAdapter = class {
+  client;
+  constructor(connectionString) {
+    this.client = new Client({ connectionString });
+  }
+  async connect() {
+    await this.client.connect();
+  }
+  async disconnect() {
+    await this.client.end();
+  }
+  async begin() {
+    await this.client.query("BEGIN");
+  }
+  async commit() {
+    await this.client.query("COMMIT");
+  }
+  async rollback() {
+    await this.client.query("ROLLBACK");
+  }
+  async introspectSchema() {
+    const schema = { tables: {} };
+    const tablesRes = await this.client.query(`
+      SELECT table_name, table_schema
+      FROM information_schema.tables
+      WHERE table_schema NOT IN ('information_schema', 'pg_catalog')
+      AND table_type = 'BASE TABLE'
+    `);
+    for (const row of tablesRes.rows) {
+      const tableName = row.table_name;
+      schema.tables[tableName] = {
+        name: tableName,
+        schema: row.table_schema,
+        columns: {},
+        foreignKeys: [],
+        uniqueConstraints: []
+      };
+    }
+    const columnsRes = await this.client.query(`
+      SELECT table_name, column_name, data_type, is_nullable, column_default, udt_name
+      FROM information_schema.columns
+      WHERE table_schema NOT IN ('information_schema', 'pg_catalog')
+    `);
+    for (const row of columnsRes.rows) {
+      if (!schema.tables[row.table_name]) continue;
+      const col = {
+        name: row.column_name,
+        type: this.mapPostgresType(row.data_type, row.udt_name),
+        rawType: row.data_type,
+        nullable: row.is_nullable === "YES",
+        defaultValue: row.column_default,
+        isAutoIncrement: row.column_default?.includes("nextval") || false
+      };
+      schema.tables[row.table_name].columns[row.column_name] = col;
+    }
+    const fkRes = await this.client.query(`
+      SELECT
+          tc.table_name, kcu.column_name, 
+          ccu.table_name AS foreign_table_name,
+          ccu.column_name AS foreign_column_name 
+      FROM 
+          information_schema.table_constraints AS tc 
+          JOIN information_schema.key_column_usage AS kcu
+            ON tc.constraint_name = kcu.constraint_name
+          JOIN information_schema.constraint_column_usage AS ccu
+            ON ccu.constraint_name = tc.constraint_name
+      WHERE tc.constraint_type = 'FOREIGN KEY'
+    `);
+    for (const row of fkRes.rows) {
+      if (!schema.tables[row.table_name]) continue;
+      schema.tables[row.table_name].foreignKeys.push({
+        columns: [row.column_name],
+        referencedTable: row.foreign_table_name,
+        referencedColumns: [row.foreign_column_name]
+      });
+    }
+    const pkRes = await this.client.query(`
+      SELECT tc.table_name, kcu.column_name
+      FROM information_schema.table_constraints tc
+      JOIN information_schema.key_column_usage kcu ON tc.constraint_name = kcu.constraint_name
+      WHERE tc.constraint_type = 'PRIMARY KEY'
+    `);
+    for (const row of pkRes.rows) {
+      if (!schema.tables[row.table_name]) continue;
+      if (!schema.tables[row.table_name].primaryKey) {
+        schema.tables[row.table_name].primaryKey = { columns: [] };
+      }
+      schema.tables[row.table_name].primaryKey.columns.push(row.column_name);
+    }
+    return schema;
+  }
+  async insertBatch(batch) {
+    const { tableName, rows } = batch;
+    if (rows.length === 0) return;
+    const columns = Object.keys(rows[0]);
+    const placeholders = rows.map(
+      (_, rowIndex) => `(${columns.map((_2, colIndex) => `$${rowIndex * columns.length + colIndex + 1}`).join(", ")})`
+    ).join(", ");
+    const values = rows.flatMap((row) => columns.map((col) => row[col]));
+    const query = `INSERT INTO "${tableName}" (${columns.map((c) => `"${c}"`).join(", ")}) VALUES ${placeholders}`;
+    await this.client.query(query, values);
+  }
+  async truncateTables(tableNames) {
+    if (tableNames.length === 0) return;
+    const quotedNames = tableNames.map((t) => `"${t}"`).join(", ");
+    await this.client.query(`TRUNCATE ${quotedNames} RESTART IDENTITY CASCADE`);
+  }
+  capabilities = {
+    enums: true,
+    deferrableConstraints: true,
+    returning: true,
+    identityInsert: false
+  };
+  mapPostgresType(pgType, udtName) {
+    const type = pgType.toLowerCase();
+    const udt = udtName.toLowerCase();
+    if (type.includes("int") || type.includes("serial")) {
+      if (type.includes("big") || udt.includes("int8")) return NormalizedSqlType.BIGINT;
+      return NormalizedSqlType.INT;
+    }
+    if (type.includes("char") || type.includes("text") || type === "character varying") return NormalizedSqlType.STRING;
+    if (type.includes("bool")) return NormalizedSqlType.BOOLEAN;
+    if (type.includes("timestamp") || type.includes("date")) return NormalizedSqlType.DATETIME;
+    if (type.includes("json")) return NormalizedSqlType.JSON;
+    if (type.includes("numeric") || type.includes("decimal")) return NormalizedSqlType.DECIMAL;
+    if (type.includes("float") || type.includes("real")) return NormalizedSqlType.FLOAT;
+    if (udt === "uuid") return NormalizedSqlType.UUID;
+    return NormalizedSqlType.STRING;
+  }
+};
+function createPostgresAdapter(connectionString) {
+  return new PostgresAdapter(connectionString);
+}
+export {
+  PostgresAdapter,
+  createPostgresAdapter
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import pg from 'pg'\nconst { Client } = pg\nimport {\n    SqlAdapter,\n    SchemaGraph,\n    SeedBatch,\n    NormalizedSqlType,\n    TableSchema,\n    ColumnSchema\n} from 'schema-seed-core'\n\nexport class PostgresAdapter implements SqlAdapter {\n    private client: pg.Client\n\n    constructor(connectionString: string) {\n        this.client = new Client({ connectionString })\n    }\n\n    async connect(): Promise<void> {\n        await this.client.connect()\n    }\n\n    async disconnect(): Promise<void> {\n        await this.client.end()\n    }\n\n    async begin(): Promise<void> {\n        await this.client.query('BEGIN')\n    }\n\n    async commit(): Promise<void> {\n        await this.client.query('COMMIT')\n    }\n\n    async rollback(): Promise<void> {\n        await this.client.query('ROLLBACK')\n    }\n\n    async introspectSchema(): Promise<SchemaGraph> {\n        const schema: SchemaGraph = { tables: {} }\n\n        // Get tables\n        const tablesRes = await this.client.query(`\n      SELECT table_name, table_schema\n      FROM information_schema.tables\n      WHERE table_schema NOT IN ('information_schema', 'pg_catalog')\n      AND table_type = 'BASE TABLE'\n    `)\n\n        for (const row of tablesRes.rows) {\n            const tableName = row.table_name\n            schema.tables[tableName] = {\n                name: tableName,\n                schema: row.table_schema,\n                columns: {},\n                foreignKeys: [],\n                uniqueConstraints: []\n            }\n        }\n\n        // Get columns\n        const columnsRes = await this.client.query(`\n      SELECT table_name, column_name, data_type, is_nullable, column_default, udt_name\n      FROM information_schema.columns\n      WHERE table_schema NOT IN ('information_schema', 'pg_catalog')\n    `)\n\n        for (const row of columnsRes.rows) {\n            if (!schema.tables[row.table_name]) continue\n\n            const col: ColumnSchema = {\n                name: row.column_name,\n                type: this.mapPostgresType(row.data_type, row.udt_name),\n                rawType: row.data_type,\n                nullable: row.is_nullable === 'YES',\n                defaultValue: row.column_default,\n                isAutoIncrement: row.column_default?.includes('nextval') || false\n            }\n            schema.tables[row.table_name].columns[row.column_name] = col\n        }\n\n        // Get Foreign Keys\n        const fkRes = await this.client.query(`\n      SELECT\n          tc.table_name, kcu.column_name, \n          ccu.table_name AS foreign_table_name,\n          ccu.column_name AS foreign_column_name \n      FROM \n          information_schema.table_constraints AS tc \n          JOIN information_schema.key_column_usage AS kcu\n            ON tc.constraint_name = kcu.constraint_name\n          JOIN information_schema.constraint_column_usage AS ccu\n            ON ccu.constraint_name = tc.constraint_name\n      WHERE tc.constraint_type = 'FOREIGN KEY'\n    `)\n\n        for (const row of fkRes.rows) {\n            if (!schema.tables[row.table_name]) continue\n            schema.tables[row.table_name].foreignKeys.push({\n                columns: [row.column_name],\n                referencedTable: row.foreign_table_name,\n                referencedColumns: [row.foreign_column_name]\n            })\n        }\n\n        // Get Primary Keys\n        const pkRes = await this.client.query(`\n      SELECT tc.table_name, kcu.column_name\n      FROM information_schema.table_constraints tc\n      JOIN information_schema.key_column_usage kcu ON tc.constraint_name = kcu.constraint_name\n      WHERE tc.constraint_type = 'PRIMARY KEY'\n    `)\n\n        for (const row of pkRes.rows) {\n            if (!schema.tables[row.table_name]) continue\n            if (!schema.tables[row.table_name].primaryKey) {\n                schema.tables[row.table_name].primaryKey = { columns: [] }\n            }\n            schema.tables[row.table_name].primaryKey!.columns.push(row.column_name)\n        }\n\n        return schema\n    }\n\n    async insertBatch(batch: SeedBatch): Promise<void> {\n        const { tableName, rows } = batch\n        if (rows.length === 0) return\n\n        const columns = Object.keys(rows[0])\n        const placeholders = rows.map((_, rowIndex) =>\n            `(${columns.map((_, colIndex) => `$${rowIndex * columns.length + colIndex + 1}`).join(', ')})`\n        ).join(', ')\n\n        const values = rows.flatMap(row => columns.map(col => row[col]))\n        const query = `INSERT INTO \"${tableName}\" (${columns.map(c => `\"${c}\"`).join(', ')}) VALUES ${placeholders}`\n\n        await this.client.query(query, values)\n    }\n\n    async truncateTables(tableNames: string[]): Promise<void> {\n        if (tableNames.length === 0) return\n        const quotedNames = tableNames.map(t => `\"${t}\"`).join(', ')\n        await this.client.query(`TRUNCATE ${quotedNames} RESTART IDENTITY CASCADE`)\n    }\n\n    readonly capabilities = {\n        enums: true,\n        deferrableConstraints: true,\n        returning: true,\n        identityInsert: false\n    }\n\n    private mapPostgresType(pgType: string, udtName: string): NormalizedSqlType {\n        const type = pgType.toLowerCase()\n        const udt = udtName.toLowerCase()\n\n        if (type.includes('int') || type.includes('serial')) {\n            if (type.includes('big') || udt.includes('int8')) return NormalizedSqlType.BIGINT\n            return NormalizedSqlType.INT\n        }\n        if (type.includes('char') || type.includes('text') || type === 'character varying') return NormalizedSqlType.STRING\n        if (type.includes('bool')) return NormalizedSqlType.BOOLEAN\n        if (type.includes('timestamp') || type.includes('date')) return NormalizedSqlType.DATETIME\n        if (type.includes('json')) return NormalizedSqlType.JSON\n        if (type.includes('numeric') || type.includes('decimal')) return NormalizedSqlType.DECIMAL\n        if (type.includes('float') || type.includes('real')) return NormalizedSqlType.FLOAT\n        if (udt === 'uuid') return NormalizedSqlType.UUID\n\n        return NormalizedSqlType.STRING\n    }\n}\n\nexport function createPostgresAdapter(connectionString: string): PostgresAdapter {\n    return new PostgresAdapter(connectionString)\n}\n"],"mappings":";AAAA,OAAO,QAAQ;AAEf;AAAA,EAII;AAAA,OAGG;AARP,IAAM,EAAE,OAAO,IAAI;AAUZ,IAAM,kBAAN,MAA4C;AAAA,EACvC;AAAA,EAER,YAAY,kBAA0B;AAClC,SAAK,SAAS,IAAI,OAAO,EAAE,iBAAiB,CAAC;AAAA,EACjD;AAAA,EAEA,MAAM,UAAyB;AAC3B,UAAM,KAAK,OAAO,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAM,aAA4B;AAC9B,UAAM,KAAK,OAAO,IAAI;AAAA,EAC1B;AAAA,EAEA,MAAM,QAAuB;AACzB,UAAM,KAAK,OAAO,MAAM,OAAO;AAAA,EACnC;AAAA,EAEA,MAAM,SAAwB;AAC1B,UAAM,KAAK,OAAO,MAAM,QAAQ;AAAA,EACpC;AAAA,EAEA,MAAM,WAA0B;AAC5B,UAAM,KAAK,OAAO,MAAM,UAAU;AAAA,EACtC;AAAA,EAEA,MAAM,mBAAyC;AAC3C,UAAM,SAAsB,EAAE,QAAQ,CAAC,EAAE;AAGzC,UAAM,YAAY,MAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA,KAK7C;AAEG,eAAW,OAAO,UAAU,MAAM;AAC9B,YAAM,YAAY,IAAI;AACtB,aAAO,OAAO,SAAS,IAAI;AAAA,QACvB,MAAM;AAAA,QACN,QAAQ,IAAI;AAAA,QACZ,SAAS,CAAC;AAAA,QACV,aAAa,CAAC;AAAA,QACd,mBAAmB,CAAC;AAAA,MACxB;AAAA,IACJ;AAGA,UAAM,aAAa,MAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA,KAI9C;AAEG,eAAW,OAAO,WAAW,MAAM;AAC/B,UAAI,CAAC,OAAO,OAAO,IAAI,UAAU,EAAG;AAEpC,YAAM,MAAoB;AAAA,QACtB,MAAM,IAAI;AAAA,QACV,MAAM,KAAK,gBAAgB,IAAI,WAAW,IAAI,QAAQ;AAAA,QACtD,SAAS,IAAI;AAAA,QACb,UAAU,IAAI,gBAAgB;AAAA,QAC9B,cAAc,IAAI;AAAA,QAClB,iBAAiB,IAAI,gBAAgB,SAAS,SAAS,KAAK;AAAA,MAChE;AACA,aAAO,OAAO,IAAI,UAAU,EAAE,QAAQ,IAAI,WAAW,IAAI;AAAA,IAC7D;AAGA,UAAM,QAAQ,MAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,KAYzC;AAEG,eAAW,OAAO,MAAM,MAAM;AAC1B,UAAI,CAAC,OAAO,OAAO,IAAI,UAAU,EAAG;AACpC,aAAO,OAAO,IAAI,UAAU,EAAE,YAAY,KAAK;AAAA,QAC3C,SAAS,CAAC,IAAI,WAAW;AAAA,QACzB,iBAAiB,IAAI;AAAA,QACrB,mBAAmB,CAAC,IAAI,mBAAmB;AAAA,MAC/C,CAAC;AAAA,IACL;AAGA,UAAM,QAAQ,MAAM,KAAK,OAAO,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA,KAKzC;AAEG,eAAW,OAAO,MAAM,MAAM;AAC1B,UAAI,CAAC,OAAO,OAAO,IAAI,UAAU,EAAG;AACpC,UAAI,CAAC,OAAO,OAAO,IAAI,UAAU,EAAE,YAAY;AAC3C,eAAO,OAAO,IAAI,UAAU,EAAE,aAAa,EAAE,SAAS,CAAC,EAAE;AAAA,MAC7D;AACA,aAAO,OAAO,IAAI,UAAU,EAAE,WAAY,QAAQ,KAAK,IAAI,WAAW;AAAA,IAC1E;AAEA,WAAO;AAAA,EACX;AAAA,EAEA,MAAM,YAAY,OAAiC;AAC/C,UAAM,EAAE,WAAW,KAAK,IAAI;AAC5B,QAAI,KAAK,WAAW,EAAG;AAEvB,UAAM,UAAU,OAAO,KAAK,KAAK,CAAC,CAAC;AACnC,UAAM,eAAe,KAAK;AAAA,MAAI,CAAC,GAAG,aAC9B,IAAI,QAAQ,IAAI,CAACA,IAAG,aAAa,IAAI,WAAW,QAAQ,SAAS,WAAW,CAAC,EAAE,EAAE,KAAK,IAAI,CAAC;AAAA,IAC/F,EAAE,KAAK,IAAI;AAEX,UAAM,SAAS,KAAK,QAAQ,SAAO,QAAQ,IAAI,SAAO,IAAI,GAAG,CAAC,CAAC;AAC/D,UAAM,QAAQ,gBAAgB,SAAS,MAAM,QAAQ,IAAI,OAAK,IAAI,CAAC,GAAG,EAAE,KAAK,IAAI,CAAC,YAAY,YAAY;AAE1G,UAAM,KAAK,OAAO,MAAM,OAAO,MAAM;AAAA,EACzC;AAAA,EAEA,MAAM,eAAe,YAAqC;AACtD,QAAI,WAAW,WAAW,EAAG;AAC7B,UAAM,cAAc,WAAW,IAAI,OAAK,IAAI,CAAC,GAAG,EAAE,KAAK,IAAI;AAC3D,UAAM,KAAK,OAAO,MAAM,YAAY,WAAW,2BAA2B;AAAA,EAC9E;AAAA,EAES,eAAe;AAAA,IACpB,OAAO;AAAA,IACP,uBAAuB;AAAA,IACvB,WAAW;AAAA,IACX,gBAAgB;AAAA,EACpB;AAAA,EAEQ,gBAAgB,QAAgB,SAAoC;AACxE,UAAM,OAAO,OAAO,YAAY;AAChC,UAAM,MAAM,QAAQ,YAAY;AAEhC,QAAI,KAAK,SAAS,KAAK,KAAK,KAAK,SAAS,QAAQ,GAAG;AACjD,UAAI,KAAK,SAAS,KAAK,KAAK,IAAI,SAAS,MAAM,EAAG,QAAO,kBAAkB;AAC3E,aAAO,kBAAkB;AAAA,IAC7B;AACA,QAAI,KAAK,SAAS,MAAM,KAAK,KAAK,SAAS,MAAM,KAAK,SAAS,oBAAqB,QAAO,kBAAkB;AAC7G,QAAI,KAAK,SAAS,MAAM,EAAG,QAAO,kBAAkB;AACpD,QAAI,KAAK,SAAS,WAAW,KAAK,KAAK,SAAS,MAAM,EAAG,QAAO,kBAAkB;AAClF,QAAI,KAAK,SAAS,MAAM,EAAG,QAAO,kBAAkB;AACpD,QAAI,KAAK,SAAS,SAAS,KAAK,KAAK,SAAS,SAAS,EAAG,QAAO,kBAAkB;AACnF,QAAI,KAAK,SAAS,OAAO,KAAK,KAAK,SAAS,MAAM,EAAG,QAAO,kBAAkB;AAC9E,QAAI,QAAQ,OAAQ,QAAO,kBAAkB;AAE7C,WAAO,kBAAkB;AAAA,EAC7B;AACJ;AAEO,SAAS,sBAAsB,kBAA2C;AAC7E,SAAO,IAAI,gBAAgB,gBAAgB;AAC/C;","names":["_"]}

package/package.json

@@ -0,0 +1,58 @@
+{
+  "name": "schema-seed-adapter-postgres",
+  "version": "0.1.0",
+  "type": "module",
+  "description": "PostgreSQL adapter for schema-seed",
+  "author": "Ali Nazar",
+  "license": "MIT",
+  "private": false,
+  "sideEffects": false,
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/alinazar-111/schema-seed.git"
+  },
+  "homepage": "https://github.com/alinazar-111/schema-seed#readme",
+  "bugs": {
+    "url": "https://github.com/alinazar-111/schema-seed/issues"
+  },
+  "keywords": [
+    "database",
+    "seeding",
+    "test-data",
+    "adapter",
+    "schema-seed"
+  ],
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "dependencies": {
+    "pg": "^8.13.1",
+    "schema-seed-core": "0.1.0"
+  },
+  "devDependencies": {
+    "@types/pg": "^8.11.10",
+    "tsup": "^8.3.5",
+    "typescript": "^5.7.2",
+    "vitest": "^2.1.8"
+  },
+  "scripts": {
+    "build": "tsup",
+    "clean": "rm -rf dist",
+    "dev": "tsup --watch",
+    "lint": "eslint .",
+    "test": "vitest run --passWithNoTests",
+    "typecheck": "tsc --noEmit"
+  }
+}