uilint 0.2.58 → 0.2.60

package/dist/index.js

@@ -1404,7 +1404,7 @@ async function update(options) {
 }
 
 // src/commands/serve.ts
-import { existsSync as existsSync5, statSync as statSync3, readdirSync, readFileSync } from "fs";
+import { existsSync as existsSync5, statSync as statSync3, readdirSync, readFileSync, mkdirSync as mkdirSync4, writeFileSync as writeFileSync4 } from "fs";
 import { createRequire } from "module";
 import { dirname as dirname5, resolve as resolve5, relative, join as join3, parse as parse2 } from "path";
 import { WebSocketServer, WebSocket } from "ws";
@@ -1897,7 +1897,13 @@ async function handleMessage(ws, data) {
       )}`
     );
   } else if (message.type === "vision:analyze") {
+  } else if (message.type === "vision:check") {
   } else if (message.type === "config:set") {
+  } else if (message.type === "screenshot:save") {
+    const rid = message.requestId;
+    logInfo(
+      `${pc.dim("[ws]")} ${pc.bold("screenshot:save")} ${pc.dim(message.route)}${rid ? ` ${pc.dim(`(req ${rid})`)}` : ""}`
+    );
   }
   switch (message.type) {
     case "lint:file": {
@@ -2159,6 +2165,29 @@ ${stack}` : ""
       }
       break;
     }
+    case "vision:check": {
+      const { requestId } = message;
+      logInfo(
+        `${pc.dim("[ws]")} ${pc.bold("vision:check")}${requestId ? ` ${pc.dim(`(req ${requestId})`)}` : ""}`
+      );
+      try {
+        const analyzer = getVisionAnalyzerInstance();
+        const model = typeof analyzer.getModel === "function" ? analyzer.getModel() : void 0;
+        sendMessage(ws, {
+          type: "vision:status",
+          available: true,
+          model,
+          requestId
+        });
+      } catch (error) {
+        sendMessage(ws, {
+          type: "vision:status",
+          available: false,
+          requestId
+        });
+      }
+      break;
+    }
     case "config:set": {
       const { key, value } = message;
       handleConfigSet(key, value);
@@ -2234,6 +2263,75 @@ ${stack}` : ""
       }
       break;
     }
+    case "screenshot:save": {
+      const { dataUrl, route, timestamp, requestId } = message;
+      try {
+        if (!dataUrl || typeof dataUrl !== "string") {
+          sendMessage(ws, {
+            type: "screenshot:error",
+            error: "Invalid dataUrl: must be a non-empty string",
+            requestId
+          });
+          break;
+        }
+        const dataUrlPattern = /^data:image\/png;base64,/;
+        if (!dataUrlPattern.test(dataUrl)) {
+          sendMessage(ws, {
+            type: "screenshot:error",
+            error: "Invalid dataUrl: must be a base64 PNG data URL (data:image/png;base64,...)",
+            requestId
+          });
+          break;
+        }
+        const base64Data = dataUrl.replace(dataUrlPattern, "");
+        const sanitizedRoute = route.replace(/^\//, "").replace(/\//g, "-").replace(/[^a-zA-Z0-9_-]/g, "_") || "root";
+        const filename = `uilint-${timestamp}-${sanitizedRoute}.png`;
+        if (!isValidScreenshotFilename(filename)) {
+          sendMessage(ws, {
+            type: "screenshot:error",
+            error: `Generated filename is invalid: ${filename}`,
+            requestId
+          });
+          break;
+        }
+        const screenshotsDir = join3(serverAppRootForVision, ".uilint", "screenshots");
+        if (!existsSync5(screenshotsDir)) {
+          mkdirSync4(screenshotsDir, { recursive: true });
+        }
+        const imagePath = join3(screenshotsDir, filename);
+        const imageBuffer = Buffer.from(base64Data, "base64");
+        writeFileSync4(imagePath, imageBuffer);
+        const sidecarFilename = filename.replace(/\.png$/, ".json");
+        const sidecarPath = join3(screenshotsDir, sidecarFilename);
+        const sidecarData = {
+          route,
+          timestamp,
+          filename,
+          savedAt: (/* @__PURE__ */ new Date()).toISOString()
+        };
+        writeFileSync4(sidecarPath, JSON.stringify(sidecarData, null, 2));
+        logInfo(
+          `${pc.dim("[ws]")} screenshot:saved ${pc.dim(filename)} ${pc.dim(
+            `(${Math.round(imageBuffer.length / 1024)}kb)`
+          )}`
+        );
+        sendMessage(ws, {
+          type: "screenshot:saved",
+          filename,
+          path: imagePath,
+          requestId
+        });
+      } catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        logError(`${pc.dim("[ws]")} screenshot:error ${errorMessage}`);
+        sendMessage(ws, {
+          type: "screenshot:error",
+          error: errorMessage,
+          requestId
+        });
+      }
+      break;
+    }
   }
 }
 function handleDisconnect(ws) {