fix: resolve warnings when building with sourcemaps enabled (#344)

Author: szebest

src/plugins/pluginAddEntry.ts

@@ -1,6 +1,7 @@
 import * as fs from 'fs';
 import * as path from 'pathe';
 import { Plugin } from 'vite';
+import { mapCodeToCodeWithSourcemap } from '../utils/mapCodeToCodeWithSourcemap';
 
 interface AddEntryOptions {
   entryName: string;
@@ -153,7 +154,7 @@ const addEntry = ({
           const injection = `
           import ${JSON.stringify(entryPath)};
           `;
-          return injection + code;
+          return mapCodeToCodeWithSourcemap(injection + code);
         }
       },
     },

src/plugins/pluginProxyRemoteEntry.ts

@@ -1,6 +1,8 @@
 import { createFilter } from '@rollup/pluginutils';
 import { Plugin } from 'vite';
+import { mapCodeToCodeWithSourcemap } from '../utils/mapCodeToCodeWithSourcemap';
 import { getNormalizeModuleFederationOptions } from '../utils/normalizeModuleFederationOptions';
+import { resolvePublicPath } from '../utils/publicPath';
 import {
   generateExposes,
   generateRemoteEntry,
@@ -9,7 +11,6 @@ import {
   VIRTUAL_EXPOSES,
 } from '../virtualModules';
 import { parsePromise } from './pluginModuleParseEnd';
-import { resolvePublicPath } from '../utils/publicPath';
 
 const filter: (id: string) => boolean = createFilter();
 
@@ -46,25 +47,28 @@ export default function (): Plugin {
         return id;
       }
     },
-    async transform(code: string, id: string) {
-      if (!filter(id)) return;
-      if (id.includes(REMOTE_ENTRY_ID)) {
-        return parsePromise.then((_) => generateRemoteEntry(getNormalizeModuleFederationOptions()));
-      }
-      if (id === VIRTUAL_EXPOSES) {
-        return generateExposes();
-      }
-      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';
-          const publicPath = JSON.stringify(
-            resolvePublicPath(options, viteConfig.base) + options.filename
+    transform(code: string, id: string) {
+      const transformedCode = (() => {
+        if (!filter(id)) return;
+        if (id.includes(REMOTE_ENTRY_ID)) {
+          return parsePromise.then((_) =>
+            generateRemoteEntry(getNormalizeModuleFederationOptions())
           );
-          return `
+        }
+        if (id === VIRTUAL_EXPOSES) {
+          return generateExposes();
+        }
+        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';
+            const publicPath = JSON.stringify(
+              resolvePublicPath(options, viteConfig.base) + options.filename
+            );
+            return `
           const origin = (window && ${!options.ignoreOrigin}) ? window.origin : "//${host}:${viteConfig.server?.port}"
           const remoteEntryPromise = await import(origin + ${publicPath})
           // __tla only serves as a hack for vite-plugin-top-level-await.
@@ -74,9 +78,12 @@ export default function (): Plugin {
               .then(remoteEntry.init).catch(remoteEntry.init)
           })
           `;
+          }
+          return code;
         }
-        return code;
-      }
+      })();
+
+      return mapCodeToCodeWithSourcemap(transformedCode);
     },
   };
 }

src/plugins/pluginProxySharedModule_preBuild.ts

@@ -1,6 +1,8 @@
 import { Plugin, UserConfig } from 'vite';
+import { mapCodeToCodeWithSourcemap } from '../utils/mapCodeToCodeWithSourcemap';
 import { NormalizedShared } from '../utils/normalizeModuleFederationOptions';
 import { PromiseStore } from '../utils/PromiseStore';
+import VirtualModule, { assertModuleFound } from '../utils/VirtualModule';
 import {
   addUsedShares,
   generateLocalSharedImportMap,
@@ -12,7 +14,6 @@ import {
   writePreBuildLibPath,
 } from '../virtualModules';
 import { parsePromise } from './pluginModuleParseEnd';
-import VirtualModule, { assertModuleFound } from '../utils/VirtualModule';
 
 export function proxySharedModule(options: {
   shared?: NormalizedShared;
@@ -30,9 +31,11 @@ export function proxySharedModule(options: {
           return parsePromise.then((_) => generateLocalSharedImportMap());
         }
       },
-      transform(code, id) {
+      transform(_, id) {
         if (id.includes(getLocalSharedImportMapPath())) {
-          return parsePromise.then((_) => generateLocalSharedImportMap());
+          return mapCodeToCodeWithSourcemap(
+            parsePromise.then((_) => generateLocalSharedImportMap())
+          );
         }
       },
     },

src/utils/mapCodeToCodeWithSourcemap.ts

@@ -0,0 +1,16 @@
+import MagicString from 'magic-string';
+
+export async function mapCodeToCodeWithSourcemap(code?: string | Promise<string>) {
+  const resolvedCode = await code;
+
+  if (resolvedCode === undefined) {
+    return;
+  }
+
+  const s = new MagicString(resolvedCode);
+
+  return {
+    code: s.toString(),
+    map: s.generateMap({ hires: true }),
+  };
+}