improve snippet functions in code emit

storybookjs · Jun 11, 2024 · 5f4f46b · 5f4f46b
1 parent d7c4e87
commit 5f4f46b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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);
   }