From e2874f2a9bd4e9874c857fb65846462ce66c633f Mon Sep 17 00:00:00 2001
From: Mickael Kerjean <mickael@kerjean.me>
Date: Wed, 14 Sep 2022 23:55:11 +1000
Subject: [PATCH] improve (404): 404 errors

---
 server/ctrl/static.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/server/ctrl/static.go b/server/ctrl/static.go
index 6506fb7a..edf67fe2 100644
--- a/server/ctrl/static.go
+++ b/server/ctrl/static.go
@@ -70,8 +70,12 @@ func IndexHandler(_path string) func(*App, http.ResponseWriter, *http.Request) {
 }
 
 func NotFoundHandler(ctx *App, res http.ResponseWriter, req *http.Request) {
-	res.WriteHeader(http.StatusNotFound)
-	res.Write(HtmlPage404)
+	if strings.Contains(req.Header.Get("accept"), "text/html") {
+		res.WriteHeader(http.StatusNotFound)
+		res.Write(HtmlPage404)
+		return
+	}
+	SendErrorResult(res, ErrNotFound)
 }
 
 func PreflightCorsOK(ctx *App, res http.ResponseWriter, req *http.Request) {