command/policies: read a single policy

This commit is contained in:
Mitchell Hashimoto
2015-04-01 18:50:43 -07:00
parent 4930e3885d
commit 79995a4496
3 changed files with 53 additions and 5 deletions

View File

@@ -18,13 +18,19 @@ func (c *PolicyListCommand) Run(args []string) int {
} }
args = flags.Args() args = flags.Args()
if len(args) != 0 { if len(args) == 1 {
return c.read(args[0])
} else if len(args) == 0 {
return c.list()
} else {
flags.Usage() flags.Usage()
c.Ui.Error(fmt.Sprintf( c.Ui.Error(fmt.Sprintf(
"\npolicy-list expects zero arguments")) "\npolicies expects zero or one arguments"))
return 1 return 1
} }
}
func (c *PolicyListCommand) list() int {
client, err := c.Client() client, err := c.Client()
if err != nil { if err != nil {
c.Ui.Error(fmt.Sprintf( c.Ui.Error(fmt.Sprintf(
@@ -46,17 +52,37 @@ func (c *PolicyListCommand) Run(args []string) int {
return 0 return 0
} }
func (c *PolicyListCommand) read(n string) int {
client, err := c.Client()
if err != nil {
c.Ui.Error(fmt.Sprintf(
"Error initializing client: %s", err))
return 2
}
rules, err := client.Sys().GetPolicy(n)
if err != nil {
c.Ui.Error(fmt.Sprintf(
"Error: %s", err))
return 1
}
c.Ui.Output(rules)
return 0
}
func (c *PolicyListCommand) Synopsis() string { func (c *PolicyListCommand) Synopsis() string {
return "List the policies on the server" return "List the policies on the server"
} }
func (c *PolicyListCommand) Help() string { func (c *PolicyListCommand) Help() string {
helpText := ` helpText := `
Usage: vault policy-list [options] Usage: vault policies [options] [name]
List the policies that are available. List the policies that are available or read a single policy.
This command lists the policies that are written to the Vault server. This command lists the policies that are written to the Vault server.
If a name of a policy is specified, that policy is outputted.
General Options: General Options:

View File

@@ -28,3 +28,25 @@ func TestPolicyList(t *testing.T) {
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String()) t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
} }
} }
func TestPolicyRead(t *testing.T) {
core, _, token := vault.TestCoreUnsealed(t)
ln, addr := http.TestServer(t, core)
defer ln.Close()
ui := new(cli.MockUi)
c := &PolicyListCommand{
Meta: Meta{
ClientToken: token,
Ui: ui,
},
}
args := []string{
"-address", addr,
"root",
}
if code := c.Run(args); code != 0 {
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
}
}

View File

@@ -43,7 +43,7 @@ func init() {
}, nil }, nil
}, },
"policy-list": func() (cli.Command, error) { "policies": func() (cli.Command, error) {
return &command.PolicyListCommand{ return &command.PolicyListCommand{
Meta: meta, Meta: meta,
}, nil }, nil