82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package command
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// CapabilitiesCommand is a Command that enables a new endpoint.
|
|
type CapabilitiesCommand struct {
|
|
Meta
|
|
}
|
|
|
|
func (c *CapabilitiesCommand) Run(args []string) int {
|
|
flags := c.Meta.FlagSet("capabilities", FlagSetDefault)
|
|
flags.Usage = func() { c.Ui.Error(c.Help()) }
|
|
if err := flags.Parse(args); err != nil {
|
|
return 1
|
|
}
|
|
|
|
args = flags.Args()
|
|
if len(args) > 2 {
|
|
flags.Usage()
|
|
c.Ui.Error(fmt.Sprintf(
|
|
"\ncapabilities expects at most two arguments"))
|
|
return 1
|
|
}
|
|
|
|
var token string
|
|
var path string
|
|
switch len(args) {
|
|
case 1:
|
|
// only path is provided
|
|
path = args[0]
|
|
case 2:
|
|
// both token and path are provided
|
|
token = args[0]
|
|
path = args[1]
|
|
default:
|
|
}
|
|
|
|
client, err := c.Client()
|
|
if err != nil {
|
|
c.Ui.Error(fmt.Sprintf(
|
|
"Error initializing client: %s", err))
|
|
return 2
|
|
}
|
|
|
|
var capabilities []string
|
|
if token == "" {
|
|
capabilities, err = client.Sys().CapabilitiesSelf(path)
|
|
} else {
|
|
capabilities, err = client.Sys().Capabilities(token, path)
|
|
}
|
|
if err != nil {
|
|
c.Ui.Error(fmt.Sprintf(
|
|
"Error retrieving capabilities: %s", err))
|
|
return 1
|
|
}
|
|
|
|
c.Ui.Output(fmt.Sprintf("Capabilities: %s", capabilities))
|
|
return 0
|
|
}
|
|
|
|
func (c *CapabilitiesCommand) Synopsis() string {
|
|
return "Fetch the capabilities of a given token on a given path"
|
|
}
|
|
|
|
func (c *CapabilitiesCommand) Help() string {
|
|
helpText := `
|
|
Usage: vault capabilities [options] [token] path
|
|
|
|
Fetch the capabilities of a token on a given path.
|
|
If a token is given to the command '/sys/capabilities' will be called with
|
|
the given token; otherwise '/sys/capabilities-self' will be called with the
|
|
client token.
|
|
|
|
General Options:
|
|
|
|
` + generalOptionsUsage()
|
|
return strings.TrimSpace(helpText)
|
|
}
|