change ensure from file to present, add custom_config parameter, switch rsyslog::server to parameterized class

1 files changed, 7 insertions, 4 deletions

diff --git a/manifests/server.pp b/manifests/server.pp
index dceb252..cf7075c 100644
--- a/manifests/server.pp
+++ b/manifests/server.pp
@@ -1,10 +1,13 @@
-class rsyslog::server inherits rsyslog {
+class rsyslog::server ($enable_tcp = true, $enable_udp = true, $server_dir = '/srv/log/', $custom_config = undef) inherits rsyslog {
     file { $rsyslog::params::server_conf:
+        ensure  => present,
         owner   => root,
         group   => root,
-        ensure  => file,
-        content => template("${module_name}/server.conf.erb"),
-        require => Class['rsyslog::install'],
+        content => $custom_config ? {
+            ''      => template("${module_name}/server.conf.erb"),
+            default => template($custom_config),
+        },
+        require => Class['rsyslog::config'],
         notify  => Class['rsyslog::service'],
     }
 }