77 lines
2.7 KiB
Rust
77 lines
2.7 KiB
Rust
use rocket::http::ContentType;
|
|
use rocket::response::{self, Responder};
|
|
use rocket::{Request, Response};
|
|
use rocket_okapi::okapi::openapi3::{self, MediaType, RefOr};
|
|
use schemars::schema::{InstanceType, SchemaObject, SingleOrVec};
|
|
|
|
pub struct CachedFile((ContentType, Vec<u8>));
|
|
|
|
pub static CACHE_CONTROL: &str = "public, max-age=31536000, immutable";
|
|
|
|
impl<'r> Responder<'r, 'static> for CachedFile {
|
|
fn respond_to(self, req: &'r Request<'_>) -> response::Result<'static> {
|
|
Response::build_from(self.0.respond_to(req)?)
|
|
.raw_header("Cache-Control", CACHE_CONTROL)
|
|
.ok()
|
|
}
|
|
}
|
|
|
|
impl rocket_okapi::response::OpenApiResponderInner for CachedFile {
|
|
fn responses(
|
|
_gen: &mut rocket_okapi::gen::OpenApiGenerator,
|
|
) -> std::result::Result<openapi3::Responses, rocket_okapi::OpenApiError> {
|
|
let mut responses = schemars::Map::new();
|
|
let mut content = schemars::Map::new();
|
|
|
|
content.insert(
|
|
"image/png".to_owned(),
|
|
MediaType {
|
|
schema: Some(SchemaObject {
|
|
instance_type: Some(SingleOrVec::Single(Box::new(InstanceType::String))),
|
|
format: Some("binary".to_owned()),
|
|
..Default::default()
|
|
}),
|
|
..Default::default()
|
|
},
|
|
);
|
|
|
|
responses.insert(
|
|
"200".to_string(),
|
|
RefOr::Object(openapi3::Response {
|
|
description: "Default Avatar Picture".to_string(),
|
|
content,
|
|
..Default::default()
|
|
}),
|
|
);
|
|
|
|
Ok(openapi3::Responses {
|
|
responses,
|
|
..Default::default()
|
|
})
|
|
}
|
|
}
|
|
|
|
/// # Fetch Default Avatar
|
|
///
|
|
/// This returns a default avatar based on the given id.
|
|
#[openapi(tag = "User Information")]
|
|
#[get("/<target>/default_avatar")]
|
|
pub async fn req(target: String) -> CachedFile {
|
|
CachedFile((
|
|
ContentType::PNG,
|
|
match target.chars().last().unwrap() {
|
|
// 0123456789ABCDEFGHJKMNPQRSTVWXYZ
|
|
'0' | '1' | '2' | '3' | 'S' | 'Z' => {
|
|
include_bytes!(crate::asset!("user/2.png")).to_vec()
|
|
}
|
|
'4' | '5' | '6' | '7' | 'T' => include_bytes!(crate::asset!("user/3.png")).to_vec(),
|
|
'8' | '9' | 'A' | 'B' => include_bytes!(crate::asset!("user/4.png")).to_vec(),
|
|
'C' | 'D' | 'E' | 'F' | 'V' => include_bytes!(crate::asset!("user/5.png")).to_vec(),
|
|
'G' | 'H' | 'J' | 'K' | 'W' => include_bytes!(crate::asset!("user/6.png")).to_vec(),
|
|
'M' | 'N' | 'P' | 'Q' | 'X' => include_bytes!(crate::asset!("user/7.png")).to_vec(),
|
|
/*'0' | '1' | '2' | '3' | 'R' | 'Y'*/
|
|
_ => include_bytes!(crate::asset!("user/1.png")).to_vec(),
|
|
},
|
|
))
|
|
}
|