package gredis

import (
	"time"

	"github.com/gomodule/redigo/redis"
)

type Conf struct {
	Addr        string
	MaxIdle     int
	MaxActive   int
	IdleTimeout time.Duration
	Wait        bool
	DB          string
	PassWord    string
}
type ReidsConn struct {
	redis.Conn
}

func (c *Conf) SetPool() redis.Pool {
	pool := redis.Pool{
		MaxIdle:     c.MaxIdle,
		MaxActive:   c.MaxActive,
		IdleTimeout: c.IdleTimeout,
		Wait:        c.Wait,
		Dial: func() (redis.Conn, error) {
			conn, err := redis.Dial("tcp", c.Addr)
			if err != nil {
				return conn, err
			}

			if c.PassWord != "" {
				_, err = conn.Do("auth", c.PassWord)
				if err != nil {
					return conn, err
				}
			}
			if c.DB != "" {
				_, err = conn.Do("select", c.DB)
				if err != nil {
					return conn, err
				}
			}

			return conn, err
		},
	}

	return pool
}

func SetConn(pool redis.Pool) ReidsConn {
	defer pool.Close()
	conn := ReidsConn{
		pool.Get(),
	}
	return conn
}