/* * Copyright (c) 2022-2024 Niko Bonnieure, Par le Peuple, NextGraph.org developers * All rights reserved. * Licensed under the Apache License, Version 2.0 * * or the MIT license , * at your option. All files in the project carrying such * notice may not be copied, modified, or distributed except * according to those terms. */ use std::collections::HashMap; use std::net::IpAddr; use std::str::FromStr; use std::sync::Arc; use once_cell::sync::Lazy; use serde::{Deserialize, Serialize}; #[allow(unused_imports)] use serde_json::json; // #[cfg(target_arch = "wasm32")] // use js_sys::Reflect; use async_std::stream::StreamExt; use wasm_bindgen::prelude::*; #[cfg(target_arch = "wasm32")] use wasm_bindgen_futures::JsFuture; use ng_repo::errors::NgError; use ng_repo::log::*; use ng_repo::types::*; use ng_net::broker::*; use ng_net::types::{ClientInfo, ClientInfoV0, ClientType, CreateAccountBSP, IP}; use ng_net::utils::{decode_invitation_string, spawn_and_log_error, Receiver, ResultSend}; #[cfg(target_arch = "wasm32")] use ng_net::utils::{retrieve_local_bootstrap, retrieve_local_url}; use ng_net::WS_PORT; #[cfg(target_arch = "wasm32")] use ng_client_ws::remote_ws_wasm::ConnectionWebSocket; use ng_wallet::types::*; use ng_wallet::*; use nextgraph::local_broker::*; use nextgraph::verifier::types::*; #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn get_local_bootstrap(location: String, invite: JsValue) -> JsValue { let res = retrieve_local_bootstrap(location, invite.as_string(), false).await; if res.is_some() { serde_wasm_bindgen::to_value(&res.unwrap()).unwrap() } else { JsValue::FALSE } } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn get_local_bootstrap_with_public(location: String, invite: JsValue) -> JsValue { let res = retrieve_local_bootstrap(location, invite.as_string(), true).await; if res.is_some() { serde_wasm_bindgen::to_value(&res.unwrap()).unwrap() } else { JsValue::FALSE } } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn decode_invitation(invite: String) -> JsValue { let res = decode_invitation_string(invite); if res.is_some() { serde_wasm_bindgen::to_value(&res.unwrap()).unwrap() } else { JsValue::FALSE } } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn get_local_url(location: String) -> JsValue { let res = retrieve_local_url(location).await; if res.is_some() { serde_wasm_bindgen::to_value(&res.unwrap()).unwrap() } else { JsValue::FALSE } } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn get_ngone_url_of_invitation(invitation_string: String) -> JsValue { let res = decode_invitation_string(invitation_string); if res.is_some() { serde_wasm_bindgen::to_value(&res.unwrap().get_urls()[0]).unwrap() } else { JsValue::FALSE } } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub fn wallet_gen_shuffle_for_pazzle_opening(pazzle_length: u8) -> JsValue { let res = gen_shuffle_for_pazzle_opening(pazzle_length); serde_wasm_bindgen::to_value(&res).unwrap() } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub fn wallet_gen_shuffle_for_pin() -> Vec { gen_shuffle_for_pin() } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub fn wallet_open_with_pazzle( js_wallet: JsValue, pazzle: Vec, js_pin: JsValue, ) -> Result { let encrypted_wallet = serde_wasm_bindgen::from_value::(js_wallet) .map_err(|_| "Deserialization error of wallet")?; let pin = serde_wasm_bindgen::from_value::<[u8; 4]>(js_pin) .map_err(|_| "Deserialization error of pin")?; let res = nextgraph::local_broker::wallet_open_with_pazzle(&encrypted_wallet, pazzle, pin); match res { Ok(r) => Ok(r .serialize(&serde_wasm_bindgen::Serializer::new().serialize_maps_as_objects(true)) .unwrap()), Err(e) => Err(e.to_string().into()), } } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub fn wallet_update(js_wallet_id: JsValue, js_operations: JsValue) -> Result { let _wallet = serde_wasm_bindgen::from_value::(js_wallet_id) .map_err(|_| "Deserialization error of WalletId")?; let _operations = serde_wasm_bindgen::from_value::>(js_operations) .map_err(|_| "Deserialization error of operations")?; unimplemented!(); // match res { // Ok(r) => Ok(serde_wasm_bindgen::to_value(&r).unwrap()), // Err(e) => Err(e.to_string().into()), // } } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn get_wallets() -> Result { init_local_broker_with_lazy(&INIT_LOCAL_BROKER).await; let res = wallets_get_all().await.map_err(|e| { log_err!("{}", e.to_string()); }); if res.is_ok() { return Ok(serde_wasm_bindgen::to_value(&res.unwrap()).unwrap()); } Ok(JsValue::UNDEFINED) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn session_start(wallet_name: String, user_js: JsValue) -> Result { let user_id = serde_wasm_bindgen::from_value::(user_js) .map_err(|_| "Deserialization error of user_id")?; let config = SessionConfig::new_save(&user_id, &wallet_name); let res = nextgraph::local_broker::session_start(config) .await .map_err(|e: NgError| e.to_string())?; Ok(serde_wasm_bindgen::to_value(&res).unwrap()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn session_start_remote( wallet_name: String, user_js: JsValue, peer_id_js: JsValue, ) -> Result { let user_id = serde_wasm_bindgen::from_value::(user_js) .map_err(|_| "Deserialization error of user_id")?; let peer_id = serde_wasm_bindgen::from_value::>(peer_id_js) .map_err(|_| "Deserialization error of peer_id")?; let config = SessionConfig::new_remote(&user_id, &wallet_name, peer_id); let res = nextgraph::local_broker::session_start(config) .await .map_err(|e: NgError| e.to_string())?; Ok(serde_wasm_bindgen::to_value(&res).unwrap()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn wallets_reload() -> Result<(), String> { init_local_broker_with_lazy(&INIT_LOCAL_BROKER).await; nextgraph::local_broker::wallets_reload() .await .map_err(|e: NgError| e.to_string()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn add_in_memory_wallet(lws_js: JsValue) -> Result<(), String> { let lws = serde_wasm_bindgen::from_value::(lws_js) .map_err(|_| "Deserialization error of lws")?; if !lws.in_memory { return Err("This is not an in memory wallet".to_string()); } nextgraph::local_broker::wallet_add(lws) .await .map_err(|e: NgError| e.to_string()) } #[cfg(not(wasmpack_target = "nodejs"))] #[wasm_bindgen(module = "/js/browser.js")] extern "C" { fn session_save(key: String, value: String) -> Option; fn session_get(key: String) -> Option; fn session_remove(key: String); fn local_save(key: String, value: String) -> Option; fn local_get(key: String) -> Option; fn is_browser() -> bool; fn storage_clear(); } #[cfg(wasmpack_target = "nodejs")] #[wasm_bindgen(module = "/js/node.js")] extern "C" { fn session_save(key: String, value: String) -> Option; fn session_get(key: String) -> Option; fn session_remove(key: String); fn local_save(key: String, value: String) -> Option; fn local_get(key: String) -> Option; fn is_browser() -> bool; fn storage_clear(); } #[cfg(target_arch = "wasm32")] fn local_read(key: String) -> Result { local_get(key).ok_or(NgError::JsStorageReadError) } #[cfg(target_arch = "wasm32")] fn local_write(key: String, value: String) -> Result<(), NgError> { match local_save(key, value) { Some(err) => Err(NgError::JsStorageWriteError(err)), None => Ok(()), } } #[cfg(target_arch = "wasm32")] fn session_read(key: String) -> Result { session_get(key).ok_or(NgError::JsStorageReadError) } #[cfg(target_arch = "wasm32")] fn session_write(key: String, value: String) -> Result<(), NgError> { match session_save(key, value) { Some(err) => Err(NgError::JsStorageWriteError(err)), None => Ok(()), } } #[cfg(target_arch = "wasm32")] fn session_del(key: String) -> Result<(), NgError> { session_remove(key); Ok(()) } #[cfg(target_arch = "wasm32")] fn clear() { storage_clear(); } #[cfg(target_arch = "wasm32")] static INIT_LOCAL_BROKER: Lazy> = Lazy::new(|| { Box::new(|| { LocalBrokerConfig::JsStorage(JsStorageConfig { local_read: Box::new(local_read), local_write: Box::new(local_write), session_read: Arc::new(Box::new(session_read)), session_write: Arc::new(Box::new(session_write)), session_del: Arc::new(Box::new(session_del)), clear: Arc::new(Box::new(clear)), is_browser: is_browser(), }) }) }); #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn wallet_create(js_params: JsValue) -> Result { init_local_broker_with_lazy(&INIT_LOCAL_BROKER).await; let mut params = serde_wasm_bindgen::from_value::(js_params) .map_err(|_| "Deserialization error of args")?; params.result_with_wallet_file = true; let res = nextgraph::local_broker::wallet_create_v0(params).await; match res { Ok(r) => Ok(serde_wasm_bindgen::to_value(&r).unwrap()), Err(e) => Err(e.to_string().into()), } } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn wallet_get_file(wallet_name: String) -> Result { init_local_broker_with_lazy(&INIT_LOCAL_BROKER).await; let res = nextgraph::local_broker::wallet_get_file(&wallet_name).await; match res { Ok(r) => Ok(serde_wasm_bindgen::to_value(&serde_bytes::ByteBuf::from(r)).unwrap()), Err(e) => Err(e.to_string().into()), } } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn wallet_read_file(js_file: JsValue) -> Result { init_local_broker_with_lazy(&INIT_LOCAL_BROKER).await; let file = serde_wasm_bindgen::from_value::(js_file) .map_err(|_| "Deserialization error of file".to_string())?; let wallet = nextgraph::local_broker::wallet_read_file(file.into_vec()) .await .map_err(|e: NgError| e.to_string())?; Ok(serde_wasm_bindgen::to_value(&wallet).unwrap()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn wallet_was_opened( js_opened_wallet: JsValue, //SensitiveWallet ) -> Result { let opened_wallet = serde_wasm_bindgen::from_value::(js_opened_wallet) .map_err(|_| "Deserialization error of SensitiveWallet".to_string())?; let client = nextgraph::local_broker::wallet_was_opened(opened_wallet) .await .map_err(|e: NgError| e.to_string())?; Ok(serde_wasm_bindgen::to_value(&client).unwrap()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn wallet_import( js_encrypted_wallet: JsValue, //Wallet, js_opened_wallet: JsValue, //SensitiveWallet in_memory: bool, ) -> Result { let encrypted_wallet = serde_wasm_bindgen::from_value::(js_encrypted_wallet) .map_err(|_| "Deserialization error of Wallet".to_string())?; let opened_wallet = serde_wasm_bindgen::from_value::(js_opened_wallet) .map_err(|_| "Deserialization error of SensitiveWallet".to_string())?; let client = nextgraph::local_broker::wallet_import(encrypted_wallet, opened_wallet, in_memory) .await .map_err(|e: NgError| e.to_string())?; Ok(serde_wasm_bindgen::to_value(&client).unwrap()) } #[cfg(wasmpack_target = "nodejs")] #[wasm_bindgen(module = "/js/node.js")] extern "C" { fn client_details() -> String; } #[cfg(wasmpack_target = "nodejs")] #[wasm_bindgen(module = "/js/node.js")] extern "C" { fn version() -> String; } #[cfg(wasmpack_target = "nodejs")] #[wasm_bindgen] pub fn client_info() -> JsValue { let res = ClientInfo::V0(client_info_()); //res serde_wasm_bindgen::to_value(&res).unwrap() } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub fn encode_create_account(payload: JsValue) -> JsValue { //log_debug!("{:?}", payload); let create_account = serde_wasm_bindgen::from_value::(payload).unwrap(); //log_debug!("create_account {:?}", create_account); let res = create_account.encode(); //log_debug!("res {:?}", res); serde_wasm_bindgen::to_value(&res).unwrap() } #[cfg(not(wasmpack_target = "nodejs"))] #[wasm_bindgen(module = "/js/browser.js")] extern "C" { fn client_details() -> String; } #[cfg(not(wasmpack_target = "nodejs"))] #[wasm_bindgen(module = "/js/bowser.js")] extern "C" { type Bowser; #[wasm_bindgen(static_method_of = Bowser)] fn parse(UA: String) -> JsValue; } #[cfg(not(wasmpack_target = "nodejs"))] #[wasm_bindgen(module = "/js/browser.js")] extern "C" { fn client_details2(val: JsValue, version: String) -> String; } #[cfg(all(not(wasmpack_target = "nodejs"), target_arch = "wasm32"))] pub fn client_info_() -> ClientInfoV0 { let ua = client_details(); let bowser = Bowser::parse(ua); //log_debug!("{:?}", bowser); let details_string = client_details2(bowser, env!("CARGO_PKG_VERSION").to_string()); let res = ClientInfoV0 { client_type: ClientType::Web, details: details_string, version: "".to_string(), timestamp_install: 0, timestamp_updated: 0, }; res //serde_wasm_bindgen::to_value(&res).unwrap() } #[cfg(all(wasmpack_target = "nodejs", target_arch = "wasm32"))] pub fn client_info_() -> ClientInfoV0 { let res = ClientInfoV0 { client_type: ClientType::NodeService, details: client_details(), version: version(), timestamp_install: 0, timestamp_updated: 0, }; res //serde_wasm_bindgen::to_value(&res).unwrap() } #[cfg(all(not(wasmpack_target = "nodejs"), target_arch = "wasm32"))] #[wasm_bindgen] pub fn client_info() -> JsValue { let res = ClientInfo::V0(client_info_()); serde_wasm_bindgen::to_value(&res).unwrap() } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn test() { init_local_broker_with_lazy(&INIT_LOCAL_BROKER).await; //log_debug!("test is {}", BROKER.read().await.test()); #[cfg(debug_assertions)] let client_info = client_info(); log_debug!("{:?}", client_info); } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn app_request_stream( js_session_id: JsValue, js_request: JsValue, callback: &js_sys::Function, ) -> Result { let session_id: u64 = serde_wasm_bindgen::from_value::(js_session_id) .map_err(|_| "Deserialization error of session_id".to_string())?; let request = serde_wasm_bindgen::from_value::(js_request) .map_err(|_| "Deserialization error of AppRequest".to_string())?; let (reader, cancel) = nextgraph::local_broker::app_request_stream(session_id, request) .await .map_err(|e: NgError| e.to_string())?; async fn inner_task( mut reader: Receiver, callback: js_sys::Function, ) -> ResultSend<()> { while let Some(app_response) = reader.next().await { let response_js = serde_wasm_bindgen::to_value(&app_response).unwrap(); let this = JsValue::null(); match callback.call1(&this, &response_js) { Ok(jsval) => { let promise_res: Result = jsval.dyn_into(); match promise_res { Ok(promise) => { let _ = JsFuture::from(promise).await; } Err(_) => {} } } Err(e) => { log_err!("JS callback for app_request_stream failed with {:?}", e); } } } //log_info!("END OF LOOP"); Ok(()) } spawn_and_log_error(inner_task(reader, callback.clone())); let cb = Closure::once(move || { log_info!("cancelling"); //sender.close_channel() cancel(); }); //Closure::wrap(Box::new(move |sender| sender.close_channel()) as Box)>); let ret = cb.as_ref().clone(); cb.forget(); Ok(ret) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn app_request(js_session_id: JsValue, js_request: JsValue) -> Result { let session_id: u64 = serde_wasm_bindgen::from_value::(js_session_id) .map_err(|_| "Deserialization error of session_id".to_string())?; let request = serde_wasm_bindgen::from_value::(js_request) .map_err(|_| "Deserialization error of AppRequest".to_string())?; let response = nextgraph::local_broker::app_request(session_id, request) .await .map_err(|e: NgError| e.to_string())?; Ok(serde_wasm_bindgen::to_value(&response).unwrap()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn upload_chunk( js_session_id: JsValue, js_upload_id: JsValue, js_chunk: JsValue, js_nuri: JsValue, ) -> Result { //log_debug!("upload_chunk {:?}", js_nuri); let session_id: u64 = serde_wasm_bindgen::from_value::(js_session_id) .map_err(|_| "Deserialization error of session_id".to_string())?; let upload_id: u32 = serde_wasm_bindgen::from_value::(js_upload_id) .map_err(|_| "Deserialization error of upload_id".to_string())?; let chunk: serde_bytes::ByteBuf = serde_wasm_bindgen::from_value::(js_chunk) .map_err(|_| "Deserialization error of chunk".to_string())?; let nuri: NuriV0 = serde_wasm_bindgen::from_value::(js_nuri) .map_err(|_| "Deserialization error of nuri".to_string())?; let request = AppRequest::V0(AppRequestV0 { command: AppRequestCommandV0::FilePut, nuri, payload: Some(AppRequestPayload::V0( AppRequestPayloadV0::RandomAccessFilePutChunk((upload_id, chunk)), )), }); let response = nextgraph::local_broker::app_request(session_id, request) .await .map_err(|e: NgError| e.to_string())?; Ok(serde_wasm_bindgen::to_value(&response).unwrap()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn doc_fetch_private_subscribe() -> Result { let request = AppRequest::V0(AppRequestV0 { command: AppRequestCommandV0::Fetch(AppFetchContentV0::get_or_subscribe(true)), nuri: NuriV0::new_private_store_target(), payload: None, }); Ok(serde_wasm_bindgen::to_value(&request).unwrap()) } // #[cfg(target_arch = "wasm32")] // #[wasm_bindgen] // pub async fn get_readcap() -> Result { // let request = ObjectRef::nil(); // Ok(serde_wasm_bindgen::to_value(&request).unwrap()) // } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn disconnections_subscribe(callback: &js_sys::Function) -> Result { init_local_broker_with_lazy(&INIT_LOCAL_BROKER).await; let reader = nextgraph::local_broker::take_disconnections_receiver() .await .map_err(|_e: NgError| false)?; async fn inner_task( mut reader: Receiver, callback: js_sys::Function, ) -> ResultSend<()> { while let Some(user_id) = reader.next().await { let this = JsValue::null(); let user_id_js = serde_wasm_bindgen::to_value(&user_id).unwrap(); match callback.call1(&this, &user_id_js) { Ok(jsval) => { let promise_res: Result = jsval.dyn_into(); match promise_res { Ok(promise) => { let _ = JsFuture::from(promise).await; } Err(_) => {} } } Err(e) => { log_err!( "JS callback for disconnections_subscribe failed with {:?}", e ); } } } log_debug!("END OF disconnections reader"); Ok(()) } spawn_and_log_error(inner_task(reader, callback.clone())); Ok(true.into()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn probe() { let _res = BROKER .write() .await .probe( Box::new(ConnectionWebSocket {}), IP::try_from(&IpAddr::from_str("127.0.0.1").unwrap()).unwrap(), WS_PORT, ) .await; log_debug!("broker.probe : {:?}", _res); let _ = Broker::join_shutdown_with_timeout(std::time::Duration::from_secs(5)).await; } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn start() { async fn inner_task() -> ResultSend<()> { Ok(()) } spawn_and_log_error(inner_task()).await; } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn session_stop(user_id_js: JsValue) -> Result<(), String> { let user_id = serde_wasm_bindgen::from_value::(user_id_js) .map_err(|_| "serde error on user_id")?; nextgraph::local_broker::session_stop(&user_id) .await .map_err(|e: NgError| e.to_string()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn user_disconnect(user_id_js: JsValue) -> Result<(), String> { let user_id = serde_wasm_bindgen::from_value::(user_id_js) .map_err(|_| "serde error on user_id")?; nextgraph::local_broker::user_disconnect(&user_id) .await .map_err(|e: NgError| e.to_string()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn wallet_close(wallet_name: String) -> Result<(), String> { nextgraph::local_broker::wallet_close(&wallet_name) .await .map_err(|e: NgError| e.to_string()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn user_connect( client_info_js: JsValue, user_id_js: JsValue, location: Option, ) -> Result { let info = serde_wasm_bindgen::from_value::(client_info_js) .map_err(|_| "serde error on info")?; let user_id = serde_wasm_bindgen::from_value::(user_id_js) .map_err(|_| "serde error on user_id")?; #[derive(Serialize, Deserialize)] struct ConnectionInfo { pub server_id: String, pub server_ip: String, pub error: Option, #[serde(with = "serde_wasm_bindgen::preserve")] pub since: js_sys::Date, } let mut opened_connections: HashMap = HashMap::new(); let results = nextgraph::local_broker::user_connect_with_device_info(info, &user_id, location) .await .map_err(|e: NgError| e.to_string())?; log_debug!("{:?}", results); for result in results { let date = js_sys::Date::new_0(); date.set_time(result.4); opened_connections.insert( result.0, ConnectionInfo { server_id: result.1, server_ip: result.2, error: result.3, since: date, }, ); } //BROKER.read().await.print_status(); Ok(opened_connections .serialize(&serde_wasm_bindgen::Serializer::new().serialize_maps_as_objects(true)) .unwrap()) } #[cfg(target_arch = "wasm32")] #[cfg(test)] mod test { use wasm_bindgen_test::*; wasm_bindgen_test_configure!(run_in_browser); use crate::probe; use crate::start; #[wasm_bindgen_test] pub async fn test_connection() { //probe().await; start().await; } }