drizzle-orm 0.36.4-166fb8d → 0.36.4-7f37974

package/mysql-core/dialect.cjs

@@ -1,7 +1,9 @@
 "use strict";
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -15,12 +17,21 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 var dialect_exports = {};
 __export(dialect_exports, {
   MySqlDialect: () => MySqlDialect
 });
 module.exports = __toCommonJS(dialect_exports);
+var V1 = __toESM(require("../_relations.cjs"), 1);
 var import_alias = require("../alias.cjs");
 var import_casing = require("../casing.cjs");
 var import_column = require("../column.cjs");
@@ -386,7 +397,8 @@ class MySqlDialect {
       invokeSource
     });
   }
-  buildRelationalQuery({
+  /** @deprecated */
+  _buildRelationalQuery({
     fullSchema,
     schema,
     tableNamesMap,
@@ -415,7 +427,7 @@ class MySqlDialect {
         Object.entries(tableConfig.columns).map(([key, value]) => [key, (0, import_alias.aliasedTableColumn)(value, tableAlias)])
       );
       if (config.where) {
-        const whereSql = typeof config.where === "function" ? config.where(aliasedColumns, (0, import_relations.getOperators)()) : config.where;
+        const whereSql = typeof config.where === "function" ? config.where(aliasedColumns, V1.getOperators()) : config.where;
         where = whereSql && (0, import_alias.mapColumnsInSQLToAlias)(whereSql, tableAlias);
       }
       const fieldsSelection = [];
@@ -467,7 +479,7 @@ class MySqlDialect {
           selection: []
         });
       }
-      let orderByOrig = typeof config.orderBy === "function" ? config.orderBy(aliasedColumns, (0, import_relations.getOrderByOperators)()) : config.orderBy ?? [];
+      let orderByOrig = typeof config.orderBy === "function" ? config.orderBy(aliasedColumns, V1.getOrderByOperators()) : config.orderBy ?? [];
       if (!Array.isArray(orderByOrig)) {
         orderByOrig = [orderByOrig];
       }
@@ -484,7 +496,7 @@ class MySqlDialect {
         queryConfig: selectedRelationConfigValue,
         relation
       } of selectedRelations) {
-        const normalizedRelation = (0, import_relations.normalizeRelation)(schema, tableNamesMap, relation);
+        const normalizedRelation = V1.normalizeRelation(schema, tableNamesMap, relation);
         const relationTableName = (0, import_table.getTableUniqueName)(relation.referencedTable);
         const relationTableTsName = tableNamesMap[relationTableName];
         const relationTableAlias = `${tableAlias}_${selectedRelationTsKey}`;
@@ -496,13 +508,13 @@ class MySqlDialect {
             )
           )
         );
-        const builtRelation = this.buildRelationalQuery({
+        const builtRelation = this._buildRelationalQuery({
           fullSchema,
           schema,
           tableNamesMap,
           table: fullSchema[relationTableTsName],
           tableConfig: schema[relationTableTsName],
-          queryConfig: (0, import_entity.is)(relation, import_relations.One) ? selectedRelationConfigValue === true ? { limit: 1 } : { ...selectedRelationConfigValue, limit: 1 } : selectedRelationConfigValue,
+          queryConfig: (0, import_entity.is)(relation, V1.One) ? selectedRelationConfigValue === true ? { limit: 1 } : { ...selectedRelationConfigValue, limit: 1 } : selectedRelationConfigValue,
           tableAlias: relationTableAlias,
           joinOn: joinOn2,
           nestedQueryRelation: relation
@@ -537,7 +549,7 @@ class MySqlDialect {
         ),
         import_sql.sql`, `
       )})`;
-      if ((0, import_entity.is)(nestedQueryRelation, import_relations.Many)) {
+      if ((0, import_entity.is)(nestedQueryRelation, V1.Many)) {
         field = import_sql.sql`coalesce(json_arrayagg(${field}), json_array())`;
       }
       const nestedSelection = [{
@@ -611,7 +623,8 @@ class MySqlDialect {
       selection
     };
   }
-  buildRelationalQueryWithoutLateralSubqueries({
+  /** @deprecated */
+  _buildRelationalQueryWithoutLateralSubqueries({
     fullSchema,
     schema,
     tableNamesMap,
@@ -639,7 +652,7 @@ class MySqlDialect {
         Object.entries(tableConfig.columns).map(([key, value]) => [key, (0, import_alias.aliasedTableColumn)(value, tableAlias)])
       );
       if (config.where) {
-        const whereSql = typeof config.where === "function" ? config.where(aliasedColumns, (0, import_relations.getOperators)()) : config.where;
+        const whereSql = typeof config.where === "function" ? config.where(aliasedColumns, V1.getOperators()) : config.where;
         where = whereSql && (0, import_alias.mapColumnsInSQLToAlias)(whereSql, tableAlias);
       }
       const fieldsSelection = [];
@@ -691,7 +704,7 @@ class MySqlDialect {
           selection: []
         });
       }
-      let orderByOrig = typeof config.orderBy === "function" ? config.orderBy(aliasedColumns, (0, import_relations.getOrderByOperators)()) : config.orderBy ?? [];
+      let orderByOrig = typeof config.orderBy === "function" ? config.orderBy(aliasedColumns, V1.getOrderByOperators()) : config.orderBy ?? [];
       if (!Array.isArray(orderByOrig)) {
         orderByOrig = [orderByOrig];
       }
@@ -708,7 +721,7 @@ class MySqlDialect {
         queryConfig: selectedRelationConfigValue,
         relation
       } of selectedRelations) {
-        const normalizedRelation = (0, import_relations.normalizeRelation)(schema, tableNamesMap, relation);
+        const normalizedRelation = V1.normalizeRelation(schema, tableNamesMap, relation);
         const relationTableName = (0, import_table.getTableUniqueName)(relation.referencedTable);
         const relationTableTsName = tableNamesMap[relationTableName];
         const relationTableAlias = `${tableAlias}_${selectedRelationTsKey}`;
@@ -720,19 +733,19 @@ class MySqlDialect {
             )
           )
         );
-        const builtRelation = this.buildRelationalQueryWithoutLateralSubqueries({
+        const builtRelation = this._buildRelationalQueryWithoutLateralSubqueries({
           fullSchema,
           schema,
           tableNamesMap,
           table: fullSchema[relationTableTsName],
           tableConfig: schema[relationTableTsName],
-          queryConfig: (0, import_entity.is)(relation, import_relations.One) ? selectedRelationConfigValue === true ? { limit: 1 } : { ...selectedRelationConfigValue, limit: 1 } : selectedRelationConfigValue,
+          queryConfig: (0, import_entity.is)(relation, V1.One) ? selectedRelationConfigValue === true ? { limit: 1 } : { ...selectedRelationConfigValue, limit: 1 } : selectedRelationConfigValue,
           tableAlias: relationTableAlias,
           joinOn: joinOn2,
           nestedQueryRelation: relation
         });
         let fieldSql = import_sql.sql`(${builtRelation.sql})`;
-        if ((0, import_entity.is)(relation, import_relations.Many)) {
+        if ((0, import_entity.is)(relation, V1.Many)) {
           fieldSql = import_sql.sql`coalesce(${fieldSql}, json_array())`;
         }
         const field = fieldSql.as(selectedRelationTsKey);
@@ -760,7 +773,7 @@ class MySqlDialect {
         ),
         import_sql.sql`, `
       )})`;
-      if ((0, import_entity.is)(nestedQueryRelation, import_relations.Many)) {
+      if ((0, import_entity.is)(nestedQueryRelation, V1.Many)) {
         field = import_sql.sql`json_arrayagg(${field})`;
       }
       const nestedSelection = [{
@@ -832,6 +845,187 @@ class MySqlDialect {
       selection
     };
   }
+  unwrapAllColumns = (table, selection) => {
+    return import_sql.sql.join(
+      Object.entries(table[import_table.Columns]).map(([k, v]) => {
+        selection.push({
+          key: k,
+          field: v
+        });
+        return import_sql.sql`${v} as ${import_sql.sql.identifier(k)}`;
+      }),
+      import_sql.sql`, `
+    );
+  };
+  getSelectedTableColumns = (table, columns) => {
+    const selectedColumns = [];
+    if (!columns) {
+      for (const [k, v] of Object.entries(table[import_table.Columns])) {
+        selectedColumns.push({
+          column: v,
+          tsName: k
+        });
+      }
+      return selectedColumns;
+    }
+    const entries = Object.entries(columns);
+    let colSelectionMode;
+    for (const [k, v] of entries) {
+      if (colSelectionMode === void 0)
+        colSelectionMode = v;
+      else if (v !== void 0 && colSelectionMode !== v) {
+        throw new Error("Columns cannot be both true and false at the same time");
+      }
+      if (v) {
+        selectedColumns.push({
+          column: table[import_table.Columns][k],
+          tsName: k
+        });
+      }
+    }
+    if (colSelectionMode === false) {
+      for (const [k, v] of Object.entries(table[import_table.Columns])) {
+        if (columns[k] === false)
+          continue;
+        selectedColumns.push({
+          column: v,
+          tsName: k
+        });
+      }
+    }
+    return selectedColumns;
+  };
+  buildWithColumns = (tableConfig, withContainer, jsonColumns, columnIdentifiers) => {
+    for (const [k, v] of Object.entries(
+      withContainer
+    )) {
+      if (!v)
+        continue;
+      const relation = tableConfig.relations[k];
+      if (relation.$brand === "AggregatedField")
+        continue;
+      const table = relation.targetTable;
+      const columns = this.getSelectedTableColumns(table, typeof v === "boolean" ? void 0 : v.columns).map(
+        (c) => import_sql.sql`${import_sql.sql.raw(this.escapeString(c.tsName))}, ${import_sql.sql.identifier(c.tsName)}`
+      );
+      if (typeof v === "object" && v.extras) {
+        if (typeof v.extras === "function") {
+          v.extras = v.extras(table[import_table.Columns], { sql: import_sql.sql });
+        }
+        for (const [ek, ev] of Object.entries(v.extras)) {
+          if (!ev)
+            continue;
+          columns.push(
+            import_sql.sql`${import_sql.sql.raw(this.escapeString(ek))}, ${import_sql.sql.identifier(ek)}`
+          );
+        }
+      }
+      columnIdentifiers.push(
+        import_sql.sql`${table}.${import_sql.sql.identifier("r")} as ${import_sql.sql.identifier(k)}`
+      );
+      jsonColumns[k] = import_sql.sql.join(
+        columns,
+        import_sql.sql`, `
+      );
+    }
+  };
+  buildColumns = (table, tableConfig, selection, params) => params?.columns ? (() => {
+    const jsonColumns = {};
+    const columnIdentifiers = [];
+    const selectedColumns = this.getSelectedTableColumns(table, params?.columns);
+    for (const column of selectedColumns) {
+      columnIdentifiers.push(
+        import_sql.sql`${table[column.tsName]} as ${import_sql.sql.identifier(column.tsName)}`
+      );
+    }
+    if (params?.with) {
+      this.buildWithColumns(tableConfig, params.with, jsonColumns, columnIdentifiers);
+    }
+    return {
+      columns: columnIdentifiers.length ? import_sql.sql.join(columnIdentifiers, import_sql.sql`, `) : this.unwrapAllColumns(table, selection),
+      jsonColumns
+    };
+  })() : (() => {
+    const columnIdentifiers = [this.unwrapAllColumns(table, selection)];
+    const jsonColumns = {};
+    if (params?.with) {
+      this.buildWithColumns(tableConfig, params.with, jsonColumns, columnIdentifiers);
+    }
+    return {
+      columns: import_sql.sql.join(columnIdentifiers, import_sql.sql`, `),
+      jsonColumns
+    };
+  })();
+  buildRelationalQuery({
+    tables,
+    schema,
+    tableNamesMap,
+    table,
+    tableConfig,
+    queryConfig: config,
+    relationWhere,
+    mode
+  }) {
+    const selection = [];
+    const isSingle = mode === "first";
+    const params = config === true ? void 0 : config;
+    const limit = isSingle ? 1 : params?.limit;
+    const offset = params?.offset;
+    const { columns, jsonColumns } = this.buildColumns(table, tableConfig, selection, params);
+    const where = params?.where && relationWhere ? (0, import_expressions.and)((0, import_relations.relationFilterToSQL)(table, params.where), relationWhere) : params?.where ? (0, import_relations.relationFilterToSQL)(table, params.where) : relationWhere;
+    const order = params?.orderBy ? (0, import_relations.relationsOrderToSQL)(table, params.orderBy) : void 0;
+    const extras = params?.extras ? (0, import_relations.relationExtrasToSQL)(table, params.extras) : void 0;
+    if (extras)
+      selection.push(...extras.selection);
+    const selectionSet = import_sql.sql.join([columns, extras].filter((e) => !!e), import_sql.sql`, `);
+    const joins = params ? (() => {
+      const { with: joins2 } = params;
+      if (!joins2)
+        return;
+      const withEntries = Object.entries(joins2).filter(([_, v]) => v);
+      if (!withEntries.length)
+        return;
+      return import_sql.sql.join(
+        withEntries.map(([k, join]) => {
+          const relation = tableConfig.relations[k];
+          if (relation.$brand === "AggregatedField") {
+            relation.onTable(table);
+            const query2 = relation.getSQL();
+            selection.push({
+              key: k,
+              field: relation
+            });
+            return import_sql.sql`, lateral(${query2}) as ${import_sql.sql.identifier(k)}`;
+          }
+          const isSingle2 = (0, import_entity.is)(relation, import_relations.One);
+          const targetTable = relation.targetTable;
+          const relationFilter = (0, import_relations.relationToSQL)(relation);
+          const innerQuery = this.buildRelationalQuery({
+            table: targetTable,
+            mode: isSingle2 ? "first" : "many",
+            schema,
+            queryConfig: join,
+            tableConfig: schema[tableNamesMap[(0, import_table.getTableUniqueName)(targetTable)]],
+            tableNamesMap,
+            tables,
+            relationWhere: relationFilter
+          });
+          selection.push({
+            field: targetTable,
+            key: k,
+            selection: innerQuery.selection,
+            isArray: !isSingle2
+          });
+          return import_sql.sql`, lateral(select ${isSingle2 ? import_sql.sql`json_object(${jsonColumns[k]}) as ${import_sql.sql.identifier("r")}` : import_sql.sql`coalesce(json_arrayagg(json_object(${jsonColumns[k]})), json_array()) as ${import_sql.sql.identifier("r")}`} from (${innerQuery}) as ${import_sql.sql.identifier("t")}) as ${schema[k]}`;
+        })
+      );
+    })() : void 0;
+    const query = import_sql.sql`select ${selectionSet} from ${table}${import_sql.sql`${joins}`.if(joins)}${import_sql.sql` where ${where}`.if(where)}${import_sql.sql` order by ${order}`.if(order)}${import_sql.sql` limit ${limit}`.if(limit !== void 0)}${import_sql.sql` offset ${offset}`.if(offset !== void 0)}`;
+    return {
+      sql: query,
+      selection
+    };
+  }
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {