refactor: createTransportWithInvoke

hi-ogawa · Nov 7, 2024 · 2dc6e17 · 2dc6e17
1 parent c04460e
commit 2dc6e17
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 18 deletions.
diff --git a/examples/browser-cli/src/cli.ts b/examples/browser-cli/src/cli.ts
@@ -2,6 +2,7 @@ import repl from "node:repl";
 import { createManualPromise, tinyassert } from "@hiogawa/utils";
 import { chromium } from "@playwright/test";
 import {
+  DevEnvironment,
   type Plugin,
   type PluginOption,
   createServer,

diff --git a/examples/web-worker/src/lib/fetch-module-server.ts b/examples/web-worker/src/lib/fetch-module-server.ts
@@ -1,4 +1,4 @@
-import { DevEnvironment, type DevEnvironmentOptions, type Plugin } from "vite";
+import { type HotChannel, type Plugin } from "vite";
 
 export function vitePluginFetchModuleServer(): Plugin {
   return {
@@ -9,7 +9,7 @@ export function vitePluginFetchModuleServer(): Plugin {
         if (url.pathname === "/@vite/invoke") {
           const [name, ...args] = JSON.parse(url.searchParams.get("payload")!);
           const devEnv = server.environments[name]!;
-          const result = await (devEnv as any).__invoke(...args);
+          const result = devEnv.hot.api.invoke(...args);
           res.end(JSON.stringify(result));
           return;
         }
@@ -19,21 +19,17 @@ export function vitePluginFetchModuleServer(): Plugin {
   };
 }
 
-// expose `DevEnvironment.__invoke`
-export const createEnvironmentWithInvoke: NonNullable<
-  DevEnvironmentOptions["createEnvironment"]
-> = (name, config) => {
+// expose `hot.api.invoke`
+export function createTransportWithInvoke(): HotChannel {
   let invokeHandler!: Function;
-  const devEnv = new DevEnvironment(name, config, {
-    hot: false,
-    transport: {
-      setInvokeHandler(invokeHandler_) {
-        if (invokeHandler_) {
-          invokeHandler = invokeHandler_;
-        }
-      },
+  return {
+    setInvokeHandler(invokeHandler_) {
+      if (invokeHandler_) {
+        invokeHandler = invokeHandler_;
+      }
+    },
+    api: {
+      invoke: (...args: any[]) => invokeHandler(...args),
     },
-  });
-  Object.assign(devEnv, { __invoke: invokeHandler });
-  return devEnv;
-};
+  };
+}