mirror of
https://github.com/jkaninda/mysql-bkup.git
synced 2025-12-06 21:49:40 +01:00
115 lines
2.3 KiB
Go
115 lines
2.3 KiB
Go
package utils
|
|
|
|
/*****
|
|
* MySQL Backup & Restore
|
|
* @author Jonas Kaninda
|
|
* @license MIT License <https://opensource.org/licenses/MIT>
|
|
* @link https://github.com/jkaninda/mysql-bkup
|
|
**/
|
|
import (
|
|
"fmt"
|
|
"github.com/spf13/cobra"
|
|
"io/fs"
|
|
"os"
|
|
"os/exec"
|
|
)
|
|
|
|
func Info(v ...any) {
|
|
fmt.Println("⒤ ", fmt.Sprint(v...))
|
|
}
|
|
func Done(v ...any) {
|
|
fmt.Println("✔ ", fmt.Sprint(v...))
|
|
}
|
|
func Fatal(v ...any) {
|
|
fmt.Println("✘ ", fmt.Sprint(v...))
|
|
os.Exit(1)
|
|
}
|
|
func Fatalf(msg string, v ...any) {
|
|
fmt.Printf("✘ "+msg, v...)
|
|
os.Exit(1)
|
|
}
|
|
|
|
func FileExists(filename string) bool {
|
|
info, err := os.Stat(filename)
|
|
if os.IsNotExist(err) {
|
|
return false
|
|
}
|
|
return !info.IsDir()
|
|
}
|
|
|
|
func WriteToFile(filePath, content string) error {
|
|
file, err := os.Create(filePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
_, err = file.WriteString(content)
|
|
return err
|
|
}
|
|
func ChangePermission(filePath string, mod int) {
|
|
if err := os.Chmod(filePath, fs.FileMode(mod)); err != nil {
|
|
Fatalf("Error changing permissions of %s: %v\n", filePath, err)
|
|
}
|
|
|
|
}
|
|
func IsDirEmpty(name string) (bool, error) {
|
|
f, err := os.Open(name)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer f.Close()
|
|
|
|
_, err = f.Readdirnames(1)
|
|
if err == nil {
|
|
return false, nil
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
// TestDatabaseConnection tests the database connection
|
|
func TestDatabaseConnection() {
|
|
Info("Testing database connection...")
|
|
// Test database connection
|
|
cmd := exec.Command("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"), "-e", "quit")
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
Fatal("Error testing database connection:", err)
|
|
|
|
}
|
|
|
|
}
|
|
func GetEnv(cmd *cobra.Command, flagName, envName string) string {
|
|
value, _ := cmd.Flags().GetString(flagName)
|
|
if value != "" {
|
|
err := os.Setenv(envName, value)
|
|
if err != nil {
|
|
return value
|
|
}
|
|
}
|
|
return os.Getenv(envName)
|
|
}
|
|
func FlagGetString(cmd *cobra.Command, flagName string) string {
|
|
value, _ := cmd.Flags().GetString(flagName)
|
|
if value != "" {
|
|
return value
|
|
|
|
}
|
|
return ""
|
|
}
|
|
func FlagGetBool(cmd *cobra.Command, flagName string) bool {
|
|
value, _ := cmd.Flags().GetBool(flagName)
|
|
return value
|
|
}
|
|
|
|
func SetEnv(key, value string) {
|
|
|
|
err := os.Setenv(key, value)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
func ShowHistory() {
|
|
}
|