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