package storage

import (
	"context"
	"github.com/go-redis/redis/v9"
	"github.com/spf13/viper"
)

type RedisHandler struct {
	client *redis.Client
}

func NewRedisHandler(cfg *viper.Viper) (*RedisHandler, error) {
	redisClient := redis.NewClient(&redis.Options{
		Addr: cfg.GetString("storage.kv.redis.addr"),
		DB:   cfg.GetInt("storage.kv.redis.db"),
	})

	_, err := redisClient.Ping(context.Background()).Result()
	if err != nil {
		return nil, err
	}

	return &RedisHandler{
		client: redisClient,
	}, nil
}

func (rh *RedisHandler) Put(k string, v any) error {
	return rh.client.Set(context.Background(), k, v, 0).Err()
}

func (rh *RedisHandler) Get(k string) (any, error) {
	value, err := rh.client.Get(context.Background(), k).Result()
	if err != nil {
		return nil, err
	}
	return value, nil
}

func (rh *RedisHandler) Delete(k string) error {
	return rh.client.Del(context.Background(), k).Err()
}