From cf7153bd9880d79db1c1df7afca7ae3c2505b2ca Mon Sep 17 00:00:00 2001
From: Allan Lasser <lasser.allan@gmail.com>
Date: Wed, 27 Nov 2024 13:34:24 -0500
Subject: [PATCH] Fix context in storybooks

---
 .../stories/NoteHighlights.stories.svelte          |  7 +++++--
 .../stories/PageHighlights.stories.svelte          | 14 ++++++++++++--
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/src/lib/components/documents/stories/NoteHighlights.stories.svelte b/src/lib/components/documents/stories/NoteHighlights.stories.svelte
index 0362c7258..a5fb0ede3 100644
--- a/src/lib/components/documents/stories/NoteHighlights.stories.svelte
+++ b/src/lib/components/documents/stories/NoteHighlights.stories.svelte
@@ -1,6 +1,4 @@
 <script context="module" lang="ts">
-  import { setContext } from "svelte";
-  import { writable } from "svelte/store";
   import { Story } from "@storybook/addon-svelte-csf";
 
   import type { Document } from "$lib/api/types";
@@ -18,6 +16,11 @@
     tags: ["autodocs"],
     parameters: { layout: "fullscreen" },
   };
+</script>
+
+<script lang="ts">
+  import { setContext } from "svelte";
+  import { writable } from "svelte/store";
 
   setContext(
     "highlightState",
diff --git a/src/lib/components/documents/stories/PageHighlights.stories.svelte b/src/lib/components/documents/stories/PageHighlights.stories.svelte
index 65197166a..928e125e1 100644
--- a/src/lib/components/documents/stories/PageHighlights.stories.svelte
+++ b/src/lib/components/documents/stories/PageHighlights.stories.svelte
@@ -1,6 +1,4 @@
 <script context="module" lang="ts">
-  import { setContext } from "svelte";
-  import { writable } from "svelte/store";
   import { Story } from "@storybook/addon-svelte-csf";
 
   import type { Document } from "$lib/api/types";
@@ -28,6 +26,18 @@
   );
 </script>
 
+<script lang="ts">
+  import { setContext } from "svelte";
+  import { writable } from "svelte/store";
+
+  setContext(
+    "highlightState",
+    writable({
+      allOpen: true,
+    }),
+  );
+</script>
+
 <Story name="closed">
   <PageHighlights {document} />
 </Story>