pqb 0.33.0 → 0.33.2

package/dist/index.mjs

@@ -1,4 +1,4 @@
-import { ExpressionTypeMethod, Expression, RawSQLBase, emptyObject, isTemplateLiteralArgs, ColumnTypeBase, setColumnData, pushColumnData, quoteObjectKey, toArray, singleQuote, addCode, singleQuoteArray, objectHasValues, columnDefaultArgumentToCode, columnErrorMessagesToCode, getValueKey, addValue, isExpression, joinTruthy, numberDataToCode, stringDataToCode, getDefaultLanguage, dateDataToCode, pushOrNewArrayToObject, returnArg as returnArg$1, noop, arrayDataToCode, emptyArray, callWithThis, setParserToQuery, applyTransforms, isRawSQL, pushOrNewArray, setDefaultNowFn, setDefaultLanguage, makeTimestampsHelpers, setCurrentColumnName, setAdapterConnectRetry, isObjectEmpty, ValExpression, applyMixins, toSnakeCase, snakeCaseKey } from 'orchid-core';
+import { ExpressionTypeMethod, Expression, RawSQLBase, emptyObject, isTemplateLiteralArgs, ColumnTypeBase, setColumnData, pushColumnData, quoteObjectKey, toArray, singleQuote, addCode, singleQuoteArray, objectHasValues, columnDefaultArgumentToCode, columnErrorMessagesToCode, getValueKey, addValue, isExpression, joinTruthy, numberDataToCode, stringDataToCode, getDefaultLanguage, dateDataToCode, pushOrNewArrayToObject, returnArg as returnArg$1, noop, arrayDataToCode, emptyArray, callWithThis, applyTransforms, setParserToQuery, isRawSQL, pushOrNewArray, setDefaultNowFn, setDefaultLanguage, makeTimestampsHelpers, setCurrentColumnName, setAdapterConnectRetry, isObjectEmpty, ValExpression, applyMixins, toSnakeCase, snakeCaseKey } from 'orchid-core';
 import pg from 'pg';
 import { inspect } from 'node:util';
 import { AsyncLocalStorage } from 'node:async_hooks';
@@ -3363,9 +3363,7 @@ const then = async (q, adapter, trx, beforeHooks, afterHooks, afterCommitHooks,
         result = filterResult(q, returnType, queryResult, hookSelect, result);
     }
     if (query.transform) {
-      for (const fn of query.transform) {
-        result = fn(result);
-      }
+      result = applyTransforms(returnType, query.transform, result);
     }
     return resolve == null ? void 0 : resolve(result);
   } catch (err) {
@@ -3583,10 +3581,8 @@ const addParserForSelectItem = (q, as, key, arg) => {
         setParserToQuery(q.q, key, (item) => {
           const t = query.returnType || "all";
           subQueryResult.rows = t === "value" || t === "valueOrThrow" ? [[item]] : t === "one" || t === "oneOrThrow" ? [item] : item;
-          return applyTransforms(
-            query.transform,
-            query.handleResult(arg, t, subQueryResult, true)
-          );
+          const result = query.handleResult(arg, t, subQueryResult, true);
+          return query.transform ? applyTransforms(t, query.transform, result) : result;
         });
       }
       if (query.returnType === "valueOrThrow" || query.returnType === "oneOrThrow") {
@@ -6711,7 +6707,10 @@ class Create {
    * @param args - object with columns list and raw SQL for values
    */
   createRaw(...args) {
-    return _queryCreateRaw(this.clone(), args);
+    return _queryCreateRaw(
+      this.clone(),
+      args
+    );
   }
   /**
    * Works exactly as {@link createRaw}, except that it returns inserted row count by default.
@@ -6719,7 +6718,10 @@ class Create {
    * @param args - object with columns list and raw SQL for values
    */
   insertRaw(...args) {
-    return _queryInsertRaw(this.clone(), args);
+    return _queryInsertRaw(
+      this.clone(),
+      args
+    );
   }
   /**
    * `createManyRaw` and `insertManyRaw` are for creating many record with raw SQL expressions.
@@ -6740,7 +6742,10 @@ class Create {
    * @param args - object with columns list and array of raw SQL for values
    */
   createManyRaw(...args) {
-    return _queryCreateManyRaw(this.clone(), args);
+    return _queryCreateManyRaw(
+      this.clone(),
+      args
+    );
   }
   /**
    * Works exactly as {@link createManyRaw}, except that it returns inserted row count by default.
@@ -6748,7 +6753,10 @@ class Create {
    * @param args - object with columns list and array of raw SQL for values
    */
   insertManyRaw(...args) {
-    return _queryInsertManyRaw(this.clone(), args);
+    return _queryInsertManyRaw(
+      this.clone(),
+      args
+    );
   }
   /**
    * These methods are for creating a single record, for batch creating see {@link createManyFrom}.
@@ -6790,7 +6798,11 @@ class Create {
    * @param data - additionally you can set some columns
    */
   createFrom(query, data) {
-    return _queryCreateFrom(this.clone(), query, data);
+    return _queryCreateFrom(
+      this.clone(),
+      query,
+      data
+    );
   }
   /**
    * Works exactly as {@link createFrom}, except that it returns inserted row count by default.
@@ -6799,7 +6811,11 @@ class Create {
    * @param data - additionally you can set some columns
    */
   insertFrom(query, data) {
-    return _queryInsertFrom(this.clone(), query, data);
+    return _queryInsertFrom(
+      this.clone(),
+      query,
+      data
+    );
   }
   /**
    * Similar to `createFrom`, but intended to create many records.
@@ -6815,7 +6831,10 @@ class Create {
    * @param query - query to create new records from
    */
   createManyFrom(query) {
-    return _queryCreateManyFrom(this.clone(), query);
+    return _queryCreateManyFrom(
+      this.clone(),
+      query
+    );
   }
   /**
    * Works exactly as {@link createManyFrom}, except that it returns inserted row count by default.
@@ -6823,7 +6842,10 @@ class Create {
    * @param query - query to create new records from
    */
   insertManyFrom(query) {
-    return _queryInsertManyFrom(this.clone(), query);
+    return _queryInsertManyFrom(
+      this.clone(),
+      query
+    );
   }
   /**
    * `defaults` allows setting values that will be used later in `create`.
@@ -6848,7 +6870,10 @@ class Create {
    * @param data - default values for `create` and `createMany` which will follow `defaults`
    */
   defaults(data) {
-    return _queryDefaults(this.clone(), data);
+    return _queryDefaults(
+      this.clone(),
+      data
+    );
   }
   /**
    * By default, violating unique constraint will cause the creative query to throw,
@@ -7310,9 +7335,7 @@ class Having {
     return pushQueryValue(
       q,
       "having",
-      args.map(
-        (arg) => arg(q).q.expr
-      )
+      args.map((arg) => arg(q).q.expr)
     );
   }
   /**
@@ -10362,6 +10385,8 @@ class TransformMethods {
    *
    * `transform` method should be called in the last order, other methods can't be chained after calling it.
    *
+   * It is meant to transform the whole result of a query, for transforming individual records consider using {@link QueryMap.map}.
+   *
    * The [hooks](/guide/hooks.html) that are going to run after the query will receive the query result **before** transferring.
    *
    * Consider the following example of a cursor-based pagination by `id`:
@@ -10413,6 +10438,52 @@ class TransformMethods {
   }
 }
 
+class QueryMap {
+  /**
+   * Use `map` to transform individual records of a query result.
+   *
+   * It accepts a single record and should return a single transformed record.
+   *
+   * For transforming the whole result of a query, consider using [transform](#transform) instead.
+   *
+   * The [hooks](/guide/hooks) that are going to run after the query will receive the query result **before** transformation.
+   *
+   * ```ts
+   * // add a `titleLength` to every post
+   * const posts = await db.post.limit(10).map((post) => ({
+   *   ...post,
+   *   titleLength: post.title.length,
+   * }));
+   *
+   * posts[0].titleLength; // number
+   *
+   * // using the exact same `map` function to transform a single post
+   * const singlePost = await db.post.find(id).map((post) => ({
+   *   ...post,
+   *   titleLength: post.title.length,
+   * }));
+   *
+   * singlePost.titleLength; // number
+   *
+   * // can be used in sub-queries
+   * const postsWithComments = await db.post.select('title', {
+   *   comments: (q) =>
+   *     q.comments.map((comment) => ({
+   *       ...comment,
+   *       truncatedContent: comment.content.slice(0, 100),
+   *     })),
+   * });
+   *
+   * postsWithComments[0].comments[0].truncatedContent; // string
+   * ```
+   *
+   * @param fn - function to transform an individual record
+   */
+  map(fn) {
+    return pushQueryValue(this.clone(), "transform", { map: fn });
+  }
+}
+
 class ScopeMethods {
   /**
    * See {@link ScopeMethods}
@@ -11394,6 +11465,7 @@ applyMixins(QueryMethods, [
   MergeQueryMethods,
   SqlMethod,
   TransformMethods,
+  QueryMap,
   ScopeMethods,
   SoftDeleteMethods,
   ExpressionMethods