andrey
12 months ago
4 changed files with 0 additions and 253 deletions
@ -1,119 +0,0 @@ |
|||
package vault |
|||
|
|||
import ( |
|||
"context" |
|||
"encoding/json" |
|||
"fmt" |
|||
"log" |
|||
"strings" |
|||
|
|||
"github.com/hashicorp/vault/api" |
|||
"gitoa.ru/go-4devs/config" |
|||
"gitoa.ru/go-4devs/config/value" |
|||
) |
|||
|
|||
const ( |
|||
Name = "vault" |
|||
Separator = "/" |
|||
Prefix = "secret/data/" |
|||
ValueName = "value" |
|||
) |
|||
|
|||
var _ config.Provider = (*SecretKV2)(nil) |
|||
|
|||
type SecretOption func(*SecretKV2) |
|||
|
|||
func WithSecretResolve(f func(key []string) (string, string)) SecretOption { |
|||
return func(s *SecretKV2) { s.resolve = f } |
|||
} |
|||
|
|||
func NewSecretKV2(namespace, appName string, client *api.Client, opts ...SecretOption) *SecretKV2 { |
|||
prov := SecretKV2{ |
|||
client: client, |
|||
resolve: func(key []string) (string, string) { |
|||
keysLen := len(key) |
|||
if keysLen == 1 { |
|||
return "", key[0] |
|||
} |
|||
|
|||
return strings.Join(key[:keysLen-1], Separator), key[keysLen-1] |
|||
}, |
|||
name: Name, |
|||
prefix: Prefix + namespace + Separator + appName, |
|||
} |
|||
|
|||
for _, opt := range opts { |
|||
opt(&prov) |
|||
} |
|||
|
|||
return &prov |
|||
} |
|||
|
|||
type SecretKV2 struct { |
|||
client *api.Client |
|||
resolve func(key []string) (string, string) |
|||
name string |
|||
prefix string |
|||
} |
|||
|
|||
func (p *SecretKV2) Name() string { |
|||
return p.name |
|||
} |
|||
func (p *SecretKV2) Key(in []string) (string, string) { |
|||
path, val := p.resolve(in) |
|||
if path == "" { |
|||
return p.prefix, val |
|||
} |
|||
|
|||
return p.prefix + Separator + path, val |
|||
} |
|||
func (p *SecretKV2) read(path, key string) (*api.Secret, error) { |
|||
secret, err := p.client.Logical().Read(path) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
if secret == nil && key != ValueName { |
|||
return p.read(path+Separator+key, ValueName) |
|||
} |
|||
|
|||
return secret, err |
|||
} |
|||
|
|||
func (p *SecretKV2) Value(ctx context.Context, key ...string) (config.Value, error) { |
|||
path, field := p.Key(key) |
|||
|
|||
secret, err := p.read(path, field) |
|||
if err != nil { |
|||
return nil, fmt.Errorf("%w: path:%s, field:%s, provider:%s", err, path, field, p.Name()) |
|||
} |
|||
|
|||
if secret == nil || len(secret.Data) == 0 { |
|||
log.Println(secret == nil) |
|||
return nil, fmt.Errorf("%w: path:%s, field:%s, provider:%s", config.ErrValueNotFound, path, field, p.Name()) |
|||
} |
|||
|
|||
if len(secret.Warnings) > 0 { |
|||
return nil, |
|||
fmt.Errorf("%w: warn: %s, path:%s, field:%s, provider:%s", config.ErrValueNotFound, secret.Warnings, path, field, p.Name()) |
|||
} |
|||
|
|||
data, ok := secret.Data["data"].(map[string]interface{}) |
|||
if !ok { |
|||
return nil, fmt.Errorf("%w: path:%s, field:%s, provider:%s", config.ErrValueNotFound, path, field, p.Name()) |
|||
} |
|||
|
|||
if val, ok := data[field]; ok { |
|||
return value.JString(fmt.Sprint(val)), nil |
|||
} |
|||
|
|||
if val, ok := data[ValueName]; ok { |
|||
return value.JString(fmt.Sprint(val)), nil |
|||
} |
|||
|
|||
md, err := json.Marshal(data) |
|||
if err != nil { |
|||
return nil, fmt.Errorf("%w: %w", config.ErrInvalidValue, err) |
|||
} |
|||
|
|||
return value.JBytes(md), nil |
|||
} |
|||
@ -1,26 +0,0 @@ |
|||
package vault_test |
|||
|
|||
import ( |
|||
"testing" |
|||
"time" |
|||
|
|||
"github.com/stretchr/testify/require" |
|||
"gitoa.ru/go-4devs/config/provider/vault" |
|||
"gitoa.ru/go-4devs/config/test" |
|||
) |
|||
|
|||
func TestProvider(t *testing.T) { |
|||
t.Parallel() |
|||
|
|||
cl, err := test.NewVault() |
|||
require.NoError(t, err) |
|||
|
|||
provider := vault.NewSecretKV2("fdevs", "config", cl) |
|||
|
|||
read := []test.Read{ |
|||
test.NewReadConfig("database"), |
|||
test.NewRead(test.DSN, "db", "dsn"), |
|||
test.NewRead(time.Minute, "db", "timeout"), |
|||
} |
|||
test.Run(t, provider, read) |
|||
} |
|||
@ -1,89 +0,0 @@ |
|||
package test |
|||
|
|||
import ( |
|||
"bytes" |
|||
"context" |
|||
"encoding/json" |
|||
"net/http" |
|||
"os" |
|||
|
|||
"github.com/hashicorp/vault/api" |
|||
) |
|||
|
|||
const token = "dev" |
|||
|
|||
func NewVault() (*api.Client, error) { |
|||
address, ok := os.LookupEnv("VAULT_DEV_LISTEN_ADDRESS") |
|||
if !ok { |
|||
address = "http://127.0.0.1:8200" |
|||
} |
|||
|
|||
tokenID, ok := os.LookupEnv("VAULT_DEV_ROOT_TOKEN_ID") |
|||
if !ok { |
|||
tokenID = token |
|||
} |
|||
|
|||
cl, err := api.NewClient(&api.Config{ |
|||
Address: address, |
|||
}) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
cl.SetToken(tokenID) |
|||
|
|||
values := map[string]map[string]interface{}{ |
|||
"database": { |
|||
"duration": 1260000000000, |
|||
"enabled": true, |
|||
}, |
|||
"db": { |
|||
"dsn": DSN, |
|||
"timeout": "60s", |
|||
}, |
|||
"example": { |
|||
"dsn": DSN, |
|||
"timeout": "60s", |
|||
}, |
|||
} |
|||
|
|||
for name, val := range values { |
|||
if err := create(address, tokenID, name, val); err != nil { |
|||
return nil, err |
|||
} |
|||
} |
|||
|
|||
return cl, nil |
|||
} |
|||
|
|||
func create(host, token, path string, data map[string]interface{}) error { |
|||
type Req struct { |
|||
Data interface{} `json:"data"` |
|||
} |
|||
|
|||
b, err := json.Marshal(Req{Data: data}) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
body := bytes.NewBuffer(b) |
|||
|
|||
req, err := http.NewRequestWithContext( |
|||
context.Background(), |
|||
http.MethodPost, |
|||
host+"/v1/secret/data/fdevs/config/"+path, |
|||
body, |
|||
) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
req.Header.Set("X-Vault-Token", token) |
|||
|
|||
res, err := http.DefaultClient.Do(req) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
return res.Body.Close() |
|||
} |
|||
Loading…
Reference in new issue