feat: add /health for agent and simunet.

Author: vincentbergman

.github/workflows/release-agent.yml

@@ -64,13 +64,13 @@ jobs:
             **Docker:**
             ```bash
             # Pull the image
-            docker pull ghcr.io/${{ github.repository }}/netdriver-agent:${{ steps.version.outputs.version }}
+            docker pull ghcr.io/opensecflow/netdriver-agent:${{ steps.version.outputs.version }}
 
             # Run the agent
             docker run -d -p 8000:8000 \
               -v $(pwd)/config:/app/config \
               -v $(pwd)/logs:/app/logs \
-              ghcr.io/${{ github.repository }}/netdriver-agent:${{ steps.version.outputs.version }}
+              ghcr.io/opensecflow/netdriver-agent:${{ steps.version.outputs.version }}
             ```
 
             **Available tags:** `${{ steps.version.outputs.version }}`, `latest`

.github/workflows/release-simunet.yml

@@ -64,13 +64,13 @@ jobs:
             **Docker:**
             ```bash
             # Pull the image
-            docker pull ghcr.io/${{ github.repository }}/netdriver-simunet:${{ steps.version.outputs.version }}
+            docker pull ghcr.io/opensecflow/netdriver-simunet:${{ steps.version.outputs.version }}
 
             # Run with host network mode
             docker run -d --network host \
               -v $(pwd)/config:/app/config \
               -v $(pwd)/logs:/app/logs \
-              ghcr.io/${{ github.repository }}/netdriver-simunet:${{ steps.version.outputs.version }}
+              ghcr.io/opensecflow/netdriver-simunet:${{ steps.version.outputs.version }}
             ```
 
             **Available tags:** `${{ steps.version.outputs.version }}`, `latest`

bases/netdriver/agent/main.py

@@ -70,6 +70,15 @@ async def root() -> dict:
     }
 
 
+@app.get("/health")
+async def health() -> dict:
+    """ health check endpoint for docker """
+    return {
+        "status": "healthy",
+        "service": "netdriver-agent"
+    }
+
+
 def start():
     """Start the agent server with optional configuration file parameter."""
     parser = argparse.ArgumentParser(description="NetDriver Agent Server")

bases/netdriver/simunet/main.py

@@ -55,6 +55,23 @@ async def on_shutdown() -> None:
 app.add_event_handler("shutdown", on_shutdown)
 
 
+@app.get("/")
+async def root() -> dict:
+    """ root endpoint """
+    return {
+        "message": "Welcome to the NetDriver SimuNet",
+    }
+
+
+@app.get("/health")
+async def health() -> dict:
+    """ health check endpoint for docker """
+    return {
+        "status": "healthy",
+        "service": "netdriver-simunet"
+    }
+
+
 def start():
     """Start the simunet server with optional configuration file parameter."""
     parser = argparse.ArgumentParser(description="NetDriver SimuNet Server")

projects/simunet/Dockerfile

@@ -52,5 +52,9 @@ ENV PYTHONUNBUFFERED=1
 ENV FATSTAPI_ENV=production
 ENV NETDRIVER_CONFIG_PATH=/app/config/simunet/simunet.yml
 
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=40s --retries=3 \
+    CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:8001/health')" || exit 1
+
 # Run simunet
 CMD ["uvicorn", "netdriver.simunet.main:app", "--host", "0.0.0.0", "--port", "8001"]