package redis

import (
	"context"
	"encoding/json"
	"strings"

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

type jsonWrapper[T any] struct {
	value T
}

func (w *jsonWrapper[T]) MarshalBinary() ([]byte, error) {
	data, err := json.Marshal(w.value)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return data, nil
}

func (w *jsonWrapper[T]) UnmarshalBinary(data []byte) error {
	if err := json.Unmarshal(data, &w.value); err != nil {
		return errors.WithStack(err)
	}

	return nil
}

func (w *jsonWrapper[T]) UnmarshalText(data []byte) error {
	if err := json.Unmarshal(data, &w.value); err != nil {
		return errors.WithStack(err)
	}

	return nil
}

func (w *jsonWrapper[T]) Value() T {
	return w.value
}

func wrap[T any](v T) *jsonWrapper[T] {
	return &jsonWrapper[T]{v}
}

func unwrap[T any](v any) (T, error) {
	str, ok := v.(string)
	if !ok {
		return *new(T), errors.Errorf("could not unwrap value of type '%T'", v)
	}

	u := new(T)

	if err := json.Unmarshal([]byte(str), u); err != nil {
		return *new(T), errors.WithStack(err)
	}

	return *u, nil
}

func key(parts ...string) string {
	return strings.Join(parts, ":")
}

func WithTx(ctx context.Context, client redis.UniversalClient, key string, fn func(ctx context.Context, tx *redis.Tx) error) error {
	txf := func(tx *redis.Tx) error {
		if err := fn(ctx, tx); err != nil {
			return errors.WithStack(err)
		}

		return nil
	}

	err := client.Watch(ctx, txf, key)
	if err != nil {
		return errors.WithStack(err)
	}

	return nil
}

func contains[T ~string](values []T, v T) bool {
	for _, vv := range values {
		if vv == v {
			return true
		}
	}

	return false
}