package command import ( "encoding/base64" "encoding/json" "flag" "fmt" "strings" "github.com/hashicorp/consul/api" "github.com/mitchellh/cli" ) // KVExportCommand is a Command implementation that is used to export // a KV tree as JSON type KVExportCommand struct { Ui cli.Ui } func (c *KVExportCommand) Synopsis() string { return "Exports a tree from the KV store as JSON" } func (c *KVExportCommand) Help() string { helpText := ` Usage: consul kv export [KEY_OR_PREFIX] Retrieves key-value pairs for the given prefix from Consul's key-value store, and writes a JSON representation to stdout. This can be used with the command "consul kv import" to move entire trees between Consul clusters. $ consul kv export vault For a full list of options and examples, please see the Consul documentation. ` + apiOptsText + ` KV Export Options: None. ` return strings.TrimSpace(helpText) } func (c *KVExportCommand) Run(args []string) int { cmdFlags := flag.NewFlagSet("export", flag.ContinueOnError) datacenter := cmdFlags.String("datacenter", "", "") token := cmdFlags.String("token", "", "") stale := cmdFlags.Bool("stale", false, "") httpAddr := HTTPAddrFlag(cmdFlags) if err := cmdFlags.Parse(args); err != nil { return 1 } key := "" // Check for arg validation args = cmdFlags.Args() switch len(args) { case 0: key = "" case 1: key = args[0] default: c.Ui.Error(fmt.Sprintf("Too many arguments (expected 1, got %d)", len(args))) return 1 } // This is just a "nice" thing to do. Since pairs cannot start with a /, but // users will likely put "/" or "/foo", lets go ahead and strip that for them // here. if len(key) > 0 && key[0] == '/' { key = key[1:] } // Create and test the HTTP client conf := api.DefaultConfig() conf.Address = *httpAddr conf.Token = *token client, err := api.NewClient(conf) if err != nil { c.Ui.Error(fmt.Sprintf("Error connecting to Consul agent: %s", err)) return 1 } pairs, _, err := client.KV().List(key, &api.QueryOptions{ Datacenter: *datacenter, AllowStale: *stale, }) if err != nil { c.Ui.Error(fmt.Sprintf("Error querying Consul agent: %s", err)) return 1 } exported := make([]*kvExportEntry, len(pairs)) for i, pair := range pairs { exported[i] = toExportEntry(pair) } marshaled, err := json.MarshalIndent(exported, "", "\t") if err != nil { c.Ui.Error(fmt.Sprintf("Error exporting KV data: %s", err)) return 1 } c.Ui.Info(string(marshaled)) return 0 } type kvExportEntry struct { Key string `json:"key"` Flags uint64 `json:"flags"` Value string `json:"value"` } func toExportEntry(pair *api.KVPair) *kvExportEntry { return &kvExportEntry{ Key: pair.Key, Flags: pair.Flags, Value: base64.StdEncoding.EncodeToString(pair.Value), } }