euv_cli/server/
fn.rs

1use crate::*;
2
3/// Sets the global application state.
4///
5/// # Arguments
6///
7/// - `Arc<AppState>` - The shared application state to store globally.
8///
9/// # Returns
10///
11/// - `Result<(), EuvError>` - Indicates success or failure of the initialization.
12pub(crate) fn set_global_state(state: Arc<AppState>) -> Result<(), EuvError> {
13    APP_STATE.set(state).map_err(|_: Arc<AppState>| {
14        EuvError::Message(String::from("Global state already initialized"))
15    })
16}
17
18/// Retrieves the global application state.
19///
20/// # Returns
21///
22/// - `Option<Arc<AppState>>` - The global state if initialized.
23pub(crate) fn get_global_state() -> Option<Arc<AppState>> {
24    APP_STATE.get().cloned()
25}
26
27/// Generates `index.html` based on the build profile.
28///
29/// Uses `INDEX_HTML_RELEASE` when `is_release` is `true` (no live-reload script),
30/// otherwise uses `INDEX_HTML_DEV` (includes live-reload instrumentation).
31///
32/// Then writes the template with the import path placeholder replaced to disk.
33///
34/// # Arguments
35///
36/// - `&HtmlConfig` - The HTML generation configuration.
37///
38/// # Returns
39///
40/// - `Result<String, EuvError>` - The generated HTML content written to disk.
41pub(crate) async fn generate_html(config: &HtmlConfig) -> Result<String, EuvError> {
42    let template_content: String = if let Some(custom_path) = config.try_get_custom_index_html() {
43        let bytes: Vec<u8> =
44            read(custom_path)
45                .await
46                .map_err(|error: io::Error| EuvError::IoPath {
47                    message: String::from("Failed to read custom index.html"),
48                    path: custom_path.to_path_buf(),
49                    error,
50                })?;
51        String::from_utf8(bytes).map_err(|error: std::string::FromUtf8Error| EuvError::Utf8 {
52            message: String::from("Custom index.html is not valid UTF-8"),
53            error,
54        })?
55    } else if config.get_is_release() {
56        INDEX_HTML_RELEASE.to_string()
57    } else {
58        INDEX_HTML_DEV.to_string()
59    };
60    let html: String = template_content
61        .replace(IMPORT_PATH_PLACEHOLDER, config.get_import_path())
62        .replace(RELOAD_ROUTE_PLACEHOLDER, RELOAD_ROUTE);
63    let index_path: PathBuf = config.get_serving_root().join(INDEX_HTML_FILE_NAME);
64    create_dir_all(config.get_serving_root())
65        .await
66        .map_err(|error: io::Error| EuvError::Io {
67            message: String::from("Failed to create static directory"),
68            error,
69        })?;
70    write(&index_path, &html)
71        .await
72        .map_err(|error: io::Error| EuvError::Io {
73            message: String::from("Failed to write index.html"),
74            error,
75        })?;
76    Ok(html)
77}
78
79/// Resolves the effective www directory, handling wasm-pack nested output.
80///
81/// # Arguments
82///
83/// - `&Path` - The candidate www directory path.
84///
85/// # Returns
86///
87/// - `PathBuf` - The resolved www directory containing `index.html`.
88pub async fn resolve_www_dir(www_dir: &Path) -> PathBuf {
89    if metadata(www_dir.join(INDEX_HTML_FILE_NAME)).await.is_ok() {
90        return www_dir.to_path_buf();
91    }
92    let parent_name: Option<&str> = www_dir
93        .file_name()
94        .and_then(|file_name_os_str: &ffi::OsStr| file_name_os_str.to_str());
95    if let Some(name) = parent_name {
96        let nested: PathBuf = www_dir.join(name);
97        if metadata(nested.join(INDEX_HTML_FILE_NAME)).await.is_ok() {
98            return nested;
99        }
100    }
101    www_dir.to_path_buf()
102}
103
104/// Resolves the pkg directory for serving WASM artifacts.
105///
106/// Delegates to `resolve_out_dir` which respects `--out-dir`
107/// from wasm-pack args or defaults to `{www_dir}/pkg`.
108///
109/// # Arguments
110///
111/// - `&ModeArgs` - The CLI arguments for resolving out_dir.
112///
113/// # Returns
114///
115/// - `PathBuf` - The resolved pkg directory containing WASM build artifacts.
116pub fn resolve_pkg_dir(args: &ModeArgs) -> PathBuf {
117    resolve_out_dir(args)
118}
119
120/// Resolves a file path within a base directory with path-traversal protection.
121///
122/// Canonicalizes both the file path and the base directory, then verifies
123/// that the canonical file path starts with the canonical base directory.
124///
125/// # Arguments
126///
127/// - `&Path` - The base directory to resolve within.
128/// - `&str` - The relative path to the file.
129///
130/// # Returns
131///
132/// - `Option<PathBuf>` - The resolved file path if valid, `None` otherwise.
133pub async fn resolve_file_in_base(base: &Path, path: &str) -> Option<PathBuf> {
134    let file_path: PathBuf = base.join(path);
135    let canonical_path: PathBuf = canonicalize(&file_path).await.ok()?;
136    let base_canonical: PathBuf = canonicalize(base).await.ok()?;
137    canonical_path
138        .starts_with(&base_canonical)
139        .then_some(file_path)
140}
euv_cli/server/fn.rs

euv_cli/server/
fn.rs
