From cca655fdcc768ff5897aea15fb83511c5ae658e8 Mon Sep 17 00:00:00 2001
From: Taylor <taylor.walton@socfortress.co>
Date: Thu, 13 Jul 2023 21:22:07 -0500
Subject: [PATCH] get influxdb alerts from influxdb_alerts

---
 backend/app/routes/influxdb.py          | 12 +++++
 backend/app/services/InfluxDB/alerts.py | 66 ++++++++++++++++++++++++-
 2 files changed, 76 insertions(+), 2 deletions(-)

diff --git a/backend/app/routes/influxdb.py b/backend/app/routes/influxdb.py
index 993eb57c..672d3546 100644
--- a/backend/app/routes/influxdb.py
+++ b/backend/app/routes/influxdb.py
@@ -43,6 +43,18 @@ def get_check_query(check_id: str) -> jsonify:
     check_query = service.collect_check_query(check_id)
     return jsonify(check_query)
 
+@bp.route("/influxdb/alerts", methods=["GET"])
+def get_alerts() -> jsonify:
+    """
+    Endpoint to list all alerts from the `influxdb_alerts` table.
+
+    Returns:
+        jsonify: A JSON response containing the list of all alerts from InfluxDB.
+    """
+    logger.info("Received request to get all InfluxDB alerts")
+    service = InfluxDBAlertsService.from_connector_details("InfluxDB")
+    alerts = service.collect_alerts()
+    return jsonify(alerts)
 
 @bp.route("/influxdb/alert", methods=["POST"])
 def put_alert() -> jsonify:
diff --git a/backend/app/services/InfluxDB/alerts.py b/backend/app/services/InfluxDB/alerts.py
index 64b99e07..fefe4c8f 100644
--- a/backend/app/services/InfluxDB/alerts.py
+++ b/backend/app/services/InfluxDB/alerts.py
@@ -1,4 +1,4 @@
-from typing import Dict
+from typing import Dict, Union, List
 
 import requests
 from loguru import logger
@@ -107,7 +107,7 @@ def from_connector_details(cls, connector_name: str) -> "InfluxDBAlertsService":
 
     def validate_payload(self, data: Dict[str, object]) -> str:
         """
-        Validates the payload received from the Sublime alert webhook.
+        Validates the payload received from the influxdb alert webhook.
 
         Args:
             data (Dict[str, object]): The data received from the webhook.
@@ -137,3 +137,65 @@ def store_alerts(self, check_name: str, message: str) -> None:
         influxbd_alert = InfluxDBAlerts(check_name=check_name, message=message)
         db.session.add(influxbd_alert)
         db.session.commit()
+
+    def collect_alerts(self) -> Dict[str, Union[bool, str, List[Dict[str, str]]]]:
+        """
+        Collects alerts from `influxdb_alerts` table in the database.
+
+        Returns:
+            Dict[str, Union[bool, str, List[Dict[str, str]]]]: A dictionary containing the success status,
+                a message, and potentially the message details.
+        """
+        if not self._are_influxdb_details_collected():
+            return {
+                "message": "Failed to collect influxdb details",
+                "success": False,
+            }
+
+        alerts = self._collect_alerts_from_db()
+        if alerts["success"] is False:
+            return alerts
+
+        return {
+            "message": "Successfully collected alerts",
+            "success": True,
+            "alerts": alerts["alerts"],
+        }
+
+    def _are_influxdb_details_collected(self) -> bool:
+        """
+        Checks whether the details for the influxdb connector were successfully collected.
+
+        Returns:
+            bool: True if all details were collected, False otherwise.
+        """
+        return all([self.connector_url, self.connector_api_key])
+
+    def _collect_alerts_from_db(self) -> Dict[str, Union[bool, str, List[Dict[str, str]]]]:
+        """
+        Collects alerts from `influxdb_alerts` table in the database.
+
+        Returns:
+            Dict[str, Union[bool, str, List[Dict[str, str]]]]: A dictionary containing the success status,
+                a message, and potentially the message details.
+        """
+        try:
+            alerts = InfluxDBAlerts.query.all()
+        except Exception as e:
+            logger.error(f"Failed to collect alerts from database. Error: {e}")
+            return {
+                "message": "Failed to collect alerts from database",
+                "success": False,
+            }
+
+        return {
+            "message": "Successfully collected alerts from database",
+            "success": True,
+            "alerts": [
+                {
+                    "check_name": alert.check_name,
+                    "message": alert.message,
+                }
+                for alert in alerts
+            ],
+        }