e62rs 1.5.0

An in-terminal E621/926 browser.
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73

//! the actual media server
use {
    crate::serve::{
        cfg::ServerConfig,
        media::gallery::MediaGallery,
        routes::{
            AppState, css_handler, index_handler, js_handler, list_media_handler, stats_handler,
        },
    },
    color_eyre::eyre::Result,
    rocket::{Config, figment::Figment, fs::FileServer, routes},
    std::sync::Arc,
    tracing::info,
};

#[derive(Clone)]
/// the media server
pub struct MediaServer {
    /// the server config
    config: ServerConfig,
}

impl MediaServer {
    /// initialize the media server
    pub fn new(config: ServerConfig) -> Self {
        Self { config }
    }

    /// serve the media server
    pub async fn serve(self) -> Result<()> {
        let gallery = MediaGallery::new(
            self.config.media_directory.clone(),
            self.config.enable_metadata_filtering,
        );
        let state = Arc::new(AppState::new(gallery));

        info!("e6srv running at http://{}", self.config.bind_address);
        info!(
            "Serving media from: {}",
            self.config.media_directory.display()
        );
        info!(
            "Metadata filtering: {}",
            if self.config.enable_metadata_filtering {
                "enabled"
            } else {
                "disabled"
            }
        );

        let figment = Figment::from(Config::default())
            .merge(("address", self.config.bind_address.ip()))
            .merge(("port", self.config.bind_address.port()));

        let mut rocket = rocket::custom(figment)
            .mount(
                "/",
                routes![
                    index_handler,
                    list_media_handler,
                    stats_handler,
                    css_handler,
                    js_handler
                ],
            )
            .mount("/files", FileServer::from(&self.config.media_directory));

        rocket = rocket.manage(state);
        rocket.launch().await?;

        Ok(())
    }
}