fix: invalid routes to /.auth/** should return 404

Closes #116

Author: manekinekko

src/auth/index.ts

@@ -101,8 +101,10 @@ export async function processAuth(request: ServerRequest, response: ServerRespon
   }
 
   const statusCode = context.res.status || defaultStatus;
-  response.writeHead(statusCode);
-  response.end(context.res.body);
+  if (statusCode === 200) {
+    response.writeHead(statusCode);
+    response.end(context.res.body);
+  }
 
   return statusCode;
 }

src/proxy/server.ts

@@ -161,9 +161,14 @@ const requestHandler = (userConfig: SWAConfigFile | null) =>
     }
 
     // proxy AUTH request to AUTH emulator
-    else if (req.url?.startsWith("/.auth")) {
+    else if (req.url.startsWith("/.auth")) {
       const statusCode = await processAuth(req, res);
       console.log("auth>", req.method, `http://` + req.headers.host! + req.url, statusCode);
+
+      if (statusCode === 404) {
+        req.url = "404.html";
+        serve(SWA_PUBLIC_DIR, req, res);
+      }
     }
 
     // proxy API request to Azure Functions emulator