package apiserver

import (
	"context"

	"github.com/marmotedu/log"
	"github.com/pingcap-inc/pinsights/internal/apiserver/config"
	"github.com/pingcap-inc/pinsights/internal/apiserver/store"
	"github.com/pingcap-inc/pinsights/internal/apiserver/store/mysql"
	"github.com/pingcap-inc/pinsights/internal/pkg/authz"
	genericoptions "github.com/pingcap-inc/pinsights/internal/pkg/options"
	genericapiserver "github.com/pingcap-inc/pinsights/internal/pkg/server"
	"github.com/pingcap-inc/pinsights/pkg/shutdown"
	"github.com/pingcap-inc/pinsights/pkg/shutdown/shutdownmanagers/posixsignal"
	"github.com/pingcap-inc/pinsights/pkg/storage"
)

type apiServer struct {
	gs               *shutdown.GracefulShutdown
	genericAPIServer *genericapiserver.GenericAPIServer
	redisOptions     *genericoptions.RedisOptions
}

type preparedAPIServer struct {
	*apiServer
}

type ExtraConfig struct {
	mysqlOptions *genericoptions.MySQLOptions
}

func createAPIServer(cfg *config.Config) (*apiServer, error) {
	gs := shutdown.New()
	gs.AddShutdownManager(posixsignal.NewPosixSignalManager())

	genericConfig, err := buildGenericConfig(cfg)
	if err != nil {
		return nil, err
	}

	genericServer, err := genericConfig.Complete().New()
	if err != nil {
		return nil, err
	}

	extraConfig, err := buildExtraConfig(cfg)
	if err != nil {
		return nil, err
	}
	storeIns, _ := mysql.GetMySQLFactoryOr(extraConfig.mysqlOptions)
	store.SetClient(storeIns)

	// we just init enforcers
	authz.GetCasbinEnforcer(extraConfig.mysqlOptions)

	server := &apiServer{
		gs:               gs,
		genericAPIServer: genericServer,
		redisOptions:     cfg.RedisOptions,
	}

	return server, nil
}

func (s *apiServer) PrepareRun() preparedAPIServer {
	initRouter(s.genericAPIServer.Engine)

	s.initRedisStore()

	s.gs.AddShutdownCallback(shutdown.ShutdownFunc(func(string) error {
		s.genericAPIServer.Close()
		return nil
	}))

	return preparedAPIServer{s}
}

func (s preparedAPIServer) Run() error {
	// start shutdown managers
	if err := s.gs.Start(); err != nil {
		log.Fatalf("start shutdown manager failed: %s", err.Error())
	}

	return s.genericAPIServer.Run()
}

func buildGenericConfig(cfg *config.Config) (genericConfig *genericapiserver.Config, lastErr error) {
	genericConfig = genericapiserver.NewConfig()
	if lastErr = cfg.GenericServerRunOptions.ApplyTo(genericConfig); lastErr != nil {
		return
	}

	if lastErr = cfg.InsecureServing.ApplyTo(genericConfig); lastErr != nil {
		return
	}

	return
}

func buildExtraConfig(cfg *config.Config) (*ExtraConfig, error) {
	return &ExtraConfig{
		mysqlOptions: cfg.MySQLOptions,
	}, nil
}

func (s *apiServer) initRedisStore() {
	ctx, cancel := context.WithCancel(context.Background())
	s.gs.AddShutdownCallback(shutdown.ShutdownFunc(func(string) error {
		cancel()
		return nil
	}))

	config := &storage.Config{
		Host:                  s.redisOptions.Host,
		Port:                  s.redisOptions.Port,
		Addrs:                 s.redisOptions.Addrs,
		MasterName:            s.redisOptions.MasterName,
		Username:              s.redisOptions.Username,
		Password:              s.redisOptions.Password,
		Database:              s.redisOptions.Database,
		MaxIdle:               s.redisOptions.MaxIdle,
		MaxActive:             s.redisOptions.MaxActive,
		Timeout:               s.redisOptions.Timeout,
		EnableCluster:         s.redisOptions.EnableCluster,
		UseSSL:                s.redisOptions.UseSSL,
		SSLInsecureSkipVerify: s.redisOptions.SSLInsecureSkipVerify,
	}

	go storage.ConnectToRedis(ctx, config)
}