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

use std::collections::HashMap;
use std::path::Path;

use crate::functions::{compute_data_etag, guess_mime};
use crate::{EntityTag, Mime};

#[derive(Debug)]
/// Static resources.
pub struct StaticResources {
    resources: HashMap<&'static str, (Mime, &'static [u8], EntityTag)>,
}

impl StaticResources {
    #[inline]
    /// Create an instance of `StaticResources`.
    pub fn new() -> StaticResources {
        StaticResources {
            resources: HashMap::new(),
        }
    }

    #[inline]
    /// Register a static resource.
    pub fn register_resource_static<P: AsRef<Path>>(
        &mut self,
        name: &'static str,
        path: P,
        data: &'static [u8],
    ) {
        let etag = compute_data_etag(data);

        let mime = guess_mime(path);

        self.resources.insert(name, (mime, data, etag));
    }

    #[inline]
    /// Get the specific resource.
    pub fn get_resource<S: AsRef<str>>(
        &self,
        name: S,
    ) -> Option<(&Mime, &'static [u8], &EntityTag)> {
        let name = name.as_ref();

        self.resources.get(name).map(|(mime, data, etag)| (mime, *data, etag))
    }
}

impl Default for StaticResources {
    #[inline]
    fn default() -> Self {
        StaticResources::new()
    }
}