package redisx

import (
	"context"
	"crypto/tls"

	redis "github.com/redis/go-redis/v9"
)

// PoolSize：连接池中的最大连接数。
// MinIdleConns：连接池中的最少空闲连接数。
// MaxConnAge：连接池中连接的最长寿命。
// PoolTimeout：当所有连接都被使用时，客户端等待一个连接被释放到连接池的最长时间。
// IdleTimeout：在连接池中保持空闲状态的最长时间，超过这个时间的空闲连接将被关闭。
// IdleCheckFrequency：检查空闲连接的频率。

// 单机
func NewRedis(ctx context.Context, opts ...Option) redis.UniversalClient {
	opt := defaultOptions()
	for _, apply := range opts {
		apply(&opt)
	}

	var tlsConfig *tls.Config
	if opt.enableTLS {
		tlsConfig = &tls.Config{InsecureSkipVerify: true}
	}

	var client redis.UniversalClient

	switch opt.mode {
	case RedisModeCluster:
		// 集群模式
		if len(opt.address) == 0 {
			panic("redis cluster mode requires at least one address")
		}
		client = redis.NewClusterClient(
			&redis.ClusterOptions{
				Addrs:     opt.address,
				Password:  opt.password, // no password set
				TLSConfig: tlsConfig,
			},
		)
	case RedisModeSentinel:
		// 哨兵模式
		if len(opt.sentinels) == 0 {
			panic("redis sentinel mode requires at least one sentinel address")
		}
		client = redis.NewFailoverClient(
			&redis.FailoverOptions{
				MasterName:    opt.masterName,
				SentinelAddrs: opt.sentinels,
				Password:      opt.password, // no password set
				TLSConfig:     tlsConfig,
				DB:            opt.db, // use default DB
				// SentinelPassword: global.Config.Redis.Password, // 哨兵认证密码（如果有）
				// SentinelUsername: global.Config.Redis.Username, // 哨兵认证用户名（如果有）
			},
		)
	default:
		// 单机模式
		if len(opt.address) == 0 {
			panic("redis single mode requires an address")
		}
		client = redis.NewClient(
			&redis.Options{
				Addr:      opt.address[0],
				Password:  opt.password, // no password set
				DB:        opt.db,       // use default DB
				TLSConfig: tlsConfig,
			},
		)
	}

	_, err := client.Ping(ctx).Result()
	if err != nil {
		panic(err)
	}
	return client
}