enable logs and status for plugins

Author: mplsgrant

src/warnet/constants.py

@@ -18,10 +18,13 @@
 KUBE_INTERNAL_NAMESPACES = ["kube-node-lease", "kube-public", "kube-system", "kubernetes-dashboard"]
 HELM_COMMAND = "helm upgrade --install"
 
+MISSION_TAG = "mission"
 TANK_MISSION = "tank"
 COMMANDER_MISSION = "commander"
 PLUGIN_MISSION = "plugin"
+LIGHTNING_MISSION = "lightning"
 
+CONTAINER_TAG = "container"
 BITCOINCORE_CONTAINER = "bitcoincore"
 COMMANDER_CONTAINER = "commander"
 

src/warnet/control.py

@@ -1,4 +1,5 @@
 import io
+import itertools
 import json
 import os
 import subprocess
@@ -24,7 +25,6 @@
     COMMANDER_CHART,
     COMMANDER_CONTAINER,
     COMMANDER_MISSION,
-    PLUGIN_MISSION,
     TANK_MISSION,
 )
 from .k8s import (
@@ -42,6 +42,7 @@
     wait_for_pod,
     write_file_to_container,
 )
+from .plugins import get_plugin_missions, get_plugin_primary_containers
 from .process import run_command, stream_command
 
 console = Console()
@@ -385,7 +386,8 @@ def format_pods(pods: list[V1Pod]) -> list[str]:
             pod_list = []
             formatted_commanders = format_pods(get_mission(COMMANDER_MISSION))
             formatted_tanks = format_pods(get_mission(TANK_MISSION))
-            formatted_plugins = format_pods(get_mission(PLUGIN_MISSION))
+            plugin_pods = [get_mission(mission) for mission in get_plugin_missions()]
+            formatted_plugins = format_pods(list(itertools.chain.from_iterable(plugin_pods)))
             pod_list.extend(formatted_commanders)
             pod_list.extend(formatted_tanks)
             pod_list.extend(formatted_plugins)
@@ -414,6 +416,7 @@ def format_pods(pods: list[V1Pod]) -> list[str]:
     try:
         pod = get_pod(pod_name, namespace=namespace)
         eligible_container_names = [BITCOINCORE_CONTAINER, COMMANDER_CONTAINER]
+        eligible_container_names.extend(get_plugin_primary_containers())
         available_container_names = [container.name for container in pod.spec.containers]
         container_name = next(
             (

src/warnet/plugins.py

@@ -14,6 +14,8 @@
 from inquirer.themes import GreenPassion
 
 from warnet.constants import (
+    CONTAINER_TAG,
+    MISSION_TAG,
     PLUGIN_YAML,
     PLUGINS_LABEL,
     WARNET_USER_DIR_ENV_VAR,
@@ -196,3 +198,11 @@ def get_plugins_with_status(plugin_dir: Optional[Path] = None) -> list[tuple[Pat
     ]
     plugins = [plugin_dir for plugin_dir in candidates if any(plugin_dir.glob("plugin.yaml"))]
     return [(plugin, check_if_plugin_enabled(plugin)) for plugin in plugins]
+
+
+def get_plugin_missions() -> list[str]:
+    return [getattr(module, MISSION_TAG.upper(), None) for module in imported_modules.values()]
+
+
+def get_plugin_primary_containers() -> list[str]:
+    return [getattr(module, CONTAINER_TAG.upper(), None) for module in imported_modules.values()]

src/warnet/status.py

@@ -1,3 +1,4 @@
+import itertools
 import sys
 
 import click
@@ -8,8 +9,10 @@
 from rich.text import Text
 from urllib3.exceptions import MaxRetryError
 
+from .constants import COMMANDER_MISSION, TANK_MISSION
 from .k8s import get_mission
 from .network import _connected
+from .plugins import get_plugin_missions
 
 
 @click.command()
@@ -20,6 +23,7 @@ def status():
     try:
         tanks = _get_tank_status()
         scenarios = _get_deployed_scenarios()
+        plugins = _get_plugin_status()
     except ConfigException as e:
         print(e)
         print(
@@ -65,6 +69,20 @@ def status():
     else:
         table.add_row("Scenario", "No active scenarios", "")
 
+    # Add a separator if there are plugins
+    if plugins:
+        table.add_row("", "", "")
+
+    # Add plugins to the table
+    active_plugins = 0
+    if plugins:
+        for plugin in plugins:
+            table.add_row("Plugin", plugin["name"], plugin["status"], plugin["namespace"])
+            if plugin["status"] == "running" or plugin["status"] == "pending":
+                active_plugins += 1
+    else:
+        table.add_row("Plugin", "No active plugins", "")
+
     # Create a panel to wrap the table
     panel = Panel(
         table,
@@ -81,12 +99,13 @@ def status():
     summary = Text()
     summary.append(f"\nTotal Tanks: {len(tanks)}", style="bold cyan")
     summary.append(f" | Active Scenarios: {active}", style="bold green")
+    summary.append(f" | Active Plugins: {active_plugins}", style="bold green")
     console.print(summary)
     _connected(end="\r")
 
 
 def _get_tank_status():
-    tanks = get_mission("tank")
+    tanks = get_mission(TANK_MISSION)
     return [
         {
             "name": tank.metadata.name,
@@ -97,8 +116,20 @@ def _get_tank_status():
     ]
 
 
+def _get_plugin_status():
+    plugin_pods = [get_mission(mission) for mission in get_plugin_missions()]
+    return [
+        {
+            "name": pod.metadata.name,
+            "status": pod.status.phase.lower(),
+            "namespace": pod.metadata.namespace,
+        }
+        for pod in list(itertools.chain.from_iterable(plugin_pods))  # flatten
+    ]
+
+
 def _get_deployed_scenarios():
-    commanders = get_mission("commander")
+    commanders = get_mission(COMMANDER_MISSION)
     return [
         {
             "name": c.metadata.name,