turbine-orm 0.19.2 → 0.22.0

package/dist/cjs/dialect.js

@@ -6,6 +6,39 @@
  * PostgreSQL-native by default, but query generation now depends on this
  * contract for the SQL primitives that vary across MySQL and SQLite.
  */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.postgresDialect = void 0;
 const errors_js_1 = require("./errors.js");
@@ -13,11 +46,17 @@ const schema_js_1 = require("./schema.js");
 /** PostgreSQL implementation of the dialect contract. */
 exports.postgresDialect = {
     name: 'postgresql',
+    resultStrategy: 'returning',
     supportsReturning: true,
     supportsILike: true,
     jsonPathSupport: 'native',
     emptyJsonArrayLiteral: "'[]'::json",
     nullJsonLiteral: 'NULL',
+    aggSupportsInlineOrderBy: true,
+    supportsVector: true,
+    supportsListenNotify: true,
+    supportsRLS: true,
+    supportsAdvisoryLock: true,
     paramPlaceholder(index) {
         return `$${index}`;
     },
@@ -35,6 +74,9 @@ exports.postgresDialect = {
         const suffix = orderBy ? ` ${orderBy}` : '';
         return `COALESCE(json_agg(${jsonObjectExpr}${suffix}), ${this.emptyJsonArrayLiteral})`;
     },
+    wrapJsonSubresult(subquery, fallback) {
+        return `COALESCE((${subquery}), ${fallback})`;
+    },
     buildReturningClause(selection = '*') {
         return ` RETURNING ${selection}`;
     },
@@ -76,6 +118,15 @@ exports.postgresDialect = {
     typeToTypeScript(dialectType, nullable) {
         return (0, schema_js_1.pgTypeToTs)(dialectType, nullable);
     },
+    castAggregate(expr, target) {
+        return `${expr}::${target}`;
+    },
+    buildInClause(expr, paramRef, negated) {
+        return negated ? `${expr} != ALL(${paramRef})` : `${expr} = ANY(${paramRef})`;
+    },
+    inClauseParam(values) {
+        return values;
+    },
     arrayType(baseType) {
         return (0, schema_js_1.pgArrayType)(baseType);
     },
@@ -131,4 +182,69 @@ exports.postgresDialect = {
     buildMigrationDeleteApplied(table) {
         return `DELETE FROM ${table} WHERE name = ${this.paramPlaceholder(1)}`;
     },
+    beginStatement(isolationLevel) {
+        return isolationLevel ? `BEGIN ISOLATION LEVEL ${isolationLevel}` : 'BEGIN';
+    },
+    commitStatement() {
+        return 'COMMIT';
+    },
+    rollbackStatement() {
+        return 'ROLLBACK';
+    },
+    savepointStatement(name) {
+        return `SAVEPOINT ${name}`;
+    },
+    releaseSavepointStatement(name) {
+        return `RELEASE SAVEPOINT ${name}`;
+    },
+    rollbackToSavepointStatement(name) {
+        return `ROLLBACK TO SAVEPOINT ${name}`;
+    },
+    buildSetSessionConfig(name, value) {
+        // set_config(name, value, is_local=true) — the parameterizable,
+        // transaction-local equivalent of `SET LOCAL` (which rejects bind params).
+        return {
+            sql: `SELECT set_config(${this.paramPlaceholder(1)}, ${this.paramPlaceholder(2)}, true)`,
+            params: [name, value],
+        };
+    },
+    async *openStream(connection, sql, params, batchSize) {
+        // Cursors require a single connection inside a transaction. Identical SQL
+        // sequence to the historical inline implementation: BEGIN → DECLARE … NO
+        // SCROLL CURSOR FOR → FETCH n (loop) → CLOSE → COMMIT; ROLLBACK on error.
+        const cursorName = `turbine_cursor_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
+        const quotedCursor = this.quoteIdentifier(cursorName);
+        await connection.query(this.beginStatement());
+        try {
+            await connection.query(`DECLARE ${quotedCursor} NO SCROLL CURSOR FOR ${sql}`, params);
+            while (true) {
+                const batch = await connection.query(`FETCH ${batchSize} FROM ${quotedCursor}`);
+                if (batch.rows.length === 0)
+                    break;
+                yield batch.rows;
+                if (batch.rows.length < batchSize)
+                    break;
+            }
+            await connection.query(`CLOSE ${quotedCursor}`);
+            await connection.query(this.commitStatement());
+        }
+        catch (err) {
+            try {
+                await connection.query(this.rollbackStatement());
+            }
+            catch {
+                // Connection may already be broken — ignore rollback error.
+            }
+            throw err;
+        }
+    },
+    // Postgres introspection wraps the information_schema / pg_catalog reader in
+    // introspect.ts. A dynamic import keeps the static graph acyclic (introspect.ts
+    // imports postgresDialect) and calls the raw catalog reader (never the router).
+    introspector: {
+        async introspect(options) {
+            const { introspectPostgresCatalog } = await Promise.resolve().then(() => __importStar(require('./introspect.js')));
+            return introspectPostgresCatalog(options);
+        },
+    },
 };

package/dist/cjs/errors.js

@@ -6,7 +6,7 @@
  * All Turbine errors extend TurbineError which includes a `code` property.
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.OptimisticLockError = exports.PipelineError = exports.ExclusionConstraintError = exports.CheckConstraintError = exports.SerializationFailureError = exports.DeadlockError = exports.NotNullViolationError = exports.ForeignKeyError = exports.UniqueConstraintError = exports.CircularRelationError = exports.MigrationError = exports.RelationError = exports.ConnectionError = exports.ValidationError = exports.TimeoutError = exports.NotFoundError = exports.TurbineError = exports.TurbineErrorCode = void 0;
+exports.UnsupportedFeatureError = exports.OptimisticLockError = exports.PipelineError = exports.ExclusionConstraintError = exports.CheckConstraintError = exports.SerializationFailureError = exports.DeadlockError = exports.NotNullViolationError = exports.ForeignKeyError = exports.UniqueConstraintError = exports.CircularRelationError = exports.MigrationError = exports.RelationError = exports.ConnectionError = exports.ValidationError = exports.TimeoutError = exports.NotFoundError = exports.TurbineError = exports.TurbineErrorCode = void 0;
 exports.setErrorMessageMode = setErrorMessageMode;
 exports.getErrorMessageMode = getErrorMessageMode;
 exports.wrapPgError = wrapPgError;
@@ -28,6 +28,7 @@ exports.TurbineErrorCode = {
     PIPELINE: 'TURBINE_E014',
     OPTIMISTIC_LOCK: 'TURBINE_E015',
     EXCLUSION_VIOLATION: 'TURBINE_E016',
+    UNSUPPORTED_FEATURE: 'TURBINE_E017',
 };
 /** Base error class for all Turbine errors */
 class TurbineError extends Error {
@@ -458,6 +459,23 @@ class OptimisticLockError extends TurbineError {
     }
 }
 exports.OptimisticLockError = OptimisticLockError;
+/**
+ * Thrown when a Postgres-only feature (pgvector distance ops, LISTEN/NOTIFY
+ * realtime, RLS session GUCs, advisory-lock migration locking, ...) is invoked
+ * on a dialect/engine whose capability flag reports it unsupported. Surfaces a
+ * clear `unsupported on <engine>` message instead of generating broken SQL.
+ */
+class UnsupportedFeatureError extends TurbineError {
+    feature;
+    dialect;
+    constructor(feature, dialect, hint) {
+        super(exports.TurbineErrorCode.UNSUPPORTED_FEATURE, `[turbine] ${feature} is unsupported on "${dialect}".${hint ? ` ${hint}` : ''}`);
+        this.name = 'UnsupportedFeatureError';
+        this.feature = feature;
+        this.dialect = dialect;
+    }
+}
+exports.UnsupportedFeatureError = UnsupportedFeatureError;
 /**
  * Parse column names out of a pg `detail` string like:
  *   "Key (email)=(foo@bar) already exists."

package/dist/cjs/generate.js

@@ -363,6 +363,10 @@ function generateMetadata(schema) {
     lines.push('  },');
     lines.push('};');
     lines.push('');
+    // Back-compat lowercase alias. `SCHEMA` is the canonical export, but docs and
+    // users frequently import `schema`; emit both so either name resolves.
+    lines.push('export const schema = SCHEMA;');
+    lines.push('');
     return lines.join('\n');
 }
 // ---------------------------------------------------------------------------

package/dist/cjs/index.js

@@ -34,8 +34,8 @@
  * ```
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.column = exports.ColumnBuilder = exports.applyManyToManyRelations = exports.snakeToPascal = exports.snakeToCamel = exports.singularize = exports.pgTypeToTs = exports.pgArrayType = exports.normalizeKeyColumns = exports.isDateType = exports.camelToSnake = exports.validateChannel = exports.QueryInterface = exports.pipelineSupported = exports.executePipeline = exports.hasRelationFields = exports.executeNestedUpdate = exports.executeNestedCreate = exports.introspect = exports.generate = exports.wrapPgError = exports.ValidationError = exports.UniqueConstraintError = exports.TurbineErrorCode = exports.TurbineError = exports.TimeoutError = exports.setErrorMessageMode = exports.SerializationFailureError = exports.RelationError = exports.PipelineError = exports.OptimisticLockError = exports.NotNullViolationError = exports.NotFoundError = exports.MigrationError = exports.getErrorMessageMode = exports.ForeignKeyError = exports.ExclusionConstraintError = exports.DeadlockError = exports.ConnectionError = exports.CircularRelationError = exports.CheckConstraintError = exports.postgresDialect = exports.withRetry = exports.TurbineClient = exports.TransactionClient = exports.yugabytedb = exports.timescale = exports.postgresql = exports.cockroachdb = exports.alloydb = void 0;
-exports.TypedSqlQuery = exports.buildTypedSql = exports.turbineHttp = exports.schemaToSQLString = exports.schemaToSQL = exports.schemaPush = exports.schemaDiff = exports.table = exports.defineSchema = void 0;
+exports.ColumnBuilder = exports.applyManyToManyRelations = exports.snakeToPascal = exports.snakeToCamel = exports.singularize = exports.pgTypeToTs = exports.pgArrayType = exports.normalizeKeyColumns = exports.isDateType = exports.camelToSnake = exports.validateChannel = exports.QueryInterface = exports.pipelineSupported = exports.executePipeline = exports.hasRelationFields = exports.executeNestedUpdate = exports.executeNestedCreate = exports.introspect = exports.generate = exports.wrapPgError = exports.ValidationError = exports.UnsupportedFeatureError = exports.UniqueConstraintError = exports.TurbineErrorCode = exports.TurbineError = exports.TimeoutError = exports.setErrorMessageMode = exports.SerializationFailureError = exports.RelationError = exports.PipelineError = exports.OptimisticLockError = exports.NotNullViolationError = exports.NotFoundError = exports.MigrationError = exports.getErrorMessageMode = exports.ForeignKeyError = exports.ExclusionConstraintError = exports.DeadlockError = exports.ConnectionError = exports.CircularRelationError = exports.CheckConstraintError = exports.postgresDialect = exports.withRetry = exports.TurbineClient = exports.TransactionClient = exports.yugabytedb = exports.timescale = exports.postgresql = exports.cockroachdb = exports.alloydb = void 0;
+exports.TypedSqlQuery = exports.buildTypedSql = exports.turbineHttp = exports.schemaToSQLString = exports.schemaToSQL = exports.schemaPush = exports.schemaDiff = exports.table = exports.defineSchema = exports.column = void 0;
 var index_js_1 = require("./adapters/index.js");
 Object.defineProperty(exports, "alloydb", { enumerable: true, get: function () { return index_js_1.alloydb; } });
 Object.defineProperty(exports, "cockroachdb", { enumerable: true, get: function () { return index_js_1.cockroachdb; } });
@@ -70,6 +70,7 @@ Object.defineProperty(exports, "TimeoutError", { enumerable: true, get: function
 Object.defineProperty(exports, "TurbineError", { enumerable: true, get: function () { return errors_js_1.TurbineError; } });
 Object.defineProperty(exports, "TurbineErrorCode", { enumerable: true, get: function () { return errors_js_1.TurbineErrorCode; } });
 Object.defineProperty(exports, "UniqueConstraintError", { enumerable: true, get: function () { return errors_js_1.UniqueConstraintError; } });
+Object.defineProperty(exports, "UnsupportedFeatureError", { enumerable: true, get: function () { return errors_js_1.UnsupportedFeatureError; } });
 Object.defineProperty(exports, "ValidationError", { enumerable: true, get: function () { return errors_js_1.ValidationError; } });
 Object.defineProperty(exports, "wrapPgError", { enumerable: true, get: function () { return errors_js_1.wrapPgError; } });
 // Code generation

package/dist/cjs/introspect.js

@@ -13,6 +13,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.introspect = introspect;
+exports.introspectPostgresCatalog = introspectPostgresCatalog;
 const pg_1 = __importDefault(require("pg"));
 const dialect_js_1 = require("./dialect.js");
 const schema_js_1 = require("./schema.js");
@@ -100,7 +101,26 @@ const SQL_ENUMS = `
 // ---------------------------------------------------------------------------
 // Main introspection function
 // ---------------------------------------------------------------------------
+/**
+ * Introspect a database into {@link SchemaMetadata}, routing through the active
+ * dialect's {@link Dialect.introspector} so each engine can override the catalog
+ * SQL. PostgreSQL is driven by {@link introspectPostgresCatalog}.
+ */
 async function introspect(options) {
+    const dialect = options.dialect ?? dialect_js_1.postgresDialect;
+    const introspector = dialect.introspector;
+    if (introspector) {
+        return introspector.introspect(options);
+    }
+    // Dialects without an introspector fall back to the Postgres catalog reader.
+    return introspectPostgresCatalog(options);
+}
+/**
+ * PostgreSQL catalog introspector: reads information_schema + pg_catalog and
+ * produces {@link SchemaMetadata}. This is the implementation wrapped by
+ * `postgresDialect.introspector`; call {@link introspect} for dialect routing.
+ */
+async function introspectPostgresCatalog(options) {
     const schema = options.schema ?? 'public';
     const dialect = dialect_js_1.postgresDialect;
     const pool = new pg_1.default.Pool({