pqb 0.35.6 → 0.36.0

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, applyTransforms, callWithThis, setParserToQuery, 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, toSnakeCase, dateDataToCode, pushOrNewArrayToObject, returnArg as returnArg$1, noop, arrayDataToCode, emptyArray, applyTransforms, callWithThis, setParserToQuery, isRawSQL, pushOrNewArray, setDefaultNowFn, setDefaultLanguage, makeTimestampsHelpers, setCurrentColumnName, setAdapterConnectRetry, isObjectEmpty, ValExpression, applyMixins, snakeCaseKey } from 'orchid-core';
 import pg from 'pg';
 import { inspect } from 'node:util';
 import { AsyncLocalStorage } from 'node:async_hooks';
@@ -378,7 +378,8 @@ class ColumnType extends ColumnTypeBase {
    * @param args - raw SQL
    */
   generated(...args) {
-    return setColumnData(this, "generated", raw(...args));
+    const sql = raw(...args);
+    return setColumnData(this, "generated", (...args2) => sql.toSQL(...args2));
   }
 }
 
@@ -459,6 +460,10 @@ const columnsShapeToCode = (shape, t) => {
   for (const key in shape) {
     if (hasTimestamps && (key === "createdAt" || key === "updatedAt"))
       continue;
+    const column = shape[key];
+    const name = column.data.name;
+    if (name === key)
+      column.data.name = void 0;
     code.push(
       ...combineCodeElements([
         `${quoteObjectKey(key)}: `,
@@ -466,6 +471,8 @@ const columnsShapeToCode = (shape, t) => {
         ","
       ])
     );
+    if (name === key)
+      column.data.name = name;
   }
   if (hasTimestamps) {
     code.push(`...${t}.timestamps(),`);
@@ -1515,26 +1522,33 @@ class TsVectorColumn extends ColumnType {
    * @param args
    */
   generated(...args) {
-    const first = args[0];
-    if (typeof first === "string" || !("raw" in first)) {
-      const target = typeof first === "string" ? args[1] : first;
-      const language = typeof first === "string" ? first : this.defaultLanguage;
-      let sql;
-      if (Array.isArray(target)) {
-        const columns = target.length === 1 ? `"${target[0]}"` : target.map((column) => `coalesce("${column}", '')`).join(` || ' ' || `);
-        sql = `to_tsvector('${language}', ${columns})`;
-      } else {
-        for (const key in target) {
-          sql = (sql ? sql + " || " : "(") + `setweight(to_tsvector('${language}', coalesce("${key}", '')), '${target[key]}')`;
+    return setColumnData(this, "generated", (ctx, quotedAs) => {
+      const first = args[0];
+      if (typeof first === "string" || !("raw" in first)) {
+        const target = typeof first === "string" ? args[1] : first;
+        const language = typeof first === "string" ? first : this.defaultLanguage;
+        const { snakeCase } = ctx;
+        let sql;
+        if (Array.isArray(target)) {
+          const columns = target.length === 1 ? `"${snakeCase ? toSnakeCase(target[0]) : target[0]}"` : target.map(
+            (column) => `coalesce("${snakeCase ? toSnakeCase(column) : column}", '')`
+          ).join(` || ' ' || `);
+          sql = `to_tsvector('${language}', ${columns})`;
+        } else {
+          for (const key in target) {
+            sql = (sql ? sql + " || " : "(") + `setweight(to_tsvector('${language}', coalesce("${snakeCase ? toSnakeCase(key) : key}", '')), '${target[key]}')`;
+          }
+          if (sql) {
+            sql += ")";
+          } else {
+            throw new Error("Empty target in the text search generated column");
+          }
         }
-        if (sql)
-          sql += ")";
+        return sql;
+      } else {
+        return raw(...args).toSQL(ctx, quotedAs);
       }
-      const arr = [sql];
-      arr.raw = arr;
-      args = [arr];
-    }
-    return super.generated(...args);
+    });
   }
 }
 class TsQueryColumn extends ColumnType {