package redis

import (
	"context"
	"gitee.com/h79/goutils/common/result"
	"gitee.com/h79/goutils/common/server"
	"github.com/go-redis/redis/v8"
	"go.uber.org/zap"
)

var _ Client = (*Connector)(nil)

type Connector struct {
	selectName string
	clientMap  map[string]*Adapter
}

func NewRedis(conf []server.Redis) (Client, error) {
	zap.L().Info("Redis", zap.Any("Config", conf))
	clients := &Connector{clientMap: map[string]*Adapter{}}

	for i := range conf {
		adapter, err := NewAdapter(conf[i].Name, &conf[i].Master, &conf[i].Sentinel, &conf[i].Cluster)
		if err != nil {
			continue
		}
		if conf[i].Logger.LogLevel > 0 {
			redis.SetLogger(&Logger{
				RedisLogger: conf[i].Logger,
			})
		}
		clients.clientMap[conf[i].Name] = adapter
	}
	return clients, nil
}

func (c *Connector) Get(name string) (Redis, error) {
	if len(name) == 0 {
		name = c.selectName
	}
	cli, exists := c.clientMap[name]
	if exists == true {
		return cli, nil
	}
	return nil, result.Error(result.ErrNotFound, "Not found")
}

func (c *Connector) Close(name string) error {
	if len(name) == 0 {
		name = c.selectName
	}
	cli, exists := c.clientMap[name]
	if exists == true {
		return cli.client.Close()
	}
	return nil
}

func (c *Connector) CloseAll() {
	for _, cli := range c.clientMap {
		_ = cli.client.Close()
	}
	c.clientMap = nil
}

func (c *Connector) Select(name string) {
	c.selectName = name
}

func (c *Connector) SelectDB(name string, index int) error {
	cli, err := c.Get(name)
	if err != nil {
		return err
	}
	_, err = cli.Rds().Do(context.Background(), "select", index).Result()
	return err
}