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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

//! routes for the gallery
use {
    crate::{
        getopt,
        serve::{
            media::{
                filter::MediaFilter, gallery::MediaGallery, item::MediaItem, stats::FilterStats,
            },
            theme::registry::ThemeRegistry,
        },
    },
    rocket::{
        State, get,
        http::Status,
        response::content::{RawCss, RawHtml, RawJavaScript},
        serde::json::Json,
    },
    std::sync::Arc,
    tokio::sync::RwLock,
};

/// the HTML code of the media gallery
const HTML_TEMPLATE: &str = include_str!("../../resources/gallery/index.html");
/// the CSS for the html template
const CSS: &str = include_str!("../../resources/gallery/styles.css");
/// the JS for the html template
const JS: &str = include_str!("../../resources/gallery/script.js");

/// the state of the app
pub struct AppState {
    /// the gallery
    pub gallery: Arc<RwLock<MediaGallery>>,
}

impl AppState {
    /// initialize a new AppState
    ///
    /// # Arguments
    ///
    /// * `gallery` - a loaded gallery
    pub fn new(gallery: MediaGallery) -> Self {
        Self {
            gallery: Arc::new(RwLock::new(gallery)),
        }
    }
}

#[get("/")]
/// the main gallery page
pub async fn index_handler() -> RawHtml<String> {
    RawHtml(HTML_TEMPLATE.to_string())
}

#[get("/styles.css")]
/// serve the dynamically themed CSS
pub async fn css_handler() -> RawCss<String> {
    let configured_theme = getopt!(gallery.theme);
    let registry = ThemeRegistry::new();
    let css_vars = registry
        .get_theme_css_vars(&configured_theme)
        .unwrap_or("".to_string());
    let css_code = CSS.replace("/* {{THEME_CSS_VARS}} */", &css_vars);
    RawCss(css_code)
}

#[get("/script.js")]
/// serve the JS
pub async fn js_handler() -> RawJavaScript<String> {
    RawJavaScript(JS.to_string())
}

#[get("/api/media?<filter..>")]
/// handler for listing media
///
/// # Arguments
///
/// * `state` - the current state of the app
/// * `filter` - any active filters
pub async fn list_media_handler(
    state: &State<Arc<AppState>>,
    filter: Option<MediaFilter>,
) -> Result<Json<Vec<MediaItem>>, Status> {
    let mut gallery = state.gallery.write().await;
    let filter = filter.unwrap_or_default();

    let items = gallery
        .get_filtered_items(&filter)
        .await
        .map_err(|_| Status::InternalServerError)?;

    Ok(Json(items))
}

#[get("/api/stats")]
/// handler for getting current filter stats
pub async fn stats_handler(state: &State<Arc<AppState>>) -> Result<Json<FilterStats>, Status> {
    let gallery = state.gallery.read().await;
    let stats = gallery.get_filter_stats();
    Ok(Json(stats))
}