From a5e5e712dcb8d65efe65c5a754feca147fba8999 Mon Sep 17 00:00:00 2001
From: Ondrej Sika <ondrej@ondrejsika.com>
Date: Tue, 28 Mar 2023 20:16:06 +0200
Subject: [PATCH] feat(pkg/task/lib/apt_install): Add support for Update param
 (for apt-get update)

---
 pkg/task/lib/apt_install/apt_install.go | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/pkg/task/lib/apt_install/apt_install.go b/pkg/task/lib/apt_install/apt_install.go
index 22cd758..09918dd 100644
--- a/pkg/task/lib/apt_install/apt_install.go
+++ b/pkg/task/lib/apt_install/apt_install.go
@@ -8,8 +8,9 @@ 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(
@@ -17,6 +18,17 @@ func Run(
 	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"
 	}