nestjs-query-mikro-orm 0.1.0 → 0.1.1

package/dist/index.cjs

@@ -840,7 +840,14 @@ var RelationQueryBuilder = class {
         const entityKeys = Object.keys(entityAsRecord);
         const matchingKey = entityKeys.find((key) => {
           const keyLower = key.toLowerCase();
-          return keyLower.endsWith("id") && relationNameLower.includes(keyLower.replace(/id$/, ""));
+          if (keyLower === "id" || !keyLower.endsWith("id")) {
+            return false;
+          }
+          const base = keyLower.replace(/id$/, "");
+          if (!base) {
+            return false;
+          }
+          return relationNameLower.includes(base);
         });
         if (matchingKey) {
           fkValue = entityAsRecord[matchingKey];
@@ -1174,6 +1181,10 @@ var RelationQueryService = class {
     if (Array.isArray(dto)) {
       return this.batchQueryRelations(RelationClass, relationName, dto, query);
     }
+    if (this.isRelationClassIdentity(RelationClass, relationName)) {
+      const relationQueryBuilder2 = this.getRelationQueryBuilder(relationName);
+      return await relationQueryBuilder2.selectAndExecute(dto, query);
+    }
     const assembler = nestjsQueryCore.AssemblerFactory.getAssembler(RelationClass, this.getRelationEntity(relationName));
     const relationQueryBuilder = this.getRelationQueryBuilder(relationName);
     return assembler.convertToDTOs(await relationQueryBuilder.selectAndExecute(dto, assembler.convertQuery(query)));
@@ -1207,6 +1218,16 @@ var RelationQueryService = class {
     if (Array.isArray(dto)) {
       return this.batchFindRelations(RelationClass, relationName, dto, opts);
     }
+    if (this.isRelationClassIdentity(RelationClass, relationName)) {
+      const relationQueryBuilder2 = this.getRelationQueryBuilder(relationName);
+      const relations2 = await relationQueryBuilder2.selectAndExecute(dto, {
+        filter: opts?.filter,
+        paging: {
+          limit: 1
+        }
+      });
+      return relations2[0];
+    }
     const assembler = nestjsQueryCore.AssemblerFactory.getAssembler(RelationClass, this.getRelationEntity(relationName));
     const relationQueryBuilder = this.getRelationQueryBuilder(relationName);
     const relations = await relationQueryBuilder.selectAndExecute(dto, {
@@ -1227,10 +1248,20 @@ var RelationQueryService = class {
   */
   async addRelations(relationName, id, relationIds, opts) {
     const entity = await this.getById(id, opts);
+    const meta = this.getRelationMeta(relationName);
     const relations = await this.getRelations(relationName, relationIds, opts?.relationFilter);
     if (!this.foundAllRelations(relationIds, relations)) {
       throw new Error(`Unable to find all ${relationName} to add to ${this.EntityClass.name}`);
     }
+    if (meta.kind === "1:m" && meta.mappedBy) {
+      for (const relation of relations) {
+        core.wrap(relation).assign({
+          [meta.mappedBy]: entity
+        });
+      }
+      await this.repo.getEntityManager().flush();
+      return entity;
+    }
     const collection = entity[relationName];
     if (collection && typeof collection.add === "function") {
       for (const relation of relations) {
@@ -1251,12 +1282,33 @@ var RelationQueryService = class {
   */
   async setRelations(relationName, id, relationIds, opts) {
     const entity = await this.getById(id, opts);
+    const meta = this.getRelationMeta(relationName);
     const relations = await this.getRelations(relationName, relationIds, opts?.relationFilter);
     if (relationIds.length) {
       if (!this.foundAllRelations(relationIds, relations)) {
         throw new Error(`Unable to find all ${relationName} to set on ${this.EntityClass.name}`);
       }
     }
+    if (meta.kind === "1:m" && meta.mappedBy) {
+      const relationQueryBuilder = this.getRelationQueryBuilder(relationName);
+      const currentRelations = await relationQueryBuilder.selectAndExecute(entity, {});
+      const nextSet = new Set(relations.map((r) => core.wrap(r).getPrimaryKey()));
+      for (const currentRelation of currentRelations) {
+        const currentPk = core.wrap(currentRelation).getPrimaryKey();
+        if (!nextSet.has(currentPk)) {
+          core.wrap(currentRelation).assign({
+            [meta.mappedBy]: null
+          });
+        }
+      }
+      for (const relation of relations) {
+        core.wrap(relation).assign({
+          [meta.mappedBy]: entity
+        });
+      }
+      await this.repo.getEntityManager().flush();
+      return entity;
+    }
     const collection = entity[relationName];
     if (collection && typeof collection.set === "function") {
       await collection.init();
@@ -1296,10 +1348,20 @@ var RelationQueryService = class {
   */
   async removeRelations(relationName, id, relationIds, opts) {
     const entity = await this.getById(id, opts);
+    const meta = this.getRelationMeta(relationName);
     const relations = await this.getRelations(relationName, relationIds, opts?.relationFilter);
     if (!this.foundAllRelations(relationIds, relations)) {
       throw new Error(`Unable to find all ${relationName} to remove from ${this.EntityClass.name}`);
     }
+    if (meta.kind === "1:m" && meta.mappedBy) {
+      for (const relation of relations) {
+        core.wrap(relation).assign({
+          [meta.mappedBy]: null
+        });
+      }
+      await this.repo.getEntityManager().flush();
+      return entity;
+    }
     const collection = entity[relationName];
     if (collection && typeof collection.remove === "function") {
       await collection.init();
@@ -1328,14 +1390,25 @@ var RelationQueryService = class {
     const meta = this.getRelationMeta(relationName);
     if (meta.kind === "1:1" || meta.kind === "m:1") {
       const fkFieldName = `${relationName}Id`;
-      const assignData = {
-        [relationName]: null
-      };
-      if (fkFieldName in entity) {
+      const assignData = {};
+      const ownDescriptor = Object.getOwnPropertyDescriptor(entity, fkFieldName);
+      const protoDescriptor = Object.getOwnPropertyDescriptor(Object.getPrototypeOf(entity), fkFieldName);
+      const descriptor = ownDescriptor ?? protoDescriptor;
+      const canAssignFk = fkFieldName in entity && (!descriptor || Boolean(descriptor.set) || descriptor.writable === true);
+      if (canAssignFk) {
         assignData[fkFieldName] = null;
+      } else {
+        assignData[relationName] = null;
       }
       core.wrap(entity).assign(assignData);
     } else {
+      if (meta.kind === "1:m" && meta.mappedBy) {
+        core.wrap(relation).assign({
+          [meta.mappedBy]: null
+        });
+        await this.repo.getEntityManager().flush();
+        return entity;
+      }
       const collection = entity[relationName];
       if (collection && typeof collection.remove === "function") {
         await collection.init();
@@ -1356,13 +1429,14 @@ var RelationQueryService = class {
   * @param query - A query to filter, page or sort relations.
   */
   async batchQueryRelations(RelationClass, relationName, entities, query) {
-    const assembler = nestjsQueryCore.AssemblerFactory.getAssembler(RelationClass, this.getRelationEntity(relationName));
+    const bypassAssembler = this.isRelationClassIdentity(RelationClass, relationName);
+    const assembler = bypassAssembler ? void 0 : nestjsQueryCore.AssemblerFactory.getAssembler(RelationClass, this.getRelationEntity(relationName));
     const relationQueryBuilder = this.getRelationQueryBuilder(relationName);
-    const convertedQuery = assembler.convertQuery(query);
+    const convertedQuery = assembler ? assembler.convertQuery(query) : query;
     const results = /* @__PURE__ */ new Map();
     await Promise.all(entities.map(async (entity) => {
       const relations = await relationQueryBuilder.selectAndExecute(entity, convertedQuery);
-      const relationDtos = await assembler.convertToDTOs(relations);
+      const relationDtos = bypassAssembler ? relations : await assembler.convertToDTOs(relations);
       if (relationDtos.length > 0) {
         results.set(entity, relationDtos);
       }
@@ -1432,6 +1506,10 @@ var RelationQueryService = class {
     });
     return results;
   }
+  isRelationClassIdentity(RelationClass, relationName) {
+    const relationEntity = this.getRelationEntity(relationName);
+    return RelationClass === relationEntity || RelationClass.name === relationEntity.name;
+  }
   getRelationMeta(relationName) {
     const em = this.repo.getEntityManager();
     const metadata = em.getMetadata().get(this.repo.getEntityName());
@@ -1442,12 +1520,14 @@ var RelationQueryService = class {
     return {
       kind: relationMeta.kind,
       type: relationMeta.type,
-      entity: relationMeta.entity
+      entity: relationMeta.entity,
+      mappedBy: relationMeta.mappedBy
     };
   }
   getRelationEntity(relationName) {
     const relationMeta = this.getRelationMeta(relationName);
-    return relationMeta.entity();
+    const entity = relationMeta.entity();
+    return entity && "class" in entity ? entity.class : entity;
   }
   async getRelations(relationName, ids, filter) {
     const em = this.repo.getEntityManager();
@@ -1509,7 +1589,7 @@ var MikroOrmQueryService = class extends RelationQueryService {
         return data;
       })(this.EntityClass);
       nestjsQueryCore.AssemblerDeserializer((d) => {
-        const entity = this.repo.getEntityManager().create(this.EntityClass, classTransformer.instanceToPlain(d));
+        const entity = this.repo.getEntityManager().merge(this.EntityClass, d);
         return entity;
       })(this.EntityClass);
     }