fix(vue): codegen setup directives

histoire-dev · Apr 9, 2024 · 92f91cc · 92f91cc
1 parent a6a9b81
commit 92f91cc
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/packages/histoire-plugin-vue/src/client/codegen.ts b/packages/histoire-plugin-vue/src/client/codegen.ts
@@ -101,6 +101,15 @@ async function printVNode (vnode: VNode, propsOverrides: Record<string, any> = n
             }
             if (dirName) break
           }
+          // Check in setup state
+          if (!dirName) {
+            for (const key in target.setupState) {
+              if (target.setupState[key] === dir.dir) {
+                dirName = key.replace(/^v(\w)/, (_, p1) => p1.toLowerCase())
+                break
+              }
+            }
+          }
           if (dirName) {
             genDirective(dirName, dir)
           }