package cache

import (
	"MossChat/config"
	"MossChat/logger"
	"MossChat/utils"
	"github.com/go-redis/redis"
	"time"
)

var DB *redis.Client

func InitRedis() {
	DB = redis.NewClient(&redis.Options{
		Addr: config.REDIS_ADDR,
		DB:   0,
	})
	_, err := DB.Ping().Result()
	if err != nil {
		logger.Log.Println("无法连接到Redis!")
		panic(err)
	}
	logger.Log.Println("已加载redis!")
}

// GenerateToken 计算新的Token
func GenerateToken(uid int, duration time.Duration) error {
	err := DB.Set(utils.ToTokenKey(utils.RandomString(16)), uid, duration).Err()
	if err != nil {
		return err
	}
	return nil
}

// 获取用户ID
func GetUidByToken(token string) (int, error) {
	uid, err := DB.Get(utils.ToTokenKey(token)).Int()
	if err != nil {
		return 0, err
	}
	return uid, nil
}

// 移除Token
func RemoveToken(token string) error {
	err := DB.Del(utils.ToTokenKey(token)).Err()
	if err != nil {
		return err
	}
	return nil
}