fix(examples/basic): update static file serving (#390)

closes #389 Signed-off-by: Logan McAnsh <[email protected]>
mcansh · Aug 1, 2024 · dec8ebc · dec8ebc
1 parent 87feccf
commit dec8ebc
Showing 1 changed file with 9 additions and 14 deletions.
diff --git a/examples/basic/server.js b/examples/basic/server.js
@@ -49,18 +49,6 @@ if (process.env.NODE_ENV === "production") {
     root: PUBLIC_DIR,
     prefix: "/",
     wildcard: false,
-    cacheControl: true,
-    dotfiles: "allow",
-    etag: true,
-    serveDotFiles: true,
-    lastModified: true,
-    setHeaders,
-  });
-} else
-  await app.register(fastifyStatic, {
-    root: BUILD_DIR,
-    prefix: "/build",
-    wildcard: true,
     decorateReply: false,
     cacheControl: true,
     dotfiles: "allow",
@@ -69,17 +57,24 @@ if (process.env.NODE_ENV === "production") {
     lastModified: true,
     setHeaders,
   });
+}
 
 await app.register(fastifyStatic, {
-  root: PUBLIC_DIR,
+  root: BUILD_DIR,
   prefix: "/",
   wildcard: false,
   cacheControl: true,
   dotfiles: "allow",
   etag: true,
   serveDotFiles: true,
   lastModified: true,
-  setHeaders,
+  setHeaders(res, filepath) {
+    let isAsset = filepath.startsWith(BUILD_DIR);
+    res.setHeader(
+      "cache-control",
+      isAsset ? ASSET_CACHE_CONTROL : DEFAULT_CACHE_CONTROL,
+    );
+  },
 });
 
 app.register(async function (childServer) {