Generating IP addresses.

1 files changed, 8 insertions, 3 deletions

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();
     }