Fix destructured mount in play

Addressed storybookjs#270
SteveALee · Jan 25, 2025 · def680f · def680f
1 parent 06dee29
commit def680f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/runtime/create-runtime-stories.ts b/src/runtime/create-runtime-stories.ts
@@ -68,7 +68,10 @@ export const createRuntimeStories = (Stories: Component, meta: Meta<Cmp>) => {
        * The 'play' function should be delegated to the real play Story function
        * in order to be run into the component scope.
        */
-      storyObj.play = (storyContext) => {
+      function params(fn) { return fn.toString().match(/[^(]*\(([^)]*)/)?.slice(1) ?? [] }
+      const isMounting = params(storyObj.play).filter((p) => /\{\s*mount\s*\}/.test(p) ).length != 0
+      storyObj.play = isMounting ? function ({ mount }) { return playDelegator(arguments[0]) } : (storyContext) => playDelegator(storyContext);
+      function playDelegator(storyContext) {
         const delegate = storyContext.playFunction?.__play;
 
         if (delegate) {