package redisx

import (
	"context"
	"errors"
	"fmt"
	"strconv"
	"time"

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

// ---------------- 基础方法 ----------------

// Set 设置任意 key
func Set(key string, value interface{}, expire time.Duration) error {
	return Client().Set(context.Background(), key, value, expire).Err()
}

// Get 获取任意 key 的值，返回 string
func Get(key string) (string, error) {
	val, err := Client().Get(context.Background(), key).Result()
	if errors.Is(err, redis.Nil) {
		return "", nil
	}
	return val, err
}

// Delete 删除任意 key
func Delete(key string) error {
	return Client().Del(context.Background(), key).Err()
}

// Exists 判断 key 是否存在
func Exists(key string) (bool, error) {
	cnt, err := Client().Exists(context.Background(), key).Result()
	if err != nil {
		return false, err
	}
	return cnt > 0, nil
}

// ================== Space User / Role ==================

const (
	RoleMember int32 = 1
	RoleAdmin  int32 = 5
	RoleOwner  int32 = 9

	SpaceUserTTL = 10 * time.Minute
)

// space:user:{spaceId}:{userId}
func SpaceUserKey(spaceId int64, userId int64) string {
	return fmt.Sprintf("space:user:%d:%d", spaceId, userId)
}

// Set / Create space user role
func SetSpaceUserRole(spaceId int64, userId int64, role int32) error {
	key := SpaceUserKey(spaceId, userId)
	return Set(key, role, SpaceUserTTL)
}

// Update space user role (same as set, semantic clarity)
func UpdateSpaceUserRole(spaceId int64, userId int64, role int32) error {
	return SetSpaceUserRole(spaceId, userId, role)
}

// Get space user role
// return 0 means not exists
func GetSpaceUserRole(spaceId int64, userId int64) (int32, error) {
	key := SpaceUserKey(spaceId, userId)
	val, err := Get(key)
	if err != nil {
		return 0, err
	}
	if val == "" {
		return 0, nil
	}

	r, err := strconv.ParseInt(val, 10, 32)
	if err != nil {
		return 0, err
	}
	return int32(r), nil
}

// Delete space user (kick / dissolve)
func DeleteSpaceUser(spaceId int64, userId int64) error {
	key := SpaceUserKey(spaceId, userId)
	return Delete(key)
}

// Exists space user
func ExistsSpaceUser(spaceId int64, userId int64) (bool, error) {
	key := SpaceUserKey(spaceId, userId)
	return Exists(key)
}

// Permission check
// owner > admin > member
func HasPermission(userRole int32, needRole int32) bool {
	return userRole >= needRole
}