orchid-orm 1.47.1 → 1.49.0

package/dist/index.mjs

@@ -1,6 +1,6 @@
-import { makeColumnTypes, QueryHooks, defaultSchemaConfig, raw, getColumnTypes, parseTableData, _queryHookAfterCreate, _queryHookAfterUpdate, getQueryAs, setQueryObjectValueImmutable, pushQueryOnForOuter, _queryWhere, _queryDefaults, VirtualColumn, pushQueryValueImmutable, _queryCreateMany, isQueryReturnsAll, _queryHookBeforeUpdate, _queryFindBy, _queryCreate, _queryRows, _queryUpdate, _queryDelete, _queryUpdateOrThrow, OrchidOrmInternalError, _queryJoinOn, _queryCreateFrom, NotFoundError, _queryFindByOptional, _querySelect, _queryWhereExists, _queryTake, _queryTakeOptional, Adapter, _initQueryBuilder, Db, getClonedQueryData } from 'pqb';
+import { makeColumnTypes, QueryHooks, defaultSchemaConfig, raw, getColumnTypes, parseTableData, _queryHookAfterCreate, _queryHookAfterUpdate, getQueryAs, setQueryObjectValueImmutable, pushQueryOnForOuter, cloneQueryBaseUnscoped, DynamicRawSQL, RawSQL, getShapeFromSelect, _queryWhere, _queryDefaults, getPrimaryKeys, VirtualColumn, pushQueryValueImmutable, _queryCreateMany, isQueryReturnsAll, _queryHookBeforeUpdate, _queryFindBy, _queryCreate, _queryRows, _queryUpdate, _queryDelete, _queryUpdateOrThrow, OrchidOrmInternalError, _queryJoinOn, _queryCreateFrom, NotFoundError, _queryFindByOptional, _querySelect, _queryWhereExists, _queryTake, _queryTakeOptional, Adapter, _initQueryBuilder, Db, getClonedQueryData } from 'pqb';
 export * from 'pqb';
-import { getStackTrace, applyMixins, getCallerFilePath, snakeCaseKey, toSnakeCase, emptyObject, emptyArray, toArray, objectHasValues, pick } from 'orchid-core';
+import { getStackTrace, applyMixins, getCallerFilePath, snakeCaseKey, toSnakeCase, emptyObject, emptyArray, isExpression, toArray, objectHasValues, pick } from 'orchid-core';
 export * from 'orchid-core';
 import { AsyncLocalStorage } from 'node:async_hooks';
 
@@ -206,7 +206,10 @@ const selectIfNotSelected = (q, columns) => {
 };
 function joinHasThrough(q, baseQuery, joiningQuery, throughRelation, sourceRelation) {
   return q.whereExists(
-    throughRelation.joinQuery(throughRelation.query, baseQuery),
+    throughRelation.joinQuery(
+      throughRelation.query,
+      baseQuery
+    ),
     () => {
       const as = getQueryAs(joiningQuery);
       return sourceRelation.joinQuery(
@@ -231,17 +234,6 @@ function joinHasRelation(baseQuery, joiningQuery, primaryKeys, foreignKeys, len)
   }
   return q;
 }
-const joinQueryChainingHOF = (reverseJoin, joinQuery) => (joiningQuery, baseQuery) => {
-  const chain = joiningQuery.q.relChain;
-  if (!chain || chain.length === 1) {
-    return joinQuery(joiningQuery, baseQuery);
-  }
-  const last = chain[chain.length - 1];
-  const query = "relationConfig" in last ? last.relationConfig.joinQuery(last, baseQuery) : last;
-  return joiningQuery.where({
-    EXISTS: { q: reverseJoin(query, joiningQuery) }
-  });
-};
 const addAutoForeignKey = (tableConfig, from, to, primaryKeys, foreignKeys, options) => {
   var _a;
   const toTable = to.table;
@@ -298,6 +290,66 @@ const getColumnKeyFromDbName = (query, name) => {
   return name;
 };
 
+const joinQueryChainHOF = (relPKeys, reverseJoin, joinQuery) => (joiningQuery, baseQuery) => {
+  const jq = joiningQuery;
+  const chain = jq.q.relChain;
+  if (!chain || chain.length === 1) {
+    return joinQuery(jq, baseQuery);
+  }
+  const last = chain[chain.length - 1];
+  const query = "relationConfig" in last ? last.relationConfig.joinQuery(last, baseQuery) : last;
+  const result = jq.join(
+    { _internalJoin: reverseJoin(query, jq) },
+    void 0
+  );
+  if (!query.q.chainMultiple) {
+    return result;
+  }
+  const item = selectRowNumber(result, relPKeys);
+  combineOrdering(result, query);
+  return wrapQuery(jq, result, item);
+};
+const selectRowNumber = (result, relPKeys) => {
+  const hookSelect = result.q.hookSelect = new Map(
+    result.q.hookSelect && [...result.q.hookSelect]
+  );
+  const as = `"${getQueryAs(result)}"`;
+  const partitionColumns = relPKeys.map(
+    (key) => `${as}."${result.shape[key]?.data.name || key}"`
+  );
+  const item = {
+    select: {
+      sql: `row_number() OVER (PARTITION BY ${partitionColumns.join(", ")})`
+    }
+  };
+  hookSelect.set("r", item);
+  return item;
+};
+const combineOrdering = (result, joined) => {
+  const { order } = joined.q;
+  if (order) {
+    const as = getQueryAs(joined);
+    const add = order.map(
+      (o) => typeof o === "string" ? `${as}.${o}` : isExpression(o) ? o : Object.fromEntries(
+        Object.entries(o).map(([key, value]) => [`${as}.${key}`, value])
+      )
+    );
+    const arr = result.q.order;
+    result.q.order = arr ? [...add, ...arr] : add;
+  }
+};
+const wrapQuery = (joiningQuery, result, item) => {
+  const baseOuterQuery = cloneQueryBaseUnscoped(joiningQuery);
+  const outer = baseOuterQuery.clone();
+  outer.q.and = [new DynamicRawSQL(() => new RawSQL(`${item.as || "r"} = 1`))];
+  outer.q.useFromLimitOffset = true;
+  outer.shape = getShapeFromSelect(result, true);
+  outer.q.select = Object.keys(outer.shape);
+  outer.q.returnType = result.q.returnType;
+  result.q.returnsOne = true;
+  return result.wrap(outer);
+};
+
 class BelongsToVirtualColumn extends VirtualColumn {
   constructor(schema, key, state) {
     super(schema);
@@ -404,9 +456,15 @@ const makeBelongsToMethod = (tableConfig, table, relation, relationName, query)
       relationName,
       state
     ),
-    joinQuery: joinQueryChainingHOF(
+    joinQuery: joinQueryChainHOF(
+      getPrimaryKeys(query),
       reverseJoin,
-      (joiningQuery, baseQuery) => join(baseQuery, joiningQuery, primaryKeys, foreignKeys)
+      (joiningQuery, baseQuery) => join(
+        baseQuery,
+        joiningQuery,
+        primaryKeys,
+        foreignKeys
+      )
     ),
     reverseJoin
   };
@@ -643,6 +701,7 @@ class HasOneVirtualColumn extends VirtualColumn {
   }
 }
 const makeHasOneMethod = (tableConfig, table, relation, relationName, query) => {
+  const relPKeys = getPrimaryKeys(query);
   if ("through" in relation.options) {
     const { through, source } = relation.options;
     const throughRelation = getThroughRelation(table, through);
@@ -670,7 +729,8 @@ const makeHasOneMethod = (tableConfig, table, relation, relationName, query) =>
         const throughQuery = table.queryRelated(through, params);
         return query.whereExists(throughQuery, whereExistsCallback);
       },
-      joinQuery: joinQueryChainingHOF(
+      joinQuery: joinQueryChainHOF(
+        relPKeys,
         reverseJoin2,
         (joiningQuery, baseQuery) => joinHasThrough(
           joiningQuery,
@@ -728,7 +788,8 @@ const makeHasOneMethod = (tableConfig, table, relation, relationName, query) =>
       relationName,
       state
     ),
-    joinQuery: joinQueryChainingHOF(
+    joinQuery: joinQueryChainHOF(
+      relPKeys,
       reverseJoin,
       (joiningQuery, baseQuery) => joinHasRelation(
         baseQuery,
@@ -911,6 +972,7 @@ class HasManyVirtualColumn extends VirtualColumn {
   }
 }
 const makeHasManyMethod = (tableConfig, table, relation, relationName, query) => {
+  const relPKeys = getPrimaryKeys(query);
   if ("through" in relation.options) {
     const { through, source } = relation.options;
     const throughRelation = getThroughRelation(table, through);
@@ -941,7 +1003,8 @@ const makeHasManyMethod = (tableConfig, table, relation, relationName, query) =>
           whereExistsCallback
         );
       },
-      joinQuery: joinQueryChainingHOF(
+      joinQuery: joinQueryChainHOF(
+        relPKeys,
         reverseJoin2,
         (joiningQuery, baseQuery) => joinHasThrough(
           joiningQuery,
@@ -999,7 +1062,8 @@ const makeHasManyMethod = (tableConfig, table, relation, relationName, query) =>
       relationName,
       state
     ),
-    joinQuery: joinQueryChainingHOF(
+    joinQuery: joinQueryChainHOF(
+      relPKeys,
       reverseJoin,
       (joiningQuery, baseQuery) => joinHasRelation(
         baseQuery,
@@ -1388,7 +1452,8 @@ const makeHasAndBelongsToManyMethod = (tableConfig, table, qb, relation, relatio
       relationName,
       state
     ),
-    joinQuery: joinQueryChainingHOF(
+    joinQuery: joinQueryChainHOF(
+      getPrimaryKeys(query),
       reverseJoin,
       (joiningQuery, baseQuery2) => joinQuery(
         joiningQuery,