refactor(server): simplify main.go startup

cmd/server/main.go

@@ -3,14 +3,12 @@ package main
 import (
 	"context"
 	"database/sql"
-	"errors"
 	"fmt"
 	"log"
 	"net/http"
 	"os"
 	"os/signal"
 	"path/filepath"
-	"strings"
 	"syscall"
 	"time"
 
@@ -24,71 +22,33 @@ import (
 )
 
 func main() {
-	dbFile := strings.TrimSpace(os.Getenv("DATABASE_FILE"))
-	if dbFile == "" {
-		dbFile = "mal.db"
-	}
-
-	dsn := fmt.Sprintf("file:%s?_foreign_keys=on", dbFile)
-	db, err := sql.Open("sqlite3", dsn)
+	db, err := sql.Open("sqlite3", fmt.Sprintf("file:%s?_foreign_keys=on", dbFile()))
 	if err != nil {
 		log.Fatalf("failed to open db: %v", err)
 	}
 	defer db.Close()
 
-	if err := db.Ping(); err != nil {
-		log.Fatalf("failed to ping db: %v", err)
-	}
-
-	if _, err := db.Exec("PRAGMA foreign_keys = ON"); err != nil {
-		log.Fatalf("failed to enforce sqlite foreign keys: %v", err)
-	}
-
-	var fkState int
-	if err := db.QueryRow("PRAGMA foreign_keys").Scan(&fkState); err != nil {
-		log.Fatalf("failed to verify sqlite foreign keys: %v", err)
-	}
-	if fkState != 1 {
-		log.Fatal("sqlite foreign keys are disabled")
-	}
-
-	migrationsDir, err := resolveMigrationsDir()
-	if err != nil {
-		log.Fatalf("failed to locate migrations directory: %v", err)
-	}
-
-	// Run migrations with tracking
+	migrationsDir := migrationsDir()
 	if err := database.RunMigrations(db, migrationsDir); err != nil {
 		log.Fatalf("failed to run migrations: %v", err)
 	}
 
-	playbackSecret := strings.TrimSpace(os.Getenv("PLAYBACK_PROXY_SECRET"))
-	if len(playbackSecret) < 32 {
-		log.Fatal("PLAYBACK_PROXY_SECRET must be set and at least 32 characters")
-	}
-
 	queries := database.New(db)
-	authService := auth.NewService(queries)
 	jikanClient := jikan.NewClient(queries)
 
-	// Start background workers
-	relationsWorker := worker.New(queries, jikanClient)
-
 	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
 	defer stop()
 
-	go relationsWorker.Start(ctx)
+	go worker.New(queries, jikanClient).Start(ctx)
 
 	app := server.Config{
 		DB:                  queries,
 		SQLDB:               db,
 		JikanClient:         jikanClient,
-		AuthService:         authService,
-		PlaybackProxySecret: playbackSecret,
+		AuthService:         auth.NewService(queries),
+		PlaybackProxySecret: playbackSecret(),
 	}
 
-	handler := server.NewRouter(app)
-
 	port := os.Getenv("PORT")
 	if port == "" {
 		port = "3000"
@@ -96,22 +56,14 @@ func main() {
 
 	httpServer := &http.Server{
 		Addr:              ":" + port,
-		Handler:           handler,
+		Handler:           server.NewRouter(app),
 		ReadHeaderTimeout: 5 * time.Second,
 		ReadTimeout:       10 * time.Second,
 		WriteTimeout:      30 * time.Second,
 		IdleTimeout:       120 * time.Second,
 	}
 
-	go func() {
-		<-ctx.Done()
-		shutdownCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
-		defer cancel()
-
-		if err := httpServer.Shutdown(shutdownCtx); err != nil {
-			log.Printf("server shutdown failed: %v", err)
-		}
-	}()
+	go gracefulShutdown(httpServer, ctx)
 
 	log.Printf("Server starting on http://localhost:%s", port)
 	if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
@@ -119,68 +71,34 @@ func main() {
 	}
 }
 
-func resolveMigrationsDir() (string, error) {
-	configured := strings.TrimSpace(os.Getenv("MIGRATIONS_DIR"))
-	if configured != "" {
-		hasFiles, err := directoryHasSQLFiles(configured)
-		if err != nil {
-			return "", err
+func dbFile() string {
+	if f := os.Getenv("DATABASE_FILE"); f != "" {
+		return f
 	}
-
-		if !hasFiles {
-			return "", fmt.Errorf("MIGRATIONS_DIR has no .sql files: %s", configured)
-		}
-
-		return configured, nil
-	}
-
-	workingDir, err := os.Getwd()
-	if err != nil {
-		return "", err
-	}
-
-	executablePath, err := os.Executable()
-	if err != nil {
-		return "", err
-	}
-
-	candidates := []string{
-		filepath.Join(workingDir, "migrations"),
-		filepath.Join(filepath.Dir(executablePath), "migrations"),
-	}
-
-	for _, candidate := range candidates {
-		hasFiles, checkErr := directoryHasSQLFiles(candidate)
-		if checkErr != nil {
-			if errors.Is(checkErr, os.ErrNotExist) {
-				continue
-			}
-
-			return "", checkErr
-		}
-
-		if hasFiles {
-			return candidate, nil
-		}
-	}
-
-	return "", errors.New("could not find migrations directory")
+	return "mal.db"
 }
 
-func directoryHasSQLFiles(dir string) (bool, error) {
-	info, err := os.Stat(dir)
-	if err != nil {
-		return false, err
+func migrationsDir() string {
+	if dir := os.Getenv("MIGRATIONS_DIR"); dir != "" {
+		return dir
+	}
+	wd, _ := os.Getwd()
+	return filepath.Join(wd, "migrations")
+}
+
+func playbackSecret() string {
+	secret := os.Getenv("PLAYBACK_PROXY_SECRET")
+	if len(secret) < 32 {
+		log.Fatal("PLAYBACK_PROXY_SECRET must be set and at least 32 characters")
+	}
+	return secret
+}
+
+func gracefulShutdown(srv *http.Server, ctx context.Context) {
+	<-ctx.Done()
+	shutdownCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+	if err := srv.Shutdown(shutdownCtx); err != nil {
+		log.Printf("server shutdown failed: %v", err)
 	}
-
-	if !info.IsDir() {
-		return false, fmt.Errorf("not a directory: %s", dir)
-	}
-
-	files, err := filepath.Glob(filepath.Join(dir, "*.sql"))
-	if err != nil {
-		return false, err
-	}
-
-	return len(files) > 0, nil
 }