fix: let resolveBuiltUrl return a __VITE_ASSET__ reference

vitejs · Apr 19, 2021 · f325977 · f325977
1 parent a02eb2c
commit f325977
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/packages/vite/src/node/plugins/asset.ts b/packages/vite/src/node/plugins/asset.ts
@@ -284,9 +284,19 @@ async function resolveBuiltUrl(
 ) {
   for (const { resolveBuiltUrl } of config.plugins) {
     if (resolveBuiltUrl) {
-      const result = await resolveBuiltUrl.call(ctx, url)
-      if (result) {
-        return result
+      const builtUrl = await resolveBuiltUrl.call(ctx, url)
+      if (builtUrl) {
+        const match = builtUrl.match(/^__VITE_ASSET__(.+?)__/)
+        if (match) {
+          let map = assetHashToFilenameMap.get(config)
+          if (!map) {
+            map = new Map()
+            assetHashToFilenameMap.set(config, map)
+          }
+          const contentHash = match[1]
+          map.set(contentHash, ctx.getFileName(contentHash))
+        }
+        return builtUrl
       }
     }
   }