feat(pkg/task/replace_string_in_local_file): Add TaskReplaceStringInLocalFile

Author: ondrejsika

pkg/task/lib/replace_string_in_local_file/replace_string_in_local_file.go

@@ -0,0 +1,43 @@
+package replace_string_in_local_file
+
+import (
+	"io/ioutil"
+	"strings"
+
+	"github.com/sikalabs/gobble/pkg/libtask"
+)
+
+type TaskReplaceStringInLocalFile struct {
+	Path    string `yaml:"path"`
+	Find    string `yaml:"find"`
+	Replace string `yaml:"replace"`
+}
+
+func Run(
+	taskInput libtask.TaskInput,
+	taskParams TaskReplaceStringInLocalFile,
+) libtask.TaskOutput {
+	err := replaceStringInFile(taskParams.Path, taskParams.Find, taskParams.Replace)
+	return libtask.TaskOutput{
+		Error: err,
+	}
+}
+
+func replaceStringInFile(filename, searchStr, replaceStr string) error {
+	// read the entire file into memory
+	content, err := ioutil.ReadFile(filename)
+	if err != nil {
+		return err
+	}
+
+	// replace the string
+	newContent := strings.ReplaceAll(string(content), searchStr, replaceStr)
+
+	// write the updated content back to the file
+	err = ioutil.WriteFile(filename, []byte(newContent), 0644)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

pkg/task/task.go

@@ -10,18 +10,20 @@ import (
 	"github.com/sikalabs/gobble/pkg/task/lib/command"
 	"github.com/sikalabs/gobble/pkg/task/lib/cp"
 	"github.com/sikalabs/gobble/pkg/task/lib/print"
+	rsilf "github.com/sikalabs/gobble/pkg/task/lib/replace_string_in_local_file"
 	"github.com/sikalabs/gobble/pkg/task/lib/template"
 )
 
 type Task struct {
-	Name          string                           `yaml:"name"`
-	AptInstall    apt_install.TaskAptInstall       `yaml:"apt_install"`
-	Cp            cp.TaskCp                        `yaml:"cp"`
-	Template      template.TaskTemplate            `yaml:"template"`
-	Command       command.TaskCommand              `yaml:"command"`
-	Chmod         chmod.TaskChmod                  `yaml:"chmod"`
-	Print         print.TaskPrint                  `yaml:"print"`
-	AuthorizedKey authorized_key.TaskAuthorizedKey `yaml:"authorized_key"`
+	Name                         string                             `yaml:"name"`
+	AptInstall                   apt_install.TaskAptInstall         `yaml:"apt_install"`
+	Cp                           cp.TaskCp                          `yaml:"cp"`
+	Template                     template.TaskTemplate              `yaml:"template"`
+	Command                      command.TaskCommand                `yaml:"command"`
+	Chmod                        chmod.TaskChmod                    `yaml:"chmod"`
+	Print                        print.TaskPrint                    `yaml:"print"`
+	AuthorizedKey                authorized_key.TaskAuthorizedKey   `yaml:"authorized_key"`
+	TaskReplaceStringInLocalFile rsilf.TaskReplaceStringInLocalFile `yaml:"replace_string_in_local_file"`
 }
 
 func Run(
@@ -43,6 +45,8 @@ func Run(
 		return print.Run(taskInput, task.Print)
 	case task.AuthorizedKey.Key != "":
 		return authorized_key.Run(taskInput, task.AuthorizedKey)
+	case task.TaskReplaceStringInLocalFile.Path != "":
+		return rsilf.Run(taskInput, task.TaskReplaceStringInLocalFile)
 	}
 	return libtask.TaskOutput{
 		Error: fmt.Errorf("task \"%s\" not found", task.Name),