refactor: pass on both Bot and BotVersion to MatchPlayer

3 files changed, 39 insertions, 18 deletions

diff --git a/planetwars-server/src/modules/bot_api.rs b/planetwars-server/src/modules/bot_api.rs
index 6324010..732aa21 100644
--- a/planetwars-server/src/modules/bot_api.rs
+++ b/planetwars-server/src/modules/bot_api.rs
@@ -115,7 +115,7 @@ impl pb::bot_api_service_server::BotApiService for BotApiServer {
         });
         let mut run_match = RunMatch::from_players(vec![
             MatchPlayer::from_bot_spec(remote_bot_spec),
-            MatchPlayer::from_code_bundle(&opponent_code_bundle),
+            MatchPlayer::from_bot_version(&opponent, &opponent_code_bundle),
         ]);
         let created_match = run_match
             .store_in_database(&conn)
diff --git a/planetwars-server/src/modules/matches.rs b/planetwars-server/src/modules/matches.rs
index 4a5a980..a8c7ca9 100644
--- a/planetwars-server/src/modules/matches.rs
+++ b/planetwars-server/src/modules/matches.rs
@@ -24,15 +24,28 @@ pub struct RunMatch {
 
 pub struct MatchPlayer {
     bot_spec: Box<dyn BotSpec>,
-    // meta that will be passed on to database
+    // metadata that will be passed on to database
     code_bundle_id: Option<i32>,
 }
 
 impl MatchPlayer {
-    pub fn from_code_bundle(code_bundle: &db::bots::BotVersion) -> Self {
+    pub fn from_bot_version(bot: &db::bots::Bot, version: &db::bots::BotVersion) -> Self {
         MatchPlayer {
-            bot_spec: code_bundle_to_botspec(code_bundle),
-            code_bundle_id: Some(code_bundle.id),
+            bot_spec: bot_version_to_botspec(bot, version),
+            code_bundle_id: Some(version.id),
+        }
+    }
+
+    /// Construct a MatchPlayer from a BotVersion that certainly contains a code bundle path.
+    /// Will panic when this is not the case.
+    pub fn from_code_bundle_version(version: &db::bots::BotVersion) -> Self {
+        let code_bundle_path = version
+            .code_bundle_path
+            .as_ref()
+            .expect("no code_bundle_path found");
+        MatchPlayer {
+            bot_spec: python_docker_bot_spec(code_bundle_path),
+            code_bundle_id: Some(version.id),
         }
     }
 
@@ -97,12 +110,24 @@ impl RunMatch {
     }
 }
 
-pub fn code_bundle_to_botspec(code_bundle: &db::bots::BotVersion) -> Box<dyn BotSpec> {
-    // TODO: get rid of this unwrap
-    let bundle_path = PathBuf::from(BOTS_DIR).join(code_bundle.code_bundle_path.as_ref().unwrap());
+pub fn bot_version_to_botspec(
+    _bot: &db::bots::Bot,
+    bot_version: &db::bots::BotVersion,
+) -> Box<dyn BotSpec> {
+    if let Some(code_bundle_path) = &bot_version.code_bundle_path {
+        python_docker_bot_spec(code_bundle_path)
+    } else if let Some(_container_digest) = &bot_version.container_digest {
+        unimplemented!()
+    } else {
+        panic!("bad bot version")
+    }
+}
+
+fn python_docker_bot_spec(code_bundle_path: &str) -> Box<dyn BotSpec> {
+    let code_bundle_abs_path = PathBuf::from(BOTS_DIR).join(code_bundle_path);
 
     Box::new(DockerBotSpec {
-        code_path: bundle_path,
+        code_path: code_bundle_abs_path,
         image: PYTHON_IMAGE.to_string(),
         argv: vec!["python".to_string(), "bot.py".to_string()],
     })
diff --git a/planetwars-server/src/modules/ranking.rs b/planetwars-server/src/modules/ranking.rs
index 751c35e..3182ce2 100644
--- a/planetwars-server/src/modules/ranking.rs
+++ b/planetwars-server/src/modules/ranking.rs
@@ -37,18 +37,14 @@ pub async fn run_ranker(db_pool: DbPool) {
 
 async fn play_ranking_match(selected_bots: Vec<Bot>, db_pool: DbPool) {
     let db_conn = db_pool.get().await.expect("could not get db pool");
-    let mut code_bundles = Vec::new();
+    let mut players = Vec::new();
     for bot in &selected_bots {
-        let code_bundle = db::bots::active_bot_version(bot.id, &db_conn)
-            .expect("could not get active code bundle");
-        code_bundles.push(code_bundle);
+        let version = db::bots::active_bot_version(bot.id, &db_conn)
+            .expect("could not get active bot version");
+        let player = MatchPlayer::from_bot_version(bot, &version);
+        players.push(player);
     }
 
-    let players = code_bundles
-        .iter()
-        .map(MatchPlayer::from_code_bundle)
-        .collect::<Vec<_>>();
-
     let mut run_match = RunMatch::from_players(players);
     run_match
         .store_in_database(&db_conn)