use post middleware

embroider-build · Mar 19, 2024 · c335877 · c335877
1 parent 5af57e5
commit c335877
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/packages/vite/src/assets.ts b/packages/vite/src/assets.ts
@@ -36,18 +36,20 @@ export function assets(): Plugin {
     configureServer(server) {
       mode = server.config.command;
       publicDir = server.config.publicDir;
-      server.middlewares.use((req, res, next) => {
-        if (req.originalUrl?.includes('?')) {
-          return next();
-        }
-        if (req.originalUrl && req.originalUrl.length > 1) {
-          const assetUrl = findPublicAsset(req.originalUrl, resolverLoader.resolver);
-          if (assetUrl) {
-            return send(req as Readable, assetUrl).pipe(res);
+      return () => {
+        server.middlewares.use((req, res, next) => {
+          if (req.originalUrl?.includes('?')) {
+            return next();
           }
-        }
-        return next();
-      });
+          if (req.originalUrl && req.originalUrl.length > 1) {
+            const assetUrl = findPublicAsset(req.originalUrl, resolverLoader.resolver);
+            if (assetUrl) {
+              return send(req as Readable, assetUrl).pipe(res);
+            }
+          }
+          return next();
+        });
+      }
     },
     async buildStart() {
       if (mode !== 'build') return;