From 3058028edc7ab794e496e12075fa79a4133cd42c Mon Sep 17 00:00:00 2001 From: Ilion Beyst Date: Sat, 10 Sep 2022 18:56:03 +0200 Subject: allow retrieving bot code --- planetwars-server/src/routes/bots.rs | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) (limited to 'planetwars-server/src/routes') 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, + Extension(config): Extension>, +) -> Result, 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) +} -- cgit v1.2.3