package main

import (
	"context"
	"fmt"
	"net"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/teacinema/media-service/internal/config"
	"github.com/teacinema/media-service/internal/infrastructure/grpc"
	httpserver "github.com/teacinema/media-service/internal/infrastructure/http"
	"github.com/teacinema/media-service/internal/infrastructure/storage"
	"github.com/teacinema/media-service/pkg/logger"
)

func main() {
	cfg := config.Load()

	logger.Init(cfg.Logging.Level)

	logger.Info("🚀 Starting media-service in %s mode", cfg.App.Env)

	var mediaStorage storage.Storage
	var err error

	mediaStorage, err = storage.NewS3Storage(cfg)
	if err != nil {
		logger.Fatal("failed to init S3 storage: %v", err)
	}
	logger.Info("✅ S3 storage connected (bucket: %s)", cfg.Storage.Bucket)

	grpcServer := grpc.NewServer(mediaStorage, cfg)
	grpcListener, err := net.Listen("tcp", fmt.Sprintf(":%s", cfg.GRPC.Port))

	if err != nil {
		logger.Fatal("failed to listen gRPC: %v", err)
	}

	httpSrv := httpserver.NewServer(mediaStorage, cfg)

	go func() {
		logger.Info("gRPC listening on :%s", cfg.GRPC.Port)
		if err := grpcServer.Serve(grpcListener); err != nil {
			logger.Fatal("gRPC serve error: %v", err)
		}
	}()

	go func() {
		logger.Info("HTTP listening on :%s", cfg.HTTP.Port)
		if err := httpSrv.Start(); err != nil {
			logger.Fatal("HTTP server error: %v", err)
		}
	}()

	go func() {
		logger.Info("📦 Background queue workers started")
	}()

	waitForShutdown(func() {
		logger.Warn("🛑 Graceful shutdown started...")
		grpcServer.GracefulStop()
		mediaStorage.Close()
		httpSrv.Stop(context.Background())
		logger.Info("✅ Shutdown complete")
	})
}

func waitForShutdown(cleanup func()) {
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
	<-quit

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

	cleanup()

	<-ctx.Done()
}