Merge pull request #14 from S-Bryce/patch-1

Add missing "connection" JSON property

main.py

@@ -89,24 +89,37 @@ def collect():
         yield metric
         metric = Metric("wazuh_agent_status", "Total Wazuh agents by status", "summary")
 
+        # Wazuh >= v4.4
+        if "connection" in agents["agent_status"]:
+            agents_path = agents["agent_status"]["connection"]
+        # Legacy Wazuh support (< v4.4)
+        else:
+            agents_path = agents["agent_status"]
+
         metric.add_sample(
-            "wazuh_active_agents", value=agents["agent_status"]["active"], labels={}
+            "wazuh_active_agents",
+            value=agents_path["active"],                
+            labels={}
         )
         metric.add_sample(
-            "wazuh_disconnected_agents",
-            value=agents["agent_status"]["disconnected"],
+           "wazuh_disconnected_agents",
+            value=agents_path["disconnected"],
             labels={},
         )
         metric.add_sample(
             "wazuh_never_connected_agents",
-            value=agents["agent_status"]["never_connected"],
+            value=agents_path["never_connected"],
             labels={},
         )
         metric.add_sample(
-            "wazuh_pending_agents", value=agents["agent_status"]["pending"], labels={}
+            "wazuh_pending_agents",
+            value=agents_path["pending"],
+            labels={}
         )
         metric.add_sample(
-            "wazuh_total_agents", value=agents["agent_status"]["total"], labels={}
+            "wazuh_total_agents",
+            value=agents_path["total"],
+            labels={}
         )
         yield metric
         metric = InfoMetricFamily("wazuh_agent_version", "Wazuh agent versions")