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

cmd/backup.go

@@ -0,0 +1,28 @@
+package cmd
+
+import (
+	"github.com/jkaninda/pg-bkup/pkg"
+	"github.com/jkaninda/pg-bkup/utils"
+	"github.com/spf13/cobra"
+)
+
+var BackupCmd = &cobra.Command{
+	Use:     "backup ",
+	Short:   "Backup database operation",
+	Example: utils.BackupExample,
+	Run: func(cmd *cobra.Command, args []string) {
+		if len(args) == 0 {
+			pkg.StartBackup(cmd)
+		} else {
+			utils.Fatal("Error, no argument required")
+		}
+	},
+}
+
+func init() {
+	//Backup
+	BackupCmd.PersistentFlags().StringP("mode", "m", "default", "Set execution mode. default or scheduled")
+	BackupCmd.PersistentFlags().StringP("period", "", "0 1 * * *", "Set schedule period time")
+	BackupCmd.PersistentFlags().BoolP("disable-compression", "", false, "Disable backup compression")
+
+}

cmd/history.go

@@ -0,0 +1,14 @@
+package cmd
+
+import (
+	"github.com/jkaninda/pg-bkup/utils"
+	"github.com/spf13/cobra"
+)
+
+var HistoryCmd = &cobra.Command{
+	Use:   "history",
+	Short: "Show the history of backup",
+	Run: func(cmd *cobra.Command, args []string) {
+		utils.ShowHistory()
+	},
+}

cmd/restore.go

@@ -0,0 +1,28 @@
+package cmd
+
+import (
+	"github.com/jkaninda/pg-bkup/pkg"
+	"github.com/jkaninda/pg-bkup/utils"
+	"github.com/spf13/cobra"
+)
+
+var RestoreCmd = &cobra.Command{
+	Use:     "restore",
+	Short:   "Restore database operation",
+	Example: utils.RestoreExample,
+	Run: func(cmd *cobra.Command, args []string) {
+		if len(args) == 0 {
+			pkg.StartRestore(cmd)
+		} else {
+			utils.Fatal("Error, no argument required")
+
+		}
+
+	},
+}
+
+func init() {
+	//Restore
+	RestoreCmd.PersistentFlags().StringP("file", "f", "", "File name of database")
+
+}

cmd/root.go

@@ -5,20 +5,32 @@ Copyright © 2024 Jonas Kaninda  <jonaskaninda@gmail.com>
 package cmd
 
 import (
-	"os"
-
+	"fmt"
+	"github.com/jkaninda/pg-bkup/utils"
 	"github.com/spf13/cobra"
+	"os"
 )
 
 // rootCmd represents the base command when called without any subcommands
 var rootCmd = &cobra.Command{
-	Use:   "pg-bkup",
-	Short: "PostgreSQL Backup tool, backup database to S3 or Object Storage",
-	Long:  `PostgreSQL Backup and Restoration tool. Backup database to AWS S3 storage or any S3 Alternatives for Object Storage.`,
-	// Uncomment the following line if your bare application
-	// has an action associated with it:
-	// Run: func(cmd *cobra.Command, args []string) { },
+	Use:     "pg-bkup [Command]",
+	Short:   "PostgreSQL Backup tool, backup database to S3 or Object Storage",
+	Long:    `PostgreSQL Database backup and restoration tool. Backup database to AWS S3 storage or any S3 Alternatives for Object Storage.`,
+	Example: utils.MainExample,
+	Version: appVersion,
+	//TODO: To remove
+	//For old user || To remove
+	Run: func(cmd *cobra.Command, args []string) {
+		if operation != "" {
+			if operation == "backup" || operation == "restore" {
+				fmt.Println(utils.Notice)
+				utils.Fatal("New config required, please check --help")
+			}
+		}
+	},
 }
+var operation = ""
+var s3Path = "/pg-bkup"
 
 // Execute adds all child commands to the root command and sets flags appropriately.
 // This is called by main.main(). It only needs to happen once to the rootCmd.
@@ -30,26 +42,21 @@ func Execute() {
 }
 
 func init() {
-	// Here you will define your flags and configuration settings.
-	// Cobra supports persistent flags, which, if defined here,
-	// will be global for your application.
-
-	// rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.mysql-bkup.yaml)")
-
-	// Cobra also supports local flags, which will only run
-	// when this action is called directly.
-	rootCmd.PersistentFlags().StringP("operation", "o", "backup", "Set operation")
 	rootCmd.PersistentFlags().StringP("storage", "s", "local", "Set storage. local or s3")
-	rootCmd.PersistentFlags().StringP("file", "f", "", "Set file name")
-	rootCmd.PersistentFlags().StringP("path", "P", "/mysql-bkup", "Set s3 path, without file name")
+	rootCmd.PersistentFlags().StringP("path", "P", s3Path, "Set s3 path, without file name. for S3 storage only")
 	rootCmd.PersistentFlags().StringP("dbname", "d", "", "Set database name")
+	rootCmd.PersistentFlags().IntP("timeout", "t", 30, "Set timeout")
+	rootCmd.PersistentFlags().IntP("port", "p", 5432, "Set database port")
+	rootCmd.PersistentFlags().StringVarP(&operation, "operation", "o", "", "Set operation, for old version only")
+
 	rootCmd.PersistentFlags().StringP("mode", "m", "default", "Set execution mode. default or scheduled")
 	rootCmd.PersistentFlags().StringP("period", "", "0 1 * * *", "Set schedule period time")
-	rootCmd.PersistentFlags().IntP("timeout", "t", 30, "Set timeout")
 	rootCmd.PersistentFlags().BoolP("disable-compression", "", false, "Disable backup compression")
-	rootCmd.PersistentFlags().IntP("port", "p", 5432, "Set database port")
-	rootCmd.PersistentFlags().BoolP("help", "h", false, "Print this help message")
-	rootCmd.PersistentFlags().BoolP("version", "v", false, "shows version information")
-	rootCmd.AddCommand(VersionCmd)
+	rootCmd.PersistentFlags().StringP("file", "f", "", "File name of database")
 
+	rootCmd.AddCommand(VersionCmd)
+	rootCmd.AddCommand(BackupCmd)
+	rootCmd.AddCommand(RestoreCmd)
+	rootCmd.AddCommand(S3MountCmd)
+	rootCmd.AddCommand(HistoryCmd)
 }

cmd/s3mount.go

@@ -0,0 +1,14 @@
+package cmd
+
+import (
+	"github.com/jkaninda/pg-bkup/pkg"
+	"github.com/spf13/cobra"
+)
+
+var S3MountCmd = &cobra.Command{
+	Use:   "s3mount",
+	Short: "Mount AWS S3 storage",
+	Run: func(cmd *cobra.Command, args []string) {
+		pkg.S3Mount()
+	},
+}

cmd/version.go

@@ -1,5 +1,9 @@
 package cmd
 
+/*
+Copyright © 2024 Jonas Kaninda  <jonaskaninda@gmail.com>
+*/
+
 import (
 	"fmt"
 	"github.com/spf13/cobra"
@@ -16,8 +20,7 @@ var VersionCmd = &cobra.Command{
 	},
 }
 
-// Version display application version
 func Version() {
 	fmt.Printf("Version: %s \n", appVersion)
-	fmt.Print()
+	fmt.Println()
 }