Add button to download ranking

Author: eliasol

backend/src/handlers/admin.rs

@@ -1,5 +1,6 @@
 use axum::Json;
 use axum::extract::{Path, State};
+use axum_extra::response::Attachment;
 use tracing::info;
 
 use super::ApiError;
@@ -34,6 +35,41 @@ pub async fn user_list(
     Ok(Json(resp))
 }
 
+pub async fn download_ranking(
+    State(state): State<AppState>,
+    _: Admin,
+) -> Result<Attachment<Vec<u8>>, ApiError> {
+    let tasks = database::get_tasks(&state.pool).await?;
+    let task_names: Vec<&str> = tasks.iter().map(|task| task.name.as_str()).collect();
+
+    let users = database::get_users(&state.pool).await?;
+    let mut resp = vec![];
+    for u in users {
+        resp.push(super::get_user_status(&state.pool, u).await?);
+    }
+
+    let mut csv = csv::Writer::from_writer(vec![]);
+
+    let mut headers = vec!["name", "surname", "token", "role"];
+    headers.append(&mut task_names.clone());
+    headers.append(&mut vec!["total_score"]);
+
+    csv.write_record(&headers)?;
+
+    for user in resp {
+        let mut row = vec![user.name, user.surname, user.token, user.role.to_string()];
+        for t in &task_names {
+            row.push(user.tasks.get(*t).map_or(0f64, |x| x.score).to_string());
+        }
+        row.push(user.total_score.to_string());
+        csv.write_record(row)?;
+    }
+
+    Ok(Attachment::new(csv.into_inner().unwrap())
+        .content_type("text/csv")
+        .filename("ranking.csv"))
+}
+
 pub async fn questions(
     State(state): State<AppState>,
     _: Admin,

backend/src/handlers/mod.rs

@@ -58,6 +58,12 @@ impl From<serde_json::Error> for ApiError {
     }
 }
 
+impl From<csv::Error> for ApiError {
+    fn from(err: csv::Error) -> Self {
+        ApiError::Internal(err.to_string())
+    }
+}
+
 impl From<sqlx::Error> for ApiError {
     fn from(err: sqlx::Error) -> Self {
         ApiError::Internal(err.to_string())

backend/src/serve.rs

@@ -88,6 +88,7 @@ fn create_router(app_state: AppState) -> Router {
             post(admin::set_extra_time),
         )
         .route("/api/admin/user_list", get(admin::user_list))
+        .route("/api/admin/ranking", get(admin::download_ranking))
         .route("/api/admin/questions", get(admin::questions))
         .route(
             "/api/admin/answer_question/{id}",

frontend/src/admin/AdminContestStatus.tsx

@@ -1,6 +1,7 @@
 import React, { useEffect, useMemo, useState } from "react";
 import { FontAwesomeIcon } from "@fortawesome/react-fontawesome";
 import {
+  faDownload,
   faHourglassStart,
   faSort,
   faSortDown,
@@ -230,6 +231,16 @@ export function ContestStatusView() {
                 <option value="30000">30s</option>
                 <option value="120000">2m</option>
               </select>
+              <a
+                role="button"
+                className="btn btn-light ml-3"
+                href={`${client.apiBaseURI}admin/ranking`}
+                download
+              >
+                <FontAwesomeIcon icon={faDownload} />
+                {" "}
+                CSV
+              </a>
             </div>
             <table className="table table-striped">
               <thead>