memory-serve 1.2.1

Fast static file serving for axum web applications
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

use memory_serve_core::{ENV_NAME, FORCE_EMBED_FEATURE, ROOT_ENV_NAME, load_names_directories};
use std::path::{Path, PathBuf};

/// Try to resolve the (relative) asset directory path
fn resolve_asset_dir(out_dir: &Path, key: &str, asset_dir: &str) -> PathBuf {
    let path = Path::new(&asset_dir);

    let path: PathBuf = if path.is_relative() {
        if let Ok(root_dir) = std::env::var(ROOT_ENV_NAME) {
            let root_dir = Path::new(&root_dir);
            root_dir.join(path)
        } else {
            // assume the out dit is in the target directory
            let crate_root = out_dir
                .parent() // memory-serve
                .and_then(|p| p.parent()) // build
                .and_then(|p| p.parent()) // debug/release
                .and_then(|p| p.parent()) // target
                .and_then(|p| p.parent()) // crate root
                .expect("Unable to get crate root directory.");

            crate_root.join(path)
        }
    } else {
        path.to_path_buf()
    };

    let path = match path.canonicalize() {
        Ok(path) => path,
        Err(e) => panic!("The path {path:?} specified by {key} is not a valid path: {e}"),
    };

    if !path.exists() {
        panic!("The path {path:?} specified by {key} does not exists!");
    }

    path
}

/// Generate the asset file when a ASSET_DIR environment variable is provided
fn main() {
    let out_dir: String = std::env::var("OUT_DIR").expect("OUT_DIR environment variable not set.");
    let out_dir = PathBuf::from(&out_dir);

    println!("cargo::rerun-if-env-changed={ENV_NAME}");

    let named_paths: Vec<(String, PathBuf)> = std::env::vars()
        .filter(|(key, _)| key.starts_with(ENV_NAME))
        .map(|(key, asset_dir)| {
            println!("cargo::rerun-if-env-changed={key}");

            let name = key.trim_start_matches(format!("{ENV_NAME}_").as_str());
            let path = resolve_asset_dir(&out_dir, &key, &asset_dir);

            (name.to_string(), path)
        })
        .collect();

    // determine whether to dynamically load assets or embed them in the binary
    let force_embed = std::env::var(FORCE_EMBED_FEATURE).unwrap_or_default();
    println!("cargo::rerun-if-env-changed={FORCE_EMBED_FEATURE}");
    let embed = !cfg!(debug_assertions) || force_embed == "1";

    load_names_directories(named_paths, embed);
}