pg-bkup/cmd/root.go

// Package cmd /*
/*
Copyright © 2024 Jonas Kaninda
*/
package cmd

import (
	"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 [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,
}
var operation = ""

// 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.
func Execute() {
	err := rootCmd.Execute()
	if err != nil {
		os.Exit(1)
	}
}

func init() {
	rootCmd.PersistentFlags().StringP("storage", "s", "local", "Set storage. local or s3")
	rootCmd.PersistentFlags().StringP("path", "P", "", "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.AddCommand(VersionCmd)
	rootCmd.AddCommand(BackupCmd)
	rootCmd.AddCommand(RestoreCmd)
	rootCmd.AddCommand(S3MountCmd)
	rootCmd.AddCommand(HistoryCmd)
}