Merge pull request #38 from nsleigh/feat/get-drive-data-pr

feat: add get_drive_data HA action for AI-powered drive health analysis

Author: DAB-LABS

custom_components/smart_sniffer/__init__.py

@@ -9,12 +9,17 @@
 from __future__ import annotations
 
 import logging
+from typing import Any
+
+import voluptuous as vol
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import Platform
-from homeassistant.core import HomeAssistant
+from homeassistant.core import HomeAssistant, ServiceCall, SupportsResponse
+from homeassistant.exceptions import ServiceValidationError
 
-from .const import DOMAIN
+from .attention import evaluate_attention
+from .const import DOMAIN, FILESYSTEMS_KEY, SERVICE_GET_DRIVE_DATA
 from .coordinator import AgentHealthCoordinator, SmartSnifferCoordinator
 
 _LOGGER = logging.getLogger(__name__)
@@ -37,6 +42,67 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
 
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
 
+    # Register the get_drive_data service once per domain.
+    if not hass.services.has_service(DOMAIN, SERVICE_GET_DRIVE_DATA):
+        async def handle_get_drive_data(call: ServiceCall) -> dict[str, Any]:
+            entry_id: str = call.data["config_entry_id"]
+            domain_data = hass.data.get(DOMAIN, {})
+            if entry_id not in domain_data:
+                raise ServiceValidationError(
+                    f"No SMART Sniffer entry found with ID: {entry_id}"
+                )
+            coord = domain_data[entry_id]["coordinator"]
+            health = domain_data[entry_id]["health_coordinator"]
+
+            drives: dict[str, Any] = {}
+            for drive_id, drive_data in coord.data.items():
+                if drive_id.startswith("_"):
+                    continue
+                state, severity, reasons = evaluate_attention(drive_data)
+                drives[drive_id] = {
+                    "model": drive_data.get("model"),
+                    "serial": drive_data.get("serial"),
+                    "protocol": drive_data.get("protocol"),
+                    "device_path": drive_data.get("device_path"),
+                    "attention": {
+                        "state": state,
+                        "severity": severity,
+                        "reasons": reasons,
+                    },
+                    "smart_data": drive_data.get("smart_data"),
+                }
+
+            raw_filesystems: list[dict[str, Any]] = coord.data.get(FILESYSTEMS_KEY, [])
+            filesystems = [
+                {
+                    "mountpoint": fs.get("mountpoint"),
+                    "total_bytes": fs.get("total_bytes"),
+                    "used_bytes": fs.get("used_bytes"),
+                    "use_percent": fs.get("use_percent"),
+                }
+                for fs in raw_filesystems
+            ]
+
+            return {
+                "agent": {
+                    "name": health.config_entry.title,
+                    "host": health.host,
+                    "os": health.data.get("os"),
+                    "uptime": health.data.get("uptime_seconds"),
+                    "version": health.data.get("version"),
+                },
+                "drives": drives,
+                "filesystems": filesystems,
+            }
+
+        hass.services.async_register(
+            DOMAIN,
+            SERVICE_GET_DRIVE_DATA,
+            handle_get_drive_data,
+            schema=vol.Schema({vol.Required("config_entry_id"): str}),
+            supports_response=SupportsResponse.ONLY,
+        )
+
     # Reload the integration when options are changed via the UI.
     entry.async_on_unload(entry.add_update_listener(_async_update_listener))
 
@@ -48,6 +114,8 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
     if unload_ok:
         hass.data[DOMAIN].pop(entry.entry_id)
+        if not hass.data[DOMAIN]:
+            hass.services.async_remove(DOMAIN, SERVICE_GET_DRIVE_DATA)
     return unload_ok
 
 

custom_components/smart_sniffer/const.py

@@ -21,3 +21,6 @@
 # Key used in coordinator data dict to store filesystem info.
 # Underscore prefix avoids collision with drive ID keys.
 FILESYSTEMS_KEY = "_filesystems"
+
+# Service name for the get_drive_data action.
+SERVICE_GET_DRIVE_DATA = "get_drive_data"

custom_components/smart_sniffer/services.yaml

@@ -0,0 +1,13 @@
+get_drive_data:
+  name: Get Drive Data
+  description: >
+    Returns the full cached SMART data for all drives monitored by a SMART Sniffer
+    agent. Useful for passing raw drive health data to an AI for analysis.
+  fields:
+    config_entry_id:
+      name: Config Entry
+      description: The SMART Sniffer agent to query.
+      required: true
+      selector:
+        config_entry:
+          integration: smart_sniffer

custom_components/smart_sniffer/strings.json

@@ -59,6 +59,18 @@
       "unknown": "An unexpected error occurred."
     }
   },
+  "services": {
+    "get_drive_data": {
+      "name": "Get Drive Data",
+      "description": "Returns the full cached SMART data for all drives monitored by a SMART Sniffer agent. Useful for passing raw drive health data to an AI for analysis.",
+      "fields": {
+        "config_entry_id": {
+          "name": "Config Entry",
+          "description": "The SMART Sniffer agent to query."
+        }
+      }
+    }
+  },
   "issues": {
     "agent_outdated": {
       "title": "SMART Sniffer: Update agent on {hostname} (v{current_version} → v{min_version})",

custom_components/smart_sniffer/translations/en.json

@@ -59,6 +59,18 @@
       "unknown": "An unexpected error occurred."
     }
   },
+  "services": {
+    "get_drive_data": {
+      "name": "Get Drive Data",
+      "description": "Returns the full cached SMART data for all drives monitored by a SMART Sniffer agent. Useful for passing raw drive health data to an AI for analysis.",
+      "fields": {
+        "config_entry_id": {
+          "name": "Config Entry",
+          "description": "The SMART Sniffer agent to query."
+        }
+      }
+    }
+  },
   "issues": {
     "agent_outdated": {
       "title": "SMART Sniffer: Update agent on {hostname} (v{current_version} → v{min_version})",