dichroism/src/main.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
55
56
57
58
59
60
61
62

#[macro_use]
extern crate lazy_static;
use actix_web::{get, post, App, HttpResponse, HttpServer, Responder};
use base64;
use listenfd::ListenFd;
use regex::Regex;

lazy_static! {
    static ref DATA_URI_RE: Regex =
        Regex::new("^data:image/(?P<format>png|jpg|jpeg);base64,(?P<data>.+)").unwrap();
}

#[get("/")]
async fn hello() -> impl Responder {
    HttpResponse::Ok().body("Hey, this is an API!")
}

#[post("/products")]
async fn create_product(req_body: String) -> impl Responder {
    let caps = if let Some(c) = DATA_URI_RE.captures(&req_body) {
        c
    } else {
        return HttpResponse::BadRequest().body("Invalid data URI");
    };

    let format = if let Some(f) = caps.name("format") {
        f.as_str()
    } else {
        return HttpResponse::BadRequest().body("Invalid image format.");
    };

    let data = if let Some(d) = caps.name("data") {
        d.as_str()
    } else {
        return HttpResponse::BadRequest().body("No encoded data.");
    };

    HttpResponse::Ok().body(format!("created\n{}\n{}", format, data))
}

#[get("/products")]
async fn get_products(_req_body: String) -> impl Responder {
    HttpResponse::Ok().body("got product!")
}

#[actix_web::main]
async fn main() -> std::io::Result<()> {
    let mut listenfd = ListenFd::from_env();
    let mut server = HttpServer::new(|| {
        App::new()
            .service(hello)
            .service(create_product)
            .service(get_products)
    });
    server = if let Some(l) = listenfd.take_tcp_listener(0).unwrap() {
        server.listen(l)?
    } else {
        server.bind("127.0.0.1:8080")?
    };

    server.run().await
}