@tpitre/story-ui 3.0.0 → 3.1.0

package/dist/templates/StoryUI/StoryUIPanel.d.ts

@@ -1,3 +1,8 @@
+declare global {
+    interface Window {
+        __STORY_UI_GENERATED_CODE__?: Record<string, string>;
+    }
+}
 declare function StoryUIPanel(): import("react/jsx-runtime").JSX.Element;
 export default StoryUIPanel;
 export { StoryUIPanel };

package/dist/templates/StoryUI/StoryUIPanel.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"StoryUIPanel.d.ts","sourceRoot":"","sources":["../../../templates/StoryUI/StoryUIPanel.tsx"],"names":[],"mappings":"AA85CA,iBAAS,YAAY,4CA8sCpB;AAED,eAAe,YAAY,CAAC;AAC5B,OAAO,EAAE,YAAY,EAAE,CAAC"}
+{"version":3,"file":"StoryUIPanel.d.ts","sourceRoot":"","sources":["../../../templates/StoryUI/StoryUIPanel.tsx"],"names":[],"mappings":"AAsSA,OAAO,CAAC,MAAM,CAAC;IACb,UAAU,MAAM;QACd,2BAA2B,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KACtD;CACF;AA6oCD,iBAAS,YAAY,4CA8sCpB;AAED,eAAe,YAAY,CAAC;AAC5B,OAAO,EAAE,YAAY,EAAE,CAAC"}

package/dist/templates/StoryUI/StoryUIPanel.js

@@ -121,12 +121,27 @@ const titleToStoryPath = (title) => {
     const kebabTitle = title.toLowerCase().replace(/[^a-z0-9]+/g, '-').replace(/^-|-$/g, '');
     return `generated-${kebabTitle}--default`;
 };
+// Helper to store generated code for the Source Code panel to display
+const storeGeneratedCode = (title, code) => {
+    const storyPath = titleToStoryPath(title);
+    const topWindow = window.top || window;
+    // Store code in the top window so it's accessible from manager frame
+    if (!topWindow.__STORY_UI_GENERATED_CODE__) {
+        topWindow.__STORY_UI_GENERATED_CODE__ = {};
+    }
+    topWindow.__STORY_UI_GENERATED_CODE__[storyPath] = code;
+    console.log(`[Story UI] Stored code for story "${storyPath}" in window cache`);
+};
 // Helper to navigate to a newly created story after generation completes
 // In dev mode with HMR, this prevents the "Couldn't find story after HMR" error
 // In all modes, this provides a better UX by auto-navigating to the new story
-const navigateToNewStory = (title, delayMs = 4000) => {
+const navigateToNewStory = (title, code, delayMs = 1500) => {
     const storyPath = titleToStoryPath(title);
     console.log(`[Story UI] Will navigate to story "${storyPath}" in ${delayMs}ms...`);
+    // Store the code for the Source Code panel if provided
+    if (code) {
+        storeGeneratedCode(title, code);
+    }
     setTimeout(() => {
         // Navigate the TOP window (parent Storybook UI), not the iframe
         // The Story UI panel runs inside an iframe, so we need window.top to escape it
@@ -1438,7 +1453,7 @@ function StoryUIPanel() {
             // Auto-navigate to the newly created story after HMR processes the file
             // This prevents the "Couldn't find story after HMR" error by refreshing
             // after the file system has been updated and HMR has processed the change
-            navigateToNewStory(chatTitle);
+            navigateToNewStory(chatTitle, completion.code);
         }
     }, [activeChatId, activeTitle, conversation.length]);
     const handleSend = async (e) => {
@@ -1631,7 +1646,7 @@ function StoryUIPanel() {
                         saveChats(chats);
                         setRecentChats(chats);
                         // Auto-navigate to the newly created story
-                        navigateToNewStory(chatTitle);
+                        navigateToNewStory(chatTitle, data.code);
                     }
                 }
                 catch (fallbackErr) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tpitre/story-ui",
-  "version": "3.0.0",
+  "version": "3.1.0",
   "description": "AI-powered Storybook story generator with dynamic component discovery",
   "type": "module",
   "main": "dist/index.js",

package/templates/StoryUI/StoryUIPanel.tsx

@@ -291,13 +291,38 @@ const titleToStoryPath = (title: string): string => {
   return `generated-${kebabTitle}--default`;
 };
 
+// Extend window to include our code cache for the Source Code panel
+declare global {
+  interface Window {
+    __STORY_UI_GENERATED_CODE__?: Record<string, string>;
+  }
+}
+
+// Helper to store generated code for the Source Code panel to display
+const storeGeneratedCode = (title: string, code: string) => {
+  const storyPath = titleToStoryPath(title);
+  const topWindow = window.top || window;
+
+  // Store code in the top window so it's accessible from manager frame
+  if (!topWindow.__STORY_UI_GENERATED_CODE__) {
+    topWindow.__STORY_UI_GENERATED_CODE__ = {};
+  }
+  topWindow.__STORY_UI_GENERATED_CODE__[storyPath] = code;
+  console.log(`[Story UI] Stored code for story "${storyPath}" in window cache`);
+};
+
 // Helper to navigate to a newly created story after generation completes
 // In dev mode with HMR, this prevents the "Couldn't find story after HMR" error
 // In all modes, this provides a better UX by auto-navigating to the new story
-const navigateToNewStory = (title: string, delayMs: number = 4000) => {
+const navigateToNewStory = (title: string, code?: string, delayMs: number = 1500) => {
   const storyPath = titleToStoryPath(title);
   console.log(`[Story UI] Will navigate to story "${storyPath}" in ${delayMs}ms...`);
 
+  // Store the code for the Source Code panel if provided
+  if (code) {
+    storeGeneratedCode(title, code);
+  }
+
   setTimeout(() => {
     // Navigate the TOP window (parent Storybook UI), not the iframe
     // The Story UI panel runs inside an iframe, so we need window.top to escape it
@@ -1927,7 +1952,7 @@ function StoryUIPanel() {
       // Auto-navigate to the newly created story after HMR processes the file
       // This prevents the "Couldn't find story after HMR" error by refreshing
       // after the file system has been updated and HMR has processed the change
-      navigateToNewStory(chatTitle);
+      navigateToNewStory(chatTitle, completion.code);
     }
   }, [activeChatId, activeTitle, conversation.length]);
 
@@ -2137,7 +2162,7 @@ function StoryUIPanel() {
             setRecentChats(chats);
 
             // Auto-navigate to the newly created story
-            navigateToNewStory(chatTitle);
+            navigateToNewStory(chatTitle, data.code);
           }
         } catch (fallbackErr: unknown) {
           const errorMessage = fallbackErr instanceof Error ? fallbackErr.message : 'Unknown error';