/* * 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 async_std::task; use once_cell::sync::Lazy; // #[cfg(target_arch = "wasm32")] // use js_sys::Reflect; use async_std::stream::StreamExt; #[cfg(target_arch = "wasm32")] use js_sys::Uint8Array; #[cfg(target_arch = "wasm32")] use ng_client_ws::remote_ws_wasm::ConnectionWebSocket; use ng_net::broker::*; use ng_net::connection::{ClientConfig, StartConfig}; use ng_net::types::{ BootstrapContent, BootstrapContentV0, ClientId, ClientInfo, ClientInfoV0, ClientType, CreateAccountBSP, IP, }; use ng_net::utils::{decode_invitation_string, spawn_and_log_error, Receiver, ResultSend, Sender}; #[cfg(target_arch = "wasm32")] use ng_net::utils::{retrieve_local_bootstrap, retrieve_local_url}; use ng_wallet::types::*; use ng_wallet::*; use nextgraph::local_broker::*; use ng_net::WS_PORT; use ng_repo::errors::NgError; use ng_repo::log::*; use ng_repo::types::*; use ng_repo::utils::generate_keypair; use serde::{Deserialize, Serialize}; use serde_json::json; use std::collections::HashMap; use std::net::IpAddr; use std::str::FromStr; use std::sync::Arc; use wasm_bindgen::prelude::*; #[cfg(target_arch = "wasm32")] use wasm_bindgen_futures::{future_to_promise, JsFuture}; #[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 mut 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_in_memory(&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; } #[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; } #[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")] 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)), 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 mut 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 in_memory: bool, ) -> Result { let mut 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 mut 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(target_arch = "wasm32")] #[wasm_bindgen] pub fn test_create_wallet() -> JsValue { let pin = [5, 2, 9, 1]; let r = CreateWalletV0::new( vec![50u8; 20], " know yourself ".to_string(), pin, 9, false, false, BootstrapContentV0::new_empty(), None, None, ); serde_wasm_bindgen::to_value(&r).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()); let client_info = client_info(); log_debug!("{:?}", client_info); } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn doc_get_file_from_store_with_object_ref( nuri: String, obj_ref_js: JsValue, ) -> Result { let obj_ref = serde_wasm_bindgen::from_value::(obj_ref_js).unwrap(); log_debug!("doc_get_file {} {:?}", nuri, obj_ref.id,); // let vec: Vec = vec![2; 10]; // let view = unsafe { Uint8Array::view(&vec) }; // let x = JsValue::from(Uint8Array::new(view.as_ref())); // let ret = ObjectContent::File(File::V0(FileV0 { // content_type: "text/plain".to_string(), // metadata: vec![], // content: vec![45; 20], // })); let obj_content = BROKER .write() .await .get_object_from_store_with_object_ref(nuri, obj_ref) .await .map_err(|e| e.to_string())?; Ok(serde_wasm_bindgen::to_value(&obj_content).unwrap()) } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn doc_sync_branch(anuri: String, callback: &js_sys::Function) -> JsValue { let vec: Vec = vec![2; 10]; let view = unsafe { Uint8Array::view(&vec) }; let x = JsValue::from(Uint8Array::new(view.as_ref())); let mut reader; let mut sender; { (reader, sender) = BROKER.write().await.doc_sync_branch(anuri.clone()).await; } async fn inner_task( mut reader: Receiver, anuri: String, callback: js_sys::Function, ) -> ResultSend<()> { while let Some(commit) = reader.next().await { let xx = serde_wasm_bindgen::to_value(&commit).unwrap(); //let xx = JsValue::from(json!(commit).to_string()); //let _ = callback.call1(&this, &xx); let this = JsValue::null(); let jsval: JsValue = callback.call1(&this, &xx).unwrap(); let promise_res: Result = jsval.dyn_into(); match promise_res { Ok(promise) => { JsFuture::from(promise).await; } Err(_) => {} } } log_debug!("END OF LOOP"); Ok(()) } spawn_and_log_error(inner_task(reader, anuri, callback.clone())); let cb = Closure::once(move || { log_debug!("close channel"); sender.close_channel() }); //Closure::wrap(Box::new(move |sender| sender.close_channel()) as Box)>); let ret = cb.as_ref().clone(); cb.forget(); return ret; } #[cfg(target_arch = "wasm32")] #[wasm_bindgen] pub async fn disconnections_subscribe(callback: &js_sys::Function) -> Result { let vec: Vec = vec![2; 10]; let view = unsafe { Uint8Array::view(&vec) }; let x = JsValue::from(Uint8Array::new(view.as_ref())); let mut reader; { reader = BROKER .write() .await .take_disconnections_receiver() .ok_or(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 xx = serde_wasm_bindgen::to_value(&user_id).unwrap(); let jsval: JsValue = callback.call1(&this, &xx).unwrap(); let promise_res: Result = jsval.dyn_into(); match promise_res { Ok(promise) => { JsFuture::from(promise).await; } Err(_) => {} } } 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); 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<()> { let server_key: PubKey = "X0nh-gOTGKSx0yL0LYJviOWRNacyqIzjQW_LKdK6opU".try_into()?; log_debug!("server_key:{}", server_key); //let keys = ng_net::utils::gen_dh_keys(); //let pub_key = PubKey::Ed25519PubKey(keys.1); let keys = generate_keypair(); let x_from_ed = keys.1.to_dh_from_ed(); log_debug!("Pub from X {}", x_from_ed); let (client_priv, client) = generate_keypair(); let (user_priv, user) = generate_keypair(); log_debug!("start connecting"); let res = BROKER .write() .await .connect( Arc::new(Box::new(ConnectionWebSocket {})), keys.0, keys.1, server_key, StartConfig::Client(ClientConfig { url: format!("ws://127.0.0.1:{}", WS_PORT), name: None, user_priv, client_priv, info: ClientInfo::V0(client_info_()), registration: None, }), ) .await; log_debug!("broker.connect : {:?}", res); if res.is_err() { return Ok(()); //panic!("Cannot connect"); } BROKER.read().await.print_status(); //res.expect_throw("assume the connection succeeds"); async fn timer_close(remote_peer_id: DirectPeerId, user: Option) -> ResultSend<()> { async move { sleep!(std::time::Duration::from_secs(3)); log_debug!("timeout"); BROKER .write() .await .close_peer_connection(&remote_peer_id, user) .await; } .await; Ok(()) } spawn_and_log_error(timer_close(server_key, Some(user))); //Broker::graceful_shutdown().await; Broker::join_shutdown_with_timeout(std::time::Duration::from_secs(5)).await; 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 mut 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; } }