yew_router/
utils.rs

1use std::cell::RefCell;
2
3use wasm_bindgen::JsCast;
4
5pub(crate) fn strip_slash_suffix(path: &str) -> &str {
6    path.strip_suffix('/').unwrap_or(path)
7}
8
9static BASE_URL_LOADED: std::sync::Once = std::sync::Once::new();
10thread_local! {
11    static BASE_URL: RefCell<Option<String>> = const { RefCell::new(None) };
12}
13
14// This exists so we can cache the base url. It costs us a `to_string` call instead of a DOM API
15// call. Considering base urls are generally short, it *should* be less expensive.
16pub fn base_url() -> Option<String> {
17    BASE_URL_LOADED.call_once(|| {
18        BASE_URL.with(|val| {
19            *val.borrow_mut() = fetch_base_url();
20        })
21    });
22    BASE_URL.with(|it| it.borrow().as_ref().map(|it| it.to_string()))
23}
24
25pub fn fetch_base_url() -> Option<String> {
26    match gloo::utils::document().query_selector("base[href]") {
27        Ok(Some(base)) => {
28            let base = base.unchecked_into::<web_sys::HtmlBaseElement>().href();
29
30            let url = web_sys::Url::new(&base).unwrap();
31            let base = url.pathname();
32
33            let base = if base != "/" {
34                strip_slash_suffix(&base)
35            } else {
36                return None;
37            };
38
39            Some(base.to_string())
40        }
41        _ => None,
42    }
43}
44
45#[cfg(all(target_arch = "wasm32", not(target_os = "wasi")))]
46pub fn compose_path(pathname: &str, query: &str) -> Option<String> {
47    gloo::utils::window()
48        .location()
49        .href()
50        .ok()
51        .and_then(|base| web_sys::Url::new_with_base(pathname, &base).ok())
52        .map(|url| {
53            url.set_search(query);
54            format!("{}{}", url.pathname(), url.search())
55        })
56}
57
58#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
59pub fn compose_path(pathname: &str, query: &str) -> Option<String> {
60    let query = query.trim();
61
62    if !query.is_empty() {
63        Some(format!("{pathname}?{query}"))
64    } else {
65        Some(pathname.to_owned())
66    }
67}
68
69// TODO: remove the cfg after wasm-bindgen-test stops emitting the function unconditionally
70#[cfg(all(
71    test,
72    target_arch = "wasm32",
73    any(target_os = "unknown", target_os = "none")
74))]
75mod tests {
76    use gloo::utils::document;
77    use wasm_bindgen_test::wasm_bindgen_test as test;
78    use yew_router::prelude::*;
79    use yew_router::utils::*;
80
81    wasm_bindgen_test::wasm_bindgen_test_configure!(run_in_browser);
82
83    #[derive(Debug, Clone, Copy, PartialEq, Routable)]
84    enum Routes {
85        #[at("/")]
86        Home,
87        #[at("/no")]
88        No,
89        #[at("/404")]
90        NotFound,
91    }
92
93    #[test]
94    fn test_base_url() {
95        document().head().unwrap().set_inner_html(r#""#);
96
97        assert_eq!(fetch_base_url(), None);
98
99        document()
100            .head()
101            .unwrap()
102            .set_inner_html(r#"<base href="/base/">"#);
103        assert_eq!(fetch_base_url(), Some("/base".to_string()));
104
105        document()
106            .head()
107            .unwrap()
108            .set_inner_html(r#"<base href="/base">"#);
109        assert_eq!(fetch_base_url(), Some("/base".to_string()));
110    }
111
112    #[test]
113    fn test_compose_path() {
114        assert_eq!(compose_path("/home", ""), Some("/home".to_string()));
115        assert_eq!(
116            compose_path("/path/to", "foo=bar"),
117            Some("/path/to?foo=bar".to_string())
118        );
119        assert_eq!(
120            compose_path("/events", "from=2019&to=2021"),
121            Some("/events?from=2019&to=2021".to_string())
122        );
123    }
124}
yew_router/utils.rs

yew_router/
utils.rs
