mirror of
				https://github.com/optim-enterprises-bv/vault.git
				synced 2025-10-31 18:48:08 +00:00 
			
		
		
		
	 6ed8b88f5f
			
		
	
	6ed8b88f5f
	
	
	
		
			
			@mitchellh suggested we fork `cli` and switch to that. Since we primarily use the interfaces in `cli`, and the new fork has not changed those, this is (mostly) a drop-in replacement. A small fix will be necessary for Vault Enterprise, I believe.
		
			
				
	
	
		
			113 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			113 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // Copyright (c) HashiCorp, Inc.
 | |
| // SPDX-License-Identifier: BUSL-1.1
 | |
| 
 | |
| package command
 | |
| 
 | |
| import (
 | |
| 	"fmt"
 | |
| 	"strings"
 | |
| 
 | |
| 	"github.com/hashicorp/cli"
 | |
| 	"github.com/hashicorp/vault/api"
 | |
| 	"github.com/posener/complete"
 | |
| )
 | |
| 
 | |
| var (
 | |
| 	_ cli.Command             = (*OperatorRaftAutopilotGetConfigCommand)(nil)
 | |
| 	_ cli.CommandAutocomplete = (*OperatorRaftAutopilotGetConfigCommand)(nil)
 | |
| )
 | |
| 
 | |
| type OperatorRaftAutopilotGetConfigCommand struct {
 | |
| 	*BaseCommand
 | |
| 	flagDRToken string
 | |
| }
 | |
| 
 | |
| func (c *OperatorRaftAutopilotGetConfigCommand) Synopsis() string {
 | |
| 	return "Returns the configuration of the autopilot subsystem under integrated storage"
 | |
| }
 | |
| 
 | |
| func (c *OperatorRaftAutopilotGetConfigCommand) Help() string {
 | |
| 	helpText := `
 | |
| Usage: vault operator raft autopilot get-config
 | |
| 
 | |
|  Returns the configuration of the autopilot subsystem under integrated storage.
 | |
| ` + c.Flags().Help()
 | |
| 
 | |
| 	return strings.TrimSpace(helpText)
 | |
| }
 | |
| 
 | |
| func (c *OperatorRaftAutopilotGetConfigCommand) Flags() *FlagSets {
 | |
| 	set := c.flagSet(FlagSetHTTP | FlagSetOutputFormat)
 | |
| 
 | |
| 	f := set.NewFlagSet("Command Options")
 | |
| 
 | |
| 	f.StringVar(&StringVar{
 | |
| 		Name:       "dr-token",
 | |
| 		Target:     &c.flagDRToken,
 | |
| 		Default:    "",
 | |
| 		EnvVar:     "",
 | |
| 		Completion: complete.PredictAnything,
 | |
| 		Usage:      "DR operation token used to authorize this request (if a DR secondary node).",
 | |
| 	})
 | |
| 
 | |
| 	return set
 | |
| }
 | |
| 
 | |
| func (c *OperatorRaftAutopilotGetConfigCommand) AutocompleteArgs() complete.Predictor {
 | |
| 	return complete.PredictAnything
 | |
| }
 | |
| 
 | |
| func (c *OperatorRaftAutopilotGetConfigCommand) AutocompleteFlags() complete.Flags {
 | |
| 	return c.Flags().Completions()
 | |
| }
 | |
| 
 | |
| func (c *OperatorRaftAutopilotGetConfigCommand) Run(args []string) int {
 | |
| 	f := c.Flags()
 | |
| 
 | |
| 	if err := f.Parse(args); err != nil {
 | |
| 		c.UI.Error(err.Error())
 | |
| 		return 1
 | |
| 	}
 | |
| 
 | |
| 	args = f.Args()
 | |
| 	switch len(args) {
 | |
| 	case 0:
 | |
| 	default:
 | |
| 		c.UI.Error(fmt.Sprintf("Incorrect arguments (expected 0, got %d)", len(args)))
 | |
| 		return 1
 | |
| 	}
 | |
| 
 | |
| 	client, err := c.Client()
 | |
| 	if err != nil {
 | |
| 		c.UI.Error(err.Error())
 | |
| 		return 2
 | |
| 	}
 | |
| 
 | |
| 	var config *api.AutopilotConfig
 | |
| 	switch {
 | |
| 	case c.flagDRToken != "":
 | |
| 		config, err = client.Sys().RaftAutopilotConfigurationWithDRToken(c.flagDRToken)
 | |
| 	default:
 | |
| 		config, err = client.Sys().RaftAutopilotConfiguration()
 | |
| 	}
 | |
| 
 | |
| 	if config == nil {
 | |
| 		return 0
 | |
| 	}
 | |
| 
 | |
| 	if Format(c.UI) != "table" {
 | |
| 		return OutputData(c.UI, config)
 | |
| 	}
 | |
| 
 | |
| 	entries := []string{"Key | Value"}
 | |
| 	entries = append(entries, fmt.Sprintf("%s | %t", "Cleanup Dead Servers", config.CleanupDeadServers))
 | |
| 	entries = append(entries, fmt.Sprintf("%s | %s", "Last Contact Threshold", config.LastContactThreshold.String()))
 | |
| 	entries = append(entries, fmt.Sprintf("%s | %s", "Dead Server Last Contact Threshold", config.DeadServerLastContactThreshold.String()))
 | |
| 	entries = append(entries, fmt.Sprintf("%s | %s", "Server Stabilization Time", config.ServerStabilizationTime.String()))
 | |
| 	entries = append(entries, fmt.Sprintf("%s | %d", "Min Quorum", config.MinQuorum))
 | |
| 	entries = append(entries, fmt.Sprintf("%s | %d", "Max Trailing Logs", config.MaxTrailingLogs))
 | |
| 	entries = append(entries, fmt.Sprintf("%s | %t", "Disable Upgrade Migration", config.DisableUpgradeMigration))
 | |
| 
 | |
| 	return OutputData(c.UI, entries)
 | |
| }
 |