chore(collections): Remove from all collections improvement

jorenn92 · Dec 11, 2023 · b871036 · b871036
1 parent 9a81f31
commit b871036
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/server/src/modules/collections/collections.service.ts b/server/src/modules/collections/collections.service.ts
@@ -365,7 +365,7 @@ export class CollectionsService {
     const handleMedia: AddCollectionMedia[] = [];
 
     const collection =
-      collectionDbId !== -1
+      collectionDbId !== -1 && collectionDbId !== undefined
         ? await this.collectionRepo.findOne({
             where: { id: collectionDbId },
           })
@@ -520,7 +520,11 @@ export class CollectionsService {
     if (action === 'add') {
       return this.addToCollection(collectionDbId, handleMedia, true);
     } else if (action === 'remove') {
-      return this.removeFromCollection(collectionDbId, handleMedia);
+      if (collectionDbId) {
+        return this.removeFromCollection(collectionDbId, handleMedia);
+      } else {
+        this.removeFromAllCollections(handleMedia);
+      }
     }
   }