sunpeak 0.10.6 → 0.11.1

package/dist/{server-CnRhUNGQ.js → server-BI9Y531R.js}

@@ -5,6 +5,8 @@ import path from "node:path";
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
 import { z } from "zod";
+const FAVICON_BASE64 = "iVBORw0KGgoAAAANSUhEUgAAAIAAAACACAYAAADDPmHLAAAACXBIWXMAAAOKAAADigGnjPUfAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAf/SURBVHgB7Z1NTFRXFMfPHb4CFBxSG4gzg48mQgsmDgutM5tqNGqTmtqFbMGoqy5EWTUxERPTrlRcdaFE3DQpMamJLsD4uQFNFw6p2JYm8gJDS1KqIxSoIHN7z+MjQpF5j3nz3rtzzy+ZzDDvDYt3/vfce//3i4FJNG2XfxYmDwH4PuWchxlwjQPzA+E6IhYJEQudcRYDX/JhHhTf0PUHCXO/TYGm7dRmk/wEsGQTBVwehBg68nzsrK4/0te8710XjBKfnDzDGWsGQloY520jQz+dfOf11b7EUj/Dk/fxIxDZgJ7PfLtXywa+lV8EtO1hCn7WYRTogBYNr7ywLANQyc96/pcJlgSAdf4Mn3oCFPxsR4igqH6xl7BUBWCDDyj4KqAtxNrAyAALqX8QsoTSojkIvf8a6kJTEBTvIeM1Y7wb1wvnxD1vlv1mfCoXxqdz4JV44efhv/PFqwD640UQHxPvw0WQTYiqoAqrglz8Q/Tzz6R2BLwLBvxA+CXUioBHq8ehLjgFVkFB4Cu48HdklXtQDP3DxdA7UAK9v5UaIpEVEXPs3jczTQuLuj//JUhGpGbCCPZ+Efj1BNwOUBA9A6XQHSsTgigByUiItkAVE92+Js7ZVZAALOnH94xCQ2TMSO1eAquLrr4yaL9TIU1mYIwfYYHKT65yxpvAw2Bpb/l8BCKixMtAr8gKl++VQ/eTMvA0jHWwwOYdTzhAGDwKNuTcSvHpEheZAKsIDxNDAbwUAqBBHjVJ+Cj4SuP3AaE0JADFyQWXWDRvVAedRzSVxqdywA1cEQDastdbfvFcX94t0EM4fP5jV/wDx6uAhugY3D79lIL/FjhGcfv0z4ar6TQ5Jf5AKzhEy8ERONswBAV5SSCWU5DH4YvtL4Q3M28kOYVjAsDAf3XgTyDWJlI94agIHKkCLjYNwjHh4RPmOCVsb3xmTsA2bd7BgVAW8gEUhwSgOCQAxSEBKI6tTiA6fKcOxoHIHBduBm11DG0TAHr7ZO9mHvQJ9p3batvYgW1VAE7ZouBnHrSN8VnbhS0CwEmaZPQ4Bz5rzAR2kLYAsN5Hj59wlotNz41qN13SbgPUhiahZwDnxEs3L156cLJs70B6z52sYMUhH0BxSACKQwJQHBKA4qyrF/D4mz4yfTwGLlvfd64OrGI5A+wPJyj4HqROdMfXYw5ZFsBxcvw8y3oMOUsCQNdPliXaKoKxseoOWhLAUSr9nsfqmIylRiBuhfIsyzZLyjZwqZkVyApWHPIBFIcEoDgkAMUx1QisC05DXeUkEPJgdiNLUwI4thf35vsLCHm4fK8CWn+oTHmfqSqgVtJt2lQmusWcLWxKAFtDlP5lA8cGzLiCKQWAGzUSclJpYtAupQCCwv8n5KTWROElJ1BxyAdQHBKA4pAAFIcEoDgpnUBcg0bIy8mOD9e8nlIAuPKXkJdUAqAqQHFIAIpDAlCclG2Azt6NQGQvZAUrDlUBikMCUBwSgOKQABTHlBNIdrCcHP2uGrpia58LmjID4BHphJzETUwLTykAPNKMkJOnJhbyphQAbkr8dLgYCLnoNxkzU43AdHejJJynP15o6j5TK4Pa75bTvgCS0W8yXmQFKw75AIpDAlAcEoDiWNokCjeJ3FD0BgjvEh8rWDi/wRyWBIArTu08r4awH9wXwIoALFUBV+5WAOFt2u+UW7rfkgDQFXTybHvCGj0DpZate8uNwPM3A0B4k/Z1ZGjLAkBbuJ/GBjwHlvxUQ7+rQU6g4pAPoDgkAMUhASgOCUBx0j46NlIzQbuIukRXrMw4wyEd0hYATjyINk7QQVIOg92+CzZ4MmlXAegONqfYhICwnwu3ArZM2LWlDYDm0GUaJ3AMfNadPfas2ratEYiKjNMU8oxjpP5bm8AubHUCQ6IdcOogDRdnEqz37VyrQVaw4pAPoDgkAMUhASgOCUBx0nYCzXCxcRAaomQXW6Gz5wM4ea0KMk1OiT/QChmmu2/er47WWD/fXkXQ6Pn6ew2cwBEBIIuTSUkEa4NzLr/9MQRO4bgPgItL2hqfQyktMFnG+FQunOmsdHxjTleMIHQMr7f8SiOIC6Czd/j8R67sxuKaE4hn2kWqJ2BDIWWCrj6/kQHcgKxgxSEfQHFIAIrjYwAJIFQlgS0PXbzC4FGwoRjaKGdvAedL9nt7cy09lzPoA+5dAeB0M+wxHNszCtHqcZABXKWLEzc8HnzRBWAxFtC2N3HOroIEoH9wdO8ofLYt4TkPAfvwOE/vyr1y17p0VmHMd4RpWtg/y/MHRV/Q+tJSF8GqYX/9C4hs+Qe2hibBDXAH1e6Y37C5ZdxMM58VlTH8ILJAm8gCJ0BSMDOgICI141AbnM6YIDDgvb+/B8+Gil01b2yBsY4/9MdHDAFUaDs1H08OQhZRF5yC4MYZY18jFEjw/RkoLZwzNrnaIN5XjkVgMF9N5xifMZ3jTtvG+1iBUZcPv8iXO+ArSDJf1aj+SGeLX8ieBQjzMM4ujQw9bsbPS0ZQHsy2iks6EFkO0/N8ha2Lfy0JQNdjiSRju0kE2QzTMca6/mDJ/FtmBRt1Akt+SSLIRpgRW4zxsm9Xu3W+UcjvA3ANiCxgvuSvDD6y6mAQ3pjPXtczxi8BITXY4MtnhfWrBd+4nuofGNkAkq3CLm4EQgqMAT7Grs0Ba3tX4N+61xyatss/C/8e4pDcJX60TQhCk809zFbmA46DeiwmevgP86D4xtsNvbX4D3pTwkq/VSHSAAAAAElFTkSuQmCC";
+const FAVICON_BUFFER = Buffer.from(FAVICON_BASE64, "base64");
 function $constructor(name, initializer2, params) {
   function init(inst, def) {
     var _a;
@@ -4820,7 +4822,7 @@ union([
   ListTasksResultSchema,
   CreateTaskResultSchema
 ]);
-function readResourceHtml(distPath) {
+function readResourceHtmlProd(distPath) {
   const htmlPath = path.resolve(distPath);
   if (!fs.existsSync(htmlPath)) {
     throw new Error(
@@ -4842,12 +4844,75 @@ ${jsContents}
 </body>
 </html>`;
 }
-function createAppServer(config2) {
-  const { name = "sunpeak-app", version: version2 = "0.1.0", simulations } = config2;
+function getViteResourceHtml(srcPath) {
+  const fileName = srcPath.split("/").pop()?.replace(/-resource\.tsx$/, "") ?? "";
+  const componentName = fileName.split("-").map((part) => part.charAt(0).toUpperCase() + part.slice(1)).join("") + "Resource";
+  const devServerUrl = "http://localhost:6766";
+  const entryParams = new URLSearchParams({ src: srcPath, component: componentName });
+  const virtualModuleUrl = `${devServerUrl}/@id/virtual:sunpeak-entry?${entryParams.toString()}`;
+  return `<!DOCTYPE html>
+<html>
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+</head>
+<body>
+  <script type="module">
+    import { injectIntoGlobalHook } from "${devServerUrl}/@react-refresh";
+    injectIntoGlobalHook(window);
+    window.$RefreshReg$ = () => {};
+    window.$RefreshSig$ = () => (type) => type;
+    window.__vite_plugin_react_preamble_installed__ = true;
+  <\/script>
+  <script type="module" src="${devServerUrl}/@vite/client"><\/script>
+  <script type="module">
+    (async () => {
+      if (!navigator.permissions?.query) return;
+      const protocol = window.location.protocol;
+      if (protocol !== 'http:' && protocol !== 'https:') return;
+      const host = window.location.hostname;
+      const isLoopback = host === 'localhost' || /^127\\.(\\d{1,3})\\.(\\d{1,3})\\.(\\d{1,3})$/.test(host) || host === '::1';
+      if (isLoopback) return;
+      try {
+        const status = await navigator.permissions.query({ name: "local-network-access" });
+        if (status.state === "denied") {
+          document.getElementById('root').innerHTML = '<div style="background:#fef2f2;border:2px solid #ef4444;border-radius:8px;padding:16px;text-align:center;font-family:system-ui,sans-serif;"><div style="color:#ef4444;font-size:18px;font-weight:600;margin-bottom:8px;">Local network access permission is denied</div><div style="color:#ef4444;font-size:14px;">Please enable it in your browser settings. <a href="https://developer.chrome.com/blog/local-network-access" target="_blank" style="color:#ef4444;text-decoration:underline;">Learn more</a></div></div>';
+        }
+      } catch (e) {}
+    })();
+  <\/script>
+  <div id="root"></div>
+  <script type="module" src="${virtualModuleUrl}"><\/script>
+</body>
+</html>`;
+}
+function getResourceHtml(simulation, viteMode) {
+  if (viteMode && simulation.srcPath) {
+    return getViteResourceHtml(simulation.srcPath);
+  }
+  return readResourceHtmlProd(simulation.distPath);
+}
+const DEV_SERVER_URL = "http://localhost:6766";
+const HMR_WS_URL = "ws://localhost:24678";
+function injectViteCSP(existingMeta) {
+  const meta = existingMeta ?? {};
+  const widgetCSP = meta["openai/widgetCSP"] ?? {};
+  const existingResourceDomains = widgetCSP["resource_domains"] ?? [];
+  const resourceDomains = existingResourceDomains.includes(DEV_SERVER_URL) ? existingResourceDomains : [...existingResourceDomains, DEV_SERVER_URL];
+  const existingConnectDomains = widgetCSP["connect_domains"] ?? [];
+  const connectDomains = existingConnectDomains.includes(HMR_WS_URL) ? existingConnectDomains : [...existingConnectDomains, HMR_WS_URL];
+  return {
+    ...meta,
+    "openai/widgetCSP": {
+      ...widgetCSP,
+      resource_domains: resourceDomains,
+      connect_domains: connectDomains
+    }
+  };
+}
+function createAppServer(config2, simulations, viteMode) {
+  const { name = "sunpeak-app", version: version2 = "0.1.0" } = config2;
   const toolInputParser = z.object({});
-  const resourceContentMap = new Map(
-    simulations.map((simulation) => [simulation.tool.name, readResourceHtml(simulation.distPath)])
-  );
   const devTimestamp = Date.now().toString(36);
   const resources = simulations.map((simulation) => {
     const resource = simulation.resource;
@@ -4871,11 +4936,13 @@ function createAppServer(config2) {
       simulation.callToolResult ?? { structuredContent: null, _meta: {} }
     ])
   );
-  const resourceMap = new Map(
+  const uriToSimulation = new Map(
+    resources.map((resource, index) => [resource.uri, simulations[index]])
+  );
+  const resourceMetaMap = new Map(
     resources.map((resource, index) => [
       resource.uri,
       {
-        content: resourceContentMap.get(simulations[index].tool.name),
         resource,
         _meta: simulations[index].resource._meta
       }
@@ -4894,24 +4961,35 @@ function createAppServer(config2) {
     }
   );
   server.setRequestHandler(ListResourcesRequestSchema, async (_request) => {
-    console.log(`[MCP] ListResources → ${resources.length} resource(s)`);
+    console.log(
+      `[MCP] ListResources → ${resources.length} resource(s)${viteMode ? " (vite mode)" : ""}`
+    );
+    if (viteMode) {
+      const resourcesWithCsp = resources.map((resource) => ({
+        ...resource,
+        _meta: injectViteCSP(resource._meta)
+      }));
+      return { resources: resourcesWithCsp };
+    }
     return { resources };
   });
   server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
     const requestedUri = request.params.uri;
-    const resourceData = resourceMap.get(requestedUri);
-    if (!resourceData) {
+    const simulation = uriToSimulation.get(requestedUri);
+    const meta = resourceMetaMap.get(requestedUri);
+    if (!simulation || !meta) {
       throw new Error(`Unknown resource: ${requestedUri}`);
     }
-    const sizeKB = (resourceData.content.length / 1024).toFixed(1);
-    console.log(`[MCP] ReadResource: ${requestedUri} → ${sizeKB}KB`);
+    const content = getResourceHtml(simulation, viteMode);
+    const sizeKB = (content.length / 1024).toFixed(1);
+    console.log(`[MCP] ReadResource: ${requestedUri} → ${sizeKB}KB${viteMode ? " (vite)" : ""}`);
     return {
       contents: [
         {
-          uri: resourceData.resource.uri,
-          mimeType: resourceData.resource.mimeType,
-          text: resourceData.content,
-          _meta: resourceData._meta
+          uri: meta.resource.uri,
+          mimeType: meta.resource.mimeType,
+          text: content,
+          _meta: viteMode ? injectViteCSP(meta._meta) : meta._meta
         }
       ]
     };
@@ -4950,9 +5028,9 @@ function createAppServer(config2) {
 const sessions = /* @__PURE__ */ new Map();
 const ssePath = "/mcp";
 const postPath = "/mcp/messages";
-async function handleSseRequest(res, config2) {
+async function handleSseRequest(res, config2, simulations, viteMode) {
   res.setHeader("Access-Control-Allow-Origin", "*");
-  const server = createAppServer(config2);
+  const server = createAppServer(config2, simulations, viteMode);
   const transport = new SSEServerTransport(postPath, res);
   const sessionId = transport.sessionId;
   sessions.set(sessionId, { server, transport });
@@ -4978,7 +5056,7 @@ async function handleSseRequest(res, config2) {
 }
 async function handlePostMessage(req, res, url) {
   res.setHeader("Access-Control-Allow-Origin", "*");
-  res.setHeader("Access-Control-Allow-Headers", "content-type");
+  res.setHeader("Access-Control-Allow-Headers", "content-type, ngrok-skip-browser-warning");
   const sessionId = url.searchParams.get("sessionId");
   if (!sessionId) {
     res.writeHead(400).end("Missing sessionId query parameter");
@@ -5001,32 +5079,74 @@ async function handlePostMessage(req, res, url) {
 function runMCPServer(config2) {
   const portEnv = Number(process.env.PORT ?? 6766);
   const port = config2.port ?? (Number.isFinite(portEnv) ? portEnv : 6766);
+  const { simulations } = config2;
+  const viteServer = config2.viteServer;
+  const viteMode = !!viteServer;
   const httpServer = createServer(async (req, res) => {
     if (!req.url) {
       res.writeHead(400).end("Missing URL");
       return;
     }
     const url = new URL$1(req.url, `http://${req.headers.host ?? "localhost"}`);
-    console.log(`[HTTP] ${req.method} ${url.pathname}`);
-    if (req.method === "OPTIONS" && (url.pathname === ssePath || url.pathname === postPath)) {
+    if (req.method === "OPTIONS") {
       res.writeHead(204, {
         "Access-Control-Allow-Origin": "*",
         "Access-Control-Allow-Methods": "GET, POST, OPTIONS",
-        "Access-Control-Allow-Headers": "content-type"
+        "Access-Control-Allow-Headers": "content-type, ngrok-skip-browser-warning"
       });
       res.end();
       return;
     }
+    if (req.method === "GET" && url.pathname === "/") {
+      res.writeHead(200, {
+        "Content-Type": "text/html",
+        "Access-Control-Allow-Origin": "*"
+      });
+      res.end(`<!DOCTYPE html>
+<html>
+<head>
+<meta charset="UTF-8" />
+<link rel="icon" type="image/png" href="/favicon.ico" />
+<title>Sunpeak MCP Server</title>
+</head>
+<body><h1>Sunpeak MCP Server</h1><p>Connect via <a href="/mcp">/mcp</a></p></body>
+</html>`);
+      return;
+    }
+    if (req.method === "GET" && url.pathname === "/favicon.ico") {
+      res.writeHead(200, {
+        "Content-Type": "image/png",
+        "Content-Length": FAVICON_BUFFER.length,
+        "Cache-Control": "public, max-age=86400",
+        "Access-Control-Allow-Origin": "*"
+      });
+      res.end(FAVICON_BUFFER);
+      return;
+    }
     if (req.method === "GET" && url.pathname === ssePath) {
-      await handleSseRequest(res, config2);
+      console.log(`[HTTP] ${req.method} ${url.pathname}`);
+      await handleSseRequest(res, config2, simulations, viteMode);
       return;
     }
     if (req.method === "POST" && url.pathname === postPath) {
+      console.log(`[HTTP] ${req.method} ${url.pathname}`);
       await handlePostMessage(req, res, url);
       return;
     }
+    if (viteServer) {
+      res.setHeader("Access-Control-Allow-Origin", "*");
+      res.setHeader("Access-Control-Allow-Headers", "content-type, ngrok-skip-browser-warning");
+      viteServer.middlewares(req, res);
+      return;
+    }
+    console.log(`[HTTP] ${req.method} ${url.pathname} → 404`);
     res.writeHead(404).end("Not Found");
   });
+  if (viteServer) {
+    httpServer.on("upgrade", (request, socket, head) => {
+      viteServer.ws.handleUpgrade(request, socket, head);
+    });
+  }
   httpServer.on("clientError", (err, socket) => {
     console.error("HTTP client error", err);
     socket.end("HTTP/1.1 400 Bad Request\r\n\r\n");
@@ -5034,9 +5154,11 @@ function runMCPServer(config2) {
   httpServer.listen(port, () => {
     console.log(`Sunpeak MCP server listening on http://localhost:${port}`);
     console.log(`  SSE stream: GET http://localhost:${port}${ssePath}`);
-    console.log(`  Message post endpoint: POST http://localhost:${port}${postPath}?sessionId=...`);
+    if (viteMode) {
+      console.log(`  Vite HMR: enabled (source files served with hot reload)`);
+    }
   });
-  const shutdown = () => {
+  const shutdown = async () => {
     console.log("\nShutting down MCP server...");
     httpServer.close(() => {
       console.log("MCP server closed");
@@ -5047,10 +5169,12 @@ function runMCPServer(config2) {
       process.exit(1);
     }, 5e3);
   };
-  process.on("SIGTERM", shutdown);
-  process.on("SIGINT", shutdown);
+  process.on("SIGTERM", () => void shutdown());
+  process.on("SIGINT", () => void shutdown());
 }
 export {
+  FAVICON_BASE64 as F,
+  FAVICON_BUFFER as a,
   runMCPServer as r
 };
-//# sourceMappingURL=server-CnRhUNGQ.js.map
+//# sourceMappingURL=server-BI9Y531R.js.map