augh

.devcontainer/devcontainer.json

@@ -5,14 +5,12 @@
 	"dockerComposeFile": "docker-compose.yml",
 	"service": "app",
 	"workspaceFolder": "/workspaces/${localWorkspaceFolderBasename}",
-
 	// Use this environment variable if you need to bind mount your local source code into a new container.
 	"remoteEnv": {
 		"LOCAL_WORKSPACE_FOLDER": "${localWorkspaceFolder}"
 	},
-
 	"features": {
-		"ghcr.io/devcontainers/features/docker-outside-of-docker:1": {
+		"ghcr.io/devcontainers/features/docker-in-docker:2": {
 			"installDockerBuildx": true,
 			"version": "latest",
 			"dockerDashComposeVersion": "none"
@@ -21,12 +19,25 @@
 			"bashCompletion": true
 		}
 	},
+	"customizations": {
+		"vscode": {
+			"extensions": [
+				"github.vscode-github-actions",
+				"charliermarsh.ruff"
+			],
+			"settings": {
+				"python.defaultInterpreterPath": ".venv/bin/python",
+				"python.terminal.activateEnvInCurrentTerminal": true,
+				"ruff.lint.preview": true,
+				"ruff.format.preview": true,
+				"git.autofetch": true
+			}
+		}
+	},
 	// Use 'forwardPorts' to make a list of ports inside the container available locally.
 	// "forwardPorts": [],
-
 	// Use 'postCreateCommand' to run commands after the container is created.
-	// "postCreateCommand": "docker --version",
-
+	"postCreateCommand": "rye sync",
 	// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
 	"remoteUser": "root" // Non-root causes issues with Rye on mounted workspaces
-}
+}

.gitignore

@@ -8,3 +8,4 @@ wheels/
 
 # venv
 .venv
+.ruff_cache

src/kumaprobe/__init__.py

@@ -1,8 +1,9 @@
-import docker
 import asyncio
+import logging
+
+import docker
 import aiohttp
 import uvloop
-import logging
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
@@ -36,11 +37,16 @@ async def check_health_and_send(container_or_service):
         logger.info(f"Checking health status for {container_or_service.name}")
 
         # Ensure "State" and "Health" exist in the attributes
-        if "State" in container_or_service.attrs and "Health" in container_or_service.attrs["State"]:
+        if (
+            "State" in container_or_service.attrs
+            and "Health" in container_or_service.attrs["State"]
+        ):
             health_status = container_or_service.attrs["State"]["Health"]["Status"]
             endpoint = labels["kumaprobe.endpoint"]
 
-            logger.info(f"Health status of {container_or_service.name}: {health_status}")
+            logger.info(
+                f"Health status of {container_or_service.name}: {health_status}"
+            )
 
             if health_status == "healthy":
                 await send_health_status(endpoint, health_status)
@@ -49,9 +55,13 @@ async def check_health_and_send(container_or_service):
                     f"{container_or_service.name} is not healthy: {health_status}"
                 )
         else:
-            logger.warning(f"No health information found for {container_or_service.name}")
+            logger.warning(
+                f"No health information found for {container_or_service.name}"
+            )
     else:
-        logger.debug(f"No 'kumaprobe.endpoint' label found for {container_or_service.name}")
+        logger.debug(
+            f"No 'kumaprobe.endpoint' label found for {container_or_service.name}"
+        )
 
 
 async def check_active_containers():
@@ -100,16 +110,23 @@ async def async_main():
         if any(label.startswith("kumaprobe") for label in labels):
             if container.id not in detected_services:
                 detected_containers.add(container.id)
-                logger.info(f"Detected container with kumaprobe label: {container.name}")
+                logger.info(
+                    f"Detected container with kumaprobe label: {container.name}"
+                )
 
     while True:
         current_containers = await check_active_containers()
         for container in current_containers:
             labels = container.attrs.get("Config", {}).get("Labels", {})
             if any(label.startswith("kumaprobe") for label in labels):
-                if container.id not in detected_containers and container.id not in detected_services:
+                if (
+                    container.id not in detected_containers
+                    and container.id not in detected_services
+                ):
                     detected_containers.add(container.id)
-                    logger.info(f"New container detected with kumaprobe label: {container.name}")
+                    logger.info(
+                        f"New container detected with kumaprobe label: {container.name}"
+                    )
 
         # Check the health of each container and send the status
         for container in current_containers:

testing/docker-compose.yaml

@@ -0,0 +1,21 @@
+name: kumaprobe-testing
+
+services: 
+  kumaprobe:
+    restart: no  
+    build: ..
+    depends_on:
+      - sample-api
+
+  sample-api: 
+    build: ./sample-api/
+    ports: 
+      - target: 5000
+        published: 8080
+        protocol: tcp
+        mode: host
+
+networks:
+  default:
+    # internal: true
+    attachable: true

testing/sample-api/Dockerfile

@@ -0,0 +1,6 @@
+FROM python:3.11-slim
+WORKDIR /app
+COPY app.py . 
+RUN pip install Flask
+EXPOSE 5000/tcp 
+CMD ["python", "app.py"]

testing/sample-api/app.py

@@ -0,0 +1,13 @@
+#!/usr/bin/env bash
+from flask import Flask
+import sys
+
+app = Flask(__name__)
+
+@app.route("/", methods=['GET'])
+def respond():
+    print("Recieved a request! Exiting.")
+    sys.exit(0)
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0")