feat: add Vault
Co-authored-by: Dmitry Fedotov <dmitry@uint32.ru> Co-committed-by: Dmitry Fedotov <dmitry@uint32.ru>
This commit is contained in:
99
internal/vault/vault.go
Normal file
99
internal/vault/vault.go
Normal file
@@ -0,0 +1,99 @@
|
||||
package vault
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/base64"
|
||||
"errors"
|
||||
|
||||
"github.com/hashicorp/vault/api"
|
||||
)
|
||||
|
||||
type Storage struct {
|
||||
client *api.Client
|
||||
path string
|
||||
}
|
||||
|
||||
func Open(token string, path string, addr string) (*Storage, error) {
|
||||
conf := &api.Config{
|
||||
Address: addr,
|
||||
}
|
||||
|
||||
c, err := api.NewClient(conf)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
c.SetToken(token)
|
||||
|
||||
return &Storage{client: c, path: path}, nil
|
||||
}
|
||||
|
||||
func (s *Storage) Save(key string, data []byte) error {
|
||||
kv := s.client.KVv1(s.path)
|
||||
|
||||
str := base64.StdEncoding.EncodeToString(data)
|
||||
m := map[string]any{
|
||||
"data": map[string]string{
|
||||
"payload": str,
|
||||
},
|
||||
}
|
||||
|
||||
if err := kv.Put(context.Background(), "testkey", m); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Storage) Load(key string) ([]byte, error) {
|
||||
kv := s.client.KVv1(s.path)
|
||||
|
||||
m, err := kv.Get(context.Background(), key)
|
||||
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 {
|
||||
kv := s.client.KVv1(s.path)
|
||||
|
||||
if err := kv.Delete(context.Background(), key); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Storage) Close() error {
|
||||
s.client.ClearToken()
|
||||
return nil
|
||||
}
|
Reference in New Issue
Block a user