From bd2208922ea6a0486074c72a57b01d7df368f885 Mon Sep 17 00:00:00 2001
From: Trygve Laugstøl <trygvis@inamo.no>
Date: Wed, 6 Jan 2021 09:12:56 +0100
Subject: Generating IP addresses.

---
 src/main/java/io/trygvis/rules/engine/Main.java | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'src/main/java/io/trygvis/rules/engine/Main.java')

diff --git a/src/main/java/io/trygvis/rules/engine/Main.java b/src/main/java/io/trygvis/rules/engine/Main.java
index 5556db7..6f04a98 100644
--- a/src/main/java/io/trygvis/rules/engine/Main.java
+++ b/src/main/java/io/trygvis/rules/engine/Main.java
@@ -3,6 +3,8 @@ package io.trygvis.rules.engine;
 import io.trygvis.rules.acme.AcmeIo;
 import io.trygvis.rules.dns.DnsEntry;
 import io.trygvis.rules.machine.Machine;
+import io.trygvis.rules.network.Ipv4Address;
+import io.trygvis.rules.network.Ipv4Cidr;
 import org.drools.core.audit.WorkingMemoryConsoleLogger;
 import org.kie.api.KieServices;
 import org.kie.api.event.rule.AgendaEventListener;
@@ -38,9 +40,12 @@ public class Main {
 
         io.dump("phase-1", session.getFactHandles());
 
-        io.dump("vpn0", session.getFactHandles(), (Object o) -> {
-            return o.getClass().getName().contains("Wg") || o instanceof Machine || o instanceof DnsEntry;
-        });
+        io.dump("vpn0", session.getFactHandles(), (Object o) ->
+                o.getClass().getName().contains("Wg") ||
+                o instanceof Machine ||
+                o instanceof DnsEntry ||
+                o instanceof Ipv4Cidr ||
+                o instanceof Ipv4Address);
 
         session.dispose();
     }
-- 
cgit v1.2.3