pqb 0.35.7 → 0.36.1

package/dist/index.d.ts

@@ -5095,9 +5095,9 @@ type SelectResultColumnsAndObj<T extends SelectSelf, Columns extends PropertyKey
         selectable: SelectAsSelectable<Obj>;
     } : K extends 'result' ? // Combine previously selected items, all columns if * was provided,
     {
-        [K in ('*' extends Columns[number] ? Exclude<Columns[number], '*'> | keyof T['shape'] : Columns[number]) | keyof Obj as K extends Columns[number] ? T['meta']['selectable'][K]['as'] : K]: K extends keyof Obj ? SelectAsValueResult<T, Obj[K]> : K extends Columns[number] ? T['meta']['selectable'][K]['column'] : never;
+        [K in ('*' extends Columns[number] ? Exclude<Columns[number], '*'> | keyof T['shape'] : Columns[number]) | keyof Obj as K extends Columns[number] ? T['meta']['selectable'][K]['as'] : K]: K extends keyof Obj ? SelectAsValueResult<T, Obj[K]> : T['meta']['selectable'][K]['column'];
     } & (T['meta']['hasSelect'] extends true ? Omit<T['result'], Columns[number]> : unknown) : K extends 'then' ? QueryThen<GetQueryResult<T, {
-        [K in ('*' extends Columns[number] ? Exclude<Columns[number], '*'> | keyof T['shape'] : Columns[number]) | keyof Obj as K extends Columns[number] ? T['meta']['selectable'][K]['as'] : K]: K extends keyof Obj ? SelectAsValueResult<T, Obj[K]> : K extends Columns[number] ? T['meta']['selectable'][K]['column'] : never;
+        [K in ('*' extends Columns[number] ? Exclude<Columns[number], '*'> | keyof T['shape'] : Columns[number]) | keyof Obj as K extends Columns[number] ? T['meta']['selectable'][K]['as'] : K]: K extends keyof Obj ? SelectAsValueResult<T, Obj[K]> : T['meta']['selectable'][K]['column'];
     } & (T['meta']['hasSelect'] extends true ? Omit<T['result'], Columns[number]> : unknown)>> : T[K];
 };
 type SelectAsSelectable<Arg> = {
@@ -7362,7 +7362,10 @@ interface ColumnData extends ColumnDataBase {
     compression?: string;
     foreignKeys?: TableData.ColumnReferences[];
     identity?: TableData.Identity;
-    generated?: RawSQLBase;
+    generated?(ctx: {
+        values: unknown[];
+        snakeCase: boolean | undefined;
+    }, quotedAs?: string): string;
 }
 interface ColumnFromDbParams {
     isNullable?: boolean;

package/dist/index.js

@@ -380,7 +380,8 @@ class ColumnType extends orchidCore.ColumnTypeBase {
    * @param args - raw SQL
    */
   generated(...args) {
-    return orchidCore.setColumnData(this, "generated", raw(...args));
+    const sql = raw(...args);
+    return orchidCore.setColumnData(this, "generated", (...args2) => sql.toSQL(...args2));
   }
 }
 
@@ -461,6 +462,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([
         `${orchidCore.quoteObjectKey(key)}: `,
@@ -468,6 +473,8 @@ const columnsShapeToCode = (shape, t) => {
         ","
       ])
     );
+    if (name === key)
+      column.data.name = name;
   }
   if (hasTimestamps) {
     code.push(`...${t}.timestamps(),`);
@@ -1517,26 +1524,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 orchidCore.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 ? orchidCore.toSnakeCase(target[0]) : target[0]}"` : target.map(
+            (column) => `coalesce("${snakeCase ? orchidCore.toSnakeCase(column) : column}", '')`
+          ).join(` || ' ' || `);
+          sql = `to_tsvector('${language}', ${columns})`;
+        } else {
+          for (const key in target) {
+            sql = (sql ? sql + " || " : "(") + `setweight(to_tsvector('${language}', coalesce("${snakeCase ? orchidCore.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 {