From bafe762ac01d16904c18404283027e426e19bc73 Mon Sep 17 00:00:00 2001
From: Trygve Laugstøl <trygvis@inamo.no>
Date: Wed, 3 Feb 2021 16:35:48 +0100
Subject: Code reorganization.

Moving main code to src, keeping modules in modules/
---
 .../main/resources/io/trygvis/rules/dba/dba.drl    | 69 ++++++++++++++++++
 .../resources/io/trygvis/rules/engine/init.drl     | 17 +++++
 .../resources/io/trygvis/rules/machine/machine.drl |  1 +
 .../io/trygvis/rules/scaleway/terraform.drl        | 81 ++++++++++++++++++++++
 .../rules/terraform/terraform-resources.drl        | 28 ++++++++
 5 files changed, 196 insertions(+)
 create mode 100644 modules/ri-base/src/main/resources/io/trygvis/rules/dba/dba.drl
 create mode 100644 modules/ri-base/src/main/resources/io/trygvis/rules/engine/init.drl
 create mode 100644 modules/ri-base/src/main/resources/io/trygvis/rules/machine/machine.drl
 create mode 100644 modules/ri-base/src/main/resources/io/trygvis/rules/scaleway/terraform.drl
 create mode 100644 modules/ri-base/src/main/resources/io/trygvis/rules/terraform/terraform-resources.drl

(limited to 'modules/ri-base/src/main/resources/io/trygvis/rules')

diff --git a/modules/ri-base/src/main/resources/io/trygvis/rules/dba/dba.drl b/modules/ri-base/src/main/resources/io/trygvis/rules/dba/dba.drl
new file mode 100644
index 0000000..c31bc57
--- /dev/null
+++ b/modules/ri-base/src/main/resources/io/trygvis/rules/dba/dba.drl
@@ -0,0 +1,69 @@
+package io.trygvis.rules.dba
+
+import io.trygvis.rules.core.Problem
+import io.trygvis.rules.machine.Machine
+import java.util.ArrayList
+import java.util.Map
+import java.util.HashMap
+import java.util.List
+import java.util.stream.Collectors
+import java.util.Collections
+
+global io.trygvis.rules.engine.TemplateEngine te;
+
+dialect  "mvel"
+
+declare DbaMachineRole
+    machine : String
+    roles   : String[]
+end
+
+rule "Assign containers to machine"
+when
+    $machine : Machine()
+    $machineRole : DbaMachineRole(machine == $machine.name)
+    $container : Container(machine == null, $machineRole.roles contains machineRole)
+then
+    System.out.println("Assigning container to machine: " + $machine.name);
+    modify ($container) {
+        machine = $machine
+    }
+end
+
+rule "Containers without hosts"
+    agenda-group "generate"
+when
+    $container : Container(machine == null)
+then
+    insert(new Problem("No machine for container", $container))
+end
+
+rule "Generate docker-compose.yaml"
+    agenda-group "generate"
+when
+    $cluster    : Cluster()
+    $containers : ArrayList(size > 0) from collect(Container(cluster == $cluster))
+then
+    System.out.println("Docker compose for cluster: " + $cluster.name + " with " + $containers.size() + " containers");
+
+    Map containersByMachine = new HashMap();
+    for (Object o : $containers) {
+        Container c = (Container) o;
+
+        List list = (List) containersByMachine.get(c.getMachine());
+        if (list == null) {
+            list = new ArrayList();
+            containersByMachine.put(c.getMachine(), list);
+        }
+        list.add(c);
+    }
+
+    System.out.println("containersByMachine = " + containersByMachine);
+
+    String path = "ansible/dba/" + $cluster.name + ".yml";
+    te.template("dba/cluster", path, Map.of(
+        "cluster", $cluster,
+        "containers", $containers,
+        "containersByMachine", containersByMachine
+    ));
+end
diff --git a/modules/ri-base/src/main/resources/io/trygvis/rules/engine/init.drl b/modules/ri-base/src/main/resources/io/trygvis/rules/engine/init.drl
new file mode 100644
index 0000000..267cc4a
--- /dev/null
+++ b/modules/ri-base/src/main/resources/io/trygvis/rules/engine/init.drl
@@ -0,0 +1,17 @@
+package io.trygvis.rules.engine;
+
+import java.util.Map
+import org.apache.commons.io.FileSystem
+import org.apache.commons.io.FileUtils
+import java.io.File
+
+global io.trygvis.rules.engine.TemplateEngine te;
+
+rule "Clean directories"
+    agenda-group "init"
+when
+    not(KeyValue(key == "rm-gen"));
+then
+    te.clean();
+    insert(new KeyValue("rm-gen", null));
+end
diff --git a/modules/ri-base/src/main/resources/io/trygvis/rules/machine/machine.drl b/modules/ri-base/src/main/resources/io/trygvis/rules/machine/machine.drl
new file mode 100644
index 0000000..0250cc6
--- /dev/null
+++ b/modules/ri-base/src/main/resources/io/trygvis/rules/machine/machine.drl
@@ -0,0 +1 @@
+package io.trygvis.rules.machine;
diff --git a/modules/ri-base/src/main/resources/io/trygvis/rules/scaleway/terraform.drl b/modules/ri-base/src/main/resources/io/trygvis/rules/scaleway/terraform.drl
new file mode 100644
index 0000000..7e0ff03
--- /dev/null
+++ b/modules/ri-base/src/main/resources/io/trygvis/rules/scaleway/terraform.drl
@@ -0,0 +1,81 @@
+package io.trygvis.rules.scaleway
+
+import io.trygvis.rules.dba.Cluster
+import io.trygvis.rules.dba.Container
+import io.trygvis.rules.machine.Machine
+import io.trygvis.rules.dns.DnsEntry
+import io.trygvis.rules.dns.DnsEntryTerraformExpression
+import java.util.ArrayList;
+import java.util.Map;
+
+global io.trygvis.rules.engine.TemplateEngine te;
+
+dialect  "mvel"
+
+declare ScalewayMachine
+    machine : Machine
+    key : String
+end
+
+declare GoogleManagedZoneTerraformExpression
+    name : String
+end
+
+rule "Terraform for Machine"
+when
+    $machine: Machine()
+then
+    ScalewayMachine scw = new ScalewayMachine();
+    scw.setKey($machine.name);
+    scw.setMachine($machine);
+
+    insert(scw);
+end
+
+rule "Create DNS entry for Terraform Machine"
+when
+    $machine : Machine(fqdn != null)
+    not(DnsEntry(fqdn == $machine.fqdn))
+then
+    DnsEntry a = DnsEntry.a($machine.fqdn);
+    insert(a);
+
+    String ipv4 = "scaleway_instance_ip.%s.address".formatted($machine.name);
+    insert(new DnsEntryTerraformExpression(a, $machine.name, ipv4));
+end
+
+rule "main-scaleway-machine.tf"
+    agenda-group "generate"
+when
+    $managedZones : ArrayList() from collect(GoogleManagedZoneTerraformExpression())
+then
+    String path = "terraform/main-scaleway-machine.tf";
+    te.template("terraform/main-scaleway-machine", path, Map.of(
+        "managedZones", $managedZones
+    ));
+end
+
+rule "TF for TerraformMachine"
+    agenda-group "generate"
+when
+    $m: Machine()
+    $scw: ScalewayMachine(machine == $m)
+then
+    String path = "terraform/scaleway-machine-%s.tf".formatted($scw.getKey());
+    te.template("terraform/machine", path, Map.of("m", $m, "scw", $scw));
+end
+
+rule "Terraform for DNS"
+    agenda-group "generate"
+when
+    $entry: DnsEntry()
+    $tf : DnsEntryTerraformExpression(entry == $entry)
+    $managedZone : GoogleManagedZoneTerraformExpression()
+then
+    String path = "terraform/dns-%s.tf".formatted($tf.key);
+    te.template("terraform/record-set", path, Map.of(
+            "entry", $entry,
+            "managedZone", $managedZone,
+            "tf", $tf)
+    );
+end
diff --git a/modules/ri-base/src/main/resources/io/trygvis/rules/terraform/terraform-resources.drl b/modules/ri-base/src/main/resources/io/trygvis/rules/terraform/terraform-resources.drl
new file mode 100644
index 0000000..40a81f0
--- /dev/null
+++ b/modules/ri-base/src/main/resources/io/trygvis/rules/terraform/terraform-resources.drl
@@ -0,0 +1,28 @@
+package io.trygvis.rules.terraform;
+
+import java.util.ArrayList
+import java.util.HashSet
+import java.util.Map
+import java.util.Set
+
+global io.trygvis.rules.engine.TemplateEngine te;
+
+dialect  "mvel"
+
+rule "terraform-resources"
+    agenda-group "generate"
+when
+    $r : TerraformResource()
+then
+    te.template("terraform/resource", $r.output, Map.of("resource", $r));
+end
+
+rule "main.tf"
+    agenda-group "generate"
+when
+    $main : TerraformMain()
+    $variables : ArrayList() from collect(TerraformInputVariable(module == $main.module))
+then
+    te.template("terraform/main", $main.module + "/main.tf", Map.of("main", $main));
+    te.template("terraform/variables", $main.module + "/vars.tf", Map.of("variables", $variables));
+end
-- 
cgit v1.2.3