backup the cluster config files

Signed-off-by: Jeff Carr <[email protected]>

2 files changed, 83 insertions, 1 deletions

diff --git a/backup.go b/backup.go
new file mode 100644
index 0000000..adaa99c
--- /dev/null
+++ b/backup.go
@@ -0,0 +1,66 @@
+package virtbuf
+
+// thank chatgpt for this because why. why write this if you can have it
+// kick this out in 30 seconds
+
+import (
+	"io"
+	"log"
+	"os"
+	"path/filepath"
+)
+
+func backupFiles(srcDir string, destDir string) error {
+	// Create the destination directory
+	err := os.MkdirAll(destDir, os.ModePerm)
+	if err != nil {
+		log.Println("Failed to create directory: %v", err)
+		return err
+	}
+
+	// Walk through the source directory
+	err = filepath.Walk(srcDir, func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+
+		// Skip if it's not a .test file or if it's a directory
+		if filepath.Ext(path) != ".json" || info.IsDir() {
+			return nil
+		}
+
+		// Destination file path
+		destPath := filepath.Join(destDir, info.Name())
+
+		// Copy the file
+		if err := copyFile(path, destPath); err != nil {
+			return err
+		}
+		return nil
+	})
+
+	if err != nil {
+		log.Println("Failed to copy files: %v", err)
+		return err
+	}
+	return nil
+}
+
+// copyFile copies a file from src to dest
+func copyFile(src, dest string) error {
+	srcFile, err := os.Open(src)
+	if err != nil {
+		return err
+	}
+	defer srcFile.Close()
+
+	destFile, err := os.Create(dest)
+	if err != nil {
+		return err
+	}
+	defer destFile.Close()
+
+	// Copy the content
+	_, err = io.Copy(destFile, srcFile)
+	return err
+}
diff --git a/config.go b/config.go
index 99ce67f..6185010 100644
--- a/config.go
+++ b/config.go
@@ -8,6 +8,7 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"time"
 
 	"google.golang.org/protobuf/encoding/protojson"
 	"google.golang.org/protobuf/encoding/prototext"
@@ -19,6 +20,12 @@ import (
 func (c *Cluster) ConfigSave() error {
 	var d *Droplets
 	d = new(Droplets)
+
+	// try to backup the current cluster config files
+	if err := backupConfigFiles(); err != nil {
+		return err
+	}
+
 	d.Droplets = c.Droplets
 	if err := ConfigWriteJSON(d, "newdroplets.json"); err != nil {
 		fmt.Println("droplets.json write failed")
@@ -71,6 +78,16 @@ func (c *Cluster) ConfigSave() error {
 	return nil
 }
 
+func backupConfigFiles() error {
+	// make a new dir to backup the files
+	now := time.Now()
+	timestamp := now.Format("2006.01.02.150405")
+	srcDir := filepath.Join(os.Getenv("VIRTIGO_HOME"))
+	destDir := filepath.Join(os.Getenv("VIRTIGO_HOME"), timestamp)
+
+	return backupFiles(srcDir, destDir)
+}
+
 func (c *Cluster) ConfigLoadOld() error {
 	if c == nil {
 		return errors.New("It's not safe to run ConfigLoad() on a nil cluster")
@@ -108,7 +125,6 @@ func (c *Cluster) ConfigLoadOld() error {
 	return nil
 }
 
-
 func (c *Cluster) ConfigLoad() error {
 	if c == nil {
 		return errors.New("It's not safe to run ConfigLoad() on a nil cluster")