2021-12-29 15:11:27 +00:00
|
|
|
use axum::extract::{Path, RawBody};
|
|
|
|
use axum::http::StatusCode;
|
2021-12-30 22:40:37 +00:00
|
|
|
use axum::response::IntoResponse;
|
2021-12-29 15:11:27 +00:00
|
|
|
use axum::Json;
|
2021-12-18 23:16:46 +00:00
|
|
|
use rand::Rng;
|
|
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
use std::io::Cursor;
|
2021-12-29 15:11:27 +00:00
|
|
|
use std::path;
|
2021-12-18 23:16:46 +00:00
|
|
|
|
|
|
|
use crate::db::bots::{self, CodeBundle};
|
|
|
|
use crate::db::users::User;
|
2021-12-29 15:11:27 +00:00
|
|
|
use crate::DatabaseConnection;
|
2021-12-18 23:16:46 +00:00
|
|
|
use bots::Bot;
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
|
|
pub struct BotParams {
|
|
|
|
name: String,
|
|
|
|
}
|
|
|
|
|
|
|
|
pub async fn create_bot(
|
2021-12-29 15:11:27 +00:00
|
|
|
conn: DatabaseConnection,
|
2021-12-18 23:16:46 +00:00
|
|
|
user: User,
|
|
|
|
params: Json<BotParams>,
|
2021-12-29 15:11:27 +00:00
|
|
|
) -> (StatusCode, Json<Bot>) {
|
|
|
|
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))
|
2021-12-18 23:16:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: handle errors
|
2021-12-30 22:40:37 +00:00
|
|
|
pub async fn get_bot(conn: DatabaseConnection, Path(bot_id): Path<i32>) -> impl IntoResponse {
|
|
|
|
bots::find_bot(bot_id, &conn)
|
|
|
|
.map(Json)
|
|
|
|
.map_err(|_| StatusCode::NOT_FOUND)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub async fn get_my_bots(
|
|
|
|
conn: DatabaseConnection,
|
|
|
|
user: User,
|
|
|
|
) -> Result<Json<Vec<Bot>>, StatusCode> {
|
|
|
|
bots::find_bots_by_owner(user.id, &conn)
|
|
|
|
.map(Json)
|
|
|
|
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)
|
2021-12-18 23:16:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: proper error handling
|
|
|
|
pub async fn upload_bot_code(
|
2021-12-29 15:11:27 +00:00
|
|
|
conn: DatabaseConnection,
|
2021-12-18 23:16:46 +00:00
|
|
|
user: User,
|
2021-12-29 15:11:27 +00:00
|
|
|
Path(bot_id): Path<i32>,
|
|
|
|
RawBody(body): RawBody,
|
|
|
|
) -> (StatusCode, Json<CodeBundle>) {
|
2021-12-18 23:16:46 +00:00
|
|
|
// TODO: put in config somewhere
|
|
|
|
let data_path = "./data/bots";
|
|
|
|
|
2021-12-29 15:11:27 +00:00
|
|
|
let bot = bots::find_bot(bot_id, &conn).expect("Bot not found");
|
2021-12-18 23:16:46 +00:00
|
|
|
|
|
|
|
assert_eq!(user.id, bot.owner_id);
|
|
|
|
|
|
|
|
// generate a random filename
|
|
|
|
let token: [u8; 16] = rand::thread_rng().gen();
|
|
|
|
let name = base64::encode(&token);
|
|
|
|
|
2021-12-29 15:11:27 +00:00
|
|
|
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();
|
2021-12-18 23:16:46 +00:00
|
|
|
|
|
|
|
zip::ZipArchive::new(Cursor::new(buf))
|
|
|
|
.unwrap()
|
|
|
|
.extract(&path)
|
|
|
|
.unwrap();
|
|
|
|
|
2021-12-29 15:11:27 +00:00
|
|
|
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");
|
2021-12-18 23:16:46 +00:00
|
|
|
|
2021-12-29 15:11:27 +00:00
|
|
|
(StatusCode::CREATED, Json(code_bundle))
|
2021-12-18 23:16:46 +00:00
|
|
|
}
|