1mod server;
2mod sparql;
3
4use futures::lock::Mutex;
5use log::error;
6use server::{handle_message, Server};
7use std::panic;
8use std::rc::Rc;
9use wasm_bindgen::prelude::*;
10use wasm_bindgen_futures::JsFuture;
11use web_sys::js_sys;
12
13pub use server::format_raw;
14
15fn send_message(writer: &web_sys::WritableStreamDefaultWriter, message: String) {
16 let _future = JsFuture::from(writer.write_with_chunk(&message.into()));
17}
18
19#[wasm_bindgen]
20pub fn init_language_server(writer: web_sys::WritableStreamDefaultWriter) -> Server {
21 wasm_logger::init(wasm_logger::Config::default());
22 panic::set_hook(Box::new(|info| {
23 let msg = info.to_string();
24 web_sys::console::error_1(&msg.into());
25 let _ = js_sys::Function::new_with_args("msg", "self.postMessage({type:'crash'});")
26 .call0(&JsValue::NULL);
27 }));
28 Server::new(move |message| send_message(&writer, message))
29}
30
31async fn read_message(
32 reader: &web_sys::ReadableStreamDefaultReader,
33) -> Result<(String, bool), String> {
34 match JsFuture::from(reader.read()).await {
35 Ok(js_object) => {
36 let value = js_sys::Reflect::get(&js_object, &"value".into())
37 .map_err(|_| "\"value\" property not present in message")?
38 .as_string()
39 .ok_or("\"value\" is not a string")?;
40 let done = js_sys::Reflect::get(&js_object, &"done".into())
41 .map_err(|_| "\"done\" property not present in message")?
42 .as_bool()
43 .ok_or("\"done\" is not a boolean")?;
44 Ok((value, done))
45 }
46 Err(_) => Err("Error while reading from input-stream".to_string()),
47 }
48}
49
50#[wasm_bindgen]
51pub async fn listen(server: Server, reader: web_sys::ReadableStreamDefaultReader) {
52 let server_rc = Rc::new(Mutex::new(server));
53 loop {
54 match read_message(&reader).await {
55 Ok((value, done)) => {
56 handle_message(server_rc.clone(), value).await;
57 if done {
58 break;
59 }
60 }
61 Err(e) => error!("{}", e),
62 }
63 }
64}