command: refactor to share output formating code

This commit is contained in:
Armon Dadgar
2015-04-27 14:55:29 -07:00
parent dfce9d75d3
commit 5aee91ccb9
4 changed files with 62 additions and 62 deletions

View File

@@ -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 {