You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
1.4 KiB
89 lines
1.4 KiB
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()
|
|
}
|
|
|