package utils

import (
	"context"
	"fmt"
	"os"
	"time"

	"github.com/pkg/errors"
	"github.com/valkey-io/valkey-go"

	"go.uber.org/zap"
)

type ConfigRedis struct {
	client valkey.Client
}

func NewConfigRedis(addr string, password string, db int) (*ConfigRedis, error) {
	if addr == "" {
		return nil, fmt.Errorf("addr is empty")
	}
	zap.L().Debug("connect redis", zap.String("addr", addr))
	client, err := valkey.NewClient(valkey.ClientOption{
		InitAddress: []string{addr},
		Password:    password,
		SelectDB:    db,
	})
	if err != nil {
		return nil, err
	}
	return &ConfigRedis{
		client: client,
	}, nil
}

func (r *ConfigRedis) Put(ctx context.Context, key string, value string, ttl time.Duration) error {
	builder := r.client.B().Set().Key(key).Value(value)
	if ttl != TtlKeep {
		builder.Ex(ttl)
	}
	return r.client.Do(ctx, builder.Build()).Error()
}

func (r *ConfigRedis) Get(ctx context.Context, key string) (string, error) {
	v, err := r.client.Do(ctx, r.client.B().Get().Key(key).Build()).ToString()
	if err != nil && errors.Is(err, valkey.Nil) {
		return "", os.ErrNotExist
	}
	return v, err
}

func (r *ConfigRedis) Delete(ctx context.Context, key string) error {
	return r.client.Do(ctx, r.client.B().Del().Key(key).Build()).Error()
}

func (r *ConfigRedis) Close() error {
	r.client.Close()
	return nil
}