From 14054b213100f173f229d7f01d4c1ed948846160 Mon Sep 17 00:00:00 2001
From: Arnout Boks <arnoutboks@gmail.com>
Date: Tue, 11 Dec 2018 20:59:55 +0100
Subject: [PATCH] Add support for dist-upgrade & autoremove action

---
 tasks/init.json | 2 +-
 tasks/init.rb   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tasks/init.json b/tasks/init.json
index 2020005..91cf951 100644
--- a/tasks/init.json
+++ b/tasks/init.json
@@ -4,7 +4,7 @@
   "parameters": {
     "action": {
       "description": "Action to perform ",
-      "type": "Enum[update, upgrade]"
+      "type": "Enum[update, upgrade, dist-upgrade, autoremove]"
     }
   }
 }
diff --git a/tasks/init.rb b/tasks/init.rb
index b4cb0dc..f8ca85e 100755
--- a/tasks/init.rb
+++ b/tasks/init.rb
@@ -5,7 +5,7 @@ require 'puppet'
 
 def apt_get(action)
   cmd = ['apt-get', action]
-  cmd << '-y' if action == 'upgrade'
+  cmd << '-y' if ['upgrade', 'dist-upgrade', 'autoremove'].include?(action)
   stdout, stderr, status = Open3.capture3(*cmd)
   raise Puppet::Error, stderr if status != 0 # rubocop:disable GetText/DecorateFunctionMessage
   { status: stdout.strip }
-- 
2.45.2