From 62861d1ee6540da8bcf65fd2b28be9239dcf8f52 Mon Sep 17 00:00:00 2001
From: Abdullah Atta <abdullahatta@streetwriters.co>
Date: Fri, 16 Feb 2024 17:32:46 +0500
Subject: [PATCH] web: clear keystore when clearing database

---
 .../src/components/error-boundary/index.tsx   | 59 ++++++++++++++-----
 1 file changed, 44 insertions(+), 15 deletions(-)

diff --git a/apps/web/src/components/error-boundary/index.tsx b/apps/web/src/components/error-boundary/index.tsx
index bc6f8d55b..2a3d715bf 100644
--- a/apps/web/src/components/error-boundary/index.tsx
+++ b/apps/web/src/components/error-boundary/index.tsx
@@ -88,10 +88,24 @@ export function ErrorComponent({ error, resetErrorBoundary }: FallbackProps) {
               <Button
                 variant="error"
                 sx={{ alignSelf: "start", px: 30, mt: 1 }}
-                onClick={() => help.fix().catch((e) => alert(e))}
+                onClick={() =>
+                  help.fix().catch((e) => {
+                    console.error(e);
+                    alert(errorToString(e));
+                  })
+                }
               >
                 Fix it
               </Button>
+              <Button
+                variant="secondary"
+                sx={{ alignSelf: "start", px: 30, mt: 1 }}
+                onClick={async () => {
+                  navigator.clipboard.writeText(errorToString(error));
+                }}
+              >
+                Copy
+              </Button>
               <Button
                 variant="secondary"
                 sx={{ alignSelf: "start", px: 30, mt: 1 }}
@@ -102,13 +116,7 @@ export function ErrorComponent({ error, resetErrorBoundary }: FallbackProps) {
                   const mailto = new URL("mailto:support@streetwriters.co");
                   mailto.searchParams.set(
                     "body",
-                    `${
-                      error instanceof Error
-                        ? error.stack
-                        : -typeof error
-                        ? error
-                        : JSON.stringify(error)
-                    }
+                    `${errorToString(error)}
 
 ---
 Device information:
@@ -142,12 +150,7 @@ ${getDeviceInfo()}`
 
 function getErrorHelp(props: FallbackProps) {
   const { error, resetErrorBoundary } = props;
-  const errorText =
-    typeof error === "string"
-      ? error
-      : error instanceof Error
-      ? error.toString()
-      : JSON.stringify(error);
+  const errorText = errorToString(error);
   if (
     errorText.includes("file is not a database") ||
     errorText.includes("unsupported file format")
@@ -157,13 +160,39 @@ function getErrorHelp(props: FallbackProps) {
       action:
         "This error can only be fixed by wiping & reseting the database. Beware that this will wipe all your data inside the database with no way to recover it later on.",
       fix: async () => {
+        const { useKeyStore } = await import("../../interfaces/key-store");
+
+        await useKeyStore.getState().clear();
+        const dialect = createDialect("notesnook");
+        const driver = dialect.createDriver();
+        if (!IS_DESKTOP_APP) await driver.init();
+        await driver.delete();
+        resetErrorBoundary();
+      }
+    };
+  } else if (errorText.includes("Could not decrypt key.")) {
+    return {
+      explanation: `This error means the at rest encryption key could not be decrypted. This can be due to data corruption or implementation change.`,
+      action:
+        "This error can only be fixed by wiping & reseting the Key Store and the database.",
+      fix: async () => {
+        const { useKeyStore } = await import("../../interfaces/key-store");
+
+        await useKeyStore.getState().clear();
         const dialect = createDialect("notesnook");
         const driver = dialect.createDriver();
         if (!IS_DESKTOP_APP) await driver.init();
         await driver.delete();
-        await driver.destroy();
         resetErrorBoundary();
       }
     };
   }
 }
+
+function errorToString(error: unknown) {
+  return error instanceof Error
+    ? [error.message, error.stack || ""].join("\n")
+    : typeof error === "string"
+    ? error
+    : JSON.stringify(error);
+}