turbine-orm 0.3.0

package/dist/introspect.js

@@ -0,0 +1,284 @@
+/**
+ * @batadata/turbine — Schema introspection
+ *
+ * Connects to a live Postgres database, reads information_schema + pg_catalog,
+ * and produces a SchemaMetadata object describing every table, column, relation,
+ * and index in the target schema.
+ *
+ * This is the foundation of `npx turbine generate`.
+ */
+import pg from 'pg';
+import { pgTypeToTs, isDateType, pgArrayType, snakeToCamel, singularize, } from './schema.js';
+// ---------------------------------------------------------------------------
+// SQL queries (all parameterized, no interpolation)
+// ---------------------------------------------------------------------------
+const SQL_TABLES = `
+  SELECT table_name
+  FROM information_schema.tables
+  WHERE table_schema = $1
+    AND table_type = 'BASE TABLE'
+  ORDER BY table_name
+`;
+const SQL_COLUMNS = `
+  SELECT
+    table_name,
+    column_name,
+    udt_name,
+    data_type,
+    is_nullable,
+    column_default,
+    ordinal_position,
+    character_maximum_length
+  FROM information_schema.columns
+  WHERE table_schema = $1
+  ORDER BY table_name, ordinal_position
+`;
+const SQL_PRIMARY_KEYS = `
+  SELECT
+    tc.table_name,
+    kcu.column_name,
+    kcu.ordinal_position
+  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.constraint_type = 'PRIMARY KEY'
+    AND tc.table_schema = $1
+  ORDER BY tc.table_name, kcu.ordinal_position
+`;
+const SQL_FOREIGN_KEYS = `
+  SELECT
+    tc.table_name AS source_table,
+    kcu.column_name AS source_column,
+    ccu.table_name AS target_table,
+    ccu.column_name AS target_column,
+    tc.constraint_name
+  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
+  JOIN information_schema.constraint_column_usage ccu
+    ON tc.constraint_name = ccu.constraint_name
+    AND tc.table_schema = ccu.table_schema
+  WHERE tc.constraint_type = 'FOREIGN KEY'
+    AND tc.table_schema = $1
+`;
+const SQL_UNIQUE_CONSTRAINTS = `
+  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
+    AND tc.table_schema = kcu.table_schema
+  WHERE tc.constraint_type = 'UNIQUE'
+    AND tc.table_schema = $1
+`;
+const SQL_INDEXES = `
+  SELECT tablename, indexname, indexdef
+  FROM pg_indexes
+  WHERE schemaname = $1
+`;
+const SQL_ENUMS = `
+  SELECT t.typname, e.enumlabel
+  FROM pg_type t
+  JOIN pg_enum e ON t.oid = e.enumtypid
+  JOIN pg_catalog.pg_namespace n ON n.oid = t.typnamespace
+  WHERE n.nspname = $1
+  ORDER BY t.typname, e.enumsortorder
+`;
+// ---------------------------------------------------------------------------
+// Main introspection function
+// ---------------------------------------------------------------------------
+export async function introspect(options) {
+    const schema = options.schema ?? 'public';
+    const pool = new pg.Pool({
+        connectionString: options.connectionString,
+        max: 1,
+        connectionTimeoutMillis: 10_000,
+    });
+    try {
+        // Run all information_schema queries in parallel
+        const [tablesResult, columnsResult, pkResult, fkResult, uniqueResult, indexResult, enumResult,] = await Promise.all([
+            pool.query(SQL_TABLES, [schema]),
+            pool.query(SQL_COLUMNS, [schema]),
+            pool.query(SQL_PRIMARY_KEYS, [schema]),
+            pool.query(SQL_FOREIGN_KEYS, [schema]),
+            pool.query(SQL_UNIQUE_CONSTRAINTS, [schema]),
+            pool.query(SQL_INDEXES, [schema]),
+            pool.query(SQL_ENUMS, [schema]),
+        ]);
+        // Filter tables by include/exclude
+        let tableNames = tablesResult.rows.map((r) => r.table_name);
+        if (options.include?.length) {
+            const includeSet = new Set(options.include);
+            tableNames = tableNames.filter((t) => includeSet.has(t));
+        }
+        if (options.exclude?.length) {
+            const excludeSet = new Set(options.exclude);
+            tableNames = tableNames.filter((t) => !excludeSet.has(t));
+        }
+        const tableSet = new Set(tableNames);
+        // ----- Group columns by table -----
+        const columnsByTable = new Map();
+        for (const row of columnsResult.rows) {
+            const tableName = row.table_name;
+            if (!tableSet.has(tableName))
+                continue;
+            const isNullable = row.is_nullable === 'YES';
+            const isArray = row.data_type === 'ARRAY';
+            const baseType = isArray ? row.udt_name.slice(1) : row.udt_name;
+            const col = {
+                name: row.column_name,
+                field: snakeToCamel(row.column_name),
+                pgType: row.udt_name,
+                tsType: pgTypeToTs(isArray ? row.udt_name : baseType, isNullable),
+                nullable: isNullable,
+                hasDefault: row.column_default !== null,
+                isArray,
+                pgArrayType: pgArrayType(baseType),
+                maxLength: row.character_maximum_length ?? undefined,
+            };
+            if (!columnsByTable.has(tableName))
+                columnsByTable.set(tableName, []);
+            columnsByTable.get(tableName).push(col);
+        }
+        // ----- Group primary keys by table -----
+        const pkByTable = new Map();
+        for (const row of pkResult.rows) {
+            if (!tableSet.has(row.table_name))
+                continue;
+            if (!pkByTable.has(row.table_name))
+                pkByTable.set(row.table_name, []);
+            pkByTable.get(row.table_name).push(row.column_name);
+        }
+        // ----- Group unique constraints by table -----
+        const uniqueByTable = new Map();
+        for (const row of uniqueResult.rows) {
+            if (!tableSet.has(row.table_name))
+                continue;
+            if (!uniqueByTable.has(row.table_name))
+                uniqueByTable.set(row.table_name, []);
+            // Each unique constraint may be multi-column; for simplicity, treat as single-col here
+            uniqueByTable.get(row.table_name).push([row.column_name]);
+        }
+        // ----- Group indexes by table -----
+        const indexesByTable = new Map();
+        for (const row of indexResult.rows) {
+            if (!tableSet.has(row.tablename))
+                continue;
+            if (!indexesByTable.has(row.tablename))
+                indexesByTable.set(row.tablename, []);
+            const isUnique = row.indexdef.includes('UNIQUE');
+            // Extract column names from indexdef (e.g. "CREATE INDEX idx ON tbl USING btree (col1, col2)")
+            const colMatch = row.indexdef.match(/\((.+)\)/);
+            const columns = colMatch
+                ? colMatch[1].split(',').map((c) => c.trim().replace(/ (ASC|DESC)/i, ''))
+                : [];
+            indexesByTable.get(row.tablename).push({
+                name: row.indexname,
+                columns,
+                unique: isUnique,
+                definition: row.indexdef,
+            });
+        }
+        // ----- Collect enums -----
+        const enums = {};
+        for (const row of enumResult.rows) {
+            if (!enums[row.typname])
+                enums[row.typname] = [];
+            enums[row.typname].push(row.enumlabel);
+        }
+        const foreignKeys = [];
+        for (const row of fkResult.rows) {
+            if (!tableSet.has(row.source_table) || !tableSet.has(row.target_table))
+                continue;
+            foreignKeys.push({
+                sourceTable: row.source_table,
+                sourceColumn: row.source_column,
+                targetTable: row.target_table,
+                targetColumn: row.target_column,
+            });
+        }
+        // ----- Build relations from foreign keys -----
+        // Count FKs per (source, target) pair for disambiguation
+        const fkCounts = new Map();
+        for (const fk of foreignKeys) {
+            const key = `${fk.sourceTable}→${fk.targetTable}`;
+            fkCounts.set(key, (fkCounts.get(key) ?? 0) + 1);
+        }
+        const relationsByTable = new Map();
+        for (const fk of foreignKeys) {
+            const pairKey = `${fk.sourceTable}→${fk.targetTable}`;
+            const needsDisambiguation = (fkCounts.get(pairKey) ?? 0) > 1;
+            // --- belongsTo on the source (child) table ---
+            // e.g. posts.user_id → users.id creates posts.user (belongsTo)
+            const belongsToName = needsDisambiguation
+                ? snakeToCamel(fk.sourceColumn.replace(/_id$/, ''))
+                : singularize(snakeToCamel(fk.targetTable));
+            if (!relationsByTable.has(fk.sourceTable))
+                relationsByTable.set(fk.sourceTable, {});
+            relationsByTable.get(fk.sourceTable)[belongsToName] = {
+                type: 'belongsTo',
+                name: belongsToName,
+                from: fk.sourceTable,
+                to: fk.targetTable,
+                foreignKey: fk.sourceColumn,
+                referenceKey: fk.targetColumn,
+            };
+            // --- hasMany on the target (parent) table ---
+            // e.g. posts.user_id → users.id creates users.posts (hasMany)
+            const hasManyName = needsDisambiguation
+                ? snakeToCamel(`${fk.sourceTable}_by_${fk.sourceColumn.replace(/_id$/, '')}`)
+                : snakeToCamel(fk.sourceTable);
+            if (!relationsByTable.has(fk.targetTable))
+                relationsByTable.set(fk.targetTable, {});
+            relationsByTable.get(fk.targetTable)[hasManyName] = {
+                type: 'hasMany',
+                name: hasManyName,
+                from: fk.targetTable,
+                to: fk.sourceTable,
+                foreignKey: fk.sourceColumn,
+                referenceKey: fk.targetColumn,
+            };
+        }
+        // ----- Assemble TableMetadata for each table -----
+        const tables = {};
+        for (const tableName of tableNames) {
+            const columns = columnsByTable.get(tableName) ?? [];
+            const columnMap = {};
+            const reverseColumnMap = {};
+            const dateColumns = new Set();
+            const pgTypes = {};
+            const allColumns = [];
+            for (const col of columns) {
+                columnMap[col.field] = col.name;
+                reverseColumnMap[col.name] = col.field;
+                allColumns.push(col.name);
+                pgTypes[col.name] = col.pgType;
+                const baseType = col.isArray ? col.pgType.slice(1) : col.pgType;
+                if (isDateType(baseType)) {
+                    dateColumns.add(col.name);
+                }
+            }
+            tables[tableName] = {
+                name: tableName,
+                columns,
+                columnMap,
+                reverseColumnMap,
+                dateColumns,
+                pgTypes,
+                allColumns,
+                primaryKey: pkByTable.get(tableName) ?? [],
+                uniqueColumns: uniqueByTable.get(tableName) ?? [],
+                relations: relationsByTable.get(tableName) ?? {},
+                indexes: indexesByTable.get(tableName) ?? [],
+            };
+        }
+        return { tables, enums };
+    }
+    finally {
+        await pool.end();
+    }
+}
+//# sourceMappingURL=introspect.js.map

package/dist/introspect.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"introspect.js","sourceRoot":"","sources":["../src/introspect.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAML,UAAU,EACV,UAAU,EACV,WAAW,EACX,YAAY,EACZ,WAAW,GACZ,MAAM,aAAa,CAAC;AAErB,8EAA8E;AAC9E,oDAAoD;AACpD,8EAA8E;AAE9E,MAAM,UAAU,GAAG;;;;;;CAMlB,CAAC;AAEF,MAAM,WAAW,GAAG;;;;;;;;;;;;;CAanB,CAAC;AAEF,MAAM,gBAAgB,GAAG;;;;;;;;;;;;CAYxB,CAAC;AAEF,MAAM,gBAAgB,GAAG;;;;;;;;;;;;;;;;CAgBxB,CAAC;AAEF,MAAM,sBAAsB,GAAG;;;;;;;;;;CAU9B,CAAC;AAEF,MAAM,WAAW,GAAG;;;;CAInB,CAAC;AAEF,MAAM,SAAS,GAAG;;;;;;;CAOjB,CAAC;AAiBF,8EAA8E;AAC9E,8BAA8B;AAC9B,8EAA8E;AAE9E,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,OAA0B;IACzD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,QAAQ,CAAC;IAC1C,MAAM,IAAI,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC;QACvB,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;QAC1C,GAAG,EAAE,CAAC;QACN,uBAAuB,EAAE,MAAM;KAChC,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,iDAAiD;QACjD,MAAM,CACJ,YAAY,EACZ,aAAa,EACb,QAAQ,EACR,QAAQ,EACR,YAAY,EACZ,WAAW,EACX,UAAU,EACX,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACpB,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC,MAAM,CAAC,CAAC;YACtC,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC,MAAM,CAAC,CAAC;YACtC,IAAI,CAAC,KAAK,CAAC,sBAAsB,EAAE,CAAC,MAAM,CAAC,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,MAAM,CAAC,CAAC;SAChC,CAAC,CAAC;QAEH,mCAAmC;QACnC,IAAI,UAAU,GAAa,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAyB,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QAC9F,IAAI,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YAC5B,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC5C,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YAC5B,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC5C,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC;QAErC,qCAAqC;QACrC,MAAM,cAAc,GAAG,IAAI,GAAG,EAA4B,CAAC;QAC3D,KAAK,MAAM,GAAG,IAAI,aAAa,CAAC,IAAI,EAAE,CAAC;YACrC,MAAM,SAAS,GAAW,GAAG,CAAC,UAAU,CAAC;YACzC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC;gBAAE,SAAS;YAEvC,MAAM,UAAU,GAAG,GAAG,CAAC,WAAW,KAAK,KAAK,CAAC;YAC7C,MAAM,OAAO,GAAG,GAAG,CAAC,SAAS,KAAK,OAAO,CAAC;YAC1C,MAAM,QAAQ,GAAW,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC;YAExE,MAAM,GAAG,GAAmB;gBAC1B,IAAI,EAAE,GAAG,CAAC,WAAW;gBACrB,KAAK,EAAE,YAAY,CAAC,GAAG,CAAC,WAAW,CAAC;gBACpC,MAAM,EAAE,GAAG,CAAC,QAAQ;gBACpB,MAAM,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE,UAAU,CAAC;gBACjE,QAAQ,EAAE,UAAU;gBACpB,UAAU,EAAE,GAAG,CAAC,cAAc,KAAK,IAAI;gBACvC,OAAO;gBACP,WAAW,EAAE,WAAW,CAAC,QAAQ,CAAC;gBAClC,SAAS,EAAE,GAAG,CAAC,wBAAwB,IAAI,SAAS;aACrD,CAAC;YAEF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC;gBAAE,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACtE,cAAc,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC3C,CAAC;QAED,0CAA0C;QAC1C,MAAM,SAAS,GAAG,IAAI,GAAG,EAAoB,CAAC;QAC9C,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;YAChC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC;gBAAE,SAAS;YAC5C,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC;gBAAE,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACtE,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAE,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACvD,CAAC;QAED,gDAAgD;QAChD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAsB,CAAC;QACpD,KAAK,MAAM,GAAG,IAAI,YAAY,CAAC,IAAI,EAAE,CAAC;YACpC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC;gBAAE,SAAS;YAC5C,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC;gBAAE,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAC9E,uFAAuF;YACvF,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;QAC7D,CAAC;QAED,qCAAqC;QACrC,MAAM,cAAc,GAAG,IAAI,GAAG,EAA2B,CAAC;QAC1D,KAAK,MAAM,GAAG,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC;YACnC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC;gBAAE,SAAS;YAC3C,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC;gBAAE,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YAE9E,MAAM,QAAQ,GAAI,GAAG,CAAC,QAAmB,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAC7D,+FAA+F;YAC/F,MAAM,QAAQ,GAAI,GAAG,CAAC,QAAmB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC5D,MAAM,OAAO,GAAG,QAAQ;gBACtB,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;gBAC1E,CAAC,CAAC,EAAE,CAAC;YAEP,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC,IAAI,CAAC;gBACtC,IAAI,EAAE,GAAG,CAAC,SAAS;gBACnB,OAAO;gBACP,MAAM,EAAE,QAAQ;gBAChB,UAAU,EAAE,GAAG,CAAC,QAAQ;aACzB,CAAC,CAAC;QACL,CAAC;QAED,4BAA4B;QAC5B,MAAM,KAAK,GAA6B,EAAE,CAAC;QAC3C,KAAK,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;YAClC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC;gBAAE,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YACjD,KAAK,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1C,CAAC;QASD,MAAM,WAAW,GAAc,EAAE,CAAC;QAClC,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;YAChC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC;gBAAE,SAAS;YACjF,WAAW,CAAC,IAAI,CAAC;gBACf,WAAW,EAAE,GAAG,CAAC,YAAY;gBAC7B,YAAY,EAAE,GAAG,CAAC,aAAa;gBAC/B,WAAW,EAAE,GAAG,CAAC,YAAY;gBAC7B,YAAY,EAAE,GAAG,CAAC,aAAa;aAChC,CAAC,CAAC;QACL,CAAC;QAED,gDAAgD;QAChD,yDAAyD;QACzD,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC3C,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;YAC7B,MAAM,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAClD,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAuC,CAAC;QAExE,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;YAC7B,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,WAAW,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YACtD,MAAM,mBAAmB,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YAE7D,gDAAgD;YAChD,+DAA+D;YAC/D,MAAM,aAAa,GAAG,mBAAmB;gBACvC,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACnD,CAAC,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;YAE9C,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC;gBAAE,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YACpF,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAE,CAAC,aAAa,CAAC,GAAG;gBACrD,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,EAAE,CAAC,WAAW;gBACpB,EAAE,EAAE,EAAE,CAAC,WAAW;gBAClB,UAAU,EAAE,EAAE,CAAC,YAAY;gBAC3B,YAAY,EAAE,EAAE,CAAC,YAAY;aAC9B,CAAC;YAEF,+CAA+C;YAC/C,8DAA8D;YAC9D,MAAM,WAAW,GAAG,mBAAmB;gBACrC,CAAC,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC,WAAW,OAAO,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC7E,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;YAEjC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC;gBAAE,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YACpF,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAE,CAAC,WAAW,CAAC,GAAG;gBACnD,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,EAAE,CAAC,WAAW;gBACpB,EAAE,EAAE,EAAE,CAAC,WAAW;gBAClB,UAAU,EAAE,EAAE,CAAC,YAAY;gBAC3B,YAAY,EAAE,EAAE,CAAC,YAAY;aAC9B,CAAC;QACJ,CAAC;QAED,oDAAoD;QACpD,MAAM,MAAM,GAAkC,EAAE,CAAC;QAEjD,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,MAAM,OAAO,GAAG,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;YACpD,MAAM,SAAS,GAA2B,EAAE,CAAC;YAC7C,MAAM,gBAAgB,GAA2B,EAAE,CAAC;YACpD,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;YACtC,MAAM,OAAO,GAA2B,EAAE,CAAC;YAC3C,MAAM,UAAU,GAAa,EAAE,CAAC;YAEhC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;gBAC1B,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC;gBAChC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC;gBACvC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAC1B,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC;gBAE/B,MAAM,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC;gBAChE,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACzB,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC;YAED,MAAM,CAAC,SAAS,CAAC,GAAG;gBAClB,IAAI,EAAE,SAAS;gBACf,OAAO;gBACP,SAAS;gBACT,gBAAgB;gBAChB,WAAW;gBACX,OAAO;gBACP,UAAU;gBACV,UAAU,EAAE,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE;gBAC1C,aAAa,EAAE,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE;gBACjD,SAAS,EAAE,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE;gBAChD,OAAO,EAAE,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE;aAC7C,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;IAC3B,CAAC;YAAS,CAAC;QACT,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC;AACH,CAAC"}

package/dist/pipeline.d.ts

@@ -0,0 +1,44 @@
+/**
+ * @batadata/turbine — Pipeline execution
+ *
+ * Pipelines batch multiple independent queries into a single database round-trip.
+ * Instead of N sequential awaits (N round-trips), you get 1 round-trip for all N queries.
+ *
+ * How it works:
+ *   1. Each query method (findUnique, count, etc.) can produce a DeferredQuery descriptor
+ *      containing the SQL, params, and a transform function.
+ *   2. pipeline() collects these descriptors, executes them in a single Postgres
+ *      pipeline/transaction, and maps each result through its transform.
+ *
+ * In the production Turbine engine, this would go through the Rust proxy which uses
+ * actual Postgres pipeline protocol (libpq PQpipelineEnter). For the TS SDK prototype,
+ * we simulate it by running queries concurrently on a single connection or via
+ * a multi-statement batch.
+ */
+import type pg from 'pg';
+import type { DeferredQuery } from './query.js';
+/**
+ * Execute multiple deferred queries in a single batch.
+ *
+ * Uses a single connection from the pool and runs all queries within
+ * a transaction to guarantee consistency and minimize round-trips.
+ *
+ * @example
+ * ```ts
+ * const [user, count, posts] = await executePipeline(pool, [
+ *   db.users.buildFindUnique({ where: { id: 1 } }),
+ *   db.posts.buildCount({ where: { orgId: 1 } }),
+ *   db.posts.buildFindMany({ where: { userId: 1 }, limit: 10 }),
+ * ]);
+ * ```
+ */
+export declare function executePipeline<T extends readonly DeferredQuery<unknown>[]>(pool: pg.Pool, queries: T): Promise<PipelineResults<T>>;
+/**
+ * Extract the result types from a tuple of DeferredQuery objects.
+ * If you pass [DeferredQuery<User>, DeferredQuery<number>, DeferredQuery<Post[]>],
+ * you get back [User, number, Post[]].
+ */
+export type PipelineResults<T extends readonly DeferredQuery<unknown>[]> = {
+    [K in keyof T]: T[K] extends DeferredQuery<infer R> ? R : never;
+};
+//# sourceMappingURL=pipeline.d.ts.map

package/dist/pipeline.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pipeline.d.ts","sourceRoot":"","sources":["../src/pipeline.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAMhD;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,eAAe,CAAC,CAAC,SAAS,SAAS,aAAa,CAAC,OAAO,CAAC,EAAE,EAC/E,IAAI,EAAE,EAAE,CAAC,IAAI,EACb,OAAO,EAAE,CAAC,GACT,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAmC7B;AAMD;;;;GAIG;AACH,MAAM,MAAM,eAAe,CAAC,CAAC,SAAS,SAAS,aAAa,CAAC,OAAO,CAAC,EAAE,IAAI;KACxE,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK;CAChE,CAAC"}

package/dist/pipeline.js

@@ -0,0 +1,69 @@
+/**
+ * @batadata/turbine — Pipeline execution
+ *
+ * Pipelines batch multiple independent queries into a single database round-trip.
+ * Instead of N sequential awaits (N round-trips), you get 1 round-trip for all N queries.
+ *
+ * How it works:
+ *   1. Each query method (findUnique, count, etc.) can produce a DeferredQuery descriptor
+ *      containing the SQL, params, and a transform function.
+ *   2. pipeline() collects these descriptors, executes them in a single Postgres
+ *      pipeline/transaction, and maps each result through its transform.
+ *
+ * In the production Turbine engine, this would go through the Rust proxy which uses
+ * actual Postgres pipeline protocol (libpq PQpipelineEnter). For the TS SDK prototype,
+ * we simulate it by running queries concurrently on a single connection or via
+ * a multi-statement batch.
+ */
+// ---------------------------------------------------------------------------
+// Pipeline executor
+// ---------------------------------------------------------------------------
+/**
+ * Execute multiple deferred queries in a single batch.
+ *
+ * Uses a single connection from the pool and runs all queries within
+ * a transaction to guarantee consistency and minimize round-trips.
+ *
+ * @example
+ * ```ts
+ * const [user, count, posts] = await executePipeline(pool, [
+ *   db.users.buildFindUnique({ where: { id: 1 } }),
+ *   db.posts.buildCount({ where: { orgId: 1 } }),
+ *   db.posts.buildFindMany({ where: { userId: 1 }, limit: 10 }),
+ * ]);
+ * ```
+ */
+export async function executePipeline(pool, queries) {
+    if (queries.length === 0) {
+        return [];
+    }
+    // Acquire a single connection for the entire batch
+    const client = await pool.connect();
+    try {
+        // Wrap in a transaction for consistency
+        await client.query('BEGIN');
+        // Execute all queries on the same connection — in sequence on a single
+        // connection this avoids pool checkout overhead, and the Postgres server
+        // processes them as a tight batch.
+        //
+        // Execute queries sequentially on the same connection to avoid the
+        // "already executing a query" deprecation in pg@8. This is still faster
+        // than separate pool checkouts because we skip N-1 acquire/release cycles.
+        // Future: use actual Postgres pipeline protocol for true pipelining.
+        const results = [];
+        for (const q of queries) {
+            const raw = await client.query(q.sql, q.params);
+            results.push(q.transform(raw));
+        }
+        await client.query('COMMIT');
+        return results;
+    }
+    catch (err) {
+        await client.query('ROLLBACK');
+        throw err;
+    }
+    finally {
+        client.release();
+    }
+}
+//# sourceMappingURL=pipeline.js.map

package/dist/pipeline.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pipeline.js","sourceRoot":"","sources":["../src/pipeline.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAKH,8EAA8E;AAC9E,oBAAoB;AACpB,8EAA8E;AAE9E;;;;;;;;;;;;;;GAcG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,IAAa,EACb,OAAU;IAEV,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,EAAmC,CAAC;IAC7C,CAAC;IAED,mDAAmD;IACnD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;IAEpC,IAAI,CAAC;QACH,wCAAwC;QACxC,MAAM,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAE5B,uEAAuE;QACvE,yEAAyE;QACzE,mCAAmC;QACnC,EAAE;QACF,mEAAmE;QACnE,wEAAwE;QACxE,2EAA2E;QAC3E,qEAAqE;QACrE,MAAM,OAAO,GAAc,EAAE,CAAC;QAC9B,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACxB,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;YAChD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;QACjC,CAAC;QAED,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAE7B,OAAO,OAA6B,CAAC;IACvC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC/B,MAAM,GAAG,CAAC;IACZ,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC;AACH,CAAC"}