moved server main to cmd/server

cmd/server/main.go

@@ -0,0 +1,107 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"log/slog"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/ewpt3ch/pkgstash/internal/cache"
+)
+
+type Server struct {
+	cfg      *Config
+	c        *cache.Cache
+	logLevel *slog.LevelVar
+}
+
+func main() {
+
+	// get options from cli flags
+	var configPath string
+	flag.StringVar(&configPath, "config", "", "path to config file")
+	logFlag := flag.String("loglevel", "INFO", "loglevel: DEBUG, INFO, WARN, ERROR")
+	flag.Parse()
+
+	// set config from flag if available
+	if len(configPath) == 0 {
+		configPath = "/etc/pkgstash/pkgstash.toml"
+	}
+
+	//set log level from flag if available
+	logLevel := new(slog.LevelVar)
+	if err := logLevel.UnmarshalText([]byte(*logFlag)); err != nil {
+		fmt.Fprintf(os.Stderr, "invalid log level %q, defaulting to INFO\n", *logFlag)
+		logLevel.Set(slog.LevelInfo)
+	}
+
+	logger := slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{
+		Level: logLevel,
+	}))
+	slog.SetDefault(logger)
+
+	cfg, err := ReadConfig(configPath)
+	if err != nil {
+		slog.Error("failed to read config", "err", err)
+		os.Exit(1)
+	}
+
+	c, err := cache.NewCache(cfg.CacheRoot, cfg.MirrorURLs, cfg.MirroredRepos)
+	if err != nil {
+		slog.Error("failed to create cache", "err", err)
+		os.Exit(1)
+	}
+	defer c.Close() //nolint:errcheck // best effort cleanup on exit
+
+	srv := &Server{
+		cfg:      cfg,
+		c:        c,
+		logLevel: logLevel,
+	}
+
+	mux := http.NewServeMux()
+	mux.HandleFunc("GET /{repo}/os/{arch}/{file}", srv.handlerPackage)
+	mux.HandleFunc("POST /api/refresh", srv.handlerRefresh)
+	mux.HandleFunc("POST /api/loglevel", srv.handlerLogLevel)
+
+	if err := srv.c.Refresh(); err != nil {
+		slog.Error("failed to refesh db files", "err", err)
+		//nolint:errcheck //already exiting
+		_ = c.Close() // best effort cleanup on exit
+		os.Exit(1)
+	}
+
+	httpServe := &http.Server{
+		Addr:              ":" + srv.cfg.Port,
+		Handler:           mux,
+		ReadHeaderTimeout: 5 * time.Second,
+	}
+
+	// gracefully quit the server
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+
+	go func() {
+		slog.Info("serving pkgstash", "root", cfg.CacheRoot, "port", cfg.Port)
+		if err = httpServe.ListenAndServe(); err != http.ErrServerClosed {
+			slog.Error("server failed", "err", err)
+			_ = c.Close() // best effort cleanup on exit
+			os.Exit(1)
+		}
+	}()
+
+	<-quit
+	slog.Info("shutting down")
+
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	if err := httpServe.Shutdown(ctx); err != nil {
+		slog.Error("shutdown failed", "err", err)
+	}
+}