package main

import (
	"context"
	"log"

	"github.com/jackc/pgx/v5/pgxpool"
	_ "github.com/jackc/pgx/v5/stdlib"
	rkboot "github.com/rookie-ninja/rk-boot/v2"
	rkentry "github.com/rookie-ninja/rk-entry/v2/entry"
	rkgrpc "github.com/rookie-ninja/rk-grpc/v2/boot"
	"google.golang.org/grpc"

	"git.techease.ru/Smart-search/smart-search-back/internal/config"
	"git.techease.ru/Smart-search/smart-search-back/internal/database"
	grpcServer "git.techease.ru/Smart-search/smart-search-back/internal/grpc"
	"git.techease.ru/Smart-search/smart-search-back/internal/repository"
	"git.techease.ru/Smart-search/smart-search-back/internal/worker"
)

func main() {
	cfg, err := config.Load("config/config.yaml")
	if err != nil {
		log.Fatalf("Failed to load config: %v", err)
	}

	ctx := context.Background()

	if err := database.RunMigrations(cfg.DatabaseURL()); err != nil {
		log.Fatalf("Failed to run migrations: %v", err)
	}

	pool, err := pgxpool.New(ctx, cfg.DatabaseURL())
	if err != nil {
		log.Fatalf("Failed to connect to database: %v", err)
	}
	defer pool.Close()

	if err := pool.Ping(ctx); err != nil {
		log.Fatalf("Failed to ping database: %v", err)
	}

	log.Println("Successfully connected to database")

	boot := rkboot.NewBoot(rkboot.WithBootConfigPath("config/boot.yaml", nil))

	grpcEntry := rkgrpc.GetGrpcEntry("smart-search-service")
	if grpcEntry == nil {
		log.Fatal("Failed to get gRPC entry from rk-boot")
	}

	loggerEntry := rkentry.GlobalAppCtx.GetLoggerEntry("smart-search-service")
	if loggerEntry == nil {
		loggerEntry = rkentry.GlobalAppCtx.GetLoggerEntryDefault()
	}
	logger := loggerEntry.Logger

	sessionRepo := repository.NewSessionRepository(pool)
	inviteRepo := repository.NewInviteRepository(pool)

	sessionCleaner := worker.NewSessionCleaner(ctx, sessionRepo)
	sessionCleaner.Start()
	defer sessionCleaner.Stop()

	inviteCleaner := worker.NewInviteCleaner(ctx, inviteRepo)
	inviteCleaner.Start()
	defer inviteCleaner.Stop()

	authHandler, userHandler, inviteHandler, requestHandler, supplierHandler := grpcServer.NewHandlers(
		pool,
		cfg.Security.JWTSecret,
		cfg.Security.CryptoSecret,
		cfg.AI.OpenAIKey,
		cfg.AI.PerplexityKey,
		logger,
	)

	grpcEntry.AddRegFuncGrpc(func(s *grpc.Server) {
		grpcServer.RegisterServices(s, authHandler, userHandler, inviteHandler, requestHandler, supplierHandler)
	})

	boot.Bootstrap(ctx)

	log.Println("gRPC server started via rk-boot")

	boot.WaitForShutdownSig(ctx)

	log.Println("Server stopped gracefully")
}