From 69331eb08a6199bfa8378e08cf378803b076eaae Mon Sep 17 00:00:00 2001
From: Ilion Beyst <ilion.beyst@gmail.com>
Date: Sun, 2 Jan 2022 17:56:52 +0100
Subject: [PATCH] serve match logs

---
 planetwars-server/src/db/matches.rs     | 21 +++++++++++++++-
 planetwars-server/src/lib.rs            |  1 +
 planetwars-server/src/routes/matches.rs | 32 +++++++++++++++++++------
 3 files changed, 46 insertions(+), 8 deletions(-)

diff --git a/planetwars-server/src/db/matches.rs b/planetwars-server/src/db/matches.rs
index 85f0631..9bf00db 100644
--- a/planetwars-server/src/db/matches.rs
+++ b/planetwars-server/src/db/matches.rs
@@ -1,5 +1,5 @@
 use chrono::NaiveDateTime;
-use diesel::{BelongingToDsl, RunQueryDsl};
+use diesel::{BelongingToDsl, QueryDsl, RunQueryDsl};
 use diesel::{Connection, GroupedBy, PgConnection, QueryResult};
 
 use crate::schema::{match_players, matches};
@@ -95,3 +95,22 @@ pub fn list_matches(conn: &PgConnection) -> QueryResult<Vec<MatchData>> {
         Ok(res)
     })
 }
+
+pub fn find_match(id: i32, conn: &PgConnection) -> QueryResult<MatchData> {
+    conn.transaction(|| {
+        let match_base = matches::table.find(id).get_result::<MatchBase>(conn)?;
+
+        let match_players = MatchPlayer::belonging_to(&match_base).load::<MatchPlayer>(conn)?;
+
+        let res = MatchData {
+            base: match_base,
+            match_players,
+        };
+
+        Ok(res)
+    })
+}
+
+pub fn find_mach_base(id: i32, conn: &PgConnection) -> QueryResult<MatchBase> {
+    matches::table.find(id).get_result::<MatchBase>(conn)
+}
diff --git a/planetwars-server/src/lib.rs b/planetwars-server/src/lib.rs
index 7eaa5a0..754d062 100644
--- a/planetwars-server/src/lib.rs
+++ b/planetwars-server/src/lib.rs
@@ -52,6 +52,7 @@ pub async fn api() -> Router {
             "/matches",
             get(routes::matches::list_matches).post(routes::matches::play_match),
         )
+        .route("/matches/:match_id", get(routes::matches::get_match_log))
         .layer(AddExtensionLayer::new(pool));
     api
 }
diff --git a/planetwars-server/src/routes/matches.rs b/planetwars-server/src/routes/matches.rs
index f2c5186..4dfd44e 100644
--- a/planetwars-server/src/routes/matches.rs
+++ b/planetwars-server/src/routes/matches.rs
@@ -1,6 +1,9 @@
-use std::path::PathBuf;
+use std::{io::Read, path::PathBuf};
 
-use axum::{extract::Extension, Json};
+use axum::{
+    extract::{Extension, Path},
+    Json,
+};
 use hyper::StatusCode;
 use planetwars_matchrunner::{run_match, MatchConfig, MatchPlayer};
 use rand::{distributions::Alphanumeric, Rng};
@@ -30,7 +33,7 @@ pub async fn play_match(
         .take(16)
         .map(char::from)
         .collect();
-    let log_path = PathBuf::from(MATCHES_DIR).join(&format!("{}.log", slug));
+    let log_file_name = format!("{}.log", slug);
 
     let mut players = Vec::new();
     let mut bot_ids = Vec::new();
@@ -58,22 +61,27 @@ pub async fn play_match(
     let match_config = MatchConfig {
         map_name: "hex".to_string(),
         map_path,
-        log_path: log_path.clone(),
+        log_path: PathBuf::from(MATCHES_DIR).join(&log_file_name),
         players: players,
     };
 
-    tokio::spawn(run_match_task(match_config, bot_ids, pool.clone()));
+    tokio::spawn(run_match_task(
+        match_config,
+        log_file_name,
+        bot_ids,
+        pool.clone(),
+    ));
     Ok(())
 }
 
 async fn run_match_task(
     config: MatchConfig,
+    log_file_name: String,
     match_players: Vec<matches::MatchPlayerData>,
     pool: ConnectionPool,
 ) {
-    let log_path = config.log_path.as_os_str().to_str().unwrap().to_string();
     let match_data = matches::NewMatch {
-        log_path: &log_path,
+        log_path: &log_file_name,
     };
 
     run_match(config).await;
@@ -119,3 +127,13 @@ pub struct BotConfig {
     pub run_command: String,
     pub build_command: Option<String>,
 }
+
+pub async fn get_match_log(
+    Path(match_id): Path<i32>,
+    conn: DatabaseConnection,
+) -> Result<Vec<u8>, StatusCode> {
+    let match_base = matches::find_mach_base(match_id, &conn).map_err(|_| StatusCode::NOT_FOUND)?;
+    let log_path = PathBuf::from(MATCHES_DIR).join(&match_base.log_path);
+    let log_contents = std::fs::read(log_path).map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
+    Ok(log_contents)
+}