From 854c2323e667e79e61cff0777a8b7afd36dda26c Mon Sep 17 00:00:00 2001
From: Abdullah Atta <abdullahatta@streetwriters.co>
Date: Thu, 22 Feb 2024 23:14:06 +0500
Subject: [PATCH] core: don't unlink vault notes on vault remove

---
 packages/core/src/api/vault.ts          | 2 +-
 packages/core/src/collections/vaults.ts | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/packages/core/src/api/vault.ts b/packages/core/src/api/vault.ts
index 8a3f3e2d4..55d433f3e 100644
--- a/packages/core/src/api/vault.ts
+++ b/packages/core/src/api/vault.ts
@@ -160,7 +160,7 @@ export default class Vault {
       await this.db.notes.remove(...lockedIds);
     }
 
-    await this.db.vaults.remove(vault.id);
+    await this.db.vaults.remove(vault.id, deleteAllLockedNotes);
     this.password = undefined;
   }
 
diff --git a/packages/core/src/collections/vaults.ts b/packages/core/src/collections/vaults.ts
index a8e761eb8..9703b5221 100644
--- a/packages/core/src/collections/vaults.ts
+++ b/packages/core/src/collections/vaults.ts
@@ -55,9 +55,9 @@ export class Vaults implements ICollection {
     return id;
   }
 
-  async remove(id: string) {
+  async remove(id: string, unlinkAllNotes = false) {
     await this.db.transaction(async () => {
-      await this.db.relations.unlinkOfType("vault", [id]);
+      if (unlinkAllNotes) await this.db.relations.unlinkOfType("vault", [id]);
       await this.collection.softDelete([id]);
     });
   }