@tpitre/story-ui 3.3.0 → 3.4.1

package/dist/mcp-server/index.js

@@ -1,6 +1,7 @@
 import dotenv from 'dotenv';
 import path from 'path';
 import { fileURLToPath } from 'url';
+import { createProxyMiddleware } from 'http-proxy-middleware';
 // Get __dirname equivalent for ES modules
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
@@ -199,6 +200,37 @@ const config = loadUserConfig();
 // Initialize URL redirect service
 const redirectService = new UrlRedirectService(process.cwd());
 const PORT = parseInt(process.env.PORT || '4001', 10);
+// Storybook proxy configuration for production deployment
+// When STORYBOOK_PROXY_ENABLED is set, proxy all non-API requests to Storybook dev server
+const storybookProxyPort = process.env.STORYBOOK_PROXY_PORT || '6006';
+const storybookProxyEnabled = process.env.STORYBOOK_PROXY_ENABLED === 'true';
+if (storybookProxyEnabled) {
+    console.log(`📖 Storybook proxy enabled - forwarding to localhost:${storybookProxyPort}`);
+    // Proxy all requests that don't match API routes to Storybook
+    // This must be added AFTER all API routes so they take precedence
+    app.use('/', createProxyMiddleware({
+        target: `http://localhost:${storybookProxyPort}`,
+        changeOrigin: true,
+        ws: true, // Enable WebSocket proxying for HMR
+        logger: console,
+        // Don't proxy API routes
+        pathFilter: (path) => {
+            const isApiRoute = path.startsWith('/mcp') ||
+                path.startsWith('/story-ui') ||
+                path.startsWith('/mcp-remote');
+            return !isApiRoute;
+        },
+        on: {
+            error: (err, req, res) => {
+                console.error('Storybook proxy error:', err.message);
+                if (res && typeof res.writeHead === 'function' && !res.headersSent) {
+                    res.writeHead(502, { 'Content-Type': 'text/plain' });
+                    res.end('Storybook is starting up, please wait...');
+                }
+            }
+        }
+    }));
+}
 // Start server
 app.listen(PORT, () => {
     console.log(`MCP server running on port ${PORT}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tpitre/story-ui",
-  "version": "3.3.0",
+  "version": "3.4.1",
   "description": "AI-powered Storybook story generator with dynamic component discovery",
   "type": "module",
   "main": "dist/index.js",
@@ -83,6 +83,7 @@
     "dotenv": "^16.3.1",
     "express": "^4.18.2",
     "glob": "^11.0.3",
+    "http-proxy-middleware": "^3.0.0",
     "inquirer": "^9.2.0",
     "node-fetch": "^2.6.7",
     "pg": "^8.16.3",