homelab-brain/homelab-ai-bot/tools/grafana.py

"""Grafana + InfluxDB Smart-Home Tool (CT 143, pve-mu-3)."""

import requests
from urllib.parse import quote

GRAFANA_URL = "http://100.66.78.56:3000"
GRAFANA_USER = "admin"
GRAFANA_PASS = "astral66"
INFLUX_URL = "http://100.66.78.56:8086"
INFLUX_DB = "iobroker"

SENSOR_MAP = {
    "aussen": ("mqtt.0.Holzvergaser_Sensoren_6.Aussenfühler.temperature", "Außen"),
    "wohnstube": ("mqtt.0.Wohnstube_Temperatur_1.Wohnstube.Wohnstube_Temperatur", "Wohnstube"),
    "kueche": ("mqtt.0.ESP_Kücheneu_2.Sensor_Küche.Küche", "Küche"),
    "garage": ("mqtt.0.ESP_Easy_3.Garage_neu.Garage", "Garage"),
    "puffer_oben": ("mqtt.0.ESP_Easy_Display_4.oben.temperature", "Puffer oben"),
    "puffer_unten": ("mqtt.0.ESP_Easy_Display_4.Unten.temperature", "Puffer unten"),
    "garten": ("mqtt.0.ESP_Easy_schwimmbad_neu.Lufttemperatur.Lufttemperatur_Garten", "Garten/Luft"),
    "schwimmbad": ("mqtt.0.ESP_Easy_schwimmbad_neu.Schwimmbad.Wassertemperatur_", "Schwimmbad"),
    "oelkessel_vl": ("mqtt.0.Oelkessel.Oelkessel_VL.Vorlauf", "Ölkessel Vorlauf"),
    "brunnen_vl": ("mqtt.0.Brunnenwasser.Oelkessel_VL.Vorlauf", "Brunnenwasser Vorlauf"),
    "holzvergaser_1": ("mqtt.0.Holzvergaser_Sensoren_6.temp1.temperature1", "Holzvergaser Sensor 1"),
    "holzvergaser_2": ("mqtt.0.Holzvergaser_Sensoren_6.temp2.temperature2", "Holzvergaser Sensor 2"),
}

ENERGIE_MEASUREMENTS = {
    "pv_leistung": ("mqtt.1.openWB.pv.W", "PV Leistung"),
    "pv_tagesertrag": ("mqtt.1.openWB.pv.DailyYieldKwh", "PV Tagesertrag"),
    "netz_bezug": ("mqtt.1.openWB.evu.W", "Netz Bezug/Einspeisung"),
    "hausverbrauch": ("mqtt.1.openWB.global.WHouseConsumption", "Hausverbrauch"),
    "batterie_soc": ("mqtt.1.openWB.housebattery.%Soc", "Batterie SoC"),
    "ueberschuss": ("mqtt.1.openWB.SmartHome.Status.uberschuss", "Überschuss"),
}

_ENERGIE_UNITS = {
    "PV Leistung": "W",
    "PV Tagesertrag": "kWh",
    "Netz Bezug/Einspeisung": "W",
    "Hausverbrauch": "W",
    "Batterie SoC": "%",
    "Überschuss": "W",
}

_HEIZUNG_KEYS = {"puffer", "vorlauf", "holz", "brunnen", "öl"}

TOOLS = [
    {
        "type": "function",
        "function": {
            "name": "get_grafana_status",
            "description": "Grafana Monitoring-Status: Health, Dashboards, Datasources, Alerts.",
            "parameters": {"type": "object", "properties": {}, "required": []},
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_temperaturen",
            "description": "Aktuelle Temperaturen aus dem Haus in Muldenstein: Räume, Außen, Puffer, Heizung. Nutze bei Fragen zu Temperatur, Heizung, Puffer, Raumklima, 'wie warm ist es'.",
            "parameters": {"type": "object", "properties": {}, "required": []},
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_energie",
            "description": "Aktuelle Energiedaten: PV-Leistung, Batterie, Hausverbrauch, Netz. Nutze bei Fragen zu Solar, PV, Strom, Energie, Batterie, Wallbox, Einspeisung.",
            "parameters": {"type": "object", "properties": {}, "required": []},
        },
    },
    {
        "type": "function",
        "function": {
            "name": "get_heizung",
            "description": "Heizungsstatus: Brenner, Puffertemperaturen, Vorlauf, Holzvergaser. Nutze bei Fragen zu Heizung, Brenner, Puffer, Holzvergaser, Ölkessel.",
            "parameters": {"type": "object", "properties": {}, "required": []},
        },
    },
]

SYSTEM_PROMPT_EXTRA = """Für Smart-Home-Daten (Temperaturen, Energie, Heizung) im Haus Muldenstein:
- get_temperaturen: alle Raumtemperaturen + Außen
- get_energie: PV, Batterie, Netz, Hausverbrauch
- get_heizung: Brenner, Puffer, Vorlauf, Holzvergaser
- get_grafana_status: Dashboard-Übersicht und Alerts
Grafana-URL: https://grafana.orbitalo.net/

WICHTIG: Die Tool-Ergebnisse sind bereits fertig formatiert.
Gib sie 1:1 weiter, NICHT umformulieren oder kürzen.
"""


def _api(path):
    try:
        r = requests.get(
            f"{GRAFANA_URL}{path}",
            auth=(GRAFANA_USER, GRAFANA_PASS),
            timeout=10,
        )
        if r.ok:
            return r.json()
    except Exception:
        pass
    return None


def _influx_query(q):
    try:
        r = requests.get(
            f"{INFLUX_URL}/query",
            params={"db": INFLUX_DB, "q": q},
            timeout=10,
        )
        if r.ok:
            return r.json()
    except Exception:
        pass
    return None


def _query_latest(measurements: dict) -> list:
    """Query last() for a dict of {key: (measurement, label)}."""
    parts = []
    for key, (meas, label) in measurements.items():
        parts.append(f'last(value) AS "{label}" FROM "{meas}"')
    q = "SELECT " + "; SELECT ".join(parts) + " WHERE time > now() - 2h"
    result = _influx_query(q)
    if not result:
        return []
    lines = []
    for stmt in result.get("results", []):
        for series in stmt.get("series", []):
            name = series["columns"][1] if len(series["columns"]) > 1 else series.get("name", "?")
            val = series["values"][0][1] if series.get("values") else None
            ts = series["values"][0][0] if series.get("values") else ""
            if val is not None:
                ts_short = ts[11:16] if len(ts) > 16 else ""
                lines.append((name, val, ts_short))
    return lines


def handle_get_grafana_status(**kw):
    health = _api("/api/health")
    if not health:
        return "Grafana nicht erreichbar (100.66.78.56:3000 / CT 143 pve-mu-3)"

    lines = [f"Grafana v{health.get('version', '?')} — DB: {health.get('database', '?')}"]
    lines.append("URL: https://grafana.orbitalo.net")
    lines.append("Server: CT 143 (Raspi-Broker) auf pve-mu-3 Muldenstein")

    datasources = _api("/api/datasources") or []
    lines.append(f"\nDatasources: {len(datasources)}")
    for ds in datasources:
        lines.append(f"  {ds.get('name', '?')} ({ds.get('type', '?')}) — DB: {ds.get('database', '-')}")

    dashboards = _api("/api/search?type=dash-db") or []
    lines.append(f"\nDashboards: {len(dashboards)}")
    for db in dashboards:
        url = f"https://grafana.orbitalo.net{db.get('url', '')}"
        lines.append(f"  {db.get('title', '?')} — {url}")

    alerts = _api("/api/alertmanager/grafana/api/v2/alerts") or []
    if alerts:
        firing = [a for a in alerts if a.get("status", {}).get("state") == "active"]
        lines.append(f"\nAlerts: {len(alerts)} gesamt, {len(firing)} aktiv")
        for a in firing[:5]:
            labels = a.get("labels", {})
            lines.append(f"  {labels.get('alertname', '?')} — {labels.get('severity', '?')}")
    else:
        lines.append("\nAlerts: keine aktiven")

    return "\n".join(lines)


def handle_get_temperaturen(**kw):
    data = _query_latest(SENSOR_MAP)
    if not data:
        return "Keine Temperaturdaten verfügbar (InfluxDB nicht erreichbar?)"

    raeume = []
    heizung = []
    for name, val, ts in data:
        entry = f"- {name}: {val:.1f} °C"
        if any(k in name.lower() for k in _HEIZUNG_KEYS):
            heizung.append(entry)
        else:
            raeume.append(entry)

    lines = ["Temperaturen Muldenstein:"]
    if raeume:
        lines.append("\nRäume & Außen:")
        lines.extend(raeume)
    if heizung:
        lines.append("\nHeizung & Puffer:")
        lines.extend(heizung)
    lines.append("\nGrafana: https://grafana.orbitalo.net/d/solar/raumtemperaturen")
    return "\n".join(lines)


def handle_get_energie(**kw):
    data = _query_latest(ENERGIE_MEASUREMENTS)
    if not data:
        return "Keine Energiedaten verfügbar (InfluxDB nicht erreichbar?)"

    lines = ["Energie Muldenstein:"]
    for name, val, ts in data:
        unit = _ENERGIE_UNITS.get(name, "")
        if unit in ("%", "kWh"):
            lines.append(f"- {name}: {val:.1f} {unit}")
        else:
            lines.append(f"- {name}: {val:.0f} {unit}")

    lines.append("\nGrafana: https://grafana.orbitalo.net/d/solar/energie-uebersicht")
    return "\n".join(lines)


def handle_get_heizung(**kw):
    heiz_sensors = {
        "puffer_oben": SENSOR_MAP["puffer_oben"],
        "puffer_unten": SENSOR_MAP["puffer_unten"],
        "oelkessel_vl": SENSOR_MAP["oelkessel_vl"],
        "brunnen_vl": SENSOR_MAP["brunnen_vl"],
        "holzvergaser_1": SENSOR_MAP["holzvergaser_1"],
        "holzvergaser_2": SENSOR_MAP["holzvergaser_2"],
        "aussen": SENSOR_MAP["aussen"],
    }
    data = _query_latest(heiz_sensors)

    brenner_q = 'SELECT last(value) FROM "brennerstatus" WHERE time > now() - 2h'
    laufzeit_q = 'SELECT last(value) FROM "brenner_heute" WHERE time > now() - 2h'
    starts_q = 'SELECT last(value) FROM "brennerstarts" WHERE time > now() - 24h'

    lines = ["Heizung Muldenstein:"]

    brenner = _influx_query(brenner_q)
    if brenner and brenner.get("results", [{}])[0].get("series"):
        val = brenner["results"][0]["series"][0]["values"][0][1]
        status = "🔥 AN" if val else "⏸ AUS"
        lines.append(f"\nBrenner: {status}")

    laufzeit = _influx_query(laufzeit_q)
    if laufzeit and laufzeit.get("results", [{}])[0].get("series"):
        val = laufzeit["results"][0]["series"][0]["values"][0][1]
        lines.append(f"Brenner heute: {val:.0f} min")

    starts = _influx_query(starts_q)
    if starts and starts.get("results", [{}])[0].get("series"):
        val = starts["results"][0]["series"][0]["values"][0][1]
        lines.append(f"Brennerstarts (24h): {val:.0f}")

    if data:
        lines.append("\nTemperaturen:")
        for name, val, ts in data:
            lines.append(f"- {name}: {val:.1f} °C")

    lines.append("\nGrafana: https://grafana.orbitalo.net/d/heizung/heizung-and-puffer")
    return "\n".join(lines)


HANDLERS = {
    "get_grafana_status": handle_get_grafana_status,
    "get_temperaturen": handle_get_temperaturen,
    "get_energie": handle_get_energie,
    "get_heizung": handle_get_heizung,
}