mysql-bkup/pkg/restore.go

package pkg

import (
	"fmt"
	"github.com/jkaninda/mysql-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(fmt.Sprintf("Error download file from s3 %s %s", 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(fmt.Sprintf("Error download file from remote server: ", 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")
	}
	// dbHVars Required environment variables for database
	var dbHVars = []string{
		"DB_HOST",
		"DB_PASSWORD",
		"DB_USERNAME",
		"DB_NAME",
	}
	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)
	}

	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 ", file, err)
			}
			//Update file name
			file = RemoveLastExtension(file)
		}

	}

	if utils.FileExists(fmt.Sprintf("%s/%s", tmpPath, file)) {

		err := os.Setenv("mysqlPASSWORD", 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) + " | mysql -h " + os.Getenv("DB_HOST") + " -P " + os.Getenv("DB_PORT") + " -u " + os.Getenv("DB_USERNAME") + " --password=" + os.Getenv("DB_PASSWORD") + " " + os.Getenv("DB_NAME")
			_, err := exec.Command("bash", "-c", str).Output()
			if err != nil {
				utils.Fatal(fmt.Sprintf("Error, in restoring the database  %s", err))
			}
			utils.Done("Database has been restored")

		} else if extension == ".sql" {
			//Restore from sql file
			str := "cat " + fmt.Sprintf("%s/%s", tmpPath, file) + " | mysql -h " + os.Getenv("DB_HOST") + " -P " + os.Getenv("DB_PORT") + " -u " + os.Getenv("DB_USERNAME") + " --password=" + os.Getenv("DB_PASSWORD") + " " + os.Getenv("DB_NAME")
			_, err := exec.Command("bash", "-c", str).Output()
			if err != nil {
				utils.Fatal(fmt.Sprintf("Error in restoring the database %s", err))
			}
			utils.Done("Database has been restored")
		} else {
			utils.Fatal(fmt.Sprintf("Unknown file extension %s", extension))
		}

	} else {
		utils.Fatal(fmt.Sprintf("File not found in %s", fmt.Sprintf("%s/%s", tmpPath, file)))
	}
}