main.go

package main

import (
	"context"
	"log"
	"net/http"
	"os"
	"os/signal"
	"path/filepath"
	"syscall"
	"time"

	"github.com/MathiasDPX/archivetube/internal/archive"
	"github.com/MathiasDPX/archivetube/internal/config"
	"github.com/MathiasDPX/archivetube/internal/metrics"
	"github.com/MathiasDPX/archivetube/internal/queue"
	"github.com/MathiasDPX/archivetube/internal/store"
	"github.com/MathiasDPX/archivetube/internal/traces"
	"github.com/MathiasDPX/archivetube/internal/web"
)

func main() {
	ctx := context.Background()

	cfg := config.Load("config.toml")

	if cfg.Observability.OTelExporter != "" {
		tracer, err := traces.Init(cfg.Observability, ctx)
		if err != nil {
			log.Fatalf("initializing tracer: %v", err)
		}
		defer func() {
			if err := tracer(ctx); err != nil {
				log.Printf("shutting down tracer: %v", err)
			}
		}()
	}

	if err := os.MkdirAll(cfg.Archive.DataDir, 0o755); err != nil {
		log.Fatalf("creating data dir: %v", err)
	}

	dbPath := filepath.Join(cfg.Archive.DataDir, "archivetube.db")
	st, err := store.New(dbPath, cfg)
	if err != nil {
		log.Fatalf("opening store: %v", err)
	}
	defer st.Close()

	if n, err := st.CountVideos(ctx); err == nil {
		metrics.SetVideosTotal(n)
		metrics.SetArchivedVideosTotal(n)
	}
	if n, err := st.CountChannels(ctx); err == nil {
		metrics.SetChannelsTotal(n)
	}
	if n, err := st.SumArchiveSize(ctx); err == nil {
		metrics.SetArchiveSizeBytes(n)
	}

	archiveSvc := archive.New(cfg.Archive.YtDlpPath, cfg.Archive.DataDir, cfg.Archive.Proxy, st, &cfg.SmartSearch)
	q := queue.New(archiveSvc.ArchiveURL)

	if sha := os.Getenv("GIT_SHA"); sha != "" {
		web.SetGitSHA(sha)
	}
	webPaths := web.DefaultWebPaths()
	tmpl, err := web.NewTemplates(webPaths.TemplateDir)
	if err != nil {
		log.Fatalf("loading templates: %v", err)
	}

	router := web.NewRouter(cfg, st, archiveSvc, q, tmpl, webPaths.StaticDir)

	srv := &http.Server{
		Addr:              cfg.Server.ListenAddr,
		Handler:           router,
		ReadHeaderTimeout: 5 * time.Second,
	}

	quit := make(chan os.Signal, 1)
	signal.Notify(quit, os.Interrupt, syscall.SIGTERM)

	go func() {
		log.Printf("archivetube listening on %s", cfg.Server.ListenAddr)
		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Fatalf("server error: %v", err)
		}
	}()

	<-quit
	log.Println("shutting down...")

	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	if err := srv.Shutdown(ctx); err != nil {
		log.Printf("server shutdown error: %v", err)
	}
}