128 lines
3.9 KiB
Go
128 lines
3.9 KiB
Go
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", "AWS_S3_PATH")
|
|
remotePath := utils.GetEnv(cmd, "path", "SSH_REMOTE_PATH")
|
|
storage = utils.GetEnv(cmd, "storage", "STORAGE")
|
|
file = utils.GetEnv(cmd, "file", "FILE_NAME")
|
|
executionMode, _ = cmd.Flags().GetString("mode")
|
|
bucket := utils.GetEnvVariable("AWS_S3_BUCKET_NAME", "BUCKET_NAME")
|
|
switch storage {
|
|
case "s3":
|
|
restoreFromS3(file, bucket, s3Path)
|
|
case "local":
|
|
utils.Info("Restore database from local")
|
|
copyToTmp(storagePath, file)
|
|
RestoreDatabase(file)
|
|
case "ssh":
|
|
restoreFromRemote(file, remotePath)
|
|
case "ftp":
|
|
utils.Fatal("Restore from FTP is not yet supported")
|
|
default:
|
|
utils.Info("Restore database from local")
|
|
RestoreDatabase(file)
|
|
}
|
|
}
|
|
|
|
func restoreFromS3(file, bucket, s3Path string) {
|
|
utils.Info("Restore database from s3")
|
|
err := utils.DownloadFile(tmpPath, file, bucket, s3Path)
|
|
if err != nil {
|
|
utils.Fatal("Error download file from s3 %s %v ", file, err)
|
|
}
|
|
RestoreDatabase(file)
|
|
}
|
|
func restoreFromRemote(file, remotePath string) {
|
|
utils.Info("Restore database from remote server")
|
|
err := CopyFromRemote(file, remotePath)
|
|
if err != nil {
|
|
utils.Fatal("Error download file from remote server: %s %v", filepath.Join(remotePath, file), err)
|
|
}
|
|
RestoreDatabase(file)
|
|
}
|
|
|
|
// RestoreDatabase restore database
|
|
func RestoreDatabase(file string) {
|
|
dbHost = os.Getenv("DB_HOST")
|
|
dbPassword = os.Getenv("DB_PASSWORD")
|
|
dbUserName = os.Getenv("DB_USERNAME")
|
|
dbName = os.Getenv("DB_NAME")
|
|
dbPort = os.Getenv("DB_PORT")
|
|
gpgPassphrase := os.Getenv("GPG_PASSPHRASE")
|
|
if file == "" {
|
|
utils.Fatal("Error, file required")
|
|
}
|
|
extension := filepath.Ext(fmt.Sprintf("%s/%s", tmpPath, file))
|
|
if extension == ".gpg" {
|
|
if gpgPassphrase == "" {
|
|
utils.Fatal("Error: GPG passphrase is required, your file seems to be a GPG file.\nYou need to provide GPG keys. GPG_PASSPHRASE environment variable is required.")
|
|
|
|
} else {
|
|
//Decrypt file
|
|
err := Decrypt(filepath.Join(tmpPath, file), gpgPassphrase)
|
|
if err != nil {
|
|
utils.Fatal("Error decrypting file %s %v", file, err)
|
|
}
|
|
//Update file name
|
|
file = RemoveLastExtension(file)
|
|
}
|
|
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
if utils.FileExists(fmt.Sprintf("%s/%s", tmpPath, file)) {
|
|
|
|
err := os.Setenv("PGPASSWORD", dbPassword)
|
|
if err != nil {
|
|
return
|
|
}
|
|
utils.TestDatabaseConnection()
|
|
|
|
extension := filepath.Ext(fmt.Sprintf("%s/%s", tmpPath, file))
|
|
// Restore from compressed file / .sql.gz
|
|
if extension == ".gz" {
|
|
str := "zcat " + fmt.Sprintf("%s/%s", tmpPath, file) + " | psql -h " + os.Getenv("DB_HOST") + " -p " + os.Getenv("DB_PORT") + " -U " + os.Getenv("DB_USERNAME") + " -v -d " + os.Getenv("DB_NAME")
|
|
_, err := exec.Command("bash", "-c", str).Output()
|
|
if err != nil {
|
|
utils.Fatal("Error, in restoring the database %v", err)
|
|
}
|
|
utils.Done("Database has been restored")
|
|
|
|
} else if extension == ".sql" {
|
|
//Restore from sql file
|
|
str := "cat " + fmt.Sprintf("%s/%s", tmpPath, file) + " | psql -h " + os.Getenv("DB_HOST") + " -p " + os.Getenv("DB_PORT") + " -U " + os.Getenv("DB_USERNAME") + " -v -d " + os.Getenv("DB_NAME")
|
|
_, err := exec.Command("bash", "-c", str).Output()
|
|
if err != nil {
|
|
utils.Fatal("Error in restoring the database %v", err)
|
|
}
|
|
utils.Done("Database has been restored")
|
|
} else {
|
|
utils.Fatal("Unknown file extension: %s", extension)
|
|
}
|
|
|
|
} else {
|
|
utils.Fatal("File not found in %s", fmt.Sprintf("%s/%s", tmpPath, file))
|
|
}
|
|
}
|