-
Notifications
You must be signed in to change notification settings - Fork 87
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Showing
4 changed files
with
100 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,80 @@ | ||
| use wasm_bindgen::{closure::Closure, JsCast, JsValue}; | ||
| use web_sys::{ErrorEvent, MessageEvent, WebSocket}; | ||
|
|
||
| /// Connects to the reload server if it's online. | ||
| pub fn connect_to_reload_server() { | ||
| // Get the host and port | ||
| let host = get_window_var("__PERSEUS_RELOAD_SERVER_HOST"); | ||
| let port = get_window_var("__PERSEUS_RELOAD_SERVER_PORT"); | ||
| let (host, port) = match (host, port) { | ||
| (Some(host), Some(port)) => (host, port), | ||
| // If either the host or port weren't set, the server almost certainly isn't online, so we won't connect | ||
| _ => return, | ||
| }; | ||
|
|
||
| // Connect to the server (it's expected that the host does not include a protocol) | ||
| let ws = match WebSocket::new(&format!("ws://{}:{}/receive", host, port)) { | ||
| Ok(ws) => ws, | ||
| Err(err) => return log(&format!("Connection failed: {:?}.", err)), | ||
| }; | ||
| // This is apparently more efficient for small bianry messages | ||
| ws.set_binary_type(web_sys::BinaryType::Arraybuffer); | ||
|
|
||
| // Set up a message handler | ||
| let onmessage_callback = Closure::wrap(Box::new(move |_| { | ||
| // With this server, if we receive any message it will be telling us to reload, so we'll do so | ||
| wasm_bindgen_futures::spawn_local(async move { | ||
| // TODO If we're using HSR, freeze the state to IndexedDB | ||
| #[cfg(feature = "hsr")] | ||
| todo!(); | ||
| // Force reload the page, getting all resources from the sevrer again (to get the new code) | ||
| log("Reloading..."); | ||
| match web_sys::window() | ||
| .unwrap() | ||
| .location() | ||
| .reload_with_forceget(true) | ||
| { | ||
| Ok(_) => (), | ||
| Err(err) => log(&format!("Reloading failed: {:?}.", err)), | ||
| }; | ||
| // We shouldn't ever get here unless there was an error, the entire page will be fully reloaded | ||
| }); | ||
| }) as Box<dyn FnMut(MessageEvent)>); | ||
| ws.set_onmessage(Some(onmessage_callback.as_ref().unchecked_ref())); | ||
| // To keep the closure alive, we need to forget about it | ||
| onmessage_callback.forget(); | ||
|
|
||
| // We should log to the console about errors from the server | ||
| let onerror_callback = Closure::wrap(Box::new(move |err: ErrorEvent| { | ||
| log(&format!("Error: {:?}.", err)); | ||
| }) as Box<dyn FnMut(ErrorEvent)>); | ||
| ws.set_onerror(Some(onerror_callback.as_ref().unchecked_ref())); | ||
| onerror_callback.forget(); | ||
|
|
||
| // We'll just log when the connection is successfully established for informational purposes | ||
| let onopen_callback = Closure::wrap(Box::new(move |_| { | ||
| log("Connected."); | ||
| }) as Box<dyn FnMut(JsValue)>); | ||
| ws.set_onopen(Some(onopen_callback.as_ref().unchecked_ref())); | ||
| onopen_callback.forget(); | ||
| } | ||
|
|
||
| fn get_window_var(name: &str) -> Option<String> { | ||
| let val_opt = web_sys::window().unwrap().get(name); | ||
| let js_obj = match val_opt { | ||
| Some(js_obj) => js_obj, | ||
| None => return None, | ||
| }; | ||
| // The object should only actually contain the string value that was injected | ||
| let state_str = match js_obj.as_string() { | ||
| Some(state_str) => state_str, | ||
| None => return None, | ||
| }; | ||
| // On the server-side, we encode a `None` value directly (otherwise it will be some convoluted stringified JSON) | ||
| Some(state_str) | ||
| } | ||
|
|
||
| /// An internal function for logging data for development reloading. | ||
| fn log(msg: &str) { | ||
| web_sys::console::log_1(&JsValue::from("[Live Reload Server]: ".to_string() + msg)); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters