fix(api-server): copy fallback fix from redwoodjs#9272 (redwoodjs#9369)

jtoar · web-flow · commit 7ab07a25cdde · 2023-11-02T06:08:12.000Z
This PR copies the fix in redwoodjs#9272 into `@redwoodjs/api-server`. The `@redwoodjs/api-server` can be accessed via `npx @redwooodjs/api-server rw-server api`. We won't have to do this once the code between the two entry points is deduplicated but till then, yeah.
diff --git a/packages/api-server/src/plugins/withWebServer.ts b/packages/api-server/src/plugins/withWebServer.ts
@@ -2,7 +2,7 @@ import fs from 'fs'
 import path from 'path'
 
 import fastifyStatic from '@fastify/static'
-import type { FastifyInstance, FastifyReply } from 'fastify'
+import type { FastifyInstance, FastifyReply, FastifyRequest } from 'fastify'
 
 import { getPaths } from '@redwoodjs/project-config'
 
@@ -54,10 +54,21 @@ const withWebServer = async (
 
   // For SPA routing fallback on unmatched routes
   // And let JS routing take over
-  fastify.setNotFoundHandler({}, function (_, reply: FastifyReply) {
-    reply.header('Content-Type', 'text/html; charset=UTF-8')
-    reply.sendFile(indexPath)
-  })
+  fastify.setNotFoundHandler(
+    {},
+    function (req: FastifyRequest, reply: FastifyReply) {
+      const requestedExtension = path.extname(req.url)
+      // If it's requesting some sort of asset, e.g. .js or .jpg files
+      // Html files should fallback to the index.html
+      if (requestedExtension !== '' && requestedExtension !== '.html') {
+        reply.code(404)
+        return reply.send('Not Found')
+      }
+
+      reply.header('Content-Type', 'text/html; charset=UTF-8')
+      return reply.sendFile(indexPath)
+    }
+  )
 
   return fastify
 }
