openmagic 0.17.0 → 0.18.0

package/dist/cli.js

@@ -912,35 +912,57 @@ You MUST respond with valid JSON in this exact format:
 8. If you cannot make the requested change, set modifications to an empty array and explain why`;
 function buildContextParts(context) {
   const parts = {};
-  if (context.selectedElement) parts.selectedElement = context.selectedElement.outerHTML;
+  if (context.selectedElement) {
+    const el = context.selectedElement;
+    parts.selectedElement = JSON.stringify({
+      cssSelector: el.cssSelector,
+      tagName: el.tagName,
+      id: el.id,
+      className: el.className,
+      outerHTML: el.outerHTML,
+      computedStyles: el.computedStyles,
+      ancestry: el.ancestry,
+      componentHint: el.componentHint
+    }, null, 2);
+  }
   if (context.files?.length) {
-    parts.filePath = context.files[0].path;
-    parts.fileContent = context.files[0].content;
+    parts.files = context.files;
   }
   if (context.projectTree) parts.projectTree = context.projectTree;
+  if (context.pageUrl) parts.pageUrl = context.pageUrl;
+  if (context.pageTitle) parts.pageTitle = context.pageTitle;
   if (context.networkLogs) parts.networkLogs = context.networkLogs.map((l) => `${l.method} ${l.url} \u2192 ${l.status || "pending"}`).join("\n");
   if (context.consoleLogs) parts.consoleLogs = context.consoleLogs.map((l) => `[${l.level}] ${l.args.join(" ")}`).join("\n");
   return parts;
 }
 function buildUserMessage(userPrompt, context) {
   const parts = [];
+  if (context.pageUrl || context.pageTitle) {
+    parts.push(`## Page Context
+URL: ${context.pageUrl || "unknown"}
+Title: ${context.pageTitle || "unknown"}`);
+  }
   if (context.projectTree) {
     parts.push(`## Project Structure
 \`\`\`
 ${context.projectTree}
 \`\`\``);
   }
-  if (context.filePath && context.fileContent) {
-    parts.push(
-      `## Source File: ${context.filePath}
+  if (context.files?.length) {
+    parts.push(`## Grounded Source Files
+${context.files.map((f) => `### ${f.path}
+\`\`\`
+${f.content}
+\`\`\``).join("\n\n")}`);
+  } else if (context.filePath && context.fileContent) {
+    parts.push(`## Source File: ${context.filePath}
 \`\`\`
 ${context.fileContent}
-\`\`\``
-    );
+\`\`\``);
   }
   if (context.selectedElement) {
-    parts.push(`## Selected Element (DOM)
-\`\`\`html
+    parts.push(`## Selected Element
+\`\`\`json
 ${context.selectedElement}
 \`\`\``);
   }
@@ -1356,7 +1378,7 @@ async function handleLlmChat(params, onChunk, onDone, onError) {
 }
 
 // src/server.ts
-var VERSION = "0.17.0";
+var VERSION = "0.18.0";
 var __dirname = dirname2(fileURLToPath(import.meta.url));
 function attachOpenMagic(httpServer, roots) {
   function handleRequest(req, res) {
@@ -1376,10 +1398,7 @@ function attachOpenMagic(httpServer, roots) {
     }
     return false;
   }
-  const wss = new WebSocketServer({
-    server: httpServer,
-    path: "/__openmagic__/ws"
-  });
+  const wss = new WebSocketServer({ noServer: true });
   const clientStates = /* @__PURE__ */ new WeakMap();
   wss.on("connection", (ws, req) => {
     const origin = req.headers.origin || "";
@@ -1411,7 +1430,17 @@ function attachOpenMagic(httpServer, roots) {
       clientStates.delete(ws);
     });
   });
-  return { wss, handleRequest };
+  function handleUpgrade(req, socket, head) {
+    const urlPath = (req.url || "").split("?")[0];
+    if (urlPath === "/__openmagic__/ws") {
+      wss.handleUpgrade(req, socket, head, (ws) => {
+        wss.emit("connection", ws, req);
+      });
+      return true;
+    }
+    return false;
+  }
+  return { wss, handleRequest, handleUpgrade };
 }
 async function handleMessage(ws, msg, state, roots) {
   switch (msg.type) {
@@ -1670,17 +1699,24 @@ ${toolbarScript}
           ${toolbarScript}
         </body></html>`
       );
+    } else if (res && typeof res.destroy === "function") {
+      try {
+        res.destroy();
+      } catch {
+      }
     }
   });
   let omHandle = null;
+  let omUpgrade = null;
   const server = http.createServer((req, res) => {
     if (omHandle && omHandle(req, res)) return;
     proxy.web(req, res);
   });
   const om = attachOpenMagic(server, roots);
   omHandle = om.handleRequest;
+  omUpgrade = om.handleUpgrade;
   server.on("upgrade", (req, socket, head) => {
-    if (req.url?.startsWith("/__openmagic__")) return;
+    if (omUpgrade && omUpgrade(req, socket, head)) return;
     proxy.ws(req, socket, head);
   });
   return server;
@@ -1875,7 +1911,7 @@ process.on("uncaughtException", (err) => {
   process.exit(1);
 });
 var childProcesses = [];
-var VERSION2 = "0.17.0";
+var VERSION2 = "0.18.0";
 function ask(question) {
   const rl = createInterface({ input: process.stdin, output: process.stdout });
   return new Promise((resolve3) => {