storage/internal/vault/vault.go

package vault

import (
	"context"
	"encoding/base64"
	"errors"

	"code.uint32.ru/tiny/storage/internal/errinternal"
	"github.com/hashicorp/vault/api"
)

var (
	ErrNotFound = errinternal.ErrNotFound
)

type Storage struct {
	kv *api.KVv1
	// TODO: kv2: *api.KVv2
}

// New returns Storage writing to the specified vault path.
// Object will be base64 encoded and written to path/key.
func New(c *api.Client, path string) *Storage {
	return &Storage{kv: c.KVv1(path)}
}

func (s *Storage) Save(key string, data []byte) error {
	str := base64.StdEncoding.EncodeToString(data)
	m := map[string]any{
		"data": map[string]string{
			"payload": str,
		},
	}

	if err := s.kv.Put(context.Background(), "testkey", m); err != nil {
		return err
	}

	return nil
}

func (s *Storage) Load(key string) ([]byte, error) {
	m, err := s.kv.Get(context.Background(), key)
	if err != nil && errors.Is(err, api.ErrSecretNotFound) {
		return nil, errors.Join(ErrNotFound, err)
	} else if err != nil {
		return nil, err
	}

	data, ok := m.Data["data"] // map[string]any
	if !ok {
		return nil, errors.New("no data found")
	}

	payloadmap, ok := data.(map[string]any)
	if !ok {
		return nil, errors.New("no payload map")
	}

	rawb, ok := payloadmap["payload"]
	if !ok {
		return nil, errors.New("no payload bytes")
	}

	str, ok := rawb.(string)
	if !ok {
		return nil, errors.New("could not convert payload to bytes")
	}

	b := []byte{}

	b, err = base64.StdEncoding.AppendDecode(b, []byte(str))
	if err != nil {
		return nil, err
	}

	return b, nil
}

func (s *Storage) Delete(key string) error {
	if err := s.kv.Delete(context.Background(), key); err != nil {
		return err
	}

	return nil
}