feat: objects in vault are stored as a single secret

Co-authored-by: Dmitry Fedotov <dmitry@uint32.ru>
Co-committed-by: Dmitry Fedotov <dmitry@uint32.ru>

internal/vault/vault.go

@@ -4,9 +4,13 @@ import (
 	"context"
 	"encoding/base64"
 	"errors"
+	"fmt"
+	"path"
+	"strings"
+
+	"github.com/hashicorp/vault/api"
 
 	"code.uint32.ru/tiny/storage/internal/errinternal"
-	"github.com/hashicorp/vault/api"
 )
 
 var (
@@ -14,25 +18,33 @@ var (
 )
 
 type Storage struct {
-	kv *api.KVv1
+	key string
+	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)}
+// Objects will be base64 encoded and written to path as a single
+// secret.
+func New(c *api.Client, secretpath string) *Storage {
+	p, k := path.Split(secretpath)
+	p = strings.Trim(p, "/")
+	k = strings.Trim(k, "/")
+
+	return &Storage{kv: c.KVv1(p), key: k}
 }
 
 func (s *Storage) Save(key string, data []byte) error {
-	str := base64.StdEncoding.EncodeToString(data)
-	m := map[string]any{
-		"data": map[string]string{
-			"payload": str,
-		},
+	datamap, err := s.getData()
+	if err != nil {
+		return err
 	}
 
-	if err := s.kv.Put(context.Background(), "testkey", m); err != nil {
+	str := base64.StdEncoding.EncodeToString(data)
+
+	datamap[key] = str
+
+	if err := s.putData(datamap); err != nil {
 		return err
 	}
 
@@ -40,9 +52,54 @@ func (s *Storage) Save(key string, data []byte) error {
 }
 
 func (s *Storage) Load(key string) ([]byte, error) {
-	m, err := s.kv.Get(context.Background(), key)
+	data, err := s.getData()
+	if err != nil {
+		return nil, err
+	}
+
+	payload, ok := data[key]
+	if !ok {
+		return nil, errors.Join(ErrNotFound, errors.New("key not found in stored secret"))
+	}
+
+	str, ok := payload.(string)
+	if !ok {
+		// returning ErrNotFound because this will fail again on retry
+		return nil, errors.Join(ErrNotFound, errors.New("could not convert payload to string"))
+	}
+
+	b := []byte{}
+
+	b, err = base64.StdEncoding.AppendDecode(b, []byte(str))
+	if err != nil {
+		return nil, errors.Join(ErrNotFound, fmt.Errorf("could not base64 decode value: %w", err))
+	}
+
+	return b, nil
+}
+
+func (s *Storage) putData(data map[string]any) error {
+	m := map[string]any{
+		"data": data,
+	}
+
+	if err := s.kv.Put(context.Background(), s.key, m); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (s *Storage) getData() (map[string]any, error) {
+	m, err := s.kv.Get(context.Background(), s.key)
 	if err != nil && errors.Is(err, api.ErrSecretNotFound) {
-		return nil, errors.Join(ErrNotFound, err)
+		// will create secret with no payload
+		m := make(map[string]any)
+		if err := s.putData(m); err != nil {
+			return nil, err
+		}
+
+		return m, nil
 	} else if err != nil {
 		return nil, err
 	}
@@ -52,33 +109,27 @@ func (s *Storage) Load(key string) ([]byte, error) {
 		return nil, errors.New("no data found")
 	}
 
-	payloadmap, ok := data.(map[string]any)
+	datamap, 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
+	return datamap, nil
 }
 
 func (s *Storage) Delete(key string) error {
-	if err := s.kv.Delete(context.Background(), key); err != nil {
+	data, err := s.getData()
+	if err != nil {
+		return err
+	}
+
+	if _, ok := data[key]; !ok {
+		return nil
+	}
+
+	delete(data, key)
+
+	if err := s.putData(data); err != nil {
 		return err
 	}
 

internal/vault/vault_test.go

@@ -38,7 +38,7 @@ func TestVaultStorage(t *testing.T) {
 	data := []byte("this is a test")
 
 	if err := st.Save(testkey, data); err != nil {
-		t.Error(err)
+		t.Fatalf("error saving data: %v", err)
 	}
 
 	b, err := st.Load(testkey)