Feat: major refactor

1 files changed, 9 insertions, 43 deletions

diff --git a/manifests/router/torrent.pp b/manifests/router/torrent.pp
index 08f4c7e..b5ac97d 100644
--- a/manifests/router/torrent.pp
+++ b/manifests/router/torrent.pp
@@ -1,48 +1,14 @@
 class firewall::router::torrent(
   $destination,
-  $zone         = 'loc',
-  $originaldest = $ipaddress,
-  $range        = lookup('firewall::torrent::range', undef, undef, '6881:6999')
+  $zone           = 'loc',
+  $originaldest   = $ipaddress,
+  $range          = lookup('firewall::torrent::range', undef, undef, '6881:6999')
+  $implementation = lookup('firewall::implementation', undef, undef, 'shorewall'),
 ) {
-  shorewall::rule { "torrent-tcp-1":
-    action          => 'DNAT',
-    source          => 'net',
-    destination     => "$zone:$destination",
-    proto           => 'tcp',
-    destinationport => "$range",
-    ratelimit       => '-',
-    order           => 200,
-  }
-
-  shorewall::rule { "torrent-tcp-2":
-    action          => 'DNAT',
-    source          => 'all',
-    destination     => "$zone:$destination",
-    proto           => 'tcp',
-    destinationport => "$range",
-    originaldest    => "$originaldest",
-    ratelimit       => '-',
-    order           => 200,
-  }
-
-  shorewall::rule { "torrent-udp-1":
-    action          => 'DNAT',
-    source          => 'net',
-    destination     => "$zone:$destination",
-    proto           => 'udp',
-    destinationport => "$range",
-    ratelimit       => '-',
-    order           => 201,
-  }
-
-  shorewall::rule { "torrent-udp-2":
-    action          => 'DNAT',
-    source          => 'all',
-    destination     => "$zone:$destination",
-    proto           => 'udp',
-    destinationport => "6881:6999",
-    originaldest    => "$originaldest",
-    ratelimit       => '-',
-    order           => 201,
+  class { "firewall::implementations::${implementation}::router::torrent":
+    destination  => $destination,
+    zone         => $zone,
+    originaldest => $originaldest,
+    range        => $range,
   }
 }