use axum::extract::{Path, RawBody}; use axum::http::StatusCode; use axum::Json; use rand::Rng; use serde::{Deserialize, Serialize}; use std::io::Cursor; use std::path; use crate::db::bots::{self, CodeBundle}; use crate::db::users::User; use crate::DatabaseConnection; use bots::Bot; #[derive(Serialize, Deserialize, Debug)] pub struct BotParams { name: String, } pub async fn create_bot( conn: DatabaseConnection, user: User, params: Json, ) -> (StatusCode, Json) { let bot_params = bots::NewBot { owner_id: user.id, name: ¶ms.name, }; let bot = bots::create_bot(&bot_params, &conn).unwrap(); (StatusCode::CREATED, Json(bot)) } // TODO: handle errors pub async fn get_bot(conn: DatabaseConnection, Path(bot_id): Path) -> Json { let bot = bots::find_bot(bot_id, &conn).unwrap(); Json(bot) } // TODO: proper error handling pub async fn upload_bot_code( conn: DatabaseConnection, user: User, Path(bot_id): Path, RawBody(body): RawBody, ) -> (StatusCode, Json) { // TODO: put in config somewhere let data_path = "./data/bots"; let bot = bots::find_bot(bot_id, &conn).expect("Bot not found"); assert_eq!(user.id, bot.owner_id); // generate a random filename let token: [u8; 16] = rand::thread_rng().gen(); let name = base64::encode(&token); let path = path::Path::new(data_path).join(name); // let capped_buf = data.open(10usize.megabytes()).into_bytes().await.unwrap(); // assert!(capped_buf.is_complete()); // let buf = capped_buf.into_inner(); let buf = hyper::body::to_bytes(body).await.unwrap(); zip::ZipArchive::new(Cursor::new(buf)) .unwrap() .extract(&path) .unwrap(); let bundle = bots::NewCodeBundle { bot_id: bot.id, path: path.to_str().unwrap(), }; let code_bundle = bots::create_code_bundle(&bundle, &conn).expect("Failed to create code bundle"); (StatusCode::CREATED, Json(code_bundle)) }