mirror of
https://github.com/optim-enterprises-bv/vault.git
synced 2025-10-29 17:52:32 +00:00
command: refactor to share output formating code
This commit is contained in:
@@ -1,13 +1,8 @@
|
||||
package command
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/hashicorp/vault/api"
|
||||
"github.com/ryanuber/columnize"
|
||||
)
|
||||
|
||||
// ReadCommand is a Command that reads data from the Vault.
|
||||
@@ -51,55 +46,7 @@ func (c *ReadCommand) Run(args []string) int {
|
||||
return 1
|
||||
}
|
||||
|
||||
return c.output(format, secret)
|
||||
}
|
||||
|
||||
func (c *ReadCommand) output(format string, secret *api.Secret) int {
|
||||
switch format {
|
||||
case "json":
|
||||
return c.formatJSON(secret)
|
||||
case "table":
|
||||
fallthrough
|
||||
default:
|
||||
return c.formatTable(secret, true)
|
||||
}
|
||||
}
|
||||
|
||||
func (c *ReadCommand) formatJSON(s *api.Secret) int {
|
||||
b, err := json.Marshal(s)
|
||||
if err != nil {
|
||||
c.Ui.Error(fmt.Sprintf(
|
||||
"Error formatting secret: %s", err))
|
||||
return 1
|
||||
}
|
||||
|
||||
var out bytes.Buffer
|
||||
json.Indent(&out, b, "", "\t")
|
||||
c.Ui.Output(out.String())
|
||||
return 0
|
||||
}
|
||||
|
||||
func (c *ReadCommand) formatTable(s *api.Secret, whitespace bool) int {
|
||||
config := columnize.DefaultConfig()
|
||||
config.Delim = "♨"
|
||||
config.Glue = "\t"
|
||||
config.Prefix = ""
|
||||
|
||||
input := make([]string, 0, 5)
|
||||
input = append(input, fmt.Sprintf("Key %s Value", config.Delim))
|
||||
|
||||
if s.LeaseID != "" && s.LeaseDuration > 0 {
|
||||
input = append(input, fmt.Sprintf("lease_id %s %s", config.Delim, s.LeaseID))
|
||||
input = append(input, fmt.Sprintf(
|
||||
"lease_duration %s %d", config.Delim, s.LeaseDuration))
|
||||
}
|
||||
|
||||
for k, v := range s.Data {
|
||||
input = append(input, fmt.Sprintf("%s %s %v", k, config.Delim, v))
|
||||
}
|
||||
|
||||
c.Ui.Output(columnize.Format(input, config))
|
||||
return 0
|
||||
return OutputSecret(c.Ui, format, secret)
|
||||
}
|
||||
|
||||
func (c *ReadCommand) Synopsis() string {
|
||||
|
||||
Reference in New Issue
Block a user