aa_fastlink 0.5.0

Share your Anna's Archive membership, without sharing your secret key!
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

// declare modules
mod config;
mod download;
mod form;
mod version;

#[tokio::main]
async fn main() {
    // import configuration constants
    use config::{BIND_IP, BIND_PORT, DOMAIN, MAX_BODY_SIZE, SECRET};

    // import warp items
    use warp::{Filter, body, get, path, post, serve};

    // ensure environment variables are set
    let _ = *SECRET;
    let _ = *DOMAIN;

    // define routes
    let routes = get()
        .map(form::render_form)
        .or(post()
            .and(body::content_length_limit(MAX_BODY_SIZE))
            .and(body::form())
            .and(path("dl"))
            .and_then(download::handle_download))
        .or(get().and(path("version")).map(version::plain));

    // initialize logger
    init_logger();

    // print version and bind address
    log::info!("Running aa-fastlink {}", env!("CARGO_PKG_VERSION"));
    log::info!("Binding to http://{}:{}", *BIND_IP, *BIND_PORT);

    // start server
    serve(routes).run((*BIND_IP, *BIND_PORT)).await;
}

// initializes logger with custom settings
pub fn init_logger() {
    use std::io::Write;

    let mut builder = env_logger::Builder::new();

    builder.format(|buf, record| writeln!(buf, "{}: {}", record.level(), record.args()));

    builder.filter_level(if *config::DEBUG_LOGGING {
        log::LevelFilter::Debug
    } else {
        log::LevelFilter::Info
    });

    builder.init();
}