aboutsummaryrefslogtreecommitdiff
path: root/planetwars-server/src/routes/bots.rs
diff options
context:
space:
mode:
Diffstat (limited to 'planetwars-server/src/routes/bots.rs')
-rw-r--r--planetwars-server/src/routes/bots.rs39
1 files changed, 39 insertions, 0 deletions
diff --git a/planetwars-server/src/routes/bots.rs b/planetwars-server/src/routes/bots.rs
index 4ab1b4e..f8087fd 100644
--- a/planetwars-server/src/routes/bots.rs
+++ b/planetwars-server/src/routes/bots.rs
@@ -7,6 +7,7 @@ use rand::distributions::Alphanumeric;
use rand::Rng;
use serde::{Deserialize, Serialize};
use serde_json::{self, json, value::Value as JsonValue};
+use std::collections::HashMap;
use std::io::Cursor;
use std::path::PathBuf;
use std::sync::Arc;
@@ -275,3 +276,41 @@ pub async fn get_code(
std::fs::read(full_bundle_path).map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
Ok(bot_code)
}
+
+#[derive(Default, Serialize, Deserialize)]
+pub struct MatchupStats {
+ win: i64,
+ loss: i64,
+ tie: i64,
+}
+
+impl MatchupStats {
+ fn update(&mut self, win: Option<bool>, count: i64) {
+ match win {
+ Some(true) => self.win += count,
+ Some(false) => self.loss += count,
+ None => self.tie += count,
+ }
+ }
+}
+
+type BotStats = HashMap<String, HashMap<String, MatchupStats>>;
+
+pub async fn get_bot_stats(
+ conn: DatabaseConnection,
+ Path(bot_name): Path<String>,
+) -> Result<Json<BotStats>, StatusCode> {
+ let stats_records = db::matches::fetch_bot_stats(&bot_name, &conn)
+ .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
+ let mut bot_stats: BotStats = HashMap::new();
+ for record in stats_records {
+ bot_stats
+ .entry(record.opponent)
+ .or_default()
+ .entry(record.map)
+ .or_default()
+ .update(record.win, record.count);
+ }
+
+ Ok(Json(bot_stats))
+}