From 7a2e6d5db19c9cad0c98dbad8db0a03376db826c Mon Sep 17 00:00:00 2001
From: Nikhil Sinha <nikhil.sinha@parseable.com>
Date: Mon, 17 Mar 2025 04:13:58 -0400
Subject: [PATCH] fix: prism home api

1. add critical severity alert count to alerts info
2. date level stats in sorted order
---
 src/alerts/mod.rs     | 5 ++++-
 src/prism/home/mod.rs | 3 ++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/alerts/mod.rs b/src/alerts/mod.rs
index db22754f5..82581ffff 100644
--- a/src/alerts/mod.rs
+++ b/src/alerts/mod.rs
@@ -886,6 +886,7 @@ pub struct AlertsInfo {
     low: u64,
     medium: u64,
     high: u64,
+    critical: u64,
 }
 
 // TODO: add RBAC
@@ -898,6 +899,7 @@ pub async fn get_alerts_info() -> Result<AlertsInfo, AlertError> {
     let mut low = 0;
     let mut medium = 0;
     let mut high = 0;
+    let mut critical = 0;
 
     for (_, alert) in alerts.iter() {
         total += 1;
@@ -911,7 +913,7 @@ pub async fn get_alerts_info() -> Result<AlertsInfo, AlertError> {
             Severity::Low => low += 1,
             Severity::Medium => medium += 1,
             Severity::High => high += 1,
-            _ => {}
+            Severity::Critical => critical += 1,
         }
     }
 
@@ -923,5 +925,6 @@ pub async fn get_alerts_info() -> Result<AlertsInfo, AlertError> {
         low,
         medium,
         high,
+        critical,
     })
 }
diff --git a/src/prism/home/mod.rs b/src/prism/home/mod.rs
index 48b062e31..95e03b8ce 100644
--- a/src/prism/home/mod.rs
+++ b/src/prism/home/mod.rs
@@ -149,7 +149,7 @@ pub async fn generate_home_response(key: &SessionKey) -> Result<HomeResponse, Pr
     let alerts_info = get_alerts_info().await?;
 
     // generate dates for date-wise stats
-    let dates = (0..7)
+    let mut dates = (0..7)
         .map(|i| {
             Utc::now()
                 .checked_sub_signed(chrono::Duration::days(i))
@@ -158,6 +158,7 @@ pub async fn generate_home_response(key: &SessionKey) -> Result<HomeResponse, Pr
         })
         .map(|date| date.format("%Y-%m-%d").to_string())
         .collect_vec();
+    dates.reverse();
 
     let mut stream_details = Vec::new();