mysql-bkup/pkg/scripts.go

// Package pkg /
/*****
@author    Jonas Kaninda
@license   MIT License <https://opensource.org/licenses/MIT>
@Copyright © 2024 Jonas Kaninda
**/
package pkg

import (
	"fmt"
	"github.com/jkaninda/mysql-bkup/utils"
	"os"
	"os/exec"
)

func CreateCrontabScript(disableCompression bool, storage string) {
	//task := "/usr/local/bin/backup_cron.sh"
	touchCmd := exec.Command("touch", backupCronFile)
	if err := touchCmd.Run(); err != nil {
		utils.Fatal("Error creating file %s: %v\n", backupCronFile, err)
	}
	var disableC = ""
	if disableCompression {
		disableC = "--disable-compression"
	}

	scriptContent := fmt.Sprintf(`#!/usr/bin/env bash
set -e
/usr/local/bin/mysql-bkup backup --dbname %s --port %s --storage %s %v
`, os.Getenv("DB_NAME"), os.Getenv("DB_PORT"), storage, disableC)

	if err := utils.WriteToFile(backupCronFile, scriptContent); err != nil {
		utils.Fatal("Error writing to %s: %v\n", backupCronFile, err)
	}

	chmodCmd := exec.Command("chmod", "+x", "/usr/local/bin/backup_cron.sh")
	if err := chmodCmd.Run(); err != nil {
		utils.Fatal("Error changing permissions of %s: %v\n", backupCronFile, err)
	}

	lnCmd := exec.Command("ln", "-s", "/usr/local/bin/backup_cron.sh", "/usr/local/bin/backup_cron")
	if err := lnCmd.Run(); err != nil {
		utils.Fatal("Error creating symbolic link: %v\n", err)

	}

	touchLogCmd := exec.Command("touch", cronLogFile)
	if err := touchLogCmd.Run(); err != nil {
		utils.Fatal("Error creating file %s: %v\n", cronLogFile, err)
	}

	cronJob := "/etc/cron.d/backup_cron"
	touchCronCmd := exec.Command("touch", cronJob)
	if err := touchCronCmd.Run(); err != nil {
		utils.Fatal("Error creating file %s: %v\n", cronJob, err)
	}

	cronContent := fmt.Sprintf(`%s root exec /bin/bash -c ". /run/supervisord.env; /usr/local/bin/backup_cron.sh >> %s"
`, os.Getenv("BACKUP_CRON_EXPRESSION"), cronLogFile)

	if err := utils.WriteToFile(cronJob, cronContent); err != nil {
		utils.Fatal("Error writing to %s: %v\n", cronJob, err)
	}
	utils.ChangePermission("/etc/cron.d/backup_cron", 0644)

	crontabCmd := exec.Command("crontab", "/etc/cron.d/backup_cron")
	if err := crontabCmd.Run(); err != nil {
		utils.Fatal("Error updating crontab: ", err)
	}
	utils.Info("Backup job created.")
}