orchid-orm 1.10.6 → 1.11.0

package/dist/index.mjs

@@ -1,17 +1,8 @@
-import { addQueryHook, columnTypes, getColumnTypes, addQueryOn, VirtualColumn, pushQueryValue, isQueryReturnsAll, getQueryAs, toSqlCacheKey, NotFoundError, relationQueryKey, Adapter, Db, anyShape, getClonedQueryData } from 'pqb';
+import { columnTypes, QueryHooks, getColumnTypes, addQueryOn, VirtualColumn, pushQueryValue, isQueryReturnsAll, getQueryAs, toSqlCacheKey, OrchidOrmInternalError, NotFoundError, relationQueryKey, Adapter, Db, anyShape, getClonedQueryData } from 'pqb';
 export { OrchidOrmError, OrchidOrmInternalError, columnTypes, testTransaction } from 'pqb';
-import { getCallerFilePath, snakeCaseKey, toSnakeCase } from 'orchid-core';
+import { getCallerFilePath, applyMixins, snakeCaseKey, toSnakeCase } from 'orchid-core';
 import { AsyncLocalStorage } from 'node:async_hooks';
 
-const addTableHooks = (table, hooks) => {
-  for (const key in hooks) {
-    addQueryHook(
-      table.baseQuery,
-      key,
-      hooks[key]
-    );
-  }
-};
 const createBaseTable = ({
   columnTypes: columnTypes$1,
   snakeCase,
@@ -39,10 +30,11 @@ const create = (columnTypes, filePath, snakeCase, nowSQL, exportAs = "BaseTable"
   const base = (_a = class {
     constructor() {
       this.snakeCase = snakeCase;
+      this.query = {};
       this.columnTypes = columnTypes;
     }
-    setHooks(callbacks) {
-      return callbacks;
+    clone() {
+      return this;
     }
     setColumns(fn) {
       if (!this.filePath) {
@@ -105,6 +97,7 @@ const create = (columnTypes, filePath, snakeCase, nowSQL, exportAs = "BaseTable"
       };
     }
   }, _a.filePath = filePath, _a.nowSQL = nowSQL, _a.exportAs = exportAs, _a);
+  applyMixins(base, [QueryHooks]);
   base.prototype.columnTypes = columnTypes;
   return base;
 };
@@ -224,9 +217,9 @@ const nestedInsert$3 = ({ query, primaryKey }) => {
     let createdI = 0;
     let connectedI = 0;
     connectOrCreatedI = 0;
-    return data.map(
-      (item) => item.connectOrCreate ? connectOrCreated[connectOrCreatedI++] || created[createdI++] : item.connect ? connected[connectedI++] : created[createdI++]
-    );
+    return data.map((item) => {
+      return item.connectOrCreate ? connectOrCreated[connectOrCreatedI++] || created[createdI++] : item.connect ? connected[connectedI++] : created[createdI++];
+    });
   };
 };
 const nestedUpdate$3 = ({ query, primaryKey, foreignKey }) => {
@@ -235,7 +228,7 @@ const nestedUpdate$3 = ({ query, primaryKey, foreignKey }) => {
     if (params.upsert && isQueryReturnsAll(q)) {
       throw new Error("`upsert` option is not allowed in a batch update");
     }
-    let idForDelete;
+    let idsForDelete;
     q._beforeUpdate(async (q2) => {
       if (params.disconnect) {
         update[foreignKey] = null;
@@ -250,7 +243,7 @@ const nestedUpdate$3 = ({ query, primaryKey, foreignKey }) => {
       } else if (params.delete) {
         const selectQuery = q2.clone();
         selectQuery.query.type = void 0;
-        idForDelete = await selectQuery._getOptional(foreignKey);
+        idsForDelete = await selectQuery._pluck(foreignKey);
         update[foreignKey] = null;
       }
     });
@@ -273,12 +266,12 @@ const nestedUpdate$3 = ({ query, primaryKey, foreignKey }) => {
         }
       });
     } else if (params.delete || params.update) {
-      q._afterQuery(async (_, data) => {
-        const id = params.delete ? idForDelete : Array.isArray(data) ? data.length === 0 ? null : {
+      q._afterUpdate([], async (data) => {
+        const id = params.delete ? { in: idsForDelete } : Array.isArray(data) ? data.length === 0 ? null : {
           in: data.map((item) => item[foreignKey]).filter((id2) => id2 !== null)
         } : data[foreignKey];
         if (id !== void 0 && id !== null) {
-          const t = query.findBy({
+          const t = query.where({
             [primaryKey]: id
           });
           if (params.delete) {
@@ -312,22 +305,20 @@ const hasRelationHandleCreate = (q, ctx, item, rowIndex, key, primaryKey, nested
     return;
   }
   q.query.wrapInTransaction = true;
-  ctx.returnTypeAll = true;
-  ctx.requiredReturning[primaryKey] = true;
   const relationData = [values];
   store.hasRelation[key] = relationData;
-  pushQueryValue(q, "afterCreate", async (q2) => {
-    const { resultAll } = ctx;
-    return nestedInsert(
+  q._afterCreate(
+    [primaryKey],
+    (rows, q2) => nestedInsert(
       q2,
       relationData.map(([rowIndex2, data]) => [
-        resultAll[rowIndex2],
+        rows[rowIndex2],
         data
       ])
-    );
-  });
+    )
+  );
 };
-const hasRelationHandleUpdate = (q, ctx, set, key, primaryKey, nestedUpdate) => {
+const hasRelationHandleUpdate = (q, set, key, primaryKey, nestedUpdate) => {
   var _a, _b;
   const value = set[key];
   if (!value.set && !("upsert" in value) && (!value.disconnect || Array.isArray(value.disconnect) && value.disconnect.length === 0) && (!value.delete || Array.isArray(value.delete) && value.delete.length === 0) && (!value.update || Array.isArray(value.update.where) && value.update.where.length === 0) && (!value.create || Array.isArray(value.create) && value.create.length === 0))
@@ -336,11 +327,10 @@ const hasRelationHandleUpdate = (q, ctx, set, key, primaryKey, nestedUpdate) =>
     q._select(primaryKey);
   }
   q.query.wrapInTransaction = true;
-  ctx.returnTypeAll = true;
-  pushQueryValue(q, "afterUpdate", (q2) => {
+  q._afterUpdate(q.primaryKeys, (rows, q2) => {
     return nestedUpdate(
       q2,
-      ctx.resultAll,
+      rows,
       value
     );
   });
@@ -384,10 +374,14 @@ class HasOneVirtualColumn extends VirtualColumn {
       this.nestedInsert
     );
   }
-  update(q, ctx, set) {
+  update(q, _, set) {
+    const params = set[this.key];
+    if ((params.set || params.create || params.upsert) && isQueryReturnsAll(q)) {
+      const key = params.set ? "set" : params.create ? "create" : "upsert";
+      throw new Error(`\`${key}\` option is not allowed in a batch update`);
+    }
     hasRelationHandleUpdate(
       q,
-      ctx,
       set,
       this.key,
       this.state.primaryKey,
@@ -500,11 +494,7 @@ const nestedInsert$2 = ({ query, primaryKey, foreignKey }) => {
   };
 };
 const nestedUpdate$2 = ({ query, primaryKey, foreignKey }) => {
-  return async (q, data, params) => {
-    if ((params.set || params.create || params.upsert) && isQueryReturnsAll(q)) {
-      const key = params.set ? "set" : params.create ? "create" : "upsert";
-      throw new Error(`\`${key}\` option is not allowed in a batch update`);
-    }
+  return async (_, data, params) => {
     const t = query.clone();
     const ids = data.map((item) => item[primaryKey]);
     const currentRelationsQuery = t.where({
@@ -577,10 +567,14 @@ class HasManyVirtualColumn extends VirtualColumn {
       this.nestedInsert
     );
   }
-  update(q, ctx, set) {
+  update(q, _, set) {
+    const params = set[this.key];
+    if ((params.set || params.create) && isQueryReturnsAll(q)) {
+      const key = params.set ? "set" : "create";
+      throw new Error(`\`${key}\` option is not allowed in a batch update`);
+    }
     hasRelationHandleUpdate(
       q,
-      ctx,
       set,
       this.key,
       this.state.primaryKey,
@@ -736,12 +730,8 @@ const nestedInsert$1 = ({ query, primaryKey, foreignKey }) => {
   };
 };
 const nestedUpdate$1 = ({ query, primaryKey, foreignKey }) => {
-  return async (q, data, params) => {
+  return async (_, data, params) => {
     var _a;
-    if ((params.set || params.create) && isQueryReturnsAll(q)) {
-      const key = params.set ? "set" : "create";
-      throw new Error(`\`${key}\` option is not allowed in a batch update`);
-    }
     const t = query.clone();
     if (params.create) {
       await t._count()._createMany(
@@ -771,7 +761,7 @@ const nestedUpdate$1 = ({ query, primaryKey, foreignKey }) => {
     }
     if (params.delete || params.update) {
       delete t.query[toSqlCacheKey];
-      const q2 = t._where(
+      const q = t._where(
         getWhereForNestedUpdate(
           data,
           params.delete || ((_a = params.update) == null ? void 0 : _a.where),
@@ -780,9 +770,9 @@ const nestedUpdate$1 = ({ query, primaryKey, foreignKey }) => {
         )
       );
       if (params.delete) {
-        await q2._delete();
+        await q._delete();
       } else if (params.update) {
-        await q2._update(params.update.data);
+        await q._update(params.update.data);
       }
     }
   };
@@ -826,10 +816,9 @@ class HasAndBelongsToManyVirtualColumn extends VirtualColumn {
       this.nestedInsert
     );
   }
-  update(q, ctx, set) {
+  update(q, _, set) {
     hasRelationHandleUpdate(
       q,
-      ctx,
       set,
       this.key,
       this.state.primaryKey,
@@ -909,24 +898,26 @@ const makeHasAndBelongsToManyMethod = (table, qb, relation, relationName, query)
     primaryKey: pk,
     modifyRelatedQuery(relationQuery) {
       const ref = {};
-      pushQueryValue(
-        relationQuery,
-        "afterCreate",
-        async (_, result) => {
-          const fromQuery = ref.query.clone();
-          fromQuery.query.select = [{ selectAs: { [fk]: pk } }];
-          const createdCount = await subQuery.count()._createFrom(
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-            fromQuery,
-            {
-              [afk]: result[apk]
-            }
+      relationQuery._afterCreate([], async (result) => {
+        if (result.length > 1) {
+          throw new OrchidOrmInternalError(
+            relationQuery,
+            "Creating multiple `hasAndBelongsToMany` records is not yet supported"
           );
-          if (createdCount === 0) {
-            throw new NotFoundError(fromQuery);
+        }
+        const fromQuery = ref.query.clone();
+        fromQuery.query.select = [{ selectAs: { [fk]: pk } }];
+        const createdCount = await subQuery.count()._createFrom(
+          // eslint-disable-next-line @typescript-eslint/no-explicit-any
+          fromQuery,
+          {
+            [afk]: result[0][apk]
           }
+        );
+        if (createdCount === 0) {
+          throw new NotFoundError(fromQuery);
         }
-      );
+      });
       return (q) => {
         ref.query = q;
       };
@@ -1437,11 +1428,16 @@ const orchidORM = (_a, tables) => {
     dbTable.db = result;
     dbTable.filePath = table.filePath;
     dbTable.name = table.constructor.name;
-    if (table.hooks)
-      addTableHooks(dbTable, table.hooks);
     result[key] = dbTable;
   }
   applyRelations(qb, tableInstances, result);
+  for (const key in tables) {
+    const table = tableInstances[key];
+    if (table.init) {
+      table.init(result);
+      Object.assign(result[key].baseQuery.query, table.query);
+    }
+  }
   return result;
 };
 
@@ -1488,5 +1484,5 @@ const createRepo = (table, methods) => {
   });
 };
 
-export { addTableHooks, createBaseTable, createRepo, orchidORM };
+export { createBaseTable, createRepo, orchidORM };
 //# sourceMappingURL=index.mjs.map