@declaro/data 2.0.0-beta.98 → 2.0.0-beta.99

package/dist/node/index.cjs

@@ -11410,8 +11410,19 @@ class MockMemoryRepository {
     if (!this.entityMetadata?.primaryKey) {
       throw new Error("Primary key is not defined in the schema metadata");
     }
-    const items = await Promise.all(inputs.map((input) => this.data.get(input[this.entityMetadata.primaryKey])));
-    return items;
+    const results = [];
+    for (const input of inputs) {
+      let item;
+      if (typeof this.args.lookup === "function") {
+        item = Array.from(this.data.values()).find((data) => this.args.lookup(data, input));
+      } else {
+        item = this.data.get(input[this.entityMetadata.primaryKey]);
+      }
+      if (item) {
+        results.push(item);
+      }
+    }
+    return results;
   }
   async search(input, options) {
     const pagination = options?.pagination || { page: 1, pageSize: 25 };
@@ -11453,24 +11464,44 @@ class MockMemoryRepository {
     if (!this.entityMetadata?.primaryKey) {
       throw new Error("Primary key is not defined in the schema metadata");
     }
-    const item = await this.data.get(lookup[this.entityMetadata.primaryKey]);
+    let item;
+    let itemKey;
+    if (typeof this.args.lookup === "function") {
+      item = Array.from(this.data.values()).find((data) => this.args.lookup(data, lookup));
+      if (item) {
+        itemKey = item[this.entityMetadata.primaryKey];
+      }
+    } else {
+      itemKey = lookup[this.entityMetadata.primaryKey];
+      item = this.data.get(itemKey);
+    }
     if (!item) {
       throw new Error("Item not found");
     }
-    this.trash.set(lookup[this.entityMetadata.primaryKey], item);
-    this.data.delete(lookup[this.entityMetadata.primaryKey]);
+    this.trash.set(itemKey, item);
+    this.data.delete(itemKey);
     return item;
   }
   async restore(lookup) {
     if (!this.entityMetadata?.primaryKey) {
       throw new Error("Primary key is not defined in the schema metadata");
     }
-    const item = await this.trash.get(lookup[this.entityMetadata.primaryKey]);
+    let item;
+    let itemKey;
+    if (typeof this.args.lookup === "function") {
+      item = Array.from(this.trash.values()).find((data) => this.args.lookup(data, lookup));
+      if (item) {
+        itemKey = item[this.entityMetadata.primaryKey];
+      }
+    } else {
+      itemKey = lookup[this.entityMetadata.primaryKey];
+      item = this.trash.get(itemKey);
+    }
     if (!item) {
       throw new Error("Item not found in trash");
     }
-    this.trash.delete(lookup[this.entityMetadata.primaryKey]);
-    this.data.set(lookup[this.entityMetadata.primaryKey], item);
+    this.trash.delete(itemKey);
+    this.data.set(itemKey, item);
     return item;
   }
   async create(input) {
@@ -11493,16 +11524,27 @@ class MockMemoryRepository {
     if (!this.entityMetadata?.primaryKey) {
       throw new Error("Primary key is not defined in the schema metadata");
     }
-    const primaryKeyValue = lookup[this.entityMetadata.primaryKey];
-    if (!primaryKeyValue) {
-      throw new Error("Primary key value must be provided");
-    }
-    const existingItem = this.data.get(primaryKeyValue);
-    if (!existingItem) {
-      throw new Error("Item not found");
+    let existingItem;
+    let itemKey;
+    if (typeof this.args.lookup === "function") {
+      existingItem = Array.from(this.data.values()).find((data) => this.args.lookup(data, lookup));
+      if (existingItem) {
+        itemKey = existingItem[this.entityMetadata.primaryKey];
+      } else {
+        throw new Error("Item not found");
+      }
+    } else {
+      itemKey = lookup[this.entityMetadata.primaryKey];
+      if (!itemKey) {
+        throw new Error("Primary key value must be provided");
+      }
+      existingItem = this.data.get(itemKey);
+      if (!existingItem) {
+        throw new Error("Item not found");
+      }
     }
     const updatedItem = this.assignInput(existingItem, input);
-    this.data.set(primaryKeyValue, updatedItem);
+    this.data.set(itemKey, updatedItem);
     return updatedItem;
   }
   async count(search, options) {
@@ -11556,5 +11598,5 @@ class MockMemoryRepository {
   }
 }
 
-//# debugId=2ABA230E346B7A7A64756E2164756E21
+//# debugId=34781135054B25FD64756E2164756E21
 //# sourceMappingURL=index.cjs.map