From 5e1bae53412b563912a5c3042c6e48bffd09d75d Mon Sep 17 00:00:00 2001
From: absolutelightning <ashesh.vidyut@hashicorp.com>
Date: Thu, 31 Aug 2023 14:48:54 +0530
Subject: [PATCH] init

---
 command/snapshot/save/snapshot_save.go | 42 +++++++++++++++++++++++---
 1 file changed, 37 insertions(+), 5 deletions(-)

diff --git a/command/snapshot/save/snapshot_save.go b/command/snapshot/save/snapshot_save.go
index 2395a4f3a038..16e473b1f016 100644
--- a/command/snapshot/save/snapshot_save.go
+++ b/command/snapshot/save/snapshot_save.go
@@ -6,7 +6,10 @@ package save
 import (
 	"flag"
 	"fmt"
+	"github.com/hashicorp/consul/version"
+	"golang.org/x/exp/slices"
 	"os"
+	"strings"
 
 	"github.com/mitchellh/cli"
 	"github.com/rboyer/safeio"
@@ -23,10 +26,18 @@ func New(ui cli.Ui) *cmd {
 }
 
 type cmd struct {
-	UI    cli.Ui
-	flags *flag.FlagSet
-	http  *flags.HTTPFlags
-	help  string
+	UI             cli.Ui
+	flags          *flag.FlagSet
+	http           *flags.HTTPFlags
+	help           string
+	appendFileName flags.StringValue
+}
+
+func (c *cmd) getAppendFileNameFlag() *flag.FlagSet {
+	fs := flag.NewFlagSet("", flag.ContinueOnError)
+	fs.Var(&c.appendFileName, "append-filename", "Append filename flag takes two possible values. "+
+		"1. version, 2. dc. It appends consul version and datacenter to filename given in command")
+	return fs
 }
 
 func (c *cmd) init() {
@@ -34,6 +45,7 @@ func (c *cmd) init() {
 	c.http = &flags.HTTPFlags{}
 	flags.Merge(c.flags, c.http.ClientFlags())
 	flags.Merge(c.flags, c.http.ServerFlags())
+	flags.Merge(c.flags, c.getAppendFileNameFlag())
 	c.help = flags.Usage(help, c.flags)
 }
 
@@ -52,12 +64,32 @@ func (c *cmd) Run(args []string) int {
 	case 1:
 		file = args[0]
 	default:
-		c.UI.Error(fmt.Sprintf("Too many arguments (expected 1, got %d)", len(args)))
+		c.UI.Error(fmt.Sprintf("Too many arguments (expected 1 or 3, got %d)", len(args)))
 		return 1
 	}
 
 	// Create and test the HTTP client
 	client, err := c.http.APIClient()
+
+	appendFileNameFlags := strings.Split(c.appendFileName.String(), ",")
+
+	if slices.Contains(appendFileNameFlags, "version") {
+		file = file + "-" + version.GetHumanVersion()
+	}
+
+	if slices.Contains(appendFileNameFlags, "dc") {
+		agentSelfResponse, err := client.Agent().Self()
+		if err != nil {
+			c.UI.Error(fmt.Sprintf("Error connecting to Consul agent and fetching datacenter: %s", err))
+			return 1
+		}
+		if config, ok := agentSelfResponse["Config"]; ok {
+			if datacenter, ok := config["Datacenter"]; ok {
+				file = file + "-" + datacenter.(string)
+			}
+		}
+	}
+
 	if err != nil {
 		c.UI.Error(fmt.Sprintf("Error connecting to Consul agent: %s", err))
 		return 1