feat(pkg/task/lib/apt_install): Add support for Update param (for apt…

…-get update)
sikalabs · Mar 28, 2023 · a5e5e71 · a5e5e71
1 parent 97520dc
commit a5e5e71
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/pkg/task/lib/apt_install/apt_install.go b/pkg/task/lib/apt_install/apt_install.go
@@ -8,15 +8,27 @@ import (
 )
 
 type TaskAptInstall struct {
-	Name  string `yaml:"name"`
-	State string `yaml:"state"`
+	Name   string `yaml:"name"`
+	State  string `yaml:"state"`
+	Update bool   `yaml:"update"`
 }
 
 func Run(
 	taskInput libtask.TaskInput,
 	taskParams TaskAptInstall,
 ) libtask.TaskOutput {
 	var err error
+	if taskParams.Update {
+		err = exec_utils.SSH(
+			taskInput,
+			"apt-get", "update",
+		)
+		if err != nil {
+			return libtask.TaskOutput{
+				Error: err,
+			}
+		}
+	}
 	if taskParams.State == "" {
 		taskParams.State = "present"
 	}