add configuration

web/handlers.go

@@ -5,16 +5,10 @@ import (
 	"net/http"
 
 	"pool-stats/stats"
-
-	"github.com/ostafen/clover/v2"
 )
 
-type Handlers struct {
-	DB *clover.DB
-}
-
-func (h Handlers) IndexHandler(w http.ResponseWriter, r *http.Request) {
-	shareStats, err := stats.GetStats(h.DB)
+func (ws *WebServer) IndexHandler(w http.ResponseWriter, r *http.Request) {
+	shareStats, err := stats.GetStats(ws.db)
 	if err != nil {
 		http.Error(w, "Failed to load stats", 500)
 		return

web/server.go

@@ -0,0 +1,28 @@
+package web
+
+import (
+	"net/http"
+
+	"fmt"
+
+	"github.com/ostafen/clover/v2"
+)
+
+type WebServer struct {
+	db   *clover.DB
+	port int
+}
+
+func NewWebServer(db *clover.DB, port int) *WebServer {
+	return &WebServer{
+		db:   db,
+		port: port,
+	}
+}
+
+func (ws *WebServer) Start() error {
+	http.HandleFunc("/", ws.IndexHandler)
+	address := ":" + fmt.Sprint(ws.port)
+	println("Listening on", address)
+	return http.ListenAndServe(address, nil)
+}