mirror of
https://github.com/jkaninda/mysql-bkup.git
synced 2025-12-06 05:29:41 +01:00
138 lines
3.9 KiB
Go
138 lines
3.9 KiB
Go
// Package pkg /
|
|
/*****
|
|
@author Jonas Kaninda
|
|
@license MIT License <https://opensource.org/licenses/MIT>
|
|
@Copyright © 2024 Jonas Kaninda
|
|
**/
|
|
package pkg
|
|
|
|
import (
|
|
"github.com/jkaninda/mysql-bkup/utils"
|
|
"github.com/spf13/cobra"
|
|
"os"
|
|
)
|
|
|
|
type Config struct {
|
|
}
|
|
|
|
type dbConfig struct {
|
|
dbHost string
|
|
dbPort string
|
|
dbName string
|
|
dbUserName string
|
|
dbPassword string
|
|
}
|
|
type targetDbConfig struct {
|
|
targetDbHost string
|
|
targetDbPort string
|
|
targetDbUserName string
|
|
targetDbPassword string
|
|
targetDbName string
|
|
}
|
|
|
|
type BackupConfig struct {
|
|
backupFileName string
|
|
backupRetention int
|
|
disableCompression bool
|
|
prune bool
|
|
encryption bool
|
|
remotePath string
|
|
passphrase string
|
|
storage string
|
|
cronExpression string
|
|
}
|
|
type RestoreConfig struct {
|
|
s3Path string
|
|
remotePath string
|
|
storage string
|
|
file string
|
|
bucket string
|
|
gpqPassphrase string
|
|
}
|
|
|
|
func initDbConfig(cmd *cobra.Command) *dbConfig {
|
|
//Set env
|
|
utils.GetEnv(cmd, "dbname", "DB_NAME")
|
|
dConf := dbConfig{}
|
|
dConf.dbHost = os.Getenv("DB_HOST")
|
|
dConf.dbPort = os.Getenv("DB_PORT")
|
|
dConf.dbName = os.Getenv("DB_NAME")
|
|
dConf.dbUserName = os.Getenv("DB_USERNAME")
|
|
dConf.dbPassword = os.Getenv("DB_PASSWORD")
|
|
|
|
err := utils.CheckEnvVars(dbHVars)
|
|
if err != nil {
|
|
utils.Error("Please make sure all required environment variables for database are set")
|
|
utils.Fatal("Error checking environment variables: %s", err)
|
|
}
|
|
return &dConf
|
|
}
|
|
func initBackupConfig(cmd *cobra.Command) *BackupConfig {
|
|
utils.SetEnv("STORAGE_PATH", storagePath)
|
|
utils.GetEnv(cmd, "cron-expression", "BACKUP_CRON_EXPRESSION")
|
|
utils.GetEnv(cmd, "period", "BACKUP_CRON_EXPRESSION")
|
|
|
|
//Get flag value and set env
|
|
remotePath := utils.GetEnv(cmd, "path", "SSH_REMOTE_PATH")
|
|
storage = utils.GetEnv(cmd, "storage", "STORAGE")
|
|
backupRetention, _ := cmd.Flags().GetInt("keep-last")
|
|
prune, _ := cmd.Flags().GetBool("prune")
|
|
disableCompression, _ = cmd.Flags().GetBool("disable-compression")
|
|
_, _ = cmd.Flags().GetString("mode")
|
|
passphrase := os.Getenv("GPG_PASSPHRASE")
|
|
_ = utils.GetEnv(cmd, "path", "AWS_S3_PATH")
|
|
cronExpression := os.Getenv("BACKUP_CRON_EXPRESSION")
|
|
|
|
if passphrase != "" {
|
|
encryption = true
|
|
}
|
|
//Initialize backup configs
|
|
config := BackupConfig{}
|
|
config.backupRetention = backupRetention
|
|
config.disableCompression = disableCompression
|
|
config.prune = prune
|
|
config.storage = storage
|
|
config.encryption = encryption
|
|
config.remotePath = remotePath
|
|
config.passphrase = passphrase
|
|
config.cronExpression = cronExpression
|
|
return &config
|
|
}
|
|
func initRestoreConfig(cmd *cobra.Command) *RestoreConfig {
|
|
utils.SetEnv("STORAGE_PATH", storagePath)
|
|
|
|
//Get flag value and set env
|
|
s3Path := utils.GetEnv(cmd, "path", "AWS_S3_PATH")
|
|
remotePath := utils.GetEnv(cmd, "path", "SSH_REMOTE_PATH")
|
|
storage = utils.GetEnv(cmd, "storage", "STORAGE")
|
|
file = utils.GetEnv(cmd, "file", "FILE_NAME")
|
|
_, _ = cmd.Flags().GetString("mode")
|
|
bucket := utils.GetEnvVariable("AWS_S3_BUCKET_NAME", "BUCKET_NAME")
|
|
gpqPassphrase := os.Getenv("GPG_PASSPHRASE")
|
|
//Initialize restore configs
|
|
rConfig := RestoreConfig{}
|
|
rConfig.s3Path = s3Path
|
|
rConfig.remotePath = remotePath
|
|
rConfig.storage = storage
|
|
rConfig.bucket = bucket
|
|
rConfig.file = file
|
|
rConfig.storage = storage
|
|
rConfig.gpqPassphrase = gpqPassphrase
|
|
return &rConfig
|
|
}
|
|
func initTargetDbConfig() *targetDbConfig {
|
|
tdbConfig := targetDbConfig{}
|
|
tdbConfig.targetDbHost = os.Getenv("TARGET_DB_HOST")
|
|
tdbConfig.targetDbPort = os.Getenv("TARGET_DB_PORT")
|
|
tdbConfig.targetDbName = os.Getenv("TARGET_DB_NAME")
|
|
tdbConfig.targetDbUserName = os.Getenv("TARGET_DB_USERNAME")
|
|
tdbConfig.targetDbPassword = os.Getenv("TARGET_DB_PASSWORD")
|
|
|
|
err := utils.CheckEnvVars(tdbRVars)
|
|
if err != nil {
|
|
utils.Error("Please make sure all required environment variables for the target database are set")
|
|
utils.Fatal("Error checking target database environment variables: %s", err)
|
|
}
|
|
return &tdbConfig
|
|
}
|