package redis

import (
	"context"
	"fmt"
	"gitee.com/h79/goutils/common/debug"
	"gitee.com/h79/goutils/common/result"
	"gitee.com/h79/goutils/dao/config"
	"gitee.com/h79/goutils/plugins"
	"github.com/go-redis/redis/v8"
	"go.uber.org/zap"
	"time"
)

var _ Redis = (*Adapter)(nil)

type Adapter struct {
	client   *redis.Client
	sentinel *redis.SentinelClient
	cluster  *redis.ClusterClient
	name     string
}

func NewAdapter(name string, cfg *config.RedisConfig, sentinelCfg *config.RedisConfig, clusterCfg *config.RedisConfig) (*Adapter, error) {
	cli, err := newClient(cfg)
	if err != nil {
		return nil, err
	}
	sentinel, err := newSentinelClient(sentinelCfg)
	if err != nil {
		zap.L().Warn("Redis: newSentinelClient", zap.Error(err))
	}
	cluster, err := newClusterClient(clusterCfg)
	if err != nil {
		zap.L().Warn("Redis: newClusterClient", zap.Error(err))
	}
	return &Adapter{client: cli, sentinel: sentinel, cluster: cluster, name: name}, nil
}

func newClient(cfg *config.RedisConfig) (*redis.Client, error) {
	if len(cfg.Host) == 0 || len(cfg.Host[0]) <= 0 {
		return nil, result.RErrParam
	}
	cli := redis.NewClient(&redis.Options{
		Network:      "tcp",
		Addr:         cfg.Host[0],
		Username:     cfg.User,
		Password:     cfg.Pwd, // no password set
		DB:           cfg.DB,  // use default DB
		DialTimeout:  cfg.DialTimeout * time.Second,
		ReadTimeout:  cfg.ReadTimeout * time.Second,
		WriteTimeout: cfg.WriteTimeout * time.Second,
		IdleTimeout:  cfg.IdleTimeout * time.Minute,
	})

	ctx := context.Background()
	if _, err := cli.Ping(ctx).Result(); err != nil {
		alarm(fmt.Sprintf("redis ping(%s)", cfg.Host), err)
	}
	return cli, nil
}

func newSentinelClient(cfg *config.RedisConfig) (*redis.SentinelClient, error) {
	if len(cfg.Host) == 0 || len(cfg.Host[0]) <= 0 {
		return nil, result.RErrParam
	}

	cli := redis.NewSentinelClient(&redis.Options{
		Network:      "tcp",
		Addr:         cfg.Host[0],
		Username:     cfg.User,
		Password:     cfg.Pwd, // no password set
		DB:           cfg.DB,  // use default DB
		DialTimeout:  cfg.DialTimeout * time.Second,
		ReadTimeout:  cfg.ReadTimeout * time.Second,
		WriteTimeout: cfg.WriteTimeout * time.Second,
		IdleTimeout:  cfg.IdleTimeout * time.Minute,
	})

	ctx := context.Background()
	if _, err := cli.Ping(ctx).Result(); err != nil {
		alarm(fmt.Sprintf("redis ping(%s)", cfg.Host), err)
	}
	return cli, nil
}

func newClusterClient(cfg *config.RedisConfig) (*redis.ClusterClient, error) {
	if len(cfg.Host) == 0 || len(cfg.Host[0]) <= 0 {
		return nil, result.RErrParam
	}

	cli := redis.NewClusterClient(&redis.ClusterOptions{
		Addrs:        cfg.Host,
		Username:     cfg.User,
		Password:     cfg.Pwd, // no password set
		DialTimeout:  cfg.DialTimeout * time.Second,
		ReadTimeout:  cfg.ReadTimeout * time.Second,
		WriteTimeout: cfg.WriteTimeout * time.Second,
		IdleTimeout:  cfg.IdleTimeout * time.Minute,
	})

	ctx := context.Background()
	if _, err := cli.Ping(ctx).Result(); err != nil {
		alarm(fmt.Sprintf("redis ping(%s)", cfg.Host), err)
	}
	return cli, nil
}

func (a *Adapter) Rds() *redis.Client {
	return a.client
}

func (a *Adapter) Sentinel() *redis.SentinelClient {
	return a.sentinel
}

func (a *Adapter) Cluster() *redis.ClusterClient {
	return a.cluster
}

func (a *Adapter) Name() string {
	return a.name
}

func alarm(detail string, err error) {
	d := debug.New(result.ErrRdsPingInternal).
		WithDetail(detail).
		WithLevel(debug.DFatalLevel).
		WithError(err)
	_ = plugins.DoWithError(plugins.KAlarm, d)
}