Adds ensure param

1 files changed, 10 insertions, 7 deletions

diff --git a/manifests/init.pp b/manifests/init.pp
index 605dd4d..f3c4836 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -17,22 +17,23 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 class syslog-ng(
-  $log_kernel_msgs = hiera('syslog-ng::log_kernel_msgs', true)
+  $log_kernel_msgs = hiera('syslog-ng::log_kernel_msgs', true),
+  $ensure          = 'present'
 ) {
   # see https://bugs.launchpad.net/ubuntu/+source/syslog-ng/+bug/1242173
   package { 'syslog-ng-core':
-    ensure => installed,
+    ensure => $ensure,
   }
 
   package { 'syslog-ng':
-    ensure  => installed,
+    ensure  => $ensure,
     require => Package['syslog-ng-core'],
   }
 
   # syslog-ng service
   service { "syslog-ng":
     enable     => true,
-    ensure     => running,
+    ensure     => $ensure,
     hasrestart => true,
     hasstatus  => false,
     # Somehow syslog-ng is not returning if it was killed due to lack of
@@ -49,17 +50,19 @@ class syslog-ng(
     owner   => "root",
     group   => "root",
     mode    => 0644,
-    ensure  => present,
+    ensure  => $ensure,
     notify  => Service["syslog-ng"],
     require => File["/etc/syslog-ng"],
   }
 
   # config folder
   file { "/etc/syslog-ng":
-    ensure  => directory,
+    ensure  => $ensure ? {
+      'present' => directory,
+      default   => absent,
+    },
     owner   => "root",
     group   => "root",
     require => Package["syslog-ng"],
   }
-
 }