bot upload PoC

2 files changed, 32 insertions, 0 deletions

diff --git a/planetwars-server/src/lib.rs b/planetwars-server/src/lib.rs
index ec12997..6cc2824 100644
--- a/planetwars-server/src/lib.rs
+++ b/planetwars-server/src/lib.rs
@@ -90,6 +90,7 @@ pub async fn api(configuration: Configuration) -> Router {
             get(routes::matches::get_match_log),
         )
         .route("/submit_bot", post(routes::demo::submit_bot))
+        .route("/save_bot", post(routes::bots::save_bot))
         .layer(AddExtensionLayer::new(db_pool));
     api
 }
diff --git a/planetwars-server/src/routes/bots.rs b/planetwars-server/src/routes/bots.rs
index 66b4d82..66479bb 100644
--- a/planetwars-server/src/routes/bots.rs
+++ b/planetwars-server/src/routes/bots.rs
@@ -1,6 +1,7 @@
 use axum::extract::{Multipart, Path};
 use axum::http::StatusCode;
 use axum::Json;
+use diesel::OptionalExtension;
 use rand::distributions::Alphanumeric;
 use rand::Rng;
 use serde::{Deserialize, Serialize};
@@ -10,10 +11,40 @@ use std::path::PathBuf;
 
 use crate::db::bots::{self, CodeBundle};
 use crate::db::users::User;
+use crate::modules::bots::save_code_bundle;
 use crate::{DatabaseConnection, BOTS_DIR};
 use bots::Bot;
 
 #[derive(Serialize, Deserialize, Debug)]
+pub struct SaveBotParams {
+    pub bot_name: String,
+    pub code: String,
+}
+pub async fn save_bot(
+    Json(params): Json<SaveBotParams>,
+    conn: DatabaseConnection,
+) -> Result<(), StatusCode> {
+    // TODO: authorization
+    let res = bots::find_bot_by_name(&params.bot_name, &conn)
+        .optional()
+        .expect("could not run query");
+    let bot = match res {
+        Some(bot) => bot,
+        None => {
+            let new_bot = bots::NewBot {
+                owner_id: None,
+                name: &params.bot_name,
+            };
+            let bot = bots::create_bot(&new_bot, &conn).expect("could not create bot");
+            bot
+        }
+    };
+    let _code_bundle =
+        save_code_bundle(&params.code, Some(bot.id), &conn).expect("failed to save code bundle");
+    Ok(())
+}
+
+#[derive(Serialize, Deserialize, Debug)]
 pub struct BotParams {
     name: String,
 }