From a6637d5e27167449a24ab406a1fb0361c3692e4c Mon Sep 17 00:00:00 2001
From: vcoppe <vianney.coppe@gmail.com>
Date: Thu, 16 May 2024 16:24:50 +0200
Subject: [PATCH] extract collapsible tree component

---
 website/src/lib/assets/layers.ts              |  4 --
 .../collapsible-tree/CollapsibleTree.svelte   | 10 ++++
 .../CollapsibleTreeNode.svelte                | 37 ++++++++++++++
 .../components/layer-control/LayerTree.svelte | 11 ++--
 .../layer-control/LayerTreeNode.svelte        | 51 ++++---------------
 5 files changed, 61 insertions(+), 52 deletions(-)
 create mode 100644 website/src/lib/components/collapsible-tree/CollapsibleTree.svelte
 create mode 100644 website/src/lib/components/collapsible-tree/CollapsibleTreeNode.svelte

diff --git a/website/src/lib/assets/layers.ts b/website/src/lib/assets/layers.ts
index 8aaed0ed..95b0626f 100644
--- a/website/src/lib/assets/layers.ts
+++ b/website/src/lib/assets/layers.ts
@@ -381,10 +381,6 @@ export const opacities: { [key: string]: number; } = {
 };
 
 export type LayerTreeType = { [key: string]: LayerTreeType | boolean; };
-export type CollapsedInfoTreeType<T> = {
-    self: T;
-    children: { [key: string]: CollapsedInfoTreeType<T>; };
-};
 
 // Hierarchy containing all basemaps
 export const basemapTree: LayerTreeType = {
diff --git a/website/src/lib/components/collapsible-tree/CollapsibleTree.svelte b/website/src/lib/components/collapsible-tree/CollapsibleTree.svelte
new file mode 100644
index 00000000..70d80316
--- /dev/null
+++ b/website/src/lib/components/collapsible-tree/CollapsibleTree.svelte
@@ -0,0 +1,10 @@
+<script lang="ts">
+	import { setContext } from 'svelte';
+	import { writable } from 'svelte/store';
+
+	let open = writable<Record<string, boolean>>({});
+
+	setContext('collapsible-tree-state', open);
+</script>
+
+<slot />
diff --git a/website/src/lib/components/collapsible-tree/CollapsibleTreeNode.svelte b/website/src/lib/components/collapsible-tree/CollapsibleTreeNode.svelte
new file mode 100644
index 00000000..52bc3d16
--- /dev/null
+++ b/website/src/lib/components/collapsible-tree/CollapsibleTreeNode.svelte
@@ -0,0 +1,37 @@
+<script lang="ts">
+	import * as Collapsible from '$lib/components/ui/collapsible';
+	import { Button } from '$lib/components/ui/button';
+	import { ChevronDown, ChevronUp } from 'lucide-svelte';
+	import { getContext } from 'svelte';
+	import type { Writable } from 'svelte/store';
+
+	export let id: string;
+
+	let open = getContext<Writable<Record<string, boolean>>>('collapsible-tree-state');
+
+	open.update((value) => {
+		if (!value.hasOwnProperty(id)) {
+			value[id] = true;
+		}
+		return value;
+	});
+</script>
+
+<Collapsible.Root bind:open={$open[id]}>
+	<Collapsible.Trigger class="w-full">
+		<Button
+			variant="ghost"
+			class="w-full flex flex-row justify-between py-0 px-1 h-fit hover:bg-background"
+		>
+			<slot name="trigger" />
+			{#if $open[id]}
+				<ChevronUp size="16" />
+			{:else}
+				<ChevronDown size="16" />
+			{/if}
+		</Button>
+	</Collapsible.Trigger>
+	<Collapsible.Content class="ml-1">
+		<slot name="content" />
+	</Collapsible.Content>
+</Collapsible.Root>
diff --git a/website/src/lib/components/layer-control/LayerTree.svelte b/website/src/lib/components/layer-control/LayerTree.svelte
index fe00f020..37e1d5e8 100644
--- a/website/src/lib/components/layer-control/LayerTree.svelte
+++ b/website/src/lib/components/layer-control/LayerTree.svelte
@@ -1,21 +1,20 @@
 <script lang="ts">
 	import LayerTreeNode from './LayerTreeNode.svelte';
-	import { type CollapsedInfoTreeType, type LayerTreeType } from '$lib/assets/layers';
+	import { type LayerTreeType } from '$lib/assets/layers';
+	import CollapsibleTree from '$lib/components/collapsible-tree/CollapsibleTree.svelte';
 
 	export let layerTree: LayerTreeType;
 	export let name: string;
 	export let selected: string | undefined = undefined;
 	export let multiple: boolean = false;
 
-	let open: CollapsedInfoTreeType<boolean> = {
-		self: true,
-		children: {}
-	};
 	export let checked: LayerTreeType = {};
 </script>
 
 <form>
 	<fieldset class="min-w-64">
-		<LayerTreeNode {name} node={layerTree} bind:selected {multiple} bind:open bind:checked />
+		<CollapsibleTree>
+			<LayerTreeNode {name} node={layerTree} bind:selected {multiple} bind:checked />
+		</CollapsibleTree>
 	</fieldset>
 </form>
diff --git a/website/src/lib/components/layer-control/LayerTreeNode.svelte b/website/src/lib/components/layer-control/LayerTreeNode.svelte
index e8de81db..29d9a47b 100644
--- a/website/src/lib/components/layer-control/LayerTreeNode.svelte
+++ b/website/src/lib/components/layer-control/LayerTreeNode.svelte
@@ -1,12 +1,9 @@
 <script lang="ts">
 	import { Label } from '$lib/components/ui/label';
 	import { Checkbox } from '$lib/components/ui/checkbox';
-	import { Button } from '$lib/components/ui/button';
-	import * as Collapsible from '$lib/components/ui/collapsible';
+	import CollapsibleTreeNode from '../collapsible-tree/CollapsibleTreeNode.svelte';
 
-	import { ChevronDown, ChevronUp } from 'lucide-svelte';
-
-	import { type CollapsedInfoTreeType, type LayerTreeType } from '$lib/assets/layers';
+	import { type LayerTreeType } from '$lib/assets/layers';
 	import { anySelectedLayer } from './utils';
 
 	import { _ } from 'svelte-i18n';
@@ -16,19 +13,8 @@
 	export let selected: string | undefined = undefined;
 	export let multiple: boolean = false;
 
-	export let open: CollapsedInfoTreeType<boolean>;
 	export let checked: LayerTreeType;
 
-	if (!Array.isArray(node)) {
-		Object.keys(node).forEach((id) => {
-			if (!open.children.hasOwnProperty(id)) {
-				open.children[id] = {
-					self: true,
-					children: {}
-				};
-			}
-		});
-	}
 	Object.keys(node).forEach((id) => {
 		if (!checked.hasOwnProperty(id)) {
 			if (typeof node[id] == 'boolean') {
@@ -40,7 +26,7 @@
 	});
 </script>
 
-<div class="flex flex-col gap-1">
+<div class="flex flex-col gap-0.5">
 	{#each Object.keys(node) as id}
 		{#if typeof node[id] == 'boolean'}
 			{#if node[id]}
@@ -62,31 +48,12 @@
 				</div>
 			{/if}
 		{:else if anySelectedLayer(node[id])}
-			<Collapsible.Root bind:open={open.children[id].self} class="ml-1">
-				<Collapsible.Trigger class="w-full"
-					><Button
-						variant="ghost"
-						class="w-full flex flex-row justify-between py-0 px-1 h-fit hover:bg-background"
-					>
-						<span class="mr-2">{$_(`layers.label.${id}`)}</span>
-						{#if open.children[id].self}
-							<ChevronUp size="16" />
-						{:else}
-							<ChevronDown size="16" />
-						{/if}
-					</Button></Collapsible.Trigger
-				>
-				<Collapsible.Content class="ml-1">
-					<svelte:self
-						node={node[id]}
-						{name}
-						bind:selected
-						{multiple}
-						bind:open={open.children[id]}
-						bind:checked={checked[id]}
-					/>
-				</Collapsible.Content>
-			</Collapsible.Root>
+			<CollapsibleTreeNode {id}>
+				<span slot="trigger">{$_(`layers.label.${id}`)}</span>
+				<div slot="content">
+					<svelte:self node={node[id]} {name} bind:selected {multiple} bind:checked={checked[id]} />
+				</div>
+			</CollapsibleTreeNode>
 		{/if}
 	{/each}
 </div>