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.rs26
1 files changed, 26 insertions, 0 deletions
diff --git a/planetwars-server/src/routes/bots.rs b/planetwars-server/src/routes/bots.rs
index f2bf202..4ab1b4e 100644
--- a/planetwars-server/src/routes/bots.rs
+++ b/planetwars-server/src/routes/bots.rs
@@ -249,3 +249,29 @@ pub async fn upload_code_multipart(
Ok(Json(code_bundle))
}
+
+pub async fn get_code(
+ conn: DatabaseConnection,
+ user: User,
+ Path(bundle_id): Path<i32>,
+ Extension(config): Extension<Arc<GlobalConfig>>,
+) -> Result<Vec<u8>, StatusCode> {
+ let version =
+ db::bots::find_bot_version(bundle_id, &conn).map_err(|_| StatusCode::NOT_FOUND)?;
+ let bot_id = version.bot_id.ok_or(StatusCode::FORBIDDEN)?;
+ let bot = db::bots::find_bot(bot_id, &conn).map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
+
+ if bot.owner_id != Some(user.id) {
+ return Err(StatusCode::FORBIDDEN);
+ }
+
+ let bundle_path = version.code_bundle_path.ok_or(StatusCode::NOT_FOUND)?;
+
+ // TODO: avoid hardcoding paths
+ let full_bundle_path = PathBuf::from(&config.bots_directory)
+ .join(&bundle_path)
+ .join("bot.py");
+ let bot_code =
+ std::fs::read(full_bundle_path).map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
+ Ok(bot_code)
+}