pqb 0.56.7 → 0.56.8

package/dist/index.mjs

@@ -1,4 +1,4 @@
-import { ExpressionTypeMethod, Expression, RawSQLBase, emptyObject, isTemplateLiteralArgs, ColumnTypeBase, setColumnData, pushColumnData, templateLiteralSQLToCode, quoteObjectKey, toArray, emptyArray, singleQuote, addCode, singleQuoteArray, objectHasValues, toSnakeCase, columnDefaultArgumentToCode, columnErrorMessagesToCode, setObjectValueImmutable, getValueKey, addValue, isExpression, dateDataToCode, joinTruthy, arrayDataToCode, numberDataToCode, noop, stringDataToCode, getDefaultLanguage, setDefaultNowFn, setDefaultLanguage, setCurrentColumnName, timestampHelpers, _getQueryAliasOrName, _getQueryOuterAliases, returnArg, pushQueryValueImmutable, NotFoundError, OrchidOrmInternalError, _setSubQueryAliases, _applyRelationAliases, isRelationQuery, getFreeAlias, _checkIfAliased, logColors, OrchidOrmError, applyTransforms, callWithThis, requirePrimaryKeys, pick, _setQueryAs, _copyQueryAliasToQuery, setParserToQuery, newDelayedRelationSelect, pushOrNewArray, getPrimaryKeys, setDelayedRelation, UnhandledTypeError, isRawSQL, pushOrNewArrayToObjectImmutable, _getQueryFreeAlias, _setQueryAlias, QueryHookUtils, MoreThanOneRowError, isObjectEmpty, ValExpression, applyMixins, _getQueryAs, QueryError, snakeCaseKey } from 'orchid-core';
+import { ExpressionTypeMethod, Expression, RawSQLBase, emptyObject, isTemplateLiteralArgs, ColumnTypeBase, setColumnData, pushColumnData, templateLiteralSQLToCode, quoteObjectKey, toArray, emptyArray, singleQuote, addCode, singleQuoteArray, objectHasValues, toSnakeCase, columnDefaultArgumentToCode, columnErrorMessagesToCode, setObjectValueImmutable, getValueKey, addValue, isIterable, isExpression, dateDataToCode, joinTruthy, arrayDataToCode, numberDataToCode, noop, stringDataToCode, getDefaultLanguage, setDefaultNowFn, setDefaultLanguage, setCurrentColumnName, timestampHelpers, _getQueryAliasOrName, _getQueryOuterAliases, returnArg, pushQueryValueImmutable, NotFoundError, OrchidOrmInternalError, _setSubQueryAliases, _applyRelationAliases, isRelationQuery, getFreeAlias, _checkIfAliased, logColors, OrchidOrmError, applyTransforms, callWithThis, requirePrimaryKeys, pick, _setQueryAs, _copyQueryAliasToQuery, setParserToQuery, newDelayedRelationSelect, pushOrNewArray, getPrimaryKeys, setDelayedRelation, UnhandledTypeError, isRawSQL, pushOrNewArrayToObjectImmutable, _getQueryFreeAlias, _setQueryAlias, QueryHookUtils, MoreThanOneRowError, isObjectEmpty, ValExpression, applyMixins, _getQueryAs, QueryError, snakeCaseKey } from 'orchid-core';
 import { inspect } from 'node:util';
 import { AsyncLocalStorage } from 'node:async_hooks';
 import { templateLiteralToSQL as templateLiteralToSQL$1 } from 'pqb';
@@ -968,7 +968,7 @@ class EnumColumn extends ColumnType {
     super(schema, schemaType);
     this.enumName = enumName;
     this.options = options;
-    this.operators = Operators.any;
+    this.operators = Operators.ordinalText;
     this.dataType = "enum";
     this.inputSchema = this.outputSchema = this.querySchema = schemaType;
   }
@@ -1069,8 +1069,8 @@ const makeVarArg = (_op) => {
 };
 const quoteValue = (arg, ctx, quotedAs, IN) => {
   if (arg && typeof arg === "object") {
-    if (IN && Array.isArray(arg)) {
-      return `(${arg.map((value) => addValue(ctx.values, value)).join(", ")})`;
+    if (IN && isIterable(arg)) {
+      return `(${(Array.isArray(arg) ? arg : [...arg]).map((value) => addValue(ctx.values, value)).join(", ")})`;
     }
     if (isExpression(arg)) {
       return arg.toSQL(ctx, quotedAs);
@@ -1111,16 +1111,7 @@ const base = {
     (key, value, ctx, quotedAs) => `NOT ${key} IN ${quoteValue(value, ctx, quotedAs, true)}`
   )
 };
-const boolean = {
-  ...base,
-  and: make(
-    (key, value, ctx, quotedAs) => `${key} AND ${value.q.expr.toSQL(ctx, quotedAs)}`
-  ),
-  or: make(
-    (key, value, ctx, quotedAs) => `(${key}) OR (${value.q.expr.toSQL(ctx, quotedAs)})`
-  )
-};
-const numeric = {
+const ord = {
   ...base,
   lt: make(
     (key, value, ctx, quotedAs) => `${key} < ${quoteValue(value, ctx, quotedAs)}`
@@ -1142,6 +1133,15 @@ const numeric = {
     )}`
   )
 };
+const boolean = {
+  ...ord,
+  and: make(
+    (key, value, ctx, quotedAs) => `${key} AND ${value.q.expr.toSQL(ctx, quotedAs)}`
+  ),
+  or: make(
+    (key, value, ctx, quotedAs) => `(${key}) OR (${value.q.expr.toSQL(ctx, quotedAs)})`
+  )
+};
 const text = {
   ...base,
   contains: make(
@@ -1163,6 +1163,10 @@ const text = {
     (key, value, ctx, quotedAs) => `${key} LIKE '%' || ${quoteLikeValue(value, ctx, quotedAs)}`
   )
 };
+const ordinalText = {
+  ...ord,
+  ...text
+};
 const encodeJsonPath = (ctx, path) => addValue(ctx.values, `{${Array.isArray(path) ? path.join(", ") : path}}`);
 const jsonPathQueryOp = (key, [path, options], ctx) => `jsonb_path_query_first(${key}, ${addValue(ctx.values, path)}${options?.vars ? `, ${addValue(ctx.values, JSON.stringify(options.vars))}${options.silent ? ", true" : ""}` : options?.silent ? ", NULL, true" : ""})`;
 const quoteJsonValue = (arg, ctx, quotedAs, IN) => {
@@ -1191,6 +1195,7 @@ const serializeJsonValue = (arg, ctx, quotedAs) => {
   return addValue(ctx.values, JSON.stringify(arg));
 };
 const json = {
+  ...ord,
   equals: make(
     (key, value, ctx, quotedAs) => value === null ? `nullif(${key}, 'null'::jsonb) IS NULL` : `${key} = ${quoteJsonValue(value, ctx, quotedAs)}`
   ),
@@ -1257,7 +1262,7 @@ const json = {
   )
 };
 const array = {
-  ...base,
+  ...ord,
   has: make(
     (key, value, ctx, quotedAs) => `${quoteValue(value, ctx, quotedAs)} = ANY(${key})`
   ),
@@ -1275,7 +1280,7 @@ const array = {
     return typeof value === "number" ? `${expr} = ${quoteValue(value, ctx, quotedAs)}` : Object.keys(value).map(
       (key2) => (
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        numeric[key2]._op(expr, value[key2], ctx, quotedAs)
+        ord[key2]._op(expr, value[key2], ctx, quotedAs)
       )
     ).join(" AND ");
   })
@@ -1283,9 +1288,10 @@ const array = {
 const Operators = {
   any: base,
   boolean,
-  number: numeric,
-  date: numeric,
-  time: numeric,
+  ordinalText,
+  number: ord,
+  date: ord,
+  time: ord,
   text,
   json,
   array
@@ -1755,6 +1761,9 @@ const defaultSchemaConfig = {
   asType() {
     return this;
   },
+  narrowType() {
+    return this;
+  },
   dateAsNumber() {
     return this.parse(Date.parse);
   },
@@ -1820,6 +1829,7 @@ class LimitedTextBaseColumn extends TextBaseColumn {
       schema,
       limit !== void 0 ? schema.stringMax(limit) : schema.stringSchema()
     );
+    this.operators = Operators.ordinalText;
     this.data.maxChars = limit;
   }
   toSQL() {
@@ -1886,6 +1896,7 @@ class TextColumn extends TextBaseColumn {
   constructor(schema) {
     super(schema, schema.stringSchema());
     this.dataType = "text";
+    this.operators = Operators.ordinalText;
   }
   static get instance() {
     return this._instance ?? (this._instance = new TextColumn(defaultSchemaConfig));
@@ -1899,7 +1910,7 @@ class ByteaColumn extends ColumnType {
   constructor(schema) {
     super(schema, schema.buffer());
     this.dataType = "bytea";
-    this.operators = Operators.text;
+    this.operators = Operators.ordinalText;
     setColumnDefaultParse(this, byteaParse);
   }
   toCode(ctx, key) {
@@ -2009,7 +2020,7 @@ class InetColumn extends ColumnType {
   constructor(schema) {
     super(schema, schema.stringSchema());
     this.dataType = "inet";
-    this.operators = Operators.text;
+    this.operators = Operators.ordinalText;
   }
   toCode(ctx, key) {
     return columnCode(this, ctx, key, `inet()`);
@@ -2019,7 +2030,7 @@ class MacAddrColumn extends ColumnType {
   constructor(schema) {
     super(schema, schema.stringSchema());
     this.dataType = "macaddr";
-    this.operators = Operators.text;
+    this.operators = Operators.ordinalText;
   }
   toCode(ctx, key) {
     return columnCode(this, ctx, key, `macaddr()`);
@@ -2029,7 +2040,7 @@ class MacAddr8Column extends ColumnType {
   constructor(schema) {
     super(schema, schema.stringSchema());
     this.dataType = "macaddr8";
-    this.operators = Operators.text;
+    this.operators = Operators.ordinalText;
   }
   toCode(ctx, key) {
     return columnCode(this, ctx, key, `macaddr8()`);
@@ -2039,7 +2050,7 @@ class BitColumn extends ColumnType {
   constructor(schema, length) {
     super(schema, schema.bit(length));
     this.dataType = "bit";
-    this.operators = Operators.text;
+    this.operators = Operators.ordinalText;
     this.data.length = length;
   }
   toCode(ctx, key) {
@@ -2057,7 +2068,7 @@ class BitVaryingColumn extends ColumnType {
   constructor(schema, length) {
     super(schema, schema.bit(length));
     this.dataType = "varbit";
-    this.operators = Operators.text;
+    this.operators = Operators.ordinalText;
     this.data.length = length;
     this.data.alias = "bitVarying";
   }
@@ -2077,7 +2088,7 @@ class TsVectorColumn extends ColumnType {
     super(schema, schema.stringSchema());
     this.defaultLanguage = defaultLanguage;
     this.dataType = "tsvector";
-    this.operators = Operators.text;
+    this.operators = Operators.ordinalText;
   }
   toCode(ctx, key) {
     return columnCode(this, ctx, key, `tsvector()`);
@@ -2169,7 +2180,7 @@ class TsQueryColumn extends ColumnType {
   constructor(schema) {
     super(schema, schema.stringSchema());
     this.dataType = "tsquery";
-    this.operators = Operators.text;
+    this.operators = Operators.ordinalText;
   }
   toCode(ctx, key) {
     return columnCode(this, ctx, key, `tsquery()`);
@@ -2181,7 +2192,7 @@ class UUIDColumn extends ColumnType {
   constructor(schema) {
     super(schema, schema.uuid());
     this.dataType = "uuid";
-    this.operators = Operators.text;
+    this.operators = Operators.ordinalText;
     this.data.defaultDefault = uuidDefault;
   }
   /**
@@ -2213,6 +2224,7 @@ class CitextColumn extends TextBaseColumn {
   constructor(schema) {
     super(schema, schema.stringSchema());
     this.dataType = "citext";
+    this.operators = Operators.ordinalText;
     this.data.extension = "citext";
   }
   toCode(ctx, key) {
@@ -3428,6 +3440,7 @@ const _queryOrNot = (q, args) => {
 const _queryWhereIn = (q, and, arg, values, not) => {
   let item;
   if (values) {
+    if (isIterable(values)) values = [...values];
     if ("length" in values && !values.length) {
       return _queryNone(q);
     }
@@ -3445,7 +3458,7 @@ const _queryWhereIn = (q, and, arg, values, not) => {
     item = {};
     for (const key in arg) {
       const values2 = arg[key];
-      if ("length" in values2 && !values2.length) {
+      if ("length" in values2 && !values2.length || "size" in values2 && !values2.size) {
         return _queryNone(q);
       }
       item[key] = { in: values2 };
@@ -3988,8 +4001,10 @@ class Where {
    *
    * ```ts
    * db.table.whereIn('column', [1, 2, 3]);
+   * db.table.whereIn('column', new Set([1, 2, 3]));
    * // the same as:
-   * db.table.where({ column: [1, 2, 3] });
+   * db.table.where({ column: { in: [1, 2, 3] } });
+   * db.table.where({ column: { in: new Set([1, 2, 3]) } });
    * ```
    *
    * `whereIn` can support a tuple of columns, that's what the `in` operator cannot support:
@@ -4020,9 +4035,9 @@ class Where {
    *
    * ```ts
    * // following queries resolves into `none`:
-   * db.table.where('id', [])
-   * db.table.where(['id', 'name'], [])
-   * db.table.where({ id: [] })
+   * db.table.whereIn('id', [])
+   * db.table.whereIn(['id', 'name'], [])
+   * db.table.whereIn({ id: [] })
    * ```
    */
   whereIn(...args) {
@@ -11781,7 +11796,7 @@ class QueryMap {
    *
    * ```ts
    * // add a `titleLength` to every post
-   * const posts = await db.post.limit(10).map((post) => ({
+   * const posts = await db.post.limit(10).map((post, i) => ({
    *   ...post,
    *   titleLength: post.title.length,
    * }));
@@ -11789,7 +11804,7 @@ class QueryMap {
    * posts[0].titleLength; // number
    *
    * // using the exact same `map` function to transform a single post
-   * const singlePost = await db.post.find(id).map((post) => ({
+   * const singlePost = await db.post.find(id).map((post, i) => ({
    *   ...post,
    *   titleLength: post.title.length,
    * }));
@@ -11799,7 +11814,7 @@ class QueryMap {
    * // can be used in sub-queries
    * const postsWithComments = await db.post.select('title', {
    *   comments: (q) =>
-   *     q.comments.map((comment) => ({
+   *     q.comments.map((comment, i) => ({
    *       ...comment,
    *       truncatedContent: comment.content.slice(0, 100),
    *     })),
@@ -11809,10 +11824,12 @@ class QueryMap {
    * ```
    *
    * @param fn - function to transform an individual record
+   * @param thisArg - same as in the native array map
    */
-  map(fn) {
+  map(fn, thisArg) {
     return pushQueryValueImmutable(_clone(this), "transform", {
-      map: fn
+      map: fn,
+      thisArg
     });
   }
 }