@skyvexsoftware/stratos-sdk 0.2.2 → 0.3.1

package/dist/shared-types/simulator.d.ts

@@ -98,6 +98,7 @@ export type FlightData = {
     simulationRate: number;
     isInMenu: boolean;
     isXPlane: boolean;
+    isHelicopter: boolean;
     stallWarning: boolean;
     overspeedWarning: boolean;
     crashed: boolean;

package/dist/vite/hmr-base-rewrite.d.ts

@@ -0,0 +1,27 @@
+/**
+ * Vite plugin that fixes cross-origin HMR for plugins loaded inside Stratos.
+ *
+ * Three problems when loading plugins via cross-origin import():
+ *
+ * 1. Absolute-path imports (/@vite/client, /@react-refresh) resolve to the
+ *    PAGE origin (the shell), not the plugin's dev server. This includes
+ *    imports injected by Vite's importAnalysis (which runs AFTER all user
+ *    transform plugins).
+ *
+ * 2. The HMR client's __BASE__ defaults to "/" so module re-fetches during
+ *    HMR go to the shell's origin.
+ *
+ * 3. @vitejs/plugin-react's preamble (which initializes React Fast Refresh)
+ *    is injected via transformIndexHtml — but plugins don't have HTML files,
+ *    so the preamble never runs. Without it, Fast Refresh silently fails.
+ *
+ * This plugin fixes all three using two hooks:
+ * - transform (enforce: post): patches __BASE__, rewrites imports our
+ *   transform can see, and injects the React Refresh preamble
+ * - configureServer middleware: intercepts FINAL module responses to rewrite
+ *   /@vite/client imports added by importAnalysis (which runs after all
+ *   user transforms)
+ */
+import type { Plugin } from "vite";
+export declare function hmrBaseRewrite(): Plugin;
+//# sourceMappingURL=hmr-base-rewrite.d.ts.map

package/dist/vite/hmr-base-rewrite.js

@@ -0,0 +1,112 @@
+/**
+ * Vite plugin that fixes cross-origin HMR for plugins loaded inside Stratos.
+ *
+ * Three problems when loading plugins via cross-origin import():
+ *
+ * 1. Absolute-path imports (/@vite/client, /@react-refresh) resolve to the
+ *    PAGE origin (the shell), not the plugin's dev server. This includes
+ *    imports injected by Vite's importAnalysis (which runs AFTER all user
+ *    transform plugins).
+ *
+ * 2. The HMR client's __BASE__ defaults to "/" so module re-fetches during
+ *    HMR go to the shell's origin.
+ *
+ * 3. @vitejs/plugin-react's preamble (which initializes React Fast Refresh)
+ *    is injected via transformIndexHtml — but plugins don't have HTML files,
+ *    so the preamble never runs. Without it, Fast Refresh silently fails.
+ *
+ * This plugin fixes all three using two hooks:
+ * - transform (enforce: post): patches __BASE__, rewrites imports our
+ *   transform can see, and injects the React Refresh preamble
+ * - configureServer middleware: intercepts FINAL module responses to rewrite
+ *   /@vite/client imports added by importAnalysis (which runs after all
+ *   user transforms)
+ */
+export function hmrBaseRewrite() {
+    let devOrigin = "";
+    return {
+        name: "stratos-hmr-base-rewrite",
+        apply: "serve",
+        enforce: "post",
+        configureServer(server) {
+            server.httpServer?.once("listening", () => {
+                const addr = server.httpServer?.address();
+                if (addr && typeof addr === "object") {
+                    const host = addr.address === "::" ||
+                        addr.address === "::1" ||
+                        addr.address === "0.0.0.0" ||
+                        addr.address === "127.0.0.1"
+                        ? "localhost"
+                        : addr.address;
+                    devOrigin = `http://${host}:${addr.port}`;
+                }
+            });
+            // Intercept module responses AFTER Vite's transformMiddleware
+            // (which includes importAnalysis) to rewrite /@vite/client imports
+            // that were injected post-transform. We wrap res.end() on every
+            // request so we can modify the final response body.
+            // Prevent Electron from caching /@vite/client across page loads.
+            // Stale cached modules break HMR because the __BASE__ and __WS_TOKEN__
+            // injections from the current server instance aren't in the cache.
+            server.middlewares.use((req, res, next) => {
+                if (req.url?.includes("@vite/client") ||
+                    req.url?.includes("@react-refresh")) {
+                    res.setHeader("Cache-Control", "no-store, no-cache, must-revalidate");
+                }
+                next();
+            });
+            // Rewrite /@vite/client imports in module responses. Vite's
+            // importAnalysis injects these AFTER all user transform plugins,
+            // so we intercept the final response via res.end().
+            server.middlewares.use((_req, res, next) => {
+                if (!devOrigin)
+                    return next();
+                const originalEnd = res.end.bind(res);
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                res.end = function (chunk, encoding, cb) {
+                    if (typeof chunk === "string" && chunk.includes("/@vite/client")) {
+                        chunk = chunk.replace(/(?<=['"])(\/@vite\/client)(?=['"])/g, `${devOrigin}/@vite/client`);
+                    }
+                    return originalEnd(chunk, encoding, cb);
+                };
+                next();
+            });
+        },
+        transform(code, id) {
+            if (!devOrigin)
+                return null;
+            // 1. Patch Vite's HMR client: replace __BASE__ with full origin
+            if (id.includes("vite/dist/client/client.mjs")) {
+                const patched = code.replace(/__BASE__/g, `"${devOrigin}/"`);
+                if (patched !== code) {
+                    return { code: patched, map: null };
+                }
+                return null;
+            }
+            // Skip node_modules
+            if (id.includes("/node_modules/"))
+                return null;
+            let patched = code;
+            // 2. Rewrite /@vite/* and /@react-refresh imports to full URLs
+            //    (catches imports from @vitejs/plugin-react and CSS transforms)
+            patched = patched.replace(/(from\s+["'])(\/(?:@vite\/[^"']+|@react-refresh))(["'])/g, `$1${devOrigin}$2$3`);
+            patched = patched.replace(/(import\s*\(\s*["'])(\/(?:@vite\/[^"']+|@react-refresh))(["'])/g, `$1${devOrigin}$2$3`);
+            // 3. Inject React Fast Refresh preamble into the entry file.
+            if (id.endsWith("/src/ui/index.tsx") || id.endsWith("/src/ui/index.ts")) {
+                const preamble = [
+                    `import __StratosRefresh__ from "${devOrigin}/@react-refresh";`,
+                    `__StratosRefresh__.injectIntoGlobalHook(window);`,
+                    `window.$RefreshReg$ = () => {};`,
+                    `window.$RefreshSig$ = () => (type) => type;`,
+                    `window.__vite_plugin_react_preamble_installed__ = true;`,
+                ].join("\n");
+                patched = preamble + "\n" + patched;
+            }
+            if (patched !== code) {
+                return { code: patched, map: null };
+            }
+            return null;
+        },
+    };
+}
+//# sourceMappingURL=hmr-base-rewrite.js.map

package/dist/vite/plugin-config.js

@@ -22,6 +22,7 @@ import { UI_EXTERNALS } from "./externals.js";
 import { serveExternals } from "./serve-externals.js";
 import { stratosDevServer } from "./stratos-dev-server.js";
 import { cssInject } from "./css-inject.js";
+import { hmrBaseRewrite } from "./hmr-base-rewrite.js";
 /** Modules external in background builds (available in Electron main process) */
 const BG_EXTERNALS = ["electron", "socket.io-client"];
 /** Node.js built-in modules to externalize in background builds */
@@ -204,30 +205,27 @@ function createUIConfig(pluginDir, entry, extraConfig) {
     const isProduction = process.env.NODE_ENV === "production";
     return {
         root: pluginDir,
+        // Skip WebSocket token validation so /@vite/client can connect from
+        // a cross-origin Electron renderer. The renderer sends its own Origin
+        // header (e.g. http://localhost:5173), which triggers Vite's token
+        // check and rejects the connection. Safe for local dev servers.
+        legacy: { skipWebSocketTokenCheck: true },
         server: {
-            cors: {
-                origin: [
-                    "http://localhost:2066",
-                    "http://127.0.0.1:2066",
-                    /^http:\/\/localhost:\d+$/,
-                ],
-            },
+            cors: true,
             hmr: {
-                // When modules are loaded via stratos-dev:// protocol, Vite's HMR
+                // When modules are loaded via cross-origin import(), Vite's HMR
                 // client needs to connect back to THIS dev server's WebSocket directly.
-                // Without this, it would try to derive the WS URL from the custom
-                // protocol scheme, which doesn't work.
                 protocol: "ws",
                 host: "localhost",
             },
         },
         optimizeDeps: {
-            exclude: [
-                ...UI_EXTERNALS,
-                "react/jsx-runtime",
-                "react/jsx-dev-runtime",
-                "react-dom/client",
-            ],
+            // Exclude non-React externals from pre-bundling (they come from the shell).
+            // React packages are NOT excluded — they need to be pre-bundled so that
+            // @vitejs/plugin-react can resolve react-refresh for Fast Refresh.
+            // serveExternals() still rewrites all source-level React imports to
+            // window.__stratos_modules__, so the pre-bundled React is never loaded.
+            exclude: UI_EXTERNALS.filter((id) => !id.startsWith("react") && id !== "react-dom"),
         },
         build: {
             outDir: path.resolve(pluginDir, "dist/ui"),
@@ -252,6 +250,7 @@ function createUIConfig(pluginDir, entry, extraConfig) {
         ...extraConfig,
         plugins: [
             serveExternals(),
+            hmrBaseRewrite(),
             stratosDevServer({ pluginDir }),
             stratosExternals(),
             cssInject(),

package/dist/vite/stratos-dev-server.js

@@ -20,12 +20,19 @@ import { io as ioClient } from "socket.io-client";
  */
 export function stratosDevServer(options) {
     const { pluginDir } = options;
-    // Shared state between configureServer and handleHotUpdate hooks
+    // Shared state between hooks
     let socket = null;
     let pluginId = "unknown";
+    let wsToken = "";
     return {
         name: "stratos-dev-server",
         apply: "serve",
+        configResolved(config) {
+            // Capture the WebSocket token generated by Vite. The shell needs this
+            // to connect to the plugin's HMR WebSocket for live updates.
+            wsToken =
+                config.webSocketToken ?? "";
+        },
         configureServer(server) {
             const stratosPort = options.stratosPort ??
                 (process.env.STRATOS_PORT
@@ -92,6 +99,8 @@ export function stratosDevServer(options) {
                         devUrl,
                         capabilities: ["ui", "hot-reload"],
                         distPath,
+                        sourceRoot: pluginDir,
+                        wsToken,
                     });
                 });
                 socket.on("dev:plugin-mounted", (data) => {
@@ -185,10 +194,10 @@ export function stratosDevServer(options) {
             process.once("SIGTERM", cleanup);
             // Start connecting after the HTTP server is listening
             server.httpServer?.once("listening", () => {
-                // Set the HMR port dynamically so Vite's HMR client connects
-                // directly to this dev server's WebSocket (cross-origin is OK for WS).
                 const addr = server.httpServer?.address();
                 if (addr && typeof addr === "object") {
+                    // Set the HMR port dynamically so Vite's HMR client connects
+                    // directly to this dev server's WebSocket.
                     server.config.server.hmr =
                         typeof server.config.server.hmr === "object"
                             ? {
@@ -201,19 +210,13 @@ export function stratosDevServer(options) {
                 connect();
             });
         },
-        // Notify Stratos when UI source files change so it can reload the plugin
-        handleHotUpdate({ file }) {
-            // Only signal for UI source changes (not background, not config files)
-            if (file.includes("/src/") && !file.includes("/src/background/")) {
-                if (socket?.connected) {
-                    console.log(`[stratos-dev-server] UI source changed: ${path.basename(file)}`);
-                    socket.emit("dev:plugin-ui-updated", {
-                        pluginId,
-                        timestamp: Date.now(),
-                    });
-                }
-            }
-        },
+        // Vite handles all HMR natively — CSS via style injection, JS/TSX via
+        // React Fast Refresh (when @vitejs/plugin-react is configured).
+        // The hmrBaseRewrite plugin patches the HMR client's base URL so
+        // cross-origin module re-fetching works correctly.
+        //
+        // Background source changes are still handled separately via
+        // fs.watch + dev:plugin-bg-updated (see configureServer above).
     };
 }
 //# sourceMappingURL=stratos-dev-server.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skyvexsoftware/stratos-sdk",
-  "version": "0.2.2",
+  "version": "0.3.1",
   "description": "Plugin SDK for Stratos — types, hooks, and UI components",
   "author": {
     "name": "Skyvex Software",