CUB-124: scaffold Control Center Go backend

go-backend/internal/router/router.go

@@ -6,14 +6,22 @@ import (
 	"net/http"
 	"time"
 
+	"code.cubecraftcreations.com/CubeCraft-Creations/Control-Center/go-backend/internal/db"
 	"code.cubecraftcreations.com/CubeCraft-Creations/Control-Center/go-backend/internal/handler"
 	"github.com/go-chi/chi/v5"
 	"github.com/go-chi/chi/v5/middleware"
 	"github.com/go-chi/cors"
 )
 
+// Dependencies carries the handler and database pool into the router.
+type Dependencies struct {
+	Handler   *handler.Handler
+	DB        *db.Pool
+	CORSOrigin string
+}
+
 // New creates a fully-configured chi router with all API routes mounted.
-func New(h *handler.Handler) *chi.Mux {
+func New(deps *Dependencies) *chi.Mux {
 	r := chi.NewRouter()
 
 	// ── Global middleware ──────────────────────────────────────────────────
@@ -23,9 +31,13 @@ func New(h *handler.Handler) *chi.Mux {
 	r.Use(middleware.Recoverer)
 	r.Use(middleware.Timeout(30 * time.Second))
 
-	// ── CORS — permissive for development ──────────────────────────────────
+	// ── CORS ───────────────────────────────────────────────────────────────
+	corsOrigin := deps.CORSOrigin
+	if corsOrigin == "" {
+		corsOrigin = "*"
+	}
 	r.Use(cors.Handler(cors.Options{
-		AllowedOrigins:   []string{"*"},
+		AllowedOrigins:   []string{corsOrigin},
 		AllowedMethods:   []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
 		AllowedHeaders:   []string{"Accept", "Authorization", "Content-Type"},
 		ExposedHeaders:   []string{"Link", "X-Total-Count"},
@@ -33,32 +45,39 @@ func New(h *handler.Handler) *chi.Mux {
 		MaxAge:           300,
 	}))
 
-	// ── Health check ───────────────────────────────────────────────────────
+	// ── Health check (with DB connectivity probe) ──────────────────────────
 	r.Get("/health", func(w http.ResponseWriter, r *http.Request) {
 		w.Header().Set("Content-Type", "application/json")
-		w.Write([]byte(`{"status":"ok"}`))
+		status := "ok"
+		if deps.DB != nil {
+			if err := deps.DB.Health(r.Context()); err != nil {
+				w.WriteHeader(http.StatusServiceUnavailable)
+				status = "db_unhealthy"
+			}
+		}
+		w.Write([]byte(`{"status":"` + status + `"}`))
 	})
 
 	// ── API v1 routes ──────────────────────────────────────────────────────
 	r.Route("/api", func(api chi.Router) {
 		// Agents CRUD
 		api.Route("/agents", func(agents chi.Router) {
-			agents.Get("/", h.ListAgents)       // GET /api/agents
-			agents.Post("/", h.CreateAgent)      // POST /api/agents
-			agents.Get("/{id}", h.GetAgent)       // GET /api/agents/{id}
-			agents.Put("/{id}", h.UpdateAgent)    // PUT /api/agents/{id}
-			agents.Delete("/{id}", h.DeleteAgent) // DELETE /api/agents/{id}
-			agents.Get("/{id}/history", h.AgentHistory) // GET /api/agents/{id}/history
+			agents.Get("/", deps.Handler.ListAgents)       // GET /api/agents
+			agents.Post("/", deps.Handler.CreateAgent)      // POST /api/agents
+			agents.Get("/{id}", deps.Handler.GetAgent)       // GET /api/agents/{id}
+			agents.Put("/{id}", deps.Handler.UpdateAgent)    // PUT /api/agents/{id}
+			agents.Delete("/{id}", deps.Handler.DeleteAgent) // DELETE /api/agents/{id}
+			agents.Get("/{id}/history", deps.Handler.AgentHistory) // GET /api/agents/{id}/history
 		})
 
 		// Sessions
-		api.Get("/sessions", h.ListSessions)
+		api.Get("/sessions", deps.Handler.ListSessions)
 
 		// Tasks
-		api.Get("/tasks", h.ListTasks)
+		api.Get("/tasks", deps.Handler.ListTasks)
 
 		// Projects
-		api.Get("/projects", h.ListProjects)
+		api.Get("/projects", deps.Handler.ListProjects)
 	})
 
 	return r