From 05b93ae3f45b74ff6984e54f0e31f39917546169 Mon Sep 17 00:00:00 2001
From: 0HyperCube <78500760+0HyperCube@users.noreply.github.com>
Date: Sat, 9 Jul 2022 23:23:49 +0100
Subject: [PATCH] Fix IndexedDb in Firefox private browsing mode (#721)

* Fix db in firefox private

* Polished wording

Co-authored-by: Keavon Chambers <keavon@keavon.com>
---
 frontend/src/io-managers/persistence.ts | 54 ++++++++++++++-----------
 1 file changed, 31 insertions(+), 23 deletions(-)

diff --git a/frontend/src/io-managers/persistence.ts b/frontend/src/io-managers/persistence.ts
index e28334bf..454fc0d1 100644
--- a/frontend/src/io-managers/persistence.ts
+++ b/frontend/src/io-managers/persistence.ts
@@ -1,4 +1,5 @@
 import { PortfolioState } from "@/state-providers/portfolio";
+import { stripIndents } from "@/utility-functions/strip-indents";
 import { Editor } from "@/wasm-communication/editor";
 import { TriggerIndexedDbWriteDocument, TriggerIndexedDbRemoveDocument } from "@/wasm-communication/messages";
 
@@ -53,8 +54,14 @@ export async function createPersistenceManager(editor: Editor, portfolio: Portfo
 		};
 
 		dbOpenRequest.onerror = (): void => {
-			// eslint-disable-next-line no-console
-			console.error("Graphite IndexedDb error:", dbOpenRequest.error);
+			const errorText = stripIndents`
+				Documents won't be saved across reloads and later visits.
+				This may be caused by Firefox's private browsing mode.
+				
+				Error on opening IndexDB:
+				${dbOpenRequest.error}
+				`;
+			editor.instance.error_dialog("Document auto-save doesn't work in this browser", errorText);
 		};
 
 		dbOpenRequest.onsuccess = (): void => {
@@ -62,29 +69,30 @@ export async function createPersistenceManager(editor: Editor, portfolio: Portfo
 		};
 	});
 
-	// Open auto-save documents
-	const db = await databaseConnection;
-	const transaction = db.transaction(GRAPHITE_AUTO_SAVE_STORE, "readonly");
-	const request = transaction.objectStore(GRAPHITE_AUTO_SAVE_STORE).getAll();
-	await new Promise((resolve): void => {
-		request.onsuccess = (): void => {
-			const previouslySavedDocuments: TriggerIndexedDbWriteDocument[] = request.result;
+	databaseConnection.then(async (db) => {
+		// Open auto-save documents
+		const transaction = db.transaction(GRAPHITE_AUTO_SAVE_STORE, "readonly");
+		const request = transaction.objectStore(GRAPHITE_AUTO_SAVE_STORE).getAll();
+		await new Promise((resolve): void => {
+			request.onsuccess = (): void => {
+				const previouslySavedDocuments: TriggerIndexedDbWriteDocument[] = request.result;
 
-			const documentOrder: string[] = JSON.parse(window.localStorage.getItem(GRAPHITE_AUTO_SAVE_ORDER_KEY) || "[]");
-			const orderedSavedDocuments = documentOrder
-				.map((id) => previouslySavedDocuments.find((autoSave) => autoSave.details.id === id))
-				.filter((x) => x !== undefined) as TriggerIndexedDbWriteDocument[];
+				const documentOrder: string[] = JSON.parse(window.localStorage.getItem(GRAPHITE_AUTO_SAVE_ORDER_KEY) || "[]");
+				const orderedSavedDocuments = documentOrder
+					.map((id) => previouslySavedDocuments.find((autoSave) => autoSave.details.id === id))
+					.filter((x) => x !== undefined) as TriggerIndexedDbWriteDocument[];
 
-			const currentDocumentVersion = editor.instance.graphite_document_version();
-			orderedSavedDocuments.forEach((doc: TriggerIndexedDbWriteDocument) => {
-				if (doc.version === currentDocumentVersion) {
-					editor.instance.open_auto_saved_document(BigInt(doc.details.id), doc.details.name, doc.details.is_saved, doc.document);
-				} else {
-					removeDocument(doc.details.id);
-				}
-			});
-			resolve(undefined);
-		};
+				const currentDocumentVersion = editor.instance.graphite_document_version();
+				orderedSavedDocuments.forEach((doc: TriggerIndexedDbWriteDocument) => {
+					if (doc.version === currentDocumentVersion) {
+						editor.instance.open_auto_saved_document(BigInt(doc.details.id), doc.details.name, doc.details.is_saved, doc.document);
+					} else {
+						removeDocument(doc.details.id);
+					}
+				});
+				resolve(undefined);
+			};
+		});
 	});
 
 	return closeDatabaseConnection;