@nuvio/vite-plugin 0.1.0 → 0.5.1

package/dist/nuvio-dev-session.js

@@ -0,0 +1,462 @@
+import {
+  buildSourceIndex,
+  extractIdsFromSource,
+  pathnameFromUpgradeUrl,
+  pickBestSourceIndex,
+  readRuntimeVersions
+} from "./chunk-KTNIIC2D.js";
+
+// src/nuvio-dev-session.ts
+import fs from "fs";
+import path from "path";
+import { watch } from "fs";
+import { WebSocket, WebSocketServer } from "ws";
+import { applyPatchToSource } from "@nuvio/ast-engine";
+import {
+  NUVIO_WS_PATH,
+  PROTOCOL_VERSION,
+  parseClientMessage,
+  serializeServerMessage
+} from "@nuvio/shared";
+import { assertPathWithinRoot } from "@nuvio/shared/secure-path";
+var APP_ENTRY_CANDIDATES = ["src/App.tsx", "src/app.tsx", "App.tsx"];
+var DEFAULT_GLOBS = [
+  "src/**/*.{tsx,jsx}",
+  "apps/**/src/**/*.{tsx,jsx}",
+  "packages/**/src/**/*.{tsx,jsx}"
+];
+function nuvioWsMessageToText(data) {
+  if (typeof data === "string") {
+    return data;
+  }
+  if (Buffer.isBuffer(data)) {
+    return data.toString("utf8");
+  }
+  if (data instanceof ArrayBuffer) {
+    return Buffer.from(data).toString("utf8");
+  }
+  if (ArrayBuffer.isView(data)) {
+    const v = data;
+    return Buffer.from(v.buffer, v.byteOffset, v.byteLength).toString("utf8");
+  }
+  return String(data);
+}
+function isAllowedOrigin(origin) {
+  if (origin === void 0 || origin === "") {
+    return true;
+  }
+  try {
+    const u = new URL(origin);
+    return (u.hostname === "localhost" || u.hostname === "127.0.0.1") && (u.protocol === "http:" || u.protocol === "https:");
+  } catch {
+    return false;
+  }
+}
+function supplementIndexFromAppTsx(serverRoot, built, classNameMode, emitWarn) {
+  if (built.entries.length > 0) {
+    return built;
+  }
+  for (const rel of APP_ENTRY_CANDIDATES) {
+    const appTsx = path.resolve(serverRoot, rel);
+    if (!fs.existsSync(appTsx)) {
+      continue;
+    }
+    try {
+      const code = fs.readFileSync(appTsx, "utf8");
+      const hits = extractIdsFromSource(appTsx, code, { classNameMode });
+      if (hits.length === 0) {
+        continue;
+      }
+      emitWarn(
+        `[Nuvio] Source index had 0 ids; supplemented from ${appTsx} (${hits.length} id(s)).`
+      );
+      return {
+        ...built,
+        entries: hits,
+        scannedFileCount: Math.max(built.scannedFileCount, 1)
+      };
+    } catch {
+    }
+  }
+  return built;
+}
+function attachNuvioDevSession(httpServer, options) {
+  const log = options.log ?? console;
+  const enabled = options.enabled ?? process.env.NUVIO !== "0";
+  const scanGlobs = options.scanGlobs ?? DEFAULT_GLOBS;
+  const verbose = options.verbose ?? false;
+  const classNameMode = options.classNameMode ?? "literal-only";
+  const serverRoot = path.resolve(options.root);
+  const fromConfigFile = options.configDir ?? "";
+  const rootCandidates = [path.resolve(fromConfigFile || serverRoot), serverRoot, process.cwd()];
+  const rootsLabel = [...new Set(rootCandidates)].join(" | ");
+  let indexVersion = 0;
+  let cachedIndexPayload = null;
+  let runtimeDiagnostics = { overlayCssMode: "self-contained" };
+  const idToEntry = /* @__PURE__ */ new Map();
+  const wss = new WebSocketServer({ noServer: true });
+  const undoStack = [];
+  const UNDO_MAX = 32;
+  const pushUndoSnapshot = (file, contents) => {
+    undoStack.push({ file, contents });
+    while (undoStack.length > UNDO_MAX) {
+      undoStack.shift();
+    }
+  };
+  const rebuildIndex = () => {
+    if (!enabled) {
+      return;
+    }
+    let built = pickBestSourceIndex(rootCandidates, scanGlobs, { classNameMode });
+    built = supplementIndexFromAppTsx(serverRoot, built, classNameMode, log.warn);
+    if (built.entries.length === 0) {
+      const fallback = buildSourceIndex(serverRoot, ["src/**/*.{tsx,jsx}"], { classNameMode });
+      if (fallback.entries.length > 0) {
+        log.warn(
+          `[Nuvio] Multi-root scan yielded 0 ids; using serverRoot-only index (${fallback.entries.length} id(s)).`
+        );
+        built = fallback;
+      }
+    }
+    indexVersion += 1;
+    idToEntry.clear();
+    for (const e of built.entries) {
+      idToEntry.set(e.id, e);
+    }
+    runtimeDiagnostics = {
+      ...readRuntimeVersions(serverRoot),
+      overlayCssMode: "self-contained"
+    };
+    cachedIndexPayload = serializeServerMessage({
+      type: "indexReady",
+      protocolVersion: PROTOCOL_VERSION,
+      indexVersion,
+      entries: built.entries,
+      duplicateErrors: built.duplicateErrors,
+      diagnostics: runtimeDiagnostics
+    });
+    if (verbose) {
+      log.info(
+        `[Nuvio] index roots=${rootsLabel} matchedFiles=${built.scannedFileCount} uniqueIds=${built.entries.length}`
+      );
+    } else {
+      log.info(`[Nuvio] index \u2014 ${built.entries.length} id(s), ${built.scannedFileCount} file(s)`);
+    }
+    if (cachedIndexPayload && wss.clients.size > 0) {
+      for (const client of wss.clients) {
+        if (client.readyState === WebSocket.OPEN) {
+          client.send(cachedIndexPayload);
+        }
+      }
+    }
+  };
+  const debouncedRebuild = /* @__PURE__ */ (() => {
+    let t;
+    return () => {
+      if (t) {
+        clearTimeout(t);
+      }
+      t = setTimeout(() => {
+        rebuildIndex();
+        t = void 0;
+      }, 120);
+    };
+  })();
+  const srcDir = path.join(serverRoot, "src");
+  let fileWatcher = null;
+  if (enabled && fs.existsSync(srcDir)) {
+    try {
+      fileWatcher = watch(srcDir, { recursive: true }, (_event, filename) => {
+        if (filename && /\.(tsx|jsx)$/.test(filename)) {
+          debouncedRebuild();
+        }
+      });
+    } catch {
+    }
+  }
+  wss.on("connection", (ws) => {
+    if (cachedIndexPayload && ws.readyState === WebSocket.OPEN) {
+      ws.send(cachedIndexPayload);
+    }
+    ws.on("message", async (data) => {
+      const text = nuvioWsMessageToText(data);
+      const msg = parseClientMessage(text);
+      if (!msg) {
+        ws.send(
+          serializeServerMessage({
+            type: "error",
+            code: "bad_message",
+            message: "Invalid client message"
+          })
+        );
+        return;
+      }
+      if (msg.protocolVersion !== PROTOCOL_VERSION) {
+        ws.send(
+          serializeServerMessage({
+            type: "error",
+            code: "bad_version",
+            message: `Expected protocolVersion ${PROTOCOL_VERSION}`,
+            requestId: "requestId" in msg ? msg.requestId : void 0
+          })
+        );
+        return;
+      }
+      if (msg.type === "ping") {
+        ws.send(
+          serializeServerMessage({
+            type: "pong",
+            protocolVersion: PROTOCOL_VERSION,
+            requestId: msg.requestId,
+            diagnostics: runtimeDiagnostics
+          })
+        );
+        if (cachedIndexPayload) {
+          ws.send(cachedIndexPayload);
+        }
+        return;
+      }
+      if (msg.type === "select") {
+        const entry = idToEntry.get(msg.id);
+        if (!entry) {
+          ws.send(
+            serializeServerMessage({
+              type: "selectAck",
+              protocolVersion: PROTOCOL_VERSION,
+              requestId: msg.requestId,
+              id: msg.id,
+              ok: false,
+              errorCode: "unknown_id",
+              errorMessage: "Id not found in dev source index"
+            })
+          );
+          return;
+        }
+        ws.send(
+          serializeServerMessage({
+            type: "selectAck",
+            protocolVersion: PROTOCOL_VERSION,
+            requestId: msg.requestId,
+            id: msg.id,
+            ok: true,
+            file: entry.file,
+            line: entry.line,
+            column: entry.column,
+            patchHostId: entry.patchHostId,
+            primaryTextTargetKey: entry.primaryTextTargetKey,
+            textTargets: entry.textTargets,
+            styleTargets: entry.styleTargets,
+            hierarchyRole: entry.hierarchyRole,
+            parentHostId: entry.parentHostId,
+            childTargetIds: entry.childTargetIds,
+            rowTargets: entry.rowTargets,
+            tableMeta: entry.tableMeta,
+            tableDataField: entry.tableDataField
+          })
+        );
+        return;
+      }
+      if (msg.type === "patchUndo") {
+        const writeGuardRoot = path.resolve(fromConfigFile || serverRoot);
+        const last = undoStack.pop();
+        if (!last) {
+          ws.send(
+            serializeServerMessage({
+              type: "patchUndoAck",
+              protocolVersion: PROTOCOL_VERSION,
+              requestId: msg.requestId,
+              ok: false,
+              errorCode: "empty_stack",
+              errorMessage: "Nothing to undo"
+            })
+          );
+          return;
+        }
+        try {
+          assertPathWithinRoot(writeGuardRoot, last.file);
+          fs.writeFileSync(last.file, last.contents, "utf8");
+        } catch (e) {
+          undoStack.push(last);
+          ws.send(
+            serializeServerMessage({
+              type: "patchUndoAck",
+              protocolVersion: PROTOCOL_VERSION,
+              requestId: msg.requestId,
+              ok: false,
+              errorCode: "undo_write_error",
+              errorMessage: String(e)
+            })
+          );
+          return;
+        }
+        ws.send(
+          serializeServerMessage({
+            type: "patchUndoAck",
+            protocolVersion: PROTOCOL_VERSION,
+            requestId: msg.requestId,
+            ok: true,
+            file: last.file,
+            undoStackDepth: undoStack.length
+          })
+        );
+        return;
+      }
+      if (msg.type === "patchApply") {
+        const entry = idToEntry.get(msg.id);
+        const writeGuardRoot = path.resolve(fromConfigFile || serverRoot);
+        const dryRun = msg.dryRun === true;
+        const patchAckExtras = dryRun ? { dryRun: true } : {};
+        if (!entry) {
+          ws.send(
+            serializeServerMessage({
+              type: "patchAck",
+              protocolVersion: PROTOCOL_VERSION,
+              requestId: msg.requestId,
+              id: msg.id,
+              ok: false,
+              errorCode: "unknown_id",
+              errorMessage: "Id not found in dev source index",
+              ...patchAckExtras
+            })
+          );
+          return;
+        }
+        try {
+          assertPathWithinRoot(writeGuardRoot, entry.file);
+        } catch (e) {
+          ws.send(
+            serializeServerMessage({
+              type: "patchAck",
+              protocolVersion: PROTOCOL_VERSION,
+              requestId: msg.requestId,
+              id: msg.id,
+              ok: false,
+              errorCode: "path_escape",
+              errorMessage: String(e),
+              ...patchAckExtras
+            })
+          );
+          return;
+        }
+        let source;
+        try {
+          source = fs.readFileSync(entry.file, "utf8");
+        } catch (e) {
+          ws.send(
+            serializeServerMessage({
+              type: "patchAck",
+              protocolVersion: PROTOCOL_VERSION,
+              requestId: msg.requestId,
+              id: msg.id,
+              ok: false,
+              errorCode: "read_error",
+              errorMessage: String(e),
+              ...patchAckExtras
+            })
+          );
+          return;
+        }
+        const result = await applyPatchToSource(source, entry.file, msg.id, msg.ops, {
+          classNameMode,
+          activeBreakpoint: msg.activeBreakpoint
+        });
+        if (!result.ok) {
+          ws.send(
+            serializeServerMessage({
+              type: "patchAck",
+              protocolVersion: PROTOCOL_VERSION,
+              requestId: msg.requestId,
+              id: msg.id,
+              ok: false,
+              errorCode: result.code,
+              errorMessage: result.message,
+              ...patchAckExtras
+            })
+          );
+          return;
+        }
+        if (dryRun) {
+          ws.send(
+            serializeServerMessage({
+              type: "patchAck",
+              protocolVersion: PROTOCOL_VERSION,
+              requestId: msg.requestId,
+              id: msg.id,
+              ok: true,
+              diffSummary: result.diffSummary,
+              dryRun: true
+            })
+          );
+          return;
+        }
+        try {
+          fs.writeFileSync(entry.file, result.source, "utf8");
+        } catch (e) {
+          ws.send(
+            serializeServerMessage({
+              type: "patchAck",
+              protocolVersion: PROTOCOL_VERSION,
+              requestId: msg.requestId,
+              id: msg.id,
+              ok: false,
+              errorCode: "write_error",
+              errorMessage: String(e)
+            })
+          );
+          return;
+        }
+        pushUndoSnapshot(entry.file, source);
+        ws.send(
+          serializeServerMessage({
+            type: "patchAck",
+            protocolVersion: PROTOCOL_VERSION,
+            requestId: msg.requestId,
+            id: msg.id,
+            ok: true,
+            diffSummary: result.diffSummary,
+            writtenFile: entry.file,
+            undoStackDepth: undoStack.length
+          })
+        );
+      }
+    });
+  });
+  const onUpgrade = (request, socket, head) => {
+    if (!enabled) {
+      return;
+    }
+    const pathname = pathnameFromUpgradeUrl(request.url);
+    if (pathname !== NUVIO_WS_PATH) {
+      return;
+    }
+    if (!isAllowedOrigin(request.headers.origin)) {
+      socket.destroy();
+      return;
+    }
+    wss.handleUpgrade(request, socket, head, (ws) => {
+      wss.emit("connection", ws, request);
+    });
+  };
+  httpServer.on("upgrade", onUpgrade);
+  if (enabled) {
+    log.info("[Nuvio] dev session attached (App Router / custom server mode)");
+    rebuildIndex();
+  } else {
+    log.info("[Nuvio] disabled (set NUVIO=1 to enable)");
+  }
+  return {
+    rebuildIndex,
+    close: () => {
+      httpServer.off("upgrade", onUpgrade);
+      fileWatcher?.close();
+      for (const client of wss.clients) {
+        client.close();
+      }
+      wss.close();
+    }
+  };
+}
+export {
+  DEFAULT_GLOBS as NUVIO_DEFAULT_SCAN_GLOBS,
+  attachNuvioDevSession
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nuvio/vite-plugin",
-  "version": "0.1.0",
+  "version": "0.5.1",
   "description": "Nuvio Vite plugin: dev WebSocket, source index for data-nuvio-id, secure patch writes.",
   "license": "MIT",
   "repository": {
@@ -29,6 +29,10 @@
     ".": {
       "types": "./dist/index.d.ts",
       "import": "./dist/index.js"
+    },
+    "./dev-session": {
+      "types": "./dist/nuvio-dev-session.d.ts",
+      "import": "./dist/nuvio-dev-session.js"
     }
   },
   "engines": {
@@ -42,8 +46,8 @@
     "@babel/traverse": "^7.26.9",
     "fast-glob": "^3.3.3",
     "ws": "^8.18.0",
-    "@nuvio/ast-engine": "0.1.0",
-    "@nuvio/shared": "0.1.0"
+    "@nuvio/ast-engine": "0.5.1",
+    "@nuvio/shared": "0.5.1"
   },
   "devDependencies": {
     "@babel/types": "^7.26.9",
@@ -56,7 +60,7 @@
     "vitest": "^3.0.6"
   },
   "scripts": {
-    "build": "tsup src/index.ts --format esm --dts --clean --external vite --external @nuvio/shared --external @nuvio/ast-engine",
+    "build": "tsup src/index.ts src/nuvio-dev-session.ts --format esm --dts --clean --external vite --external @nuvio/shared --external @nuvio/ast-engine",
     "typecheck": "tsc -p tsconfig.json --noEmit",
     "test": "vitest run"
   }