1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
use actix::Addr;
use actix_web::{get, web, HttpResponse, Responder, ResponseError};
use derive_more::Display;
use serde::Serialize;
use crate::web_server::{BotDataListRequest, BotDataRequest, BotExecutor};
#[get("/bots")]
pub async fn get_bot_list(bot: web::Data<Addr<BotExecutor>>) -> impl Responder {
let bot_datas = match bot.send(BotDataListRequest).await.unwrap() {
Ok(data) => data,
Err(_) => Vec::with_capacity(0),
};
web::Json(bot_datas)
}
#[get("/bots/{name}")]
pub async fn get_bot(bot: web::Data<Addr<BotExecutor>>, name: web::Path<String>) -> impl Responder {
if let Some(bot_data) = bot.send(BotDataRequest(name.into_inner())).await.unwrap() {
Ok(web::Json(bot_data))
} else {
Err(ApiErrorKind::NotFound)
}
}
#[derive(Serialize)]
struct ApiError {
error: String,
description: String,
}
#[derive(Debug, Display)]
enum ApiErrorKind {
#[display(fmt = "Not Found")]
NotFound,
}
impl ResponseError for ApiErrorKind {
fn error_response(&self) -> HttpResponse {
match *self {
ApiErrorKind::NotFound => HttpResponse::NotFound().json(ApiError {
error: self.to_string(),
description: String::from("The requested resource was not found"),
}),
}
}
}
|
