drizzle-orm 0.36.4-14e59f4 → 0.36.4-3c69a1f

package/relations.js

@@ -1,4 +1,5 @@
 import { getTableUniqueName, Table } from "./table.js";
+import { Columns, getTableName } from "./table.js";
 import { Column } from "./column.js";
 import { entityKind, is } from "./entity.js";
 import { PrimaryKeyBuilder } from "./pg-core/primary-keys.js";
@@ -28,289 +29,505 @@ import {
   or
 } from "./sql/expressions/index.js";
 import { SQL, sql } from "./sql/sql.js";
+import { getTableColumns } from "./utils.js";
+class Relations {
+  constructor(schema, tables, config) {
+    this.schema = schema;
+    this.tables = tables;
+    this.config = config;
+    for (const [tsName, table] of Object.entries(tables)) {
+      this.tableNamesMap[getTableUniqueName(table)] = tsName;
+      const tableConfig = this.tablesConfig[tsName] = {
+        table,
+        tsName,
+        dbName: table[Table.Symbol.Name],
+        schema: table[Table.Symbol.Schema],
+        columns: table[Table.Symbol.Columns],
+        relations: config[tsName] || {},
+        primaryKey: []
+      };
+      for (const column of Object.values(table[Table.Symbol.Columns])) {
+        if (column.primary) {
+          tableConfig.primaryKey.push(column);
+        }
+      }
+      const extraConfig = table[Table.Symbol.ExtraConfigBuilder]?.(table);
+      if (extraConfig) {
+        for (const configEntry of Object.values(extraConfig)) {
+          if (is(configEntry, PrimaryKeyBuilder)) {
+            tableConfig.primaryKey.push(...configEntry.columns);
+          }
+        }
+      }
+    }
+    for (const tableConfig of Object.values(this.tablesConfig)) {
+      for (const [relationFieldName, relation] of Object.entries(tableConfig.relations)) {
+        if (!is(relation, Relation)) {
+          continue;
+        }
+        relation.sourceTable = tableConfig.table;
+        relation.fieldName = relationFieldName;
+      }
+    }
+    for (const tableConfig of Object.values(this.tablesConfig)) {
+      for (const [relationFieldName, relation] of Object.entries(tableConfig.relations)) {
+        const relationPrintName = `relations -> ${tableConfig.tsName}.${relationFieldName}`;
+        if (!is(relation, Relation)) {
+          continue;
+        }
+        if (typeof relation.alias === "string" && !relation.alias) {
+          throw new Error(`${relationPrintName}: "alias" cannot be an empty string - omit it if you don't need it`);
+        }
+        if (relation.sourceColumns?.length === 0) {
+          throw new Error(`${relationPrintName}: "from" cannot be an empty array`);
+        }
+        if (relation.targetColumns?.length === 0) {
+          throw new Error(`${relationPrintName}: "to" cannot be an empty array`);
+        }
+        if (relation.sourceColumns && relation.targetColumns) {
+          if (relation.sourceColumns.length !== relation.targetColumns.length) {
+            throw new Error(
+              `${relationPrintName}: "from" and "to" arrays must have the same length`
+            );
+          }
+          continue;
+        }
+        if (relation.sourceColumns || relation.targetColumns) {
+          throw new Error(
+            `${relationPrintName}: relation must have either both "from" and "to" defined, or none of them`
+          );
+        }
+        let reverseRelation;
+        const targetTableTsName = this.tableNamesMap[getTableUniqueName(relation.targetTable)];
+        if (!targetTableTsName) {
+          throw new Error(
+            `Table "${getTableUniqueName(relation.targetTable)}" not found in provided TS schema`
+          );
+        }
+        const reverseTableConfig = this.tablesConfig[targetTableTsName];
+        if (!reverseTableConfig) {
+          throw new Error(
+            `${relationPrintName}: not enough data provided to build the relation - "from"/"to" are not defined, and no reverse relations of table "${targetTableTsName}" were found"`
+          );
+        }
+        if (relation.alias) {
+          const reverseRelations = Object.values(reverseTableConfig.relations).filter(
+            (it) => is(it, Relation) && it.alias === relation.alias
+          );
+          if (reverseRelations.length > 1) {
+            throw new Error(
+              `${relationPrintName}: not enough data provided to build the relation - "from"/"to" are not defined, and multiple relations with alias "${relation.alias}" found in table "${targetTableTsName}": ${reverseRelations.map((it) => `"${it.fieldName}"`).join(", ")}`
+            );
+          }
+          reverseRelation = reverseRelations[0];
+          if (!reverseRelation) {
+            throw new Error(
+              `${relationPrintName}: not enough data provided to build the relation - "from"/"to" are not defined, and there is no reverse relation of table "${targetTableTsName}" with alias "${relation.alias}"`
+            );
+          }
+        } else {
+          const reverseRelations = Object.values(reverseTableConfig.relations).filter(
+            (it) => is(it, Relation) && it.targetTable === relation.sourceTable && !it.alias
+          );
+          if (reverseRelations.length > 1) {
+            throw new Error(
+              `${relationPrintName}: not enough data provided to build the relation - "from"/"to" are not defined, and multiple relations between "${targetTableTsName}" and "${getTableUniqueName(relation.sourceTable)}" were found.
+Hint: you can specify "alias" on both sides of the relation with the same value`
+            );
+          }
+          reverseRelation = reverseRelations[0];
+          if (!reverseRelation) {
+            throw new Error(
+              `${relationPrintName}: not enough data provided to build the relation - "from"/"to" are not defined, and no reverse relation of table "${targetTableTsName}" with target table "${getTableUniqueName(relation.sourceTable)}" was found`
+            );
+          }
+        }
+        if (!reverseRelation.sourceColumns || !reverseRelation.targetColumns) {
+          throw new Error(
+            `${relationPrintName}: not enough data provided to build the relation - "from"/"to" are not defined, and reverse relation "${targetTableTsName}.${reverseRelation.fieldName}" does not have "from"/"to" defined`
+          );
+        }
+        relation.sourceColumns = reverseRelation.targetColumns;
+        relation.targetColumns = reverseRelation.sourceColumns;
+        relation.where = reverseRelation.where;
+      }
+    }
+  }
+  static [entityKind] = "RelationsV2";
+  /** table DB name -> schema table key */
+  tableNamesMap = {};
+  tablesConfig = {};
+}
 class Relation {
-  constructor(sourceTable, referencedTable, relationName) {
-    this.sourceTable = sourceTable;
-    this.referencedTable = referencedTable;
-    this.relationName = relationName;
-    this.referencedTableName = referencedTable[Table.Symbol.Name];
+  constructor(targetTable) {
+    this.targetTable = targetTable;
   }
   static [entityKind] = "Relation";
-  referencedTableName;
   fieldName;
-}
-class Relations {
-  constructor(table, config) {
-    this.table = table;
-    this.config = config;
-  }
-  static [entityKind] = "Relations";
+  sourceColumns;
+  targetColumns;
+  alias;
+  where;
+  sourceTable;
 }
 class One extends Relation {
-  constructor(sourceTable, referencedTable, config, isNullable) {
-    super(sourceTable, referencedTable, config?.relationName);
-    this.config = config;
-    this.isNullable = isNullable;
-  }
   static [entityKind] = "One";
-  withFieldName(fieldName) {
-    const relation = new One(
-      this.sourceTable,
-      this.referencedTable,
-      this.config,
-      this.isNullable
-    );
-    relation.fieldName = fieldName;
-    return relation;
+  optional;
+  constructor(targetTable, config) {
+    super(targetTable);
+    this.alias = config?.alias;
+    this.where = config?.where;
+    if (config?.from) {
+      this.sourceColumns = Array.isArray(config.from) ? config.from.map((it) => it._.column) : [config.from._.column];
+    }
+    if (config?.to) {
+      this.targetColumns = Array.isArray(config.to) ? config.to.map((it) => it._.column) : [config.to._.column];
+    }
+    this.optional = config?.optional ?? false;
   }
 }
 class Many extends Relation {
-  constructor(sourceTable, referencedTable, config) {
-    super(sourceTable, referencedTable, config?.relationName);
+  constructor(targetTable, config) {
+    super(targetTable);
     this.config = config;
+    this.alias = config?.alias;
+    this.where = config?.where;
+    if (config?.from) {
+      this.sourceColumns = Array.isArray(config.from) ? config.from.map((it) => it._.column) : [config.from._.column];
+    }
+    if (config?.to) {
+      this.targetColumns = Array.isArray(config.to) ? config.to.map((it) => it._.column) : [config.to._.column];
+    }
   }
   static [entityKind] = "Many";
-  withFieldName(fieldName) {
-    const relation = new Many(
-      this.sourceTable,
-      this.referencedTable,
-      this.config
-    );
-    relation.fieldName = fieldName;
-    return relation;
+}
+class AggregatedField {
+  static [entityKind] = "AggregatedField";
+  table;
+  onTable(table) {
+    this.table = table;
+    return this;
   }
 }
-function getOperators() {
-  return {
-    and,
-    between,
-    eq,
-    exists,
-    gt,
-    gte,
-    ilike,
-    inArray,
-    isNull,
-    isNotNull,
-    like,
-    lt,
-    lte,
-    ne,
-    not,
-    notBetween,
-    notExists,
-    notLike,
-    notIlike,
-    notInArray,
-    or,
-    sql
-  };
+class Count extends AggregatedField {
+  static [entityKind] = "AggregatedFieldCount";
+  query;
+  getSQL() {
+    if (!this.query) {
+      if (!this.table)
+        throw new Error("Table must be set before building aggregate field");
+      this.query = sql`select count(*) as ${sql.identifier("r")} from ${this.table}`.mapWith(Number);
+    }
+    return this.query;
+  }
 }
+const operators = {
+  and,
+  between,
+  eq,
+  exists,
+  gt,
+  gte,
+  ilike,
+  inArray,
+  isNull,
+  isNotNull,
+  like,
+  lt,
+  lte,
+  ne,
+  not,
+  notBetween,
+  notExists,
+  notLike,
+  notIlike,
+  notInArray,
+  or,
+  sql
+};
+const orderByOperators = {
+  sql,
+  asc,
+  desc
+};
 function getOrderByOperators() {
-  return {
-    sql,
-    asc,
-    desc
-  };
+  return orderByOperators;
 }
-function extractTablesRelationalConfig(schema, configHelpers) {
-  if (Object.keys(schema).length === 1 && "default" in schema && !is(schema["default"], Table)) {
-    schema = schema["default"];
-  }
-  const tableNamesMap = {};
-  const relationsBuffer = {};
-  const tablesConfig = {};
-  for (const [key, value] of Object.entries(schema)) {
-    if (is(value, Table)) {
-      const dbName = getTableUniqueName(value);
-      const bufferedRelations = relationsBuffer[dbName];
-      tableNamesMap[dbName] = key;
-      tablesConfig[key] = {
-        tsName: key,
-        dbName: value[Table.Symbol.Name],
-        schema: value[Table.Symbol.Schema],
-        columns: value[Table.Symbol.Columns],
-        relations: bufferedRelations?.relations ?? {},
-        primaryKey: bufferedRelations?.primaryKey ?? []
-      };
-      for (const column of Object.values(
-        value[Table.Symbol.Columns]
-      )) {
-        if (column.primary) {
-          tablesConfig[key].primaryKey.push(column);
+function mapRelationalRow(row, buildQueryResultSelection, mapColumnValue = (value) => value, parseJson = false) {
+  for (const selectionItem of buildQueryResultSelection) {
+    const field = selectionItem.field;
+    if (is(field, Table)) {
+      if (row[selectionItem.key] === null)
+        continue;
+      if (parseJson)
+        row[selectionItem.key] = JSON.parse(row[selectionItem.key]);
+      if (selectionItem.isArray) {
+        for (const item of row[selectionItem.key]) {
+          mapRelationalRow(item, selectionItem.selection, mapColumnValue);
         }
+        continue;
       }
-      const extraConfig = value[Table.Symbol.ExtraConfigBuilder]?.(value[Table.Symbol.ExtraConfigColumns]);
-      if (extraConfig) {
-        for (const configEntry of Object.values(extraConfig)) {
-          if (is(configEntry, PrimaryKeyBuilder)) {
-            tablesConfig[key].primaryKey.push(...configEntry.columns);
-          }
-        }
-      }
-    } else if (is(value, Relations)) {
-      const dbName = getTableUniqueName(value.table);
-      const tableName = tableNamesMap[dbName];
-      const relations2 = value.config(
-        configHelpers(value.table)
+      mapRelationalRow(
+        row[selectionItem.key],
+        selectionItem.selection,
+        mapColumnValue
       );
-      let primaryKey;
-      for (const [relationName, relation] of Object.entries(relations2)) {
-        if (tableName) {
-          const tableConfig = tablesConfig[tableName];
-          tableConfig.relations[relationName] = relation;
-          if (primaryKey) {
-            tableConfig.primaryKey.push(...primaryKey);
-          }
-        } else {
-          if (!(dbName in relationsBuffer)) {
-            relationsBuffer[dbName] = {
-              relations: {},
-              primaryKey
-            };
-          }
-          relationsBuffer[dbName].relations[relationName] = relation;
-        }
-      }
+      continue;
     }
+    const value = mapColumnValue(row[selectionItem.key]);
+    if (value === null)
+      continue;
+    let decoder;
+    if (is(field, Column)) {
+      decoder = field;
+    } else if (is(field, SQL)) {
+      decoder = field.decoder;
+    } else if (is(field, SQL.Aliased)) {
+      decoder = field.sql.decoder;
+    } else {
+      decoder = field.getSQL().decoder;
+    }
+    row[selectionItem.key] = decoder.mapFromDriverValue(value);
   }
-  return { tables: tablesConfig, tableNamesMap };
-}
-function relations(table, relations2) {
-  return new Relations(
-    table,
-    (helpers) => Object.fromEntries(
-      Object.entries(relations2(helpers)).map(([key, value]) => [
-        key,
-        value.withFieldName(key)
-      ])
-    )
-  );
-}
-function createOne(sourceTable) {
-  return function one(table, config) {
-    return new One(
-      sourceTable,
-      table,
-      config,
-      config?.fields.reduce((res, f) => res && f.notNull, true) ?? false
-    );
-  };
-}
-function createMany(sourceTable) {
-  return function many(referencedTable, config) {
-    return new Many(sourceTable, referencedTable, config);
-  };
+  return row;
 }
-function normalizeRelation(schema, tableNamesMap, relation) {
-  if (is(relation, One) && relation.config) {
-    return {
-      fields: relation.config.fields,
-      references: relation.config.references
+class RelationsBuilderTable {
+  static [entityKind] = "RelationsBuilderTable";
+  _;
+  constructor(table) {
+    this._ = {
+      name: getTableName(table),
+      table
     };
   }
-  const referencedTableTsName = tableNamesMap[getTableUniqueName(relation.referencedTable)];
-  if (!referencedTableTsName) {
-    throw new Error(
-      `Table "${relation.referencedTable[Table.Symbol.Name]}" not found in schema`
-    );
+  getSQL() {
+    return this._.table.getSQL();
   }
-  const referencedTableConfig = schema[referencedTableTsName];
-  if (!referencedTableConfig) {
-    throw new Error(`Table "${referencedTableTsName}" not found in schema`);
+}
+class RelationsBuilderColumn {
+  static [entityKind] = "RelationsBuilderColumn";
+  _;
+  constructor(column) {
+    this._ = {
+      tableName: getTableName(column.table),
+      data: void 0,
+      column
+    };
   }
-  const sourceTable = relation.sourceTable;
-  const sourceTableTsName = tableNamesMap[getTableUniqueName(sourceTable)];
-  if (!sourceTableTsName) {
-    throw new Error(
-      `Table "${sourceTable[Table.Symbol.Name]}" not found in schema`
-    );
+  through(column) {
+    this._.through = column;
+    return this;
   }
-  const reverseRelations = [];
-  for (const referencedTableRelation of Object.values(
-    referencedTableConfig.relations
-  )) {
-    if (relation.relationName && relation !== referencedTableRelation && referencedTableRelation.relationName === relation.relationName || !relation.relationName && referencedTableRelation.referencedTable === relation.sourceTable) {
-      reverseRelations.push(referencedTableRelation);
-    }
+  getSQL() {
+    return this._.column.getSQL();
   }
-  if (reverseRelations.length > 1) {
-    throw relation.relationName ? new Error(
-      `There are multiple relations with name "${relation.relationName}" in table "${referencedTableTsName}"`
-    ) : new Error(
-      `There are multiple relations between "${referencedTableTsName}" and "${relation.sourceTable[Table.Symbol.Name]}". Please specify relation name`
-    );
-  }
-  if (reverseRelations[0] && is(reverseRelations[0], One) && reverseRelations[0].config) {
-    return {
-      fields: reverseRelations[0].config.references,
-      references: reverseRelations[0].config.fields
+}
+class RelationsHelperStatic {
+  static [entityKind] = "RelationsHelperStatic";
+  _;
+  constructor(tables) {
+    this._ = {
+      tables
     };
+    const one = {};
+    const many = {};
+    for (const [tableName, table] of Object.entries(tables)) {
+      one[tableName] = (config) => {
+        return new One(table, config);
+      };
+      many[tableName] = (config) => {
+        return new Many(table, config);
+      };
+    }
+    this.one = one;
+    this.many = many;
   }
-  throw new Error(
-    `There is not enough information to infer relation "${sourceTableTsName}.${relation.fieldName}"`
-  );
-}
-function createTableRelationsHelpers(sourceTable) {
-  return {
-    one: createOne(sourceTable),
-    many: createMany(sourceTable)
+  one;
+  many;
+  aggs = {
+    count() {
+      return new Count();
+    }
   };
 }
-function mapRelationalRow(tablesConfig, tableConfig, row, buildQueryResultSelection, mapColumnValue = (value) => value) {
-  const result = {};
-  for (const [
-    selectionItemIndex,
-    selectionItem
-  ] of buildQueryResultSelection.entries()) {
-    if (selectionItem.isJson) {
-      const relation = tableConfig.relations[selectionItem.tsKey];
-      const rawSubRows = row[selectionItemIndex];
-      const subRows = typeof rawSubRows === "string" ? JSON.parse(rawSubRows) : rawSubRows;
-      result[selectionItem.tsKey] = is(relation, One) ? subRows && mapRelationalRow(
-        tablesConfig,
-        tablesConfig[selectionItem.relationTableTsKey],
-        subRows,
-        selectionItem.selection,
-        mapColumnValue
-      ) : subRows.map(
-        (subRow) => mapRelationalRow(
-          tablesConfig,
-          tablesConfig[selectionItem.relationTableTsKey],
-          subRow,
-          selectionItem.selection,
-          mapColumnValue
-        )
+function createRelationsHelper(schema) {
+  const schemaTables = Object.fromEntries(
+    Object.entries(schema).filter((e) => is(e[1], Table))
+  );
+  const helperStatic = new RelationsHelperStatic(schemaTables);
+  const tables = Object.entries(schema).reduce((acc, [key, value]) => {
+    if (is(value, Table)) {
+      const rTable = new RelationsBuilderTable(value);
+      const columns = Object.entries(getTableColumns(value)).reduce(
+        (acc2, [key2, column]) => {
+          const rbColumn = new RelationsBuilderColumn(column);
+          acc2[key2] = rbColumn;
+          return acc2;
+        },
+        {}
       );
-    } else {
-      const value = mapColumnValue(row[selectionItemIndex]);
-      const field = selectionItem.field;
-      let decoder;
-      if (is(field, Column)) {
-        decoder = field;
-      } else if (is(field, SQL)) {
-        decoder = field.decoder;
-      } else {
-        decoder = field.sql.decoder;
+      acc[key] = Object.assign(rTable, columns);
+    }
+    return acc;
+  }, {});
+  return Object.assign(helperStatic, tables);
+}
+function defineRelations(schema, relations) {
+  return new Relations(
+    schema,
+    schema,
+    relations(createRelationsHelper(schema))
+  );
+}
+function relationsFieldFilterToSQL(column, filter) {
+  if (typeof filter !== "object")
+    return eq(column, filter);
+  const entries = Object.entries(filter);
+  if (!entries.length)
+    return void 0;
+  const parts = [];
+  for (const [target, value] of entries) {
+    if (value === void 0)
+      continue;
+    switch (target) {
+      case "$not": {
+        const res = relationsFieldFilterToSQL(column, value);
+        if (!res)
+          continue;
+        parts.push(not(res));
+        continue;
+      }
+      case "$or": {
+        if (!value.length)
+          continue;
+        parts.push(
+          or(
+            ...value.map((subFilter) => relationsFieldFilterToSQL(column, subFilter))
+          )
+        );
+        continue;
+      }
+      default: {
+        parts.push(
+          operators[target](
+            column,
+            value
+          )
+        );
+        continue;
       }
-      result[selectionItem.tsKey] = value === null ? null : decoder.mapFromDriverValue(value);
     }
   }
-  return result;
+  if (!parts.length)
+    return void 0;
+  return and(...parts);
+}
+function relationFilterToSQL(table, filter) {
+  const entries = Object.entries(filter);
+  if (!entries.length)
+    return void 0;
+  const parts = [];
+  for (const [target, value] of entries) {
+    if (value === void 0)
+      continue;
+    switch (target) {
+      case "$raw": {
+        if (value)
+          parts.push(value(operators));
+        continue;
+      }
+      case "$or": {
+        if (!value?.length)
+          continue;
+        parts.push(
+          or(
+            ...value.map(
+              (subFilter) => relationFilterToSQL(table, subFilter)
+            )
+          )
+        );
+        continue;
+      }
+      case "$not": {
+        if (!value?.length)
+          continue;
+        parts.push(
+          not(
+            and(
+              ...value.map(
+                (subFilter) => relationFilterToSQL(table, subFilter)
+              )
+            )
+          )
+        );
+        continue;
+      }
+      default: {
+        const colFilter = relationsFieldFilterToSQL(
+          table[target],
+          value
+        );
+        if (colFilter)
+          parts.push(colFilter);
+        continue;
+      }
+    }
+  }
+  return and(...parts);
+}
+function relationsOrderToSQL(table, orders) {
+  const data = typeof orders === "function" ? orders(table[Columns], orderByOperators) : orders;
+  return is(data, SQL) ? data : Array.isArray(data) ? data.length ? sql.join(data.map((o) => is(o, SQL) ? o : asc(o)), sql`, `) : void 0 : asc(data);
+}
+function relationExtrasToSQL(table, extras) {
+  const subqueries = [];
+  const selection = [];
+  for (const [key, extra] of Object.entries(
+    typeof extras === "function" ? extras(table[Columns], { sql: operators.sql }) : extras
+  )) {
+    if (!extra)
+      continue;
+    const query = sql`(${extra.getSQL()}) as ${sql.identifier(key)}`;
+    query.decoder = extra.getSQL().decoder;
+    subqueries.push(query);
+    selection.push({
+      key,
+      field: query
+    });
+  }
+  return {
+    sql: subqueries.length ? sql.join(subqueries, sql`, `) : void 0,
+    selection
+  };
+}
+function relationToSQL(relation) {
+  const table = relation.sourceTable;
+  const columnWhere = relation.sourceColumns.map((s, i) => {
+    const t = relation.targetColumns[i];
+    return eq(s, t);
+  });
+  const targetWhere = relation.where ? and(...columnWhere, relationFilterToSQL(table, relation.where)) : and(...columnWhere);
+  return targetWhere;
 }
 export {
+  AggregatedField,
+  Count,
   Many,
   One,
   Relation,
   Relations,
-  createMany,
-  createOne,
-  createTableRelationsHelpers,
-  extractTablesRelationalConfig,
-  getOperators,
+  RelationsBuilderColumn,
+  RelationsBuilderTable,
+  RelationsHelperStatic,
+  createRelationsHelper,
+  defineRelations,
   getOrderByOperators,
   mapRelationalRow,
-  normalizeRelation,
-  relations
+  operators,
+  orderByOperators,
+  relationExtrasToSQL,
+  relationFilterToSQL,
+  relationToSQL,
+  relationsOrderToSQL
 };
 //# sourceMappingURL=relations.js.map