lens-inspector 0.1.2 → 0.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lens-inspector",
-  "version": "0.1.2",
+  "version": "0.2.0",
   "description": "Browser-based scene graph inspector for Lens Studio. See every object in your scene, live.",
   "type": "module",
   "bin": {
@@ -13,6 +13,10 @@
     "example.js",
     "README.md"
   ],
+  "scripts": {
+    "dev": "node --watch server.js",
+    "start": "node server.js"
+  },
   "dependencies": {
     "ws": "^8.16.0"
   },

package/server.js

@@ -15,7 +15,7 @@
 
 import { createServer } from "http";
 import { WebSocketServer, WebSocket } from "ws";
-import { readFileSync } from "fs";
+import { readFileSync, watch } from "fs";
 import { join, dirname } from "path";
 import { fileURLToPath } from "url";
 import { exec } from "child_process";
@@ -31,17 +31,19 @@ function arg(flag, fallback) {
 
 const PORT = parseInt(arg("--port", "8200"), 10);
 const MCP_URL = arg("--mcp", "http://localhost:8733/mcp");
+const MCP_TOKEN = arg("--mcp-token", "");
 const NO_OPEN = args.includes("--no-open");
 
 // Connected clients
 const browsers = new Set();
-const lsClients = new Set();
+let activeLS = null; // Only one LS connection at a time
+let activeIsExample = false;
 
 // Latest scene snapshot (so new browser tabs get it immediately)
 let lastScene = null;
 
-// --- HTTP server ---
-const html = readFileSync(join(__dirname, "index.html"));
+// --- HTTP server (re-reads index.html on every request for live editing) ---
+const indexPath = join(__dirname, "index.html");
 
 const http = createServer(async (req, res) => {
   res.setHeader("Access-Control-Allow-Origin", "*");
@@ -50,10 +52,10 @@ const http = createServer(async (req, res) => {
 
   if (req.method === "OPTIONS") { res.writeHead(204); res.end(); return; }
 
-  // Serve viewer
+  // Serve viewer (re-read every time for live editing)
   if (req.method === "GET" && (req.url === "/" || req.url === "/index.html")) {
     res.writeHead(200, { "Content-Type": "text/html" });
-    res.end(html);
+    res.end(readFileSync(indexPath));
     return;
   }
 
@@ -65,7 +67,7 @@ const http = createServer(async (req, res) => {
       try {
         const url = new URL(req.url, `http://localhost:${PORT}`);
         const mcpUrl = url.searchParams.get("url") || MCP_URL;
-        const token = url.searchParams.get("token") || "";
+        const token = url.searchParams.get("token") || MCP_TOKEN;
         const headers = { "Content-Type": "application/json" };
         if (token) headers["Authorization"] = "Bearer " + token;
 
@@ -92,7 +94,7 @@ const http = createServer(async (req, res) => {
   // Health
   if (req.method === "GET" && req.url === "/health") {
     res.writeHead(200, { "Content-Type": "application/json" });
-    res.end(JSON.stringify({ ls: lsClients.size, browsers: browsers.size, hasScene: !!lastScene }));
+    res.end(JSON.stringify({ ls: activeLS ? 1 : 0, lsType: activeIsExample ? 'example' : 'real', browsers: browsers.size, hasScene: !!lastScene }));
     return;
   }
 
@@ -107,22 +109,48 @@ wss.on("connection", (ws, req) => {
   const role = url.searchParams.get("role");
 
   if (role === "ls") {
-    // Lens Studio inspector component
-    lsClients.add(ws);
-    console.log(`[inspector] LS connected (${lsClients.size} total)`);
+    const isExample = url.searchParams.get("source") === "example";
+
+    // Only one LS connection at a time
+    if (activeLS && activeLS.readyState === WebSocket.OPEN) {
+      if (isExample) {
+        // Example trying to connect while real LS is active: reject
+        console.log(`[inspector] Rejecting example (real LS already connected)`);
+        ws.close(1000, "replaced");
+        return;
+      }
+      if (activeIsExample) {
+        // Real LS connecting, kick the example
+        console.log(`[inspector] Kicking example for real LS`);
+        activeLS.close(1000, "replaced");
+      } else {
+        // Another real LS trying to connect: reject the new one
+        console.log(`[inspector] Rejecting duplicate LS (one already connected)`);
+        ws.close(1000, "duplicate");
+        return;
+      }
+    }
+
+    activeLS = ws;
+    activeIsExample = isExample;
+    if (!isExample) lastScene = null;
+    console.log(`[inspector] LS connected (${isExample ? 'example' : 'real'})`);
 
     ws.on("message", (raw) => {
+      if (ws !== activeLS) return; // ignore if superseded
       const str = raw.toString();
       lastScene = str;
-      // Broadcast to all browsers
       for (const b of browsers) {
         if (b.readyState === WebSocket.OPEN) b.send(str);
       }
     });
 
     ws.on("close", () => {
-      lsClients.delete(ws);
-      console.log(`[inspector] LS disconnected (${lsClients.size} total)`);
+      if (ws === activeLS) {
+        activeLS = null;
+        activeIsExample = false;
+        console.log(`[inspector] LS disconnected`);
+      }
     });
   } else {
     // Browser viewer
@@ -143,6 +171,18 @@ wss.on("connection", (ws, req) => {
   ws.on("error", (e) => console.error("[inspector] WS error:", e.message));
 });
 
+// --- Live reload: watch index.html and tell browsers to refresh ---
+let reloadDebounce = null;
+watch(indexPath, () => {
+  clearTimeout(reloadDebounce);
+  reloadDebounce = setTimeout(() => {
+    console.log("[inspector] index.html changed, reloading browsers...");
+    for (const b of browsers) {
+      if (b.readyState === WebSocket.OPEN) b.send(JSON.stringify({ event: "reload" }));
+    }
+  }, 200);
+});
+
 http.listen(PORT, () => {
   const url = `http://localhost:${PORT}`;
   console.log(`\n  Lens Studio Scene Inspector running at ${url}`);