From 1bc07189010589ff7141b7ffc0cd53481a2fedae Mon Sep 17 00:00:00 2001
From: Sidharth Vinod <sidharthv96@gmail.com>
Date: Fri, 12 Aug 2022 22:15:36 +0530
Subject: [PATCH] Persist pan & zoom in state

---
 src/lib/components/preset.svelte |  3 +-
 src/lib/components/view.svelte   | 57 ++++++++++++++++++++------------
 src/lib/types.d.ts               |  2 ++
 src/lib/util/state.ts            | 10 +++++-
 src/routes/edit.svelte           |  4 +--
 5 files changed, 50 insertions(+), 26 deletions(-)

diff --git a/src/lib/components/preset.svelte b/src/lib/components/preset.svelte
index 2c31cbaf..310b73b4 100644
--- a/src/lib/components/preset.svelte
+++ b/src/lib/components/preset.svelte
@@ -97,7 +97,8 @@
 	const loadSampleDiagram = (diagramType: string): void => {
 		updateCode(samples[diagramType], {
 			updateDiagram: true,
-			updateEditor: true
+			updateEditor: true,
+			resetPanZoom: true
 		});
 	};
 </script>
diff --git a/src/lib/components/view.svelte b/src/lib/components/view.svelte
index 3579723d..89360b21 100644
--- a/src/lib/components/view.svelte
+++ b/src/lib/components/view.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-	import { inputStateStore, stateStore } from '$lib/util/state';
+	import { inputStateStore, stateStore, updateCodeStore } from '$lib/util/state';
 	import { onMount } from 'svelte';
 	import mermaid from 'mermaid';
 	import panzoom from 'svg-pan-zoom';
@@ -12,9 +12,18 @@
 	let outOfSync = false;
 	let hide = false;
 	let manualUpdate = true;
-	let pan: SvgPanZoom.Point;
-	let zoom: number;
 	let pzoom: SvgPanZoom.Instance;
+	let debounce: number;
+
+	const handlePanZoomChange = () => {
+		const pan = pzoom.getPan();
+		const zoom = pzoom.getZoom();
+		clearTimeout(debounce);
+		debounce = window.setTimeout(() => {
+			updateCodeStore({ pan, zoom });
+		}, 500);
+	};
+
 	onMount(() => {
 		stateStore.subscribe((state) => {
 			if (state.error !== undefined) {
@@ -40,28 +49,25 @@
 					mermaid.initialize(Object.assign({}, JSON.parse(state.mermaid)));
 					mermaid.render('graph-div', code, (svgCode) => {
 						if (svgCode.length > 0) {
-							let oldPan = pan ? { ...pan } : undefined;
-							let oldZoom = zoom;
 							pzoom?.destroy();
+							pzoom = undefined;
 							hide = true;
 							container.innerHTML = svgCode;
 							setTimeout(() => {
-								pzoom = panzoom('#graph-div', {
-									onPan: (p) => {
-										pan = p;
-										zoom = pzoom.getZoom();
-									},
-									onZoom: (z) => {
-										zoom = z;
-										pan = pzoom.getPan();
-									},
-									zoomEnabled: true,
-									panEnabled: true,
-									controlIconsEnabled: true
+								const graphDiv = document.getElementById('graph-div');
+								graphDiv.setAttribute('height', '100%');
+								graphDiv.style.maxWidth = '100%';
+								pzoom = panzoom(graphDiv, {
+									onPan: handlePanZoomChange,
+									onZoom: handlePanZoomChange,
+									controlIconsEnabled: true,
+									fit: true,
+									center: true
 								});
-								if (oldPan !== undefined && oldZoom !== undefined) {
-									pzoom.zoom(oldZoom);
-									pzoom.pan(oldPan);
+								const { pan, zoom } = state;
+								if (pan !== undefined && zoom !== undefined) {
+									pzoom.zoom(zoom);
+									pzoom.pan(pan);
 								}
 								hide = false;
 							}, 0);
@@ -79,6 +85,13 @@
 				error = true;
 			}
 		});
+		window.addEventListener('resize', () => {
+			if (pzoom) {
+				pzoom.resize();
+				pzoom.fit();
+				pzoom.center();
+			}
+		});
 	});
 </script>
 
@@ -86,8 +99,8 @@
 	<div class="p-2 text-red-600" id="errorContainer">{$stateStore.error}</div>
 {/if}
 
-<div id="view" bind:this={view} class="p-2" class:error class:outOfSync>
-	<div id="container" bind:this={container} class="flex-1 overflow-auto" class:hide />
+<div id="view" bind:this={view} class="p-2 h-full" class:error class:outOfSync>
+	<div id="container" bind:this={container} class="h-full overflow-auto" class:hide />
 </div>
 
 <style>
diff --git a/src/lib/types.d.ts b/src/lib/types.d.ts
index 8e664323..9ff6d86b 100644
--- a/src/lib/types.d.ts
+++ b/src/lib/types.d.ts
@@ -39,6 +39,8 @@ export interface State {
 	updateEditor: boolean;
 	updateDiagram: boolean;
 	autoSync: boolean;
+	pan?: { x: number; y: number };
+	zoom?: number;
 	loader?: LoaderConfig;
 }
 
diff --git a/src/lib/util/state.ts b/src/lib/util/state.ts
index e352c20d..2917e2d6 100644
--- a/src/lib/util/state.ts
+++ b/src/lib/util/state.ts
@@ -114,7 +114,11 @@ export const updateCodeStore = (newState: Partial<State>): void => {
 let prompted = false;
 export const updateCode = (
 	code: string,
-	{ updateEditor, updateDiagram = false }: { updateEditor: boolean; updateDiagram?: boolean }
+	{
+		updateEditor,
+		updateDiagram = false,
+		resetPanZoom = false
+	}: { updateEditor: boolean; updateDiagram?: boolean; resetPanZoom?: boolean }
 ): void => {
 	saveStatistics(code);
 	const lines = (code.match(/\n/g) || '').length + 1;
@@ -132,6 +136,10 @@ export const updateCode = (
 	}
 
 	inputStateStore.update((state) => {
+		if (resetPanZoom) {
+			state.pan = undefined;
+			state.zoom = undefined;
+		}
 		return { ...state, code, updateEditor, updateDiagram };
 	});
 };
diff --git a/src/routes/edit.svelte b/src/routes/edit.svelte
index e420bfea..03e3dc83 100644
--- a/src/routes/edit.svelte
+++ b/src/routes/edit.svelte
@@ -93,10 +93,10 @@
 		}
 	];
 
-	let debounce: { [key: string]: any } = {};
+	let debounce: { [key: string]: number } = {};
 	const updateHandler = (message: CustomEvent<EditorUpdateEvent>) => {
 		clearTimeout(debounce[selectedMode]);
-		debounce[selectedMode] = setTimeout(() => {
+		debounce[selectedMode] = window.setTimeout(() => {
 			const code = message.detail.text;
 			if (selectedMode === 'code') {
 				updateCode(code, {