package storage

import (
	"context"
	"github.com/go-redis/redis/v8"
	"gorm.io/gorm"
	"strconv"
	"strings"
)

var DB *gorm.DB
var Redis *redis.Client
var Registration map[int64][]byte

func SetState(ctx context.Context, id int64, state string, metadata ...string) error {
	for _, current := range metadata {
		state += "|" + current
	}
	return Redis.Set(ctx, strconv.FormatInt(id, 10), state, 0).Err()
}

func GetState(ctx context.Context, id int64) (string, []string, error) {
	cursor := Redis.Get(ctx, strconv.FormatInt(id, 10))
	if cursor.Err() == redis.Nil {
		return "", make([]string, 0), nil
	} else if cursor.Err() != nil {
		return "", nil, cursor.Err()
	}
	result := strings.Split(cursor.Val(), "|")
	return result[0], result[1:], nil
}

func ClearState(ctx context.Context, id int64) error {
	return Redis.Del(ctx, strconv.FormatInt(id, 10)).Err()
}