feat: replace --operation flag by backup and restore command

pkg/restore.go

@@ -3,11 +3,35 @@ package pkg
 import (
 	"fmt"
 	"github.com/jkaninda/pg-bkup/utils"
+	"github.com/spf13/cobra"
 	"os"
 	"os/exec"
 	"path/filepath"
 )
 
+func StartRestore(cmd *cobra.Command) {
+
+	//Set env
+	utils.SetEnv("STORAGE_PATH", storagePath)
+	utils.GetEnv(cmd, "dbname", "DB_NAME")
+	utils.GetEnv(cmd, "port", "DB_PORT")
+
+	//Get flag value and set env
+	s3Path = utils.GetEnv(cmd, "path", "S3_PATH")
+	storage = utils.GetEnv(cmd, "storage", "STORAGE")
+	file = utils.GetEnv(cmd, "file", "FILE_NAME")
+	executionMode, _ = cmd.Flags().GetString("mode")
+
+	if storage == "s3" {
+		utils.Info("Restore database from s3")
+		s3Restore(file, s3Path)
+	} else {
+		utils.Info("Restore database from local")
+		RestoreDatabase(file)
+
+	}
+}
+
 // RestoreDatabase restore database
 func RestoreDatabase(file string) {
 	dbHost = os.Getenv("DB_HOST")
@@ -16,6 +40,9 @@ func RestoreDatabase(file string) {
 	dbName = os.Getenv("DB_NAME")
 	dbPort = os.Getenv("DB_PORT")
 	storagePath = os.Getenv("STORAGE_PATH")
+	if file == "" {
+		utils.Fatal("Error required --file")
+	}
 
 	if os.Getenv("DB_HOST") == "" || os.Getenv("DB_NAME") == "" || os.Getenv("DB_USERNAME") == "" || os.Getenv("DB_PASSWORD") == "" || file == "" {
 		utils.Fatal("Please make sure all required environment variables are set")
@@ -56,3 +83,8 @@ func RestoreDatabase(file string) {
 		}
 	}
 }
+func s3Restore(file, s3Path string) {
+	// Restore database from S3
+	MountS3Storage(s3Path)
+	RestoreDatabase(file)
+}