fix: prevent unloading when it's in progress

mary-ext · Dec 23, 2023 · 3c09f44 · 3c09f44
1 parent 9ef0d58
commit 3c09f44
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/controllers/export-data-form.ts b/src/controllers/export-data-form.ts
@@ -28,6 +28,11 @@ class ExportDataForm extends HTMLElement {
 
 	private object_url: string | undefined;
 
+	handle_before_unload = (ev: BeforeUnloadEvent) => {
+		ev.preventDefault();
+		ev.returnValue = true;
+	};
+
 	connectedCallback() {
 		if (!supports_fsa) {
 			const $fsa_warning = this.fsa_warning.get()!;
@@ -82,16 +87,24 @@ class ExportDataForm extends HTMLElement {
 						return;
 					}
 
+					window.addEventListener('beforeunload', this.handle_before_unload);
+
 					this.download_archive(signal, fd, identifier, with_media).then(
 						() => {
 							// If we got here, we're still dealing with our own controller.
 							controller!.abort();
+
+							window.removeEventListener('beforeunload', this.handle_before_unload);
 						},
 						(err) => {
 							if (signal.aborted) {
 								return;
 							}
 
+							console.error(err);
+
+							window.removeEventListener('beforeunload', this.handle_before_unload);
+
 							$status.textContent = err.message;
 							$status.classList.add('text-red-500');
 							$status.classList.remove('opacity-50');

diff --git a/src/controllers/generate-archive-form.ts b/src/controllers/generate-archive-form.ts
@@ -37,6 +37,11 @@ class GenerateArchiveForm extends HTMLElement {
 
 	private fsa_large_warning = target<HTMLDivElement>(this, 'fsa_large_warning');
 
+	handle_before_unload = (ev: BeforeUnloadEvent) => {
+		ev.preventDefault();
+		ev.returnValue = true;
+	};
+
 	connectedCallback() {
 		{
 			const $picker_input = this.picker_input.get()!;
@@ -96,10 +101,14 @@ class GenerateArchiveForm extends HTMLElement {
 						return;
 					}
 
+					window.addEventListener('beforeunload', this.handle_before_unload);
+
 					this.generate_archive(signal, fd, archive, with_media).then(
 						() => {
 							// If we got here, we're still dealing with our own controller.
 							controller!.abort();
+
+							window.removeEventListener('beforeunload', this.handle_before_unload);
 						},
 						(err) => {
 							if (signal.aborted) {
@@ -108,6 +117,8 @@ class GenerateArchiveForm extends HTMLElement {
 
 							console.error(err);
 
+							window.removeEventListener('beforeunload', this.handle_before_unload);
+
 							$status.textContent = err.message;
 							$status.classList.add('text-red-500');
 							$status.classList.remove('opacity-50');