From 2337c4c11eda933f9c91b46defca63dc5557eb19 Mon Sep 17 00:00:00 2001
From: Paul Buetow <paul@buetow.org>
Date: Sat, 11 May 2024 12:59:12 +0300
Subject: move server struct to server.go

---
 internal/server/server.go | 50 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 internal/server/server.go

(limited to 'internal/server/server.go')

diff --git a/internal/server/server.go b/internal/server/server.go
new file mode 100644
index 0000000..6013c1a
--- /dev/null
+++ b/internal/server/server.go
@@ -0,0 +1,50 @@
+package server
+
+import (
+	"fmt"
+	"log"
+	"net/http"
+
+	"codeberg.org/snonux/gos/internal/config"
+	"codeberg.org/snonux/gos/internal/server/health"
+)
+
+const HealthHandlerName = `healthHandler`
+
+type Server struct {
+	Status health.Status
+	Conf   config.Config
+}
+
+type HandlerFuncWithError func(http.ResponseWriter, *http.Request) error
+
+func New(conf config.Config) Server {
+	return Server{
+		Conf:   conf,
+		Status: health.NewStatus(),
+	}
+}
+
+func (serv Server) Handle(name string, handler HandlerFuncWithError) {
+	var (
+		handlerPath = fmt.Sprintf("/%s", name)
+		handlerName = fmt.Sprintf("%sHandler", name)
+	)
+
+	http.HandleFunc(handlerPath, func(w http.ResponseWriter, r *http.Request) {
+		log.Println("Someone requested", handlerName)
+
+		// The health endpoint doesn't require an API key
+		if handlerName != HealthHandlerName && r.Header.Get("X-API-KEY") != serv.Conf.ApiKey {
+			http.Error(w, "Invalid API key", http.StatusUnauthorized)
+			log.Println("Unauthorized access attempt to", handlerName)
+			return
+		}
+
+		if err := handler(w, r); err != nil {
+			serv.Status.Set(health.Critical, handlerName, err.Error())
+			return
+		}
+		serv.Status.Clear(handlerName)
+	})
+}
-- 
cgit v1.2.3