sunpeak 0.14.3 → 0.15.4

package/dist/mcp/index.d.ts

@@ -1,3 +1,5 @@
-export { runMCPServer, type MCPServerConfig } from './server.js';
+export { runMCPServer, type MCPServerConfig, type MCPServerHandle } from './server.js';
 export type { SimulationWithDist } from './types.js';
 export { FAVICON_BASE64, FAVICON_BUFFER } from './favicon.js';
+export { registerAppTool, registerAppResource, getUiCapability, EXTENSION_ID, RESOURCE_URI_META_KEY, RESOURCE_MIME_TYPE, } from '@modelcontextprotocol/ext-apps/server';
+export type { McpUiAppToolConfig, McpUiAppResourceConfig, ToolConfig, ToolCallback, ReadResourceCallback, ResourceMetadata, } from '@modelcontextprotocol/ext-apps/server';

package/dist/mcp/index.js

@@ -2,12 +2,14 @@ import { createServer } from "node:http";
 import { URL as URL$1 } from "node:url";
 import fs from "node:fs";
 import path from "node:path";
-import { P as Protocol, o as LoggingLevelSchema, p as InitializeRequestSchema, q as InitializedNotificationSchema, S as SetLevelRequestSchema, r as mergeCapabilities, s as getObjectShape, t as isZ4Schema, u as LATEST_PROTOCOL_VERSION, E as EmptyResultSchema, v as CreateMessageResultWithToolsSchema, w as CreateMessageResultSchema, x as ElicitResultSchema, M as McpError, y as ErrorCode, z as ListRootsResultSchema, A as safeParse, c as CallToolRequestSchema, B as CreateTaskResultSchema, a as CallToolResultSchema, D as SUPPORTED_PROTOCOL_VERSIONS, F as ListToolsRequestSchema, G as normalizeObjectSchema, H as toJsonSchemaCompat, K as safeParseAsync, N as getParseErrorMessage, O as CompleteRequestSchema, Q as assertCompleteRequestResourceTemplate, U as assertCompleteRequestPrompt, d as ListResourcesRequestSchema, e as ListResourceTemplatesRequestSchema, f as ReadResourceRequestSchema, g as ListPromptsRequestSchema, V as GetPromptRequestSchema, W as objectFromShape, X as getLiteralValue, Y as getSchemaDescription, Z as isSchemaOptional, C as ContentBlockSchema, T as ToolSchema, R as RequestIdSchema, I as ImplementationSchema } from "../protocol-BQCnIrc9.js";
-import { g as getDefaultExportFromCjs } from "../_commonjsHelpers-DWwsNxpa.js";
+import { P as Protocol, p as LoggingLevelSchema, q as InitializeRequestSchema, r as InitializedNotificationSchema, S as SetLevelRequestSchema, s as mergeCapabilities, t as getObjectShape, u as isZ4Schema, v as LATEST_PROTOCOL_VERSION, E as EmptyResultSchema, w as CreateMessageResultWithToolsSchema, x as CreateMessageResultSchema, y as ElicitResultSchema, M as McpError, z as ErrorCode, A as ListRootsResultSchema, B as safeParse, c as CallToolRequestSchema, D as CreateTaskResultSchema, a as CallToolResultSchema, F as SUPPORTED_PROTOCOL_VERSIONS, L as ListToolsRequestSchema, G as normalizeObjectSchema, H as toJsonSchemaCompat, K as safeParseAsync, N as getParseErrorMessage, O as CompleteRequestSchema, Q as assertCompleteRequestResourceTemplate, U as assertCompleteRequestPrompt, e as ListResourcesRequestSchema, f as ListResourceTemplatesRequestSchema, g as ReadResourceRequestSchema, h as ListPromptsRequestSchema, V as GetPromptRequestSchema, W as objectFromShape, X as getLiteralValue, Y as getSchemaDescription, Z as isSchemaOptional, C as ContentBlockSchema, T as ToolSchema, R as RequestIdSchema, I as ImplementationSchema } from "../protocol-DFbsCx7E.js";
 import { ZodOptional } from "zod";
 import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
 const FAVICON_BASE64 = "iVBORw0KGgoAAAANSUhEUgAAAIAAAACACAYAAADDPmHLAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAAXNSR0IArs4c6QAAAARnQU1BAACxjwv8YQUAAAiLSURBVHgB7Z1NbBNHGIZfJ4GWJIAjfhKBKDYglaolGBAqUAncqNAbTTijErhUaiVIqFROJIWeeiiEnsoFEokzGE4VVJDmAFSU4oRDW/Fji6qIIFAMJKD84X7fOg5pEnt3s/Z6x/M9MPbYXhOy37vz887sjA+W2OwDZvuBl/VAyVbgdQhIBgCfH4IHSCYoFlHKxCl1AeURoJ/eu5k0+6Yv+8cceA706H56bpSAK0U7UHoUeDsOdGYUQhYBhKuAgRbKNEFQGN+RVPptWhFkEMCHQSrmL1MmAKEYiFPVXUciiE3+YBoBSPCLlPh0IpgkAAl+kROfLIIJAghTfuA+iiz4vikZiySNv8VIHBheD0T7+EXZm/f7W+ksBaAwvrEHf8UoAjWDCAVfYXn1oJH8lSMILB6i51HjWH72V4wY+cRAGRL9pak0kErxR2/hweO3EO+l9Hg2ovfKUz9EfWEEgFncuG/mF2PXxSYq+kfvQzHSAQ9QgMNr+rG19jnCtS+M17mGRRK9N8cQQtfteejsmWsIJamsGkop5tfjYwLYeJoeGqEA6aBzoHdu6kP9lkReAm4FFsGFa1XovD3XEIZiYjgB3GiiUxkic2dWHzyOb0LQG3c8HS++vQJXFT9GqhEhQcR7Z6sgBnIKh4N0Wjc20ovT8ChcnzfueILPNvdhGwlABc5f86Pj4iISg+eN070+Lxf/6eDzs4pwSdB+aSG8S0k7C+AW5UIQdCRaAjF9dCbAApARPn3xl0DQmjIUiGBNYfruXiTRX4Y+ciELgesC4P78gfpeHP/iAYQU7CHsOxZEZ/dc121m7gW49jOryH8/23LHMHSEqRw5s4TSUldFQOXO0m/hAsGaIVw9/idCK19CmB6+MObTRXLx5ny4hSslwLpVr3D28J2CefaqweMKdYdWu9IuyLsAWNVnW+96zrv3OtwuqPtmteEm5pO8C6Bx+1MIM4ODzyOO+cTVRqDgPcQI0hwRgOaIADQnZ04gT9UKrxWDJ9/wPMRbd8uRK3ImgNbdDyn9CyH/cPfwSo56BzmpAvZufyLBd5FTX8cMWz0XOBZAsHoILXT1C+7BjiqPqeQCxwJooStfLF73YYe1ueERnOLICGqkov/UwRiEwsANwvVffeDILnYkgNCKl+O3WgmFgYPP4wYzRaxgzREjSHNEAJojAtAc205gw0cJ+Mul4edFovfLceveHFvfsdUIZNPnfkc3BG/CAlj35fu2vmOrCmgRu9fTcLf8Y5szri0LIEBXPxs/grexe5FaFkC49jkE78MWsZ3b6S3fF3DqYBw1VcMQvM/gsA+dPfMsHWu5ESgDPmph1R4WK1hzxAjSHBGA5ogANMfUCm7YklB2lS7d4YmjZpNFTBuBsY4e6QEoyolz1Wg6+U7WY7JWATzbR4KvLmstrMWQVQDsLQvqEnIsAFnNQ2mMZfNNSvCsjcCta14ovBy6wIRWvsrqCooTqDniA2iOCEBzRACak7URaHejLcF7mDXwMgpApoAVB88GSnGcHMFMZBHAoNzzXwQkTAQgbYAix2wgTwSgOSIAzTGdFCp2sNrM2Apm2ApWZcNmYXpORKqNLWgykVUAPJtExgPUpsNk8+qsbYBsyhHUwGzLmZLsX66EoC68ZIwZWQXAkwl5NElQEysluGkvIHK1CssXD0FQj/NXzTeGFytYc8QI0hwRgOaIADTH8lKxPvGElcHO7XyWBcB71MhCEWpw/pofbedqLB1rWQDd9ypklrAiHDmz1PKxltsAEQt9SqHwxMi8s7PjuGUBsK14xeLqk0LhOHpmia3jbfUC7P7jgruwdc/OrR1sbRjBRctpk+FFoXD0UCltd+xGrGDNESNIc0QAmiMC0Bzbu4ZNRiziwuJ0FTdHAmiq78WBhl4IhaPh6CpHczcdCaD9l4XYTwIIynKyBYG75E4n7jpqA/AdxPt+CEJwH75z++BPy+AUx41ANofaItZGnoTcwNX+ru9WkenjuAmXm15AMynx1t1y4z8mKf+JLflc3bORMyeQF5Vq3P4UQn5hq7cty6JPdhErWHPECNIcEYDmiAA0x3k/wgTf+INgFzc268i7AFp3PzS2NBcN2KOvvwwHTy5De54n4FjePXymsFHkoz+80oiIwBoc/LpD7+Ln3+cj3+RdAAyL4Bn9Up9ueCajhyawxbul+T389c8cuIGrPgDvQnb28F0Ea2TwaDqudM/DvmMBy1u/5wLXjSB2DC9//3dqOzMpDVIkU4s5NZvs8pkPCuYE1suexOPwer52bubIJWIFa44YQZrDAkhA0JUECyAOQVfi5AT6uqkZGoIH4Uaiv3JESTs5beO62aWbAVESQLKTMnvgQXjyQ/2WPnz+yVOEa58rYSJx3BNmenVcWoDz16q8LoBf6ZSG/MCsGL3w9AIAqW3sHmLbmhdG3ktiSI49dPbMw4XrfrRfXJCT+Xr5Z7hq7DRubKOHA1CEMI0r7OGVzMfEwNWD27qYGPSu27ONfnyh+vIz5DRwY9/YedsUAEZjUBC2l1kQOzcljLy/YiT1QQ5FkRx/SNXpnRTwLgp85Op8Ra706SgNAtfjE87RhuPUK2yC4rAIeNvb0IqXqOX8oiEEaOxhXBgWMdZJpro8Gptj3HdvBL67UuGA/w8q8W80c2aCALgtMPsPknrR3unB1YW/cjSjBZ3eYs3jDTeH+KikL19PlZfh/0wqJbkqeH25mEWgNxz8kjou+tPvTLKC+YPkrtSBQnHBMeXYvgm+8e70B0tJUFxMvfLTZBgM4gOHqJ54zd1DGS1UF45dW6rOnxp8xkJPyegituKNWyjTOLyNYUZSilBX72imwKexEcww9RIG6imzjdJaSgF43D3UCA54nAr0KJXaXUBFJN3KN+M/cngKDJvF2pMAAAAASUVORK5CYII=";
 const FAVICON_BUFFER = Buffer.from(FAVICON_BASE64, "base64");
+function getDefaultExportFromCjs(x2) {
+  return x2 && x2.__esModule && Object.prototype.hasOwnProperty.call(x2, "default") ? x2["default"] : x2;
+}
 var ajv = { exports: {} };
 var core$1 = {};
 var validate = {};
@@ -14199,6 +14201,18 @@ function hk(r, i, o, t) {
 function ak(r, i, o, t, n) {
   r.registerResource(i, o, { mimeType: EI, ...t }, n);
 }
+var C6 = "io.modelcontextprotocol/ui";
+function pk(r) {
+  if (!r) return;
+  return r.extensions?.[C6];
+}
+const LOCAL_DEV_SERVER_URL = "http://localhost:8000";
+const LOCAL_HMR_WS_URL = "ws://localhost:24678";
+function needsProdBuild(headers) {
+  const ua = headers["user-agent"];
+  const userAgent = typeof ua === "string" ? ua : Array.isArray(ua) ? ua[0] : "";
+  return /claude/i.test(userAgent ?? "");
+}
 function readResourceHtmlProd(distPath) {
   const htmlPath = path.resolve(distPath);
   if (!fs.existsSync(htmlPath)) {
@@ -14211,7 +14225,7 @@ function readResourceHtmlProd(distPath) {
 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:8000";
+  const devServerUrl = LOCAL_DEV_SERVER_URL;
   const entryParams = new URLSearchParams({ src: srcPath, component: componentName });
   const virtualModuleUrl = `${devServerUrl}/@id/virtual:sunpeak-entry?${entryParams.toString()}`;
   return `<!DOCTYPE html>
@@ -14250,38 +14264,44 @@ function getViteResourceHtml(srcPath) {
 </body>
 </html>`;
 }
-function getResourceHtml(simulation, viteMode) {
-  if (viteMode && simulation.srcPath) {
+function getResourceHtml(simulation, viteMode, prodBuild) {
+  if (viteMode && simulation.srcPath && !prodBuild) {
     return getViteResourceHtml(simulation.srcPath);
   }
   return readResourceHtmlProd(simulation.distPath);
 }
-const DEV_SERVER_URL = "http://localhost:8000";
-const HMR_WS_URL = "ws://localhost:24678";
 function injectViteCSP(existingMeta) {
   const meta = existingMeta ?? {};
   const ui2 = meta.ui ?? {};
   const csp = ui2.csp ?? {};
   const existingResourceDomains = csp.resourceDomains ?? [];
-  const resourceDomains = existingResourceDomains.includes(DEV_SERVER_URL) ? existingResourceDomains : [...existingResourceDomains, DEV_SERVER_URL];
+  const resourceDomains = existingResourceDomains.includes(LOCAL_DEV_SERVER_URL) ? existingResourceDomains : [...existingResourceDomains, LOCAL_DEV_SERVER_URL];
   const existingConnectDomains = csp.connectDomains ?? [];
-  const connectDomains = existingConnectDomains.includes(HMR_WS_URL) ? existingConnectDomains : [...existingConnectDomains, HMR_WS_URL];
+  const connectDomains = [...existingConnectDomains];
+  if (!connectDomains.includes(LOCAL_DEV_SERVER_URL)) connectDomains.push(LOCAL_DEV_SERVER_URL);
+  if (!connectDomains.includes(LOCAL_HMR_WS_URL)) connectDomains.push(LOCAL_HMR_WS_URL);
   return {
     ...meta,
     ui: {
       ...ui2,
-      csp: {
-        ...csp,
-        resourceDomains,
-        connectDomains
-      }
+      csp: { ...csp, resourceDomains, connectDomains }
     }
   };
 }
 function createAppServer(config, simulations, viteMode) {
   const { name = "sunpeak-app", version = "0.1.0" } = config;
   const mcpServer = new McpServer(
-    { name, version },
+    {
+      name,
+      version,
+      icons: [
+        {
+          src: `data:image/png;base64,${FAVICON_BASE64}`,
+          mimeType: "image/png",
+          sizes: ["128x128"]
+        }
+      ]
+    },
     { capabilities: { resources: {}, tools: {} } }
   );
   const registeredResources = /* @__PURE__ */ new Set();
@@ -14294,32 +14314,39 @@ function createAppServer(config, simulations, viteMode) {
     const toolMeta = tool._meta ?? {};
     if (!registeredResources.has(uri2)) {
       registeredResources.add(uri2);
+      const listMeta = viteMode ? injectViteCSP(resourceMeta) : resourceMeta;
+      console.log(`[MCP] RegisterResource: ${uri2}`);
       ak(
         mcpServer,
         resource.name,
         uri2,
         {
           description: resource.description,
-          _meta: viteMode ? { ui: injectViteCSP(resourceMeta)?.ui } : resourceMeta
+          _meta: listMeta
         },
-        async () => {
-          console.log(`[MCP] ReadResource: ${uri2}${viteMode ? " (vite)" : ""}`);
+        async (_uri, extra) => {
+          const prodBuild = needsProdBuild(
+            extra?.requestInfo?.headers ?? {}
+          );
+          const readMeta = viteMode && !prodBuild ? injectViteCSP(resourceMeta) : resourceMeta;
           let content;
           try {
-            content = getResourceHtml(simulation, viteMode);
+            content = getResourceHtml(simulation, viteMode, prodBuild);
           } catch (error) {
             console.error(`[MCP] ReadResource error for ${uri2}:`, error);
             throw error;
           }
           const sizeKB = (content.length / 1024).toFixed(1);
-          console.log(`[MCP] ReadResource: ${uri2} → ${sizeKB}KB`);
+          console.log(
+            `[MCP] ReadResource: ${uri2} → ${sizeKB}KB${prodBuild ? " (prod build)" : " (vite)"}`
+          );
           return {
             contents: [
               {
                 uri: uri2,
                 mimeType: EI,
                 text: content,
-                _meta: viteMode ? injectViteCSP(resourceMeta) : resourceMeta
+                _meta: readMeta
               }
             ]
           };
@@ -14333,7 +14360,11 @@ function createAppServer(config, simulations, viteMode) {
         description: tool.description,
         _meta: {
           ...toolMeta,
-          ui: { resourceUri: uri2 }
+          ui: {
+            resourceUri: uri2,
+            // Preserve tool visibility from simulation metadata if declared
+            ...toolMeta.ui?.visibility ? { visibility: toolMeta.ui.visibility } : {}
+          }
         }
       },
       async (extra) => {
@@ -14362,16 +14393,24 @@ function createAppServer(config, simulations, viteMode) {
   );
   return mcpServer;
 }
+function isLocalConnection(req) {
+  const addr = req.socket.remoteAddress;
+  return addr === "127.0.0.1" || addr === "::1" || addr === "::ffff:127.0.0.1";
+}
 const sessions = /* @__PURE__ */ new Map();
 const ssePath = "/mcp";
 const postPath = "/mcp/messages";
-async function handleSseRequest(res, config, simulations, viteMode) {
+async function handleSseRequest(req, res, config, simulations, viteMode) {
   res.setHeader("Access-Control-Allow-Origin", "*");
   const server = createAppServer(config, simulations, viteMode);
   const transport = new SSEServerTransport(postPath, res);
   const sessionId = transport.sessionId;
-  sessions.set(sessionId, { server, transport });
-  console.log(`[MCP] Session started: ${sessionId.substring(0, 8)}... (${sessions.size} active)`);
+  const isLocal = isLocalConnection(req);
+  sessions.set(sessionId, { server, transport, isLocal });
+  const origin = isLocal ? "local" : "tunnel";
+  console.log(
+    `[MCP] Session started: ${sessionId.substring(0, 8)}... (${origin}, ${sessions.size} active)`
+  );
   transport.onclose = async () => {
     if (!sessions.has(sessionId)) return;
     sessions.delete(sessionId);
@@ -14479,7 +14518,7 @@ function runMCPServer(config) {
     }
     if (req.method === "GET" && url.pathname === ssePath) {
       console.log(`[HTTP] ${req.method} ${url.pathname}`);
-      await handleSseRequest(res, config, simulations, viteMode);
+      await handleSseRequest(req, res, config, simulations, viteMode);
       return;
     }
     if (req.method === "POST" && url.pathname === postPath) {
@@ -14525,10 +14564,30 @@ function runMCPServer(config) {
   };
   process.on("SIGTERM", () => void shutdown());
   process.on("SIGINT", () => void shutdown());
+  return {
+    invalidateResources() {
+      let notified = 0;
+      for (const [, session] of sessions) {
+        if (!session.isLocal) {
+          session.server.sendResourceListChanged();
+          notified++;
+        }
+      }
+      if (notified > 0) {
+        console.log(`[MCP] Notified ${notified} session(s) of resource changes`);
+      }
+    }
+  };
 }
 export {
+  C6 as EXTENSION_ID,
   FAVICON_BASE64,
   FAVICON_BUFFER,
+  EI as RESOURCE_MIME_TYPE,
+  je as RESOURCE_URI_META_KEY,
+  pk as getUiCapability,
+  ak as registerAppResource,
+  hk as registerAppTool,
   runMCPServer
 };
 //# sourceMappingURL=index.js.map