services: add simln option for both backends

Author: pinheadmz

src/backends/compose/compose_backend.py

@@ -464,5 +464,6 @@ def service_from_json(self, obj) -> object:
             "devices": obj.get("devices", []),
             "command": obj.get("args", []),
             "environment": obj.get("environment", []),
+            "restart": "on-failure",
             "networks": [self.network_name]
         }

src/backends/kubernetes/kubernetes_backend.py

@@ -15,6 +15,7 @@
 from kubernetes.dynamic import DynamicClient
 from kubernetes.dynamic.exceptions import ResourceNotFoundError
 from kubernetes.stream import stream
+from warnet.services import services
 from warnet.status import RunningStatus
 from warnet.tank import Tank
 from warnet.utils import parse_raw_messages
@@ -73,6 +74,10 @@ def down(self, warnet) -> bool:
 
         self.remove_prometheus_service_monitors(warnet.tanks)
 
+        for service_name in warnet.services:
+            if "k8s" in services[service_name]["backends"]:
+                self.client.delete_namespaced_pod(services[service_name]["container_name_suffix"], self.namespace)
+
         return True
 
     def get_file(self, tank_index: int, service: ServiceType, file_path: str):
@@ -668,6 +673,10 @@ def deploy_pods(self, warnet):
         if self.check_logging_crds_installed():
             self.apply_prometheus_service_monitors(warnet.tanks)
 
+        for service_name in warnet.services:
+            if "k8s" in services[service_name]["backends"]:
+                self.service_from_json(services[service_name])
+
         self.log.debug("Containers and services created. Configuring IP addresses")
         # now that the pods have had a second to create,
         # get the ips and set them on the tanks
@@ -701,4 +710,34 @@ def wait_for_healthy_tanks(self, warnet, timeout=30):
         pass
 
     def service_from_json(self, obj):
-        pass
+        env = []
+        for pair in obj.get("environment", []):
+            name, value = pair.split("=")
+            env.append(client.V1EnvVar(name=name, value=value))
+        service_container = client.V1Container(
+            name=obj["container_name_suffix"],
+            image=obj["image"],
+            env=env,
+            security_context=client.V1SecurityContext(
+                privileged=True,
+                capabilities=client.V1Capabilities(add=["NET_ADMIN", "NET_RAW"]),
+            ),
+        )
+        service_pod = client.V1Pod(
+            api_version="v1",
+            kind="Pod",
+            metadata=client.V1ObjectMeta(
+                name=obj["container_name_suffix"],
+                namespace=self.namespace,
+                labels={
+                    "app": obj["container_name_suffix"],
+                    "network": self.network_name,
+                },
+            ),
+            spec=client.V1PodSpec(
+                restart_policy="OnFailure",
+                containers=[service_container],
+                volumes=[],
+            ),
+        )
+        self.client.create_namespaced_pod(namespace=self.namespace, body=service_pod)

src/warnet/lnnode.py

@@ -13,7 +13,8 @@
     "--bitcoin.active",
     "--bitcoin.regtest",
     "--bitcoin.node=bitcoind",
-    "--maxpendingchannels=64"
+    "--maxpendingchannels=64",
+    "--trickledelay=1"
 ])
 
 class LNNode:

src/warnet/services.py

@@ -78,5 +78,11 @@
         "container_port": "9090",
         "config_files": [f"/{PROM_CONF_NAME}:/etc/prometheus/prometheus.yml"],
         "args": ["--config.file=/etc/prometheus/prometheus.yml"]
-    }
+    },
+    "simln": {
+        "backends": ["compose", "k8s"],
+        "image": "bitcoindevproject/simln:0.2.0",
+        "container_name_suffix": "simln",
+        "args": ["--log-level=debug"]
+    },
 }

test/data/services.graphml

@@ -15,11 +15,12 @@
   <key id="source_policy"   attr.name="source_policy"    attr.type="string"   for="edge" />
   <key id="target_policy"   attr.name="target_policy"    attr.type="string"   for="edge" />
   <graph edgedefault="directed">
-    <data key="services">cadvisor forkobserver grafana nodeexporter prometheus</data>
+    <data key="services">cadvisor forkobserver grafana nodeexporter prometheus simln</data>
     <node id="0">
         <data key="version">26.0</data>
         <data key="bitcoin_config">-uacomment=w0 -debug=validation</data>
         <data key="exporter">true</data>
+        <data key="ln">lnd</data>
     </node>
   </graph>
 </graphml>