introduce parametrized class for ports so we can pass the ports from the sshd module

1 files changed, 10 insertions, 9 deletions

diff --git a/manifests/rules/ssh.pp b/manifests/rules/ssh.pp
index f587259..0eebcb4 100644
--- a/manifests/rules/ssh.pp
+++ b/manifests/rules/ssh.pp
@@ -1,10 +1,11 @@
-class shorewall::rules::ssh {
-    shorewall::rule { 'net-me-tcp_ssh':
-        source          => 'net',
-        destination     => '$FW',
-        proto           => 'tcp',
-        destinationport => 'ssh',
-        order           => 240,
-        action          => 'ACCEPT';
-    }
+class shorewall::rules::ssh($ports) {
+  $flatted_ports = join($ports,',')
+  shorewall::rule { 'net-me-tcp_ssh':
+    source          => 'net',
+    destination     => '$FW',
+    proto           => 'tcp',
+    destinationport => $flatted_ports,
+    order           => 240,
+    action          => 'ACCEPT';
+  }
 }