package command import ( "fmt" "strings" "github.com/mitchellh/cli" "github.com/posener/complete" ) var ( _ cli.Command = (*KVUndeleteCommand)(nil) _ cli.CommandAutocomplete = (*KVUndeleteCommand)(nil) ) type KVUndeleteCommand struct { *BaseCommand flagVersions []string flagMount string } func (c *KVUndeleteCommand) Synopsis() string { return "Undeletes versions in the KV store" } func (c *KVUndeleteCommand) Help() string { helpText := ` Usage: vault kv undelete [options] KEY Undeletes the data for the provided version and path in the key-value store. This restores the data, allowing it to be returned on get requests. To undelete version 3 of key "foo": $ vault kv undelete -mount=secret -versions=3 foo The deprecated path-like syntax can also be used, but this should be avoided, as the fact that it is not actually the full API path to the secret (secret/data/foo) can cause confusion: $ vault kv undelete -versions=3 secret/foo Additional flags and more advanced use cases are detailed below. ` + c.Flags().Help() return strings.TrimSpace(helpText) } func (c *KVUndeleteCommand) Flags() *FlagSets { set := c.flagSet(FlagSetHTTP | FlagSetOutputFormat) // Common Options f := set.NewFlagSet("Common Options") f.StringSliceVar(&StringSliceVar{ Name: "versions", Target: &c.flagVersions, Default: nil, Usage: `Specifies the version numbers to undelete.`, }) f.StringVar(&StringVar{ Name: "mount", Target: &c.flagMount, Default: "", // no default, because the handling of the next arg is determined by whether this flag has a value Usage: `Specifies the path where the KV backend is mounted. If specified, the next argument will be interpreted as the secret path. If this flag is not specified, the next argument will be interpreted as the combined mount path and secret path, with /data/ automatically appended between KV v2 secrets.`, }) return set } func (c *KVUndeleteCommand) AutocompleteArgs() complete.Predictor { return nil } func (c *KVUndeleteCommand) AutocompleteFlags() complete.Flags { return c.Flags().Completions() } func (c *KVUndeleteCommand) 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 { case len(args) < 1: c.UI.Error(fmt.Sprintf("Not enough arguments (expected 1, got %d)", len(args))) return 1 case len(args) > 1: c.UI.Error(fmt.Sprintf("Too many arguments (expected 1, got %d)", len(args))) return 1 } if len(c.flagVersions) == 0 { c.UI.Error("No versions provided, use the \"-versions\" flag to specify the version to undelete.") return 1 } client, err := c.Client() if err != nil { c.UI.Error(err.Error()) return 2 } // If true, we're working with "-mount=secret foo" syntax. // If false, we're using "secret/foo" syntax. mountFlagSyntax := (c.flagMount != "") var ( mountPath string partialPath string v2 bool ) // Parse the paths and grab the KV version if mountFlagSyntax { // In this case, this arg is the secret path (e.g. "foo"). partialPath = sanitizePath(args[0]) mountPath = sanitizePath(c.flagMount) _, v2, err = isKVv2(mountPath, client) if err != nil { c.UI.Error(err.Error()) return 2 } } else { // In this case, this arg is a path-like combination of mountPath/secretPath. // (e.g. "secret/foo") partialPath = sanitizePath(args[0]) mountPath, v2, err = isKVv2(partialPath, client) if err != nil { c.UI.Error(err.Error()) return 2 } } if !v2 { c.UI.Error("Undelete not supported on KV Version 1") return 1 } undeletePath := addPrefixToKVPath(partialPath, mountPath, "undelete") data := map[string]interface{}{ "versions": kvParseVersionsFlags(c.flagVersions), } secret, err := client.Logical().Write(undeletePath, data) if err != nil { c.UI.Error(fmt.Sprintf("Error writing data to %s: %s", undeletePath, err)) if secret != nil { OutputSecret(c.UI, secret) } return 2 } if secret == nil { // Don't output anything unless using the "table" format if Format(c.UI) == "table" { c.UI.Info(fmt.Sprintf("Success! Data written to: %s", undeletePath)) } return 0 } return OutputSecret(c.UI, secret) }