From 344c6526a8d6f490fc7628ddc7d2dd06ed1a07c1 Mon Sep 17 00:00:00 2001
From: JP Appel <jeanpierre.appel01@gmail.com>
Date: Tue, 22 Jul 2025 15:41:03 -0400
Subject: Separate program entry point from commands

---
 pkg/server/server.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

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

diff --git a/pkg/server/server.go b/pkg/server/server.go
index c08d5a4..a7a5395 100644
--- a/pkg/server/server.go
+++ b/pkg/server/server.go
@@ -2,6 +2,7 @@ package server
 
 import (
 	"bytes"
+	"context"
 	"io"
 	"log/slog"
 	"net/http"
@@ -14,6 +15,11 @@ import (
 	"github.com/jpappel/atlas/pkg/query"
 )
 
+type Server interface {
+	ListenAndServe() error
+	Shutdown(context.Context) error
+}
+
 func info(w http.ResponseWriter, r *http.Request) {
 	w.Write([]byte(`
 	<h1>Atlas Server</h1>
@@ -22,7 +28,7 @@ func info(w http.ResponseWriter, r *http.Request) {
 	`))
 }
 
-func New(db *data.Query) *http.ServeMux {
+func NewMux(db *data.Query) *http.ServeMux {
 	mux := http.NewServeMux()
 
 	outputBufPool := &sync.Pool{}
-- 
cgit v1.2.3