src/api/image.rs


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
49
50
51
52
53
54

use axum::{extract::Query, response::Response, routing::get, Router, http::StatusCode};
use serde::Deserialize;

use crate::types::http::ResponseCode;


#[derive(Deserialize)]
struct AvatarRequest {
    user_id: u64,
}

async fn avatar(params: Option<Query<AvatarRequest>>) -> Response {
    
    let Some(params) = params else {
        return ResponseCode::BadRequest.text("Missing query paramaters");
    };

    let custom = format!("/image/custom/avatar/{}.png", params.user_id);
    let default = format!("/image/default/{}.png", params.user_id % 25);

    let file = ResponseCode::Success.file(&custom).await;
    if file.status() != StatusCode::OK {
        return ResponseCode::Success.file(&default).await
    }
    file
}

#[derive(Deserialize)]
struct BannerRequest {
    user_id: u64,
}

async fn banner(params: Option<Query<BannerRequest>>) -> Response {
    
    let Some(params) = params else {
        return ResponseCode::BadRequest.text("Missing query paramaters");
    };

    let custom = format!("/image/custom/banner/{}.png", params.user_id);

    let file = ResponseCode::Success.file(&custom).await;
    if file.status() != StatusCode::OK {
        return ResponseCode::NotFound.text("User does not have a custom banner")
    }
    file
}


pub fn router() -> Router {
    Router::new()
        .route("/avatar", get(avatar))
        .route("/banner", get(banner))
}