@onlook/storybook-plugin 0.3.2-beta.1 → 0.3.2-beta.2

package/dist/index.d.ts

@@ -7,12 +7,17 @@ type OnlookPluginOptions = {
     /** Additional allowed origins for CORS (merged with defaults) */
     allowedOrigins?: string[];
     /**
-     * Set when this plugin runs inside an E2B sandbox. Enables the
-     * WSS:443 HMR override required for Vite's HMR client to talk back
-     * through E2B's HTTPS port-proxy. Leave false (default) for local
-     * `bun dev:storybook` so HMR uses Vite defaults and actually fires.
+     * Override Vite's HMR client config. Set when Storybook is fronted by
+     * an HTTPS reverse proxy (sandbox runtime, ngrok, Cloudflare tunnel,
+     * etc.) so the in-page HMR client connects via WSS on the public port
+     * instead of `ws://localhost:<port>`. Plugin fills in the internal
+     * `port` so callers only specify what changes (`protocol`, `clientPort`).
+     * Leave undefined locally so Vite defaults apply and HMR fires.
      */
-    e2b?: boolean;
+    hmr?: {
+        protocol?: 'ws' | 'wss';
+        clientPort?: number;
+    };
 };
 declare function storybookOnlookPlugin(options?: OnlookPluginOptions): PluginOption[];
 

package/dist/index.js

@@ -17191,27 +17191,26 @@ function storybookOnlookPlugin(options = {}) {
   }
   const port = options.port ?? 6006;
   const allowedOrigins = [...DEFAULT_ALLOWED_ORIGINS, ...options.allowedOrigins ?? []];
-  const isE2B = options.e2b === true;
+  const hmrOverride = options.hmr;
   console.log("[STORYBOOK_PLUGIN] Plugin initialized", {
     port,
     allowedOrigins,
     storybookLocation,
     repoRoot,
-    e2b: isE2B
+    hmr: hmrOverride ?? null
   });
   const mainPlugin = {
     name: "storybook-onlook-plugin",
     config() {
       return {
         server: {
-          // HMR override only applies inside E2B sandboxes — they front
-          // Storybook over HTTPS:443, so the Vite HMR client must connect
-          // via WSS:443. Locally, omitting this lets Vite use defaults
+          // HMR override applies only when the caller explicitly opts in
+          // (typically: Storybook fronted over HTTPS by a reverse proxy).
+          // Locally, omitting this lets Vite use defaults
           // (ws://localhost:<port>) and HMR works as expected.
-          ...isE2B && {
+          ...hmrOverride && {
             hmr: {
-              protocol: "wss",
-              clientPort: 443,
+              ...hmrOverride,
               port
             }
           },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@onlook/storybook-plugin",
-  "version": "0.3.2-beta.1",
+  "version": "0.3.2-beta.2",
   "type": "module",
   "bin": {
     "onlook-storybook": "./dist/cli/index.js"