From 11e04323f346c0fc0ca4c9d1874b32945261f490 Mon Sep 17 00:00:00 2001
From: Guillaume Chau <guillaume.b.chau@gmail.com>
Date: Sat, 12 Nov 2022 02:57:44 +0100
Subject: [PATCH] fix(svelte): dynamic source, fix #344

---
 examples/svelte3/src/BaseButton.story.svelte        | 12 +++++++++++-
 .../src/client/RenderStory.svelte                   | 13 ++++++++++++-
 .../src/client/RenderVariant.svelte                 | 13 +++++++++++--
 3 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/examples/svelte3/src/BaseButton.story.svelte b/examples/svelte3/src/BaseButton.story.svelte
index d80e9ceb..687c3d80 100644
--- a/examples/svelte3/src/BaseButton.story.svelte
+++ b/examples/svelte3/src/BaseButton.story.svelte
@@ -7,9 +7,19 @@
 
   let disabled = false
   let size = 'medium'
+
+  let source;
+
+  $: {
+    source = `<BaseButton`;
+    if (disabled) {
+      source += ` disabled`;
+    }
+    source += `>Click me !</BaseButton>`;
+  }
 </script>
   
-<Hst.Story title="BaseButton">
+<Hst.Story title="BaseButton" {source}>
   <BaseButton {disabled} {size} on:click={event => logEvent('click', event)}>
     Click me!
   </BaseButton>
diff --git a/packages/histoire-plugin-svelte/src/client/RenderStory.svelte b/packages/histoire-plugin-svelte/src/client/RenderStory.svelte
index 8538bb04..59488219 100644
--- a/packages/histoire-plugin-svelte/src/client/RenderStory.svelte
+++ b/packages/histoire-plugin-svelte/src/client/RenderStory.svelte
@@ -1,12 +1,23 @@
 <script lang="ts">
-import { Story } from '@histoire/shared'
+import type { Story, Variant } from '@histoire/shared'
 import { getContext, setContext } from 'svelte'
 
 const story: Story = getContext('__hstStory')
+const currentVariant: Variant = getContext('__hstVariant')
 const slotName: string = getContext('__hstSlot')
 
 let index = { value: 0 }
 setContext('__hstIndex', index)
+
+export let source: string = null
+
+$: {
+  if (source != null) {
+    Object.assign(currentVariant, {
+      source,
+    })
+  }
+}
 </script>
 
 {#if slotName === 'controls'}
diff --git a/packages/histoire-plugin-svelte/src/client/RenderVariant.svelte b/packages/histoire-plugin-svelte/src/client/RenderVariant.svelte
index b77e23f2..ced79b8e 100644
--- a/packages/histoire-plugin-svelte/src/client/RenderVariant.svelte
+++ b/packages/histoire-plugin-svelte/src/client/RenderVariant.svelte
@@ -1,7 +1,6 @@
 <script lang="ts">
-import { Story } from '@histoire/shared'
+import type { Story, Variant } from '@histoire/shared'
 import { getContext } from 'svelte'
-import Variant from '../collect/Variant.svelte';
 
 const story: Story = getContext('__hstStory')
 const currentVariant: Variant = getContext('__hstVariant')
@@ -12,6 +11,16 @@ const variant = story.variants[index.value]
 index.value++
 
 $: shouldRender = currentVariant.id === variant.id
+
+export let source: string = null
+
+$: {
+  if (source != null) {
+    Object.assign(currentVariant, {
+      source,
+    })
+  }
+}
 </script>
 
 {#if shouldRender}