kuma-strapper/backend/services/monitors.py

from dataclasses import dataclass
from typing import Optional

from services.kuma_client import get_kuma_client, Monitor
from services.claude_agent import MonitorSuggestion


@dataclass
class DefaultMonitorProfile:
    """A default monitoring profile that doesn't require approval."""

    name: str
    description: str
    monitors: list[Monitor]


def create_host_health_monitors(hostname: str, ssh_port: int = 22) -> list[Monitor]:
    """Create default host health monitors."""
    return [
        Monitor(
            type="ping",
            name=f"{hostname} - Ping",
            hostname=hostname,
            interval=60,
        ),
        Monitor(
            type="tcp",
            name=f"{hostname} - SSH",
            hostname=hostname,
            port=ssh_port,
            interval=120,
        ),
    ]


def create_web_server_monitors(hostname: str, port: int = 80, https: bool = False) -> list[Monitor]:
    """Create monitors for a detected web server."""
    protocol = "https" if https else "http"
    return [
        Monitor(
            type="http",
            name=f"{hostname} - Web ({port})",
            url=f"{protocol}://{hostname}:{port}/",
            interval=60,
        ),
    ]


def create_docker_container_monitors(hostname: str, containers: list[dict]) -> list[Monitor]:
    """Create monitors for detected Docker containers."""
    monitors = []
    for container in containers:
        name = container.get("name", container.get("id", "unknown"))
        monitors.append(
            Monitor(
                type="docker",
                name=f"{hostname} - Container: {name}",
                docker_container=name,
                docker_host=hostname,
                interval=60,
            )
        )
    return monitors


class MonitorService:
    """Service for managing monitors in Uptime Kuma."""

    def __init__(self):
        self.created_monitors: list[dict] = []

    def create_default_monitors(
        self,
        hostname: str,
        ssh_port: int = 22,
        has_docker: bool = False,
        containers: Optional[list[dict]] = None,
        web_ports: Optional[list[int]] = None,
    ) -> list[dict]:
        """
        Create default monitors for a host.
        These are built-in and never require approval.
        """
        kuma = get_kuma_client()
        created = []

        # Host health monitors
        health_monitors = create_host_health_monitors(hostname, ssh_port)
        for monitor in health_monitors:
            try:
                result = kuma.create_monitor(monitor)
                created.append({
                    "monitor": monitor.name,
                    "type": monitor.type,
                    "status": "created",
                    "result": result,
                })
            except Exception as e:
                created.append({
                    "monitor": monitor.name,
                    "type": monitor.type,
                    "status": "failed",
                    "error": str(e),
                })

        # Web server monitors
        if web_ports:
            for port in web_ports:
                https = port == 443 or port == 8443
                web_monitors = create_web_server_monitors(hostname, port, https)
                for monitor in web_monitors:
                    try:
                        result = kuma.create_monitor(monitor)
                        created.append({
                            "monitor": monitor.name,
                            "type": monitor.type,
                            "status": "created",
                            "result": result,
                        })
                    except Exception as e:
                        created.append({
                            "monitor": monitor.name,
                            "type": monitor.type,
                            "status": "failed",
                            "error": str(e),
                        })

        # Docker container monitors
        if has_docker and containers:
            docker_monitors = create_docker_container_monitors(hostname, containers)
            for monitor in docker_monitors:
                try:
                    result = kuma.create_monitor(monitor)
                    created.append({
                        "monitor": monitor.name,
                        "type": monitor.type,
                        "status": "created",
                        "result": result,
                    })
                except Exception as e:
                    created.append({
                        "monitor": monitor.name,
                        "type": monitor.type,
                        "status": "failed",
                        "error": str(e),
                    })

        self.created_monitors.extend(created)
        return created

    def create_from_suggestion(self, suggestion: MonitorSuggestion, hostname: str) -> dict:
        """
        Create a monitor from a Claude suggestion.
        In production mode, this executes automatically.
        In dev mode, this should only be called after approval.
        """
        kuma = get_kuma_client()

        # Build monitor from suggestion
        monitor = Monitor(
            type=suggestion.type,
            name=suggestion.name,
            interval=suggestion.interval,
        )

        # Set type-specific fields
        if suggestion.type == "http" or suggestion.type == "keyword":
            monitor.url = suggestion.target
            if suggestion.keyword:
                monitor.keyword = suggestion.keyword
        elif suggestion.type == "tcp":
            monitor.hostname = suggestion.target
            monitor.port = suggestion.port
        elif suggestion.type == "ping":
            monitor.hostname = suggestion.target
        elif suggestion.type == "docker":
            monitor.docker_container = suggestion.target
            monitor.docker_host = hostname

        try:
            result = kuma.create_monitor(monitor)
            return {
                "monitor": monitor.name,
                "type": monitor.type,
                "status": "created",
                "result": result,
                "reason": suggestion.reason,
            }
        except Exception as e:
            return {
                "monitor": monitor.name,
                "type": monitor.type,
                "status": "failed",
                "error": str(e),
                "reason": suggestion.reason,
            }

    def get_existing_monitors(self) -> list[dict]:
        """Get all existing monitors from Uptime Kuma."""
        kuma = get_kuma_client()
        return kuma.get_monitors()


def parse_web_ports_from_scan(open_ports: str) -> list[int]:
    """Extract web server ports from port scan output."""
    common_web_ports = [80, 443, 8080, 8443, 3000, 5000, 8000]
    found_ports = []

    for port in common_web_ports:
        if f":{port}" in open_ports or f" {port} " in open_ports:
            found_ports.append(port)

    return found_ports


def parse_docker_containers_from_scan(docker_output: str) -> list[dict]:
    """Parse Docker container info from scan output."""
    containers = []

    if "Docker not available" in docker_output or not docker_output.strip():
        return containers

    for line in docker_output.strip().split("\n"):
        if not line.strip():
            continue

        parts = line.split("\t")
        if len(parts) >= 2:
            containers.append({
                "id": parts[0] if len(parts) > 0 else "",
                "name": parts[1] if len(parts) > 1 else "",
                "image": parts[2] if len(parts) > 2 else "",
                "status": parts[3] if len(parts) > 3 else "",
                "ports": parts[4] if len(parts) > 4 else "",
            })

    return containers


# Global monitor service instance
_monitor_service: Optional[MonitorService] = None


def get_monitor_service() -> MonitorService:
    """Get the global monitor service instance."""
    global _monitor_service
    if _monitor_service is None:
        _monitor_service = MonitorService()
    return _monitor_service