metalcheck/app/extras/pseudographic.py

from rich.console import Console
from rich.table import Table
from rich.progress import Progress
from database import fetch_all
from kubernetes import client, config
import requests

# Define constants
THINK_K8S_URL = "http://localhost:8000/think/k8s"

console = Console()

# Helper functions for conversions
def convert_cpu_to_cores(cpu):
    if "n" in cpu:
        return round(int(cpu.replace("n", "")) / 1e9, 4)
    elif "u" in cpu:
        return round(int(cpu.replace("u", "")) / 1e6, 4)
    elif "m" in cpu:
        return round(int(cpu.replace("m", "")) / 1000, 4)
    return float(cpu)

def convert_memory_to_mib(memory):
    if "Ki" in memory:
        return int(memory.replace("Ki", "")) / 1024
    elif "Mi" in memory:
        return int(memory.replace("Mi", ""))
    elif "Gi" in memory:
        return int(memory.replace("Gi", "")) * 1024
    return float(memory)

def convert_memory_to_gb(memory):
    if "Ki" in memory:
        return int(memory.replace("Ki", "")) / (1024 ** 2)
    elif "Mi" in memory:
        return int(memory.replace("Mi", "")) / 1024
    elif "Gi" in memory:
        return int(memory.replace("Gi", ""))
    return float(memory)

# Display tables
def display_metal_nodes():
    table = Table(title="🖥️ Metal Nodes", style="bold green")
    table.add_column("ID", justify="right", style="cyan")
    table.add_column("Name", style="magenta")
    table.add_column("Location", style="white")
    table.add_column("Vendor", style="green")
    table.add_column("CPU", justify="right", style="yellow")
    table.add_column("Memory (GB)", justify="right", style="cyan")
    table.add_column("Storage", style="magenta")

    nodes = fetch_all("metal_nodes")
    for node in nodes:
        table.add_row(
            f"{node[0]}",
            node[1],
            node[2],
            node[3],
            f"{node[4]}",
            node[5],
            node[6]
        )

    console.print(table)

def display_virtual_machines():
    table = Table(title="💻 Virtual Machines", style="bold blue")
    table.add_column("ID", justify="right", style="cyan")
    table.add_column("Name", style="magenta")
    table.add_column("Location", style="white")
    table.add_column("CPU", justify="right", style="yellow")
    table.add_column("Memory (GB)", justify="right", style="cyan")
    table.add_column("Storage", style="magenta")
    table.add_column("Type", style="green")

    vms = fetch_all("virtual_machines")
    for vm in vms:
        table.add_row(
            f"{vm[0]}",
            vm[1],
            vm[2],
            f"{vm[3]}",
            vm[4],
            vm[5],
            vm[6]
        )

    console.print(table)

def display_kubernetes_nodes():
    config.load_incluster_config()
    v1 = client.CoreV1Api()

    table = Table(title="📦 Kubernetes Nodes", style="bold yellow")
    table.add_column("Node Name", style="white")
    table.add_column("CPU", justify="right", style="yellow")
    table.add_column("Memory (MiB)", justify="right", style="cyan")
    table.add_column("Storage (GB)", justify="right", style="green")
    table.add_column("Instance Type", style="blue")
    table.add_column("Pods Allocatable", justify="right", style="magenta")

    nodes = v1.list_node()
    for node in nodes.items:
        ephemeral_storage = node.status.capacity.get("ephemeral-storage", "0")
        instance_type = node.metadata.labels.get("beta.kubernetes.io/instance-type", "N/A")

        table.add_row(
            node.metadata.name,
            node.status.capacity.get("cpu"),
            f"{round(convert_memory_to_mib(node.status.capacity.get('memory')), 2)}",
            f"{round(convert_memory_to_gb(ephemeral_storage), 2)}",
            instance_type,
            node.status.allocatable.get("pods")
        )

    console.print(table)

def display_namespace_usage():
    config.load_incluster_config()
    metrics_client = client.CustomObjectsApi()

    table = Table(title="📊 Namespace Resource Usage", style="bold magenta")
    table.add_column("Namespace", style="white")
    table.add_column("CPU (Cores)", justify="right", style="yellow")
    table.add_column("Memory (MiB)", justify="right", style="cyan")

    namespace_usage = {}
    pod_metrics = metrics_client.list_cluster_custom_object(
        group="metrics.k8s.io", version="v1beta1", plural="pods"
    )

    for pod in pod_metrics["items"]:
        namespace = pod["metadata"]["namespace"]
        if namespace not in namespace_usage:
            namespace_usage[namespace] = {"cpu": 0, "memory": 0}

        for container in pod["containers"]:
            cpu_usage = container["usage"]["cpu"]
            memory_usage = container["usage"]["memory"]

            namespace_usage[namespace]["cpu"] += convert_cpu_to_cores(cpu_usage)
            namespace_usage[namespace]["memory"] += convert_memory_to_mib(memory_usage)

    for namespace, usage in namespace_usage.items():
        table.add_row(
            namespace,
            f"{round(usage['cpu'], 4)}",
            f"{round(usage['memory'], 2)}"
        )

    console.print(table)

# Fetch and display AI summary
def fetch_ai_summary():
    with Progress() as progress:
        task = progress.add_task("[cyan]Thinking about Kubernetes...", total=100)

        try:
            for _ in range(10):  # Simulate progress
                progress.update(task, advance=10)
                import time; time.sleep(0.1)

            response = requests.get(THINK_K8S_URL)
            progress.update(task, completed=100)

            if response.status_code == 200:
                data = response.json()
                return data.get("summary", "No summary provided.")
            else:
                return f"Failed to fetch summary: {response.status_code} {response.text}"
        except requests.RequestException as e:
            return f"An error occurred while fetching the summary: {str(e)}"

def display_ai_summary():
    summary = fetch_ai_summary()
    console.print("\n[bold magenta]AI Summary of Kubernetes Cluster:[/bold magenta]")
    console.print(f"[green]{summary}[/green]\n")

if __name__ == "__main__":
    console.print("✨ [bold green]Welcome to the Metal Check Dashboard![/bold green] ✨\n")
    display_metal_nodes()
    display_virtual_machines()
    display_kubernetes_nodes()
    display_namespace_usage()
    display_ai_summary()