1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

extern crate alloc;
use alloc::rc::Rc;

use miden_client::{
    auth::StoreAuthenticator, rpc::WebTonicRpcClient, store::web_store::WebStore, Client,
};
use miden_objects::{crypto::rand::RpoRandomCoin, Felt};
use rand::{rngs::StdRng, Rng, SeedableRng};
use wasm_bindgen::prelude::*;

pub mod account;
pub mod export;
pub mod import;
pub mod models;
pub mod new_account;
pub mod new_transactions;
pub mod notes;
pub mod sync;
pub mod tags;
pub mod transactions;

#[wasm_bindgen]
pub struct WebClient {
    inner: Option<
        Client<
            WebTonicRpcClient,
            RpoRandomCoin,
            WebStore,
            StoreAuthenticator<RpoRandomCoin, WebStore>,
        >,
    >,
}

impl Default for WebClient {
    fn default() -> Self {
        Self::new()
    }
}

#[wasm_bindgen]
impl WebClient {
    #[wasm_bindgen(constructor)]
    pub fn new() -> Self {
        WebClient { inner: None }
    }

    pub(crate) fn get_mut_inner(
        &mut self,
    ) -> Option<
        &mut Client<
            WebTonicRpcClient,
            RpoRandomCoin,
            WebStore,
            StoreAuthenticator<RpoRandomCoin, WebStore>,
        >,
    > {
        self.inner.as_mut()
    }

    pub async fn create_client(&mut self, node_url: Option<String>) -> Result<JsValue, JsValue> {
        let mut rng = StdRng::from_entropy();
        let coin_seed: [u64; 4] = rng.gen();

        let rng = RpoRandomCoin::new(coin_seed.map(Felt::new));
        let web_store: WebStore = WebStore::new()
            .await
            .map_err(|_| JsValue::from_str("Failed to initialize WebStore"))?;
        let web_store = Rc::new(web_store);
        let authenticator: StoreAuthenticator<RpoRandomCoin, WebStore> =
            StoreAuthenticator::new_with_rng(web_store.clone(), rng);
        let web_rpc_client = WebTonicRpcClient::new(
            &node_url.unwrap_or_else(|| "http://18.203.155.106:57291".to_string()),
        );

        self.inner = Some(Client::new(web_rpc_client, rng, web_store, authenticator, false));

        Ok(JsValue::from_str("Client created successfully"))
    }
}