Merge pull request #396 from Bosun-Josh121/fix-387

Fix: Add CSV export for boost winners

Author: Marchand-Nicolas

src/endpoints/admin/quest_boost/get_boost_winners_csv.rs

@@ -0,0 +1,74 @@
+use crate::middleware::auth::auth_middleware;
+use crate::models::{AppState, BoostTable};
+use crate::utils::get_error;
+use axum::{
+    extract::{Extension, Query, State},
+    http::{header, StatusCode},
+    response::IntoResponse,
+};
+use axum_auto_routes::route;
+use mongodb::bson::doc;
+use serde::Deserialize;
+use std::sync::Arc;
+
+#[derive(Deserialize)]
+pub struct GetBoostWinnersCsvParams {
+    boost_id: i64,
+}
+
+#[route(get, "/admin/boosts/get_boost_winners_csv", auth_middleware)]
+pub async fn get_boost_winners_csv_handler(
+    State(state): State<Arc<AppState>>,
+    Extension(_sub): Extension<String>,
+    Query(params): Query<GetBoostWinnersCsvParams>,
+) -> impl IntoResponse {
+    let collection = state.db.collection::<BoostTable>("boosts");
+
+    let filter = doc! { "id": params.boost_id };
+
+    match collection.find_one(filter, None).await {
+        Ok(Some(boost_doc)) => {
+            // Convert winners data to CSV format
+            match create_csv_response(&boost_doc.winner, params.boost_id) {
+                Ok(csv_content) => {
+                    let headers = [
+                        (header::CONTENT_TYPE, "text/csv"),
+                        (
+                            header::CONTENT_DISPOSITION,
+                            &format!("attachment; filename=\"boost_{}_winners.csv\"", params.boost_id),
+                        ),
+                    ];
+
+                    (StatusCode::OK, headers, csv_content).into_response()
+                }
+                Err(e) => get_error(format!("Error creating CSV: {}", e)),
+            }
+        }
+        Ok(None) => get_error(format!("Boost with id {} not found", params.boost_id)),
+        Err(e) => get_error(format!("Error fetching boost winners: {}", e)),
+    }
+}
+
+pub(crate) fn create_csv_response(
+    winners: &Option<Vec<String>>,
+    boost_id: i64,
+) -> Result<String, Box<dyn std::error::Error>> {
+    let mut csv_content = String::new();
+    
+    // Add CSV header
+    csv_content.push_str("boost_id,winner_address,position\n");
+    
+    // Handle the Option<Vec<String>> format
+    match winners {
+        Some(winner_vec) => {
+            for (index, winner_address) in winner_vec.iter().enumerate() {
+                csv_content.push_str(&format!("{},{},{}\n", boost_id, winner_address, index + 1));
+            }
+        }
+        None => {
+            // No winners found, just return header
+        }
+    }
+    
+    Ok(csv_content)
+}

src/endpoints/admin/quest_boost/mod.rs

@@ -1,3 +1,4 @@
 pub mod create_boost;
 pub mod get_boost_winners;
+pub mod get_boost_winners_csv;
 pub mod update_boost;

src/tests/endpoints.rs

@@ -69,6 +69,60 @@ pub mod tests {
         let mut update_doc = doc! {};
         update_doc.insert("banner", bson_banner.clone());
         assert_eq!(update_doc.get("banner").unwrap(), &bson_banner);
-    }    
+    } 
+
+
+
+    // Unit tests for the CSV creation logic
+    #[tokio::test]
+    pub async fn test_csv_creation_logic() {
+        use crate::endpoints::admin::quest_boost::get_boost_winners_csv::create_csv_response;
+        
+        // Test with winners
+        let winners = Some(vec![
+            "0x1234567890abcdef".to_string(),
+            "0xfedcba0987654321".to_string(),
+        ]);
+        let boost_id = 123;
+        
+        let result = create_csv_response(&winners, boost_id).unwrap();
+        
+        let expected = "boost_id,winner_address,position\n\
+                       123,0x1234567890abcdef,1\n\
+                       123,0xfedcba0987654321,2\n";
+        
+        assert_eq!(result, expected);
+    }
+
+    #[tokio::test]
+    pub async fn test_csv_creation_no_winners() {
+        use crate::endpoints::admin::quest_boost::get_boost_winners_csv::create_csv_response;
+        
+        // Test with no winners
+        let winners = None;
+        let boost_id = 456;
+        
+        let result = create_csv_response(&winners, boost_id).unwrap();
+        
+        let expected = "boost_id,winner_address,position\n";
+        
+        assert_eq!(result, expected);
+    }
+
+    #[tokio::test]
+    pub async fn test_csv_creation_empty_winners() {
+        use crate::endpoints::admin::quest_boost::get_boost_winners_csv::create_csv_response;
+        
+        // Test with empty winners vector
+        let winners = Some(vec![]);
+        let boost_id = 789;
+        
+        let result = create_csv_response(&winners, boost_id).unwrap();
+        
+        let expected = "boost_id,winner_address,position\n";
+        
+        assert_eq!(result, expected);
+    }   
+   
 
 }