identicon-rs 7.2.1

identicon-rs is a library built around custom generation of identicon images.
Documentation 
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
55
56
57
58
59

use std::ffi::OsStr;

use axum::{
    Router,
    extract::Path,
    http::{StatusCode, header},
    response::IntoResponse,
    routing::get,
};
use identicon_rs::Identicon;

#[tokio::main]
async fn main() {
    let app = Router::new().route("/{input}", get(root));

    let listener = tokio::net::TcpListener::bind("0.0.0.0:8080").await.unwrap();
    axum::serve(listener, app).await.unwrap();
}

// basic handler that responds with a static string
async fn root(Path(input): Path<String>) -> impl IntoResponse {
    let file_path = std::path::Path::new(&input);
    match file_path.file_stem() {
        None => StatusCode::BAD_REQUEST.into_response(),
        Some(input) => match input.to_str() {
            None => StatusCode::BAD_REQUEST.into_response(),
            Some(name) => {
                let identicon: Identicon = Identicon::new(name);
                generate_image(file_path.extension(), &identicon).into_response()
            }
        },
    }
}

fn generate_image(extention: Option<&OsStr>, identicon: &Identicon) -> impl IntoResponse {
    match extention {
        Some(os_str) => match os_str.to_str() {
            Some("jpeg") | Some("jpg") => (
                StatusCode::OK,
                [(header::CONTENT_TYPE, "image/jpeg")],
                identicon.export_jpeg_data().unwrap(),
            )
                .into_response(),
            Some("png") => (
                StatusCode::OK,
                [(header::CONTENT_TYPE, "image/png")],
                identicon.export_png_data().unwrap(),
            )
                .into_response(),
            _ => StatusCode::BAD_REQUEST.into_response(),
        },
        None => (
            StatusCode::OK,
            [(header::CONTENT_TYPE, "image/png")],
            identicon.export_png_data().unwrap(),
        )
            .into_response(),
    }
}