mirror of
				https://github.com/optim-enterprises-bv/vault.git
				synced 2025-10-31 02:28:09 +00:00 
			
		
		
		
	 3189278c84
			
		
	
	3189278c84
	
	
	
		
			
			* Use Colored UI if stdout is a tty * Add format options to operator unseal * Add format test on operator unseal * Add -no-color output flag, and use BasicUi if no-color flag is provided * Move seal status formatting logic to OutputSealStatus * Apply no-color to warnings from DeprecatedCommands as well * Add OutputWithFormat to support arbitrary data, add format option to auth list * Add ability to output arbitrary list data on TableFormatter * Clear up switch logic on format * Add format option for list-related commands * Add format option to rest of commands that returns a client API response * Remove initOutputYAML and initOutputJSON, and use OutputWithFormat instead * Remove outputAsYAML and outputAsJSON, and use OutputWithFormat instead * Remove -no-color flag, use env var exclusively to toggle colored output * Fix compile * Remove -no-color flag in main.go * Add missing FlagSetOutputFormat * Fix generate-root/decode test * Migrate init functions to main.go * Add no-color flag back as hidden * Handle non-supported data types for TableFormatter.OutputList * Pull formatting much further up to remove the need to use c.flagFormat (#3950) * Pull formatting much further up to remove the need to use c.flagFormat Also remove OutputWithFormat as the logic can cause issues. * Use const for env var * Minor updates * Remove unnecessary check * Fix SSH output and some tests * Fix tests * Make race detector not run on generate root since it kills Travis these days * Update docs * Update docs * Address review feedback * Handle --format as well as -format
		
			
				
	
	
		
			129 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			129 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package command
 | |
| 
 | |
| import (
 | |
| 	"fmt"
 | |
| 	"strings"
 | |
| 
 | |
| 	"github.com/hashicorp/vault/api"
 | |
| 	"github.com/mitchellh/cli"
 | |
| 	"github.com/posener/complete"
 | |
| )
 | |
| 
 | |
| var _ cli.Command = (*TokenLookupCommand)(nil)
 | |
| var _ cli.CommandAutocomplete = (*TokenLookupCommand)(nil)
 | |
| 
 | |
| type TokenLookupCommand struct {
 | |
| 	*BaseCommand
 | |
| 
 | |
| 	flagAccessor bool
 | |
| }
 | |
| 
 | |
| func (c *TokenLookupCommand) Synopsis() string {
 | |
| 	return "Display information about a token"
 | |
| }
 | |
| 
 | |
| func (c *TokenLookupCommand) Help() string {
 | |
| 	helpText := `
 | |
| Usage: vault token lookup [options] [TOKEN | ACCESSOR]
 | |
| 
 | |
|   Displays information about a token or accessor. If a TOKEN is not provided,
 | |
|   the locally authenticated token is used.
 | |
| 
 | |
|   Get information about the locally authenticated token (this uses the
 | |
|   /auth/token/lookup-self endpoint and permission):
 | |
| 
 | |
|       $ vault token lookup
 | |
| 
 | |
|   Get information about a particular token (this uses the /auth/token/lookup
 | |
|   endpoint and permission):
 | |
| 
 | |
|       $ vault token lookup 96ddf4bc-d217-f3ba-f9bd-017055595017
 | |
| 
 | |
|   Get information about a token via its accessor:
 | |
| 
 | |
|       $ vault token lookup -accessor 9793c9b3-e04a-46f3-e7b8-748d7da248da
 | |
| 
 | |
|   For a full list of examples, please see the documentation.
 | |
| 
 | |
| ` + c.Flags().Help()
 | |
| 
 | |
| 	return strings.TrimSpace(helpText)
 | |
| }
 | |
| 
 | |
| func (c *TokenLookupCommand) Flags() *FlagSets {
 | |
| 	set := c.flagSet(FlagSetHTTP | FlagSetOutputFormat)
 | |
| 
 | |
| 	f := set.NewFlagSet("Command Options")
 | |
| 
 | |
| 	f.BoolVar(&BoolVar{
 | |
| 		Name:       "accessor",
 | |
| 		Target:     &c.flagAccessor,
 | |
| 		Default:    false,
 | |
| 		EnvVar:     "",
 | |
| 		Completion: complete.PredictNothing,
 | |
| 		Usage: "Treat the argument as an accessor instead of a token. When " +
 | |
| 			"this option is selected, the output will NOT include the token.",
 | |
| 	})
 | |
| 
 | |
| 	return set
 | |
| }
 | |
| 
 | |
| func (c *TokenLookupCommand) AutocompleteArgs() complete.Predictor {
 | |
| 	return c.PredictVaultFiles()
 | |
| }
 | |
| 
 | |
| func (c *TokenLookupCommand) AutocompleteFlags() complete.Flags {
 | |
| 	return c.Flags().Completions()
 | |
| }
 | |
| 
 | |
| func (c *TokenLookupCommand) Run(args []string) int {
 | |
| 	f := c.Flags()
 | |
| 
 | |
| 	if err := f.Parse(args); err != nil {
 | |
| 		c.UI.Error(err.Error())
 | |
| 		return 1
 | |
| 	}
 | |
| 
 | |
| 	token := ""
 | |
| 
 | |
| 	args = f.Args()
 | |
| 	switch {
 | |
| 	case c.flagAccessor && len(args) < 1:
 | |
| 		c.UI.Error(fmt.Sprintf("Not enough arguments with -accessor (expected 1, got %d)", len(args)))
 | |
| 		return 1
 | |
| 	case c.flagAccessor && len(args) > 1:
 | |
| 		c.UI.Error(fmt.Sprintf("Too many arguments with -accessor (expected 1, got %d)", len(args)))
 | |
| 		return 1
 | |
| 	case len(args) == 0:
 | |
| 		// Use the local token
 | |
| 	case len(args) == 1:
 | |
| 		token = strings.TrimSpace(args[0])
 | |
| 	case len(args) > 1:
 | |
| 		c.UI.Error(fmt.Sprintf("Too many arguments (expected 0-1, got %d)", len(args)))
 | |
| 		return 1
 | |
| 	}
 | |
| 
 | |
| 	client, err := c.Client()
 | |
| 	if err != nil {
 | |
| 		c.UI.Error(err.Error())
 | |
| 		return 2
 | |
| 	}
 | |
| 
 | |
| 	var secret *api.Secret
 | |
| 	switch {
 | |
| 	case token == "":
 | |
| 		secret, err = client.Auth().Token().LookupSelf()
 | |
| 	case c.flagAccessor:
 | |
| 		secret, err = client.Auth().Token().LookupAccessor(token)
 | |
| 	default:
 | |
| 		secret, err = client.Auth().Token().Lookup(token)
 | |
| 	}
 | |
| 
 | |
| 	if err != nil {
 | |
| 		c.UI.Error(fmt.Sprintf("Error looking up token: %s", err))
 | |
| 		return 2
 | |
| 	}
 | |
| 
 | |
| 	return OutputSecret(c.UI, secret)
 | |
| }
 |