From 5b092b1988d6384cb03576d231c288d1011db7a2 Mon Sep 17 00:00:00 2001
From: Marcel Otte <otte.mm@googlemail.com>
Date: Sun, 13 Feb 2022 00:20:45 +0100
Subject: [PATCH] Added a possibility to have scripts with arguments

---
 backup.go | 47 ++++++++++++++++++++++++++++++++++-------------
 1 file changed, 34 insertions(+), 13 deletions(-)

diff --git a/backup.go b/backup.go
index 1b6c487..cbbabbc 100644
--- a/backup.go
+++ b/backup.go
@@ -13,21 +13,20 @@ import (
 )
 
 // Mockings
-var mockExecCommand = exec.Command
 var mockCmdRun = func(c *exec.Cmd) error {
 	return c.Run()
 }
 
 // Backup contains all necessary information for executing a configured backup.
 type Backup struct {
-	Name         string `mapstructure:",omitempty"`
-	TargetDevice string `mapstructure:"targetDevice"`
-	TargetPath   string `mapstructure:"targetPath"`
-	SourcePath   string `mapstructure:"sourcePath"`
-	ScriptPath   string `mapstructure:"scriptPath"`
-	Frequency    int    `mapstructure:"frequency"`
-	ExeUser      string `mapstructure:"user,omitempty"`
-	Label        string `mapstructure:"label,omitempty"`
+	Name         string      `mapstructure:",omitempty"`
+	TargetDevice string      `mapstructure:"targetDevice"`
+	TargetPath   string      `mapstructure:"targetPath"`
+	SourcePath   string      `mapstructure:"sourcePath"`
+	ScriptPath   interface{} `mapstructure:"scriptPath"`
+	Frequency    int         `mapstructure:"frequency"`
+	ExeUser      string      `mapstructure:"user,omitempty"`
+	Label        string      `mapstructure:"label,omitempty"`
 	logger       *log.Logger
 }
 
@@ -81,7 +80,7 @@ func (b *Backup) PrepareRun() error {
 	}
 	writer := io.MultiWriter(logfile)
 	b.logger = log.New(writer, b.Name, log.LstdFlags)
-	cmd := mockExecCommand("chown", "-R", b.ExeUser, backupPath)
+	cmd := exec.Command("chown", "-R", b.ExeUser, backupPath)
 	err = mockCmdRun(cmd)
 	if err != nil {
 		b.logger.Printf("chown for backup directory failed: %s", err)
@@ -102,15 +101,37 @@ func (b *Backup) Run() error {
 		return fmt.Errorf("device %s not found", b.TargetDevice)
 	}
 	if ok && dev.IsMounted() {
-		if !strings.ContainsAny(b.ScriptPath, "/") || strings.HasPrefix(b.ScriptPath, ".") {
+		var scriptWArgs []string
+		switch slice := b.ScriptPath.(type) {
+		case []interface{}:
+			for _, v := range slice {
+				scriptWArgs = append(scriptWArgs, v.(string))
+			}
+		case []string:
+			for _, v := range slice {
+				scriptWArgs = append(scriptWArgs, v)
+			}
+		case string:
+			scriptWArgs = append(scriptWArgs, slice)
+		default:
+			log.Print("Fuck, the var is nothing we predicted...")
+		}
+		if !strings.ContainsAny(scriptWArgs[0], "/") || strings.HasPrefix(scriptWArgs[0], ".") {
 			//The scriptPath is a relative path, from the place of the config, so use the config as base
 			log.Printf("ERROR: Script path is relative, aborting.")
 			return fmt.Errorf("script path is relative, aborting")
 		}
-		cmd := mockExecCommand("/usr/bin/sh", b.ScriptPath)
+		var cmd *exec.Cmd
+		var args []string
 		if b.ExeUser != "" {
 			// setup script environment including user to use
-			cmd = mockExecCommand("sudo", "-E", "-u", b.ExeUser, "/usr/bin/sh", b.ScriptPath)
+			args = []string{"-E", "-u", b.ExeUser, "/usr/bin/sh"}
+			args = append(args, scriptWArgs...)
+			cmd = exec.Command("sudo", args...)
+		} else {
+			args = []string{}
+			args = append(args, scriptWArgs...)
+			cmd = exec.Command("/usr/bin/sh", args...)
 		}
 		b.logger.Printf("Running backup script of '%s'", b.Name)
 		b.logger.Printf("Script is: %s", b.ScriptPath)