From 5f4f46b280f93617e42f91d9e02367e08b6e49db Mon Sep 17 00:00:00 2001
From: Jeppe Reinhold <jeppe@reinhold.is>
Date: Tue, 11 Jun 2024 09:29:35 +0200
Subject: [PATCH] improve snippet functions in code emit

---
 examples/Button.stories.svelte | 3 +--
 src/runtime/emit-code.ts       | 4 ++++
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/examples/Button.stories.svelte b/examples/Button.stories.svelte
index ecd85cc..ac2329c 100644
--- a/examples/Button.stories.svelte
+++ b/examples/Button.stories.svelte
@@ -1,8 +1,7 @@
 <script context="module" lang="ts">
-  import { defineMeta } from '@storybook/addon-svelte-csf';
+  import { defineMeta, setTemplate } from '@storybook/addon-svelte-csf';
 
   import Button from './components/Button.svelte';
-  import { setTemplate } from '#index';
 
   /**
    * These are the stories for the `Button` component.
diff --git a/src/runtime/emit-code.ts b/src/runtime/emit-code.ts
index 1144d8c..9dcc3e9 100644
--- a/src/runtime/emit-code.ts
+++ b/src/runtime/emit-code.ts
@@ -100,6 +100,10 @@ const getFunctionName = (fn: Function & { getMockName?: () => string }) => {
  * convert a value to a stringified version
  */
 const valueToString = (value: any): string => {
+  if (value[Symbol.for('svelte.snippet')]) {
+    return 'snippet';
+  }
+
   if (typeof value === 'function') {
     return getFunctionName(value);
   }