main.go

package main

import (
	"log"
	"net/http"
	"os"
	"time"

	"github.com/gorilla/mux"

	"github.com/pgElephant/pgControlPlane/auth"
	"github.com/pgElephant/pgControlPlane/handlers"
	"github.com/pgElephant/pgControlPlane/pkg/store"
	"github.com/pgElephant/pgControlPlane/reconciler"
)

func main() {
	addr := ":8080"
	if v := os.Getenv("PGCP_ADDR"); v != "" {
		addr = v
	}

	r := mux.NewRouter()

	// initialize simple file-backed store for jobs (prototype)
	if err := store.Init(""); err != nil {
		log.Fatalf("failed to init store: %v", err)
	}

	// Public
	r.HandleFunc("/api/v1/login", auth.LoginHandler).Methods("POST")

	// Protected
	api := r.PathPrefix("/api/v1").Subrouter()
	api.Use(auth.JWTMiddleware)
	api.HandleFunc("/clusters", handlers.ListClustersHandler).Methods("GET")
	api.HandleFunc("/clusters", handlers.CreateClusterHandler).Methods("POST")
	api.HandleFunc("/agents/register", handlers.RegisterAgentHandler).Methods("POST")
	api.HandleFunc("/agents/heartbeat", handlers.HeartbeatAgentHandler).Methods("POST")
	api.HandleFunc("/backup", handlers.BackupRequest).Methods("POST")
	// nodes and cluster actions
	api.HandleFunc("/clusters/{cluster}/nodes", handlers.ListNodesHandler).Methods("GET")
	api.HandleFunc("/clusters/{cluster}/nodes", handlers.CreateNodeHandler).Methods("POST")
	api.HandleFunc("/clusters/{cluster}/status", handlers.ClusterStatusHandler).Methods("GET")
	api.HandleFunc("/clusters/{cluster}/promote", handlers.PromoteHandler).Methods("POST")

	stopCh := make(chan struct{})
	reconciler.StartReconciler(stopCh)
	// prune stale agent registrations periodically
	go func() {
		ticker := time.NewTicker(1 * time.Minute)
		defer ticker.Stop()
		for {
			select {
			case <-stopCh:
				return
			case <-ticker.C:
				handlers.PruneStaleAgents(5 * time.Minute)
			}
		}
	}()

	log.Printf("listening on %s", addr)
	if err := http.ListenAndServe(addr, r); err != nil {
		close(stopCh)
		log.Fatalf("server failed: %v", err)
	}
}