From 92f91cc7bb17b51e73f5a4af4b224a53a6236b51 Mon Sep 17 00:00:00 2001
From: Guillaume Chau <guillaume.b.chau@gmail.com>
Date: Tue, 9 Apr 2024 17:27:25 +0200
Subject: [PATCH] fix(vue): codegen setup directives

---
 packages/histoire-plugin-vue/src/client/codegen.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/packages/histoire-plugin-vue/src/client/codegen.ts b/packages/histoire-plugin-vue/src/client/codegen.ts
index df0d64ce..246b4fed 100644
--- 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)
           }