From a54d6e445f21d8343d2e70e5f4d747ed37260701 Mon Sep 17 00:00:00 2001
From: giteaadmin <gitea@twk95.net>
Date: Sat, 23 Aug 2025 17:34:29 -0400
Subject: [PATCH] new: 404 redirect in server.js

---
 src/server.js | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/src/server.js b/src/server.js
index 2806884..ace6419 100644
--- a/src/server.js
+++ b/src/server.js
@@ -4,13 +4,8 @@ const fs = require('fs-extra');
 const path = require('path');
 
 const app = express();
-const port = 3000; // You can change this port if needed
+const port = 3000;
 
-// Enable trust proxy to ensure Express correctly handles protocols (http/https)
-// and client IP addresses when running behind a reverse proxy.
-// This is crucial for correct URL generation in redirects and for security features
-// like rate limiting. The value 'true' trusts the X-Forwarded-* headers
-// set by the immediate upstream proxy.
 app.set('trust proxy', true);
 
 // Path to the file where view counts will be stored
@@ -23,7 +18,7 @@ if (initialData.length === 0) {
     fs.writeJsonSync(dbPath, {});
 }
 
-app.use(cors()); // Enable Cross-Origin Resource Sharing
+app.use(cors());
 app.use(express.json());
 
 // In production, the server also serves the static site from the `_site` directory.
@@ -58,6 +53,12 @@ app.post('/api/views/:slug', async (req, res) => {
     }
 });
 
+// --- 404 Handler (Catch-All) ---
+// This MUST be the last route or middleware added.
+app.use((req, res, next) => {
+  res.redirect('/404.html');
+});
+
 app.listen(port, () => {
     console.log(`View counter server listening at http://localhost:${port}`);
 });