fix: use vite server config for hostname (#196)

module-federation · Dec 3, 2024 · b377cc0 · b377cc0
1 parent e29dfda
commit b377cc0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/plugins/pluginAddEntry.ts b/src/plugins/pluginAddEntry.ts
@@ -52,8 +52,10 @@ const addEntry = ({
       },
       configureServer(server) {
         server.httpServer?.once?.('listening', () => {
-          const { port } = server.config.server;
-          fetch(`http://localhost:${port}${devEntryPath}`).catch((e) => {});
+          const { port, host: hostConfig } = server.config.server;
+          const host =
+            typeof hostConfig === 'string' && hostConfig !== '0.0.0.0' ? hostConfig : 'localhost';
+          fetch(`http://${host}:${port}${devEntryPath}`).catch((e) => {});
         });
         server.middlewares.use((req, res, next) => {
           if (!fileName) {

diff --git a/src/plugins/pluginProxyRemoteEntry.ts b/src/plugins/pluginProxyRemoteEntry.ts
@@ -56,8 +56,12 @@ export default function (): Plugin {
       if (id.includes(getHostAutoInitPath())) {
         const options = getNormalizeModuleFederationOptions();
         if (_command === 'serve') {
+          const host =
+            typeof viteConfig.server?.host === 'string' && viteConfig.server.host !== '0.0.0.0'
+              ? viteConfig.server.host
+              : 'localhost';
           return `
-          const {init} = await import("//localhost:${viteConfig.server?.port}${viteConfig.base + options.filename}")
+          const {init} = await import("//${host}:${viteConfig.server?.port}${viteConfig.base + options.filename}")
           init()
           `;
         }