Add beforeScript field (#265)

Co-authored-by: Kirill Kholodilin <[email protected]>
Clivern · May 30, 2021 · e10cc5b · e10cc5b
1 parent ddea52d
commit e10cc5b
Show file tree

Hide file tree

Showing 5 changed files with 35 additions and 1 deletion.
diff --git a/cmd/pkged.go b/cmd/pkged.go
diff --git a/core/backup/manager.go b/core/backup/manager.go
@@ -6,6 +6,8 @@ package backup
 
 import (
 	"fmt"
+	"os"
+	"os/exec"
 	"time"
 
 	"github.com/clivern/walrus/core/model"
@@ -43,6 +45,19 @@ func (m *Manager) ProcessBackup(message module.BackupMessage) error {
 		fileName,
 	)
 
+	if message.Cron.Request.BeforeScript != "" {
+
+		cmd := exec.Command("sh", "-c", message.Cron.Request.BeforeScript)
+
+		cmd.Stdin = os.Stdin
+		// cmd.Stdout = os.Stdout
+		// cmd.Stderr = os.Stderr
+		err := cmd.Run()
+		if err != nil {
+			return err
+		}
+	}
+
 	// BackupDirectory
 	if message.Cron.Request.Type == model.BackupDirectory {
 		err := m.BackupDirectory(

diff --git a/core/controller/tower/host.go b/core/controller/tower/host.go
@@ -276,6 +276,7 @@ func CreateHostCron(c *gin.Context) {
 		Request: model.Request{
 			Type:          inputs["type"],
 			Directory:     inputs["directory"],
+			BeforeScript:  inputs["beforeScript"],
 			RetentionDays: retention,
 
 			// MySQL
@@ -382,6 +383,7 @@ func UpdateHostCron(c *gin.Context) {
 
 	cronData.Name = inputs["name"]
 	cronData.Request.Directory = inputs["directory"]
+	cronData.Request.BeforeScript = inputs["beforeScript"]
 	cronData.Request.RetentionDays = retention
 	cronData.Request.Type = inputs["type"]
 
@@ -520,6 +522,7 @@ func GetHostCron(c *gin.Context) {
 		"request": gin.H{
 			"type":          cronData.Request.Type,
 			"directory":     cronData.Request.Directory,
+			"beforeScript":  cronData.Request.BeforeScript,
 			"retentionDays": cronData.Request.RetentionDays,
 
 			// MySQL

diff --git a/core/model/cron.go b/core/model/cron.go
@@ -62,6 +62,7 @@ const (
 type Request struct {
 	Type          string `json:"type"` // @BackupSQLite, @BackupMySQL, @BackupDirectory, @BackupRedis, @BackupPostgreSQL
 	Directory     string `json:"directory"`
+	BeforeScript  string `json:"beforeScript"`
 	RetentionDays int    `json:"retentionDays"`
 
 	MySQLHost         string `json:"mysqlHost"`

diff --git a/web/src/views/Host.vue b/web/src/views/Host.vue
@@ -133,6 +133,15 @@
 							</b-select>
 						</b-field>
 
+                        <b-field label="Run Script Before">
+                            <b-input
+                                type="text"
+                                v-model="form.beforeScript"
+                                placeholder="/usr/bin/innobackupex --incremental"
+                            >
+                            </b-input>
+                        </b-field>
+
 						<template v-if="form.type == '@BackupDirectory'">
 							<b-field label="Directory Path">
 								<b-input
@@ -301,6 +310,7 @@ export default {
 				name: "",
 				interval: 30,
 				retention: 10,
+				beforeScript: "",
 				directory: "",
 
 				mysqlHost: "127.0.0.1",
@@ -411,6 +421,7 @@ export default {
 			this.form.interval = 30;
 			this.form.retention = 10;
 			this.form.directory = "";
+			this.form.beforeScript = "";
 			this.form.intervalType = "@minute";
 			this.form.type = "@BackupDirectory";
 			this.form.hostId = hostId;
@@ -430,6 +441,7 @@ export default {
 							this.form.name = data.name;
 							this.form.interval = data.interval;
 							this.form.retention = data.request.retentionDays;
+							this.form.beforeScript = data.request.beforeScript;
 							this.form.directory = data.request.directory;
 
 							this.form.mysqlHost = data.request.mysqlHost;
@@ -469,6 +481,7 @@ export default {
 			this.form.interval = 30;
 			this.form.retention = 10;
 			this.form.directory = "";
+			this.form.beforeScript = "";
 			this.form.intervalType = "@minute";
 			this.form.type = "@BackupDirectory";
 			this.form.hostId = hostId;
@@ -494,6 +507,7 @@ export default {
 			this.form.interval = 30;
 			this.form.retention = 10;
 			this.form.directory = "";
+			this.form.beforeScript = "";
 			this.form.intervalType = "@minute";
 			this.form.type = "@BackupDirectory";
 			this.form.hostId = "";
@@ -526,6 +540,7 @@ export default {
 					name: this.form.name,
 					interval: this.form.interval.toString(),
 					retention: this.form.retention.toString(),
+					beforeScript: this.form.beforeScript,
 					directory: this.form.directory,
 					intervalType: this.form.intervalType,