Expose client and listen methods on Wallet

rosetta-client/src/client.rs

@@ -1,3 +1,4 @@
+#![allow(missing_docs)]
 use crate::crypto::address::Address;
 use crate::crypto::PublicKey;
 use crate::types::{
@@ -21,6 +22,7 @@ use serde_json::Value;
 use std::pin::Pin;
 use std::str::FromStr;
 
+/// Generic Client
 pub enum GenericClient {
     Bitcoin(BitcoinClient),
     Ethereum(EthereumClient),
@@ -73,6 +75,7 @@ impl GenericClient {
     }
 }
 
+/// Generic Blockchain Params
 #[derive(Deserialize, Serialize, From)]
 pub enum GenericMetadataParams {
     Bitcoin(BitcoinMetadataParams),
@@ -81,6 +84,7 @@ pub enum GenericMetadataParams {
     Polkadot(PolkadotMetadataParams),
 }
 
+/// Generic Blockchain Metadata
 #[derive(Deserialize, Serialize, From)]
 pub enum GenericMetadata {
     Bitcoin(BitcoinMetadata),

rosetta-client/src/lib.rs

@@ -5,7 +5,8 @@ use anyhow::Result;
 pub use crate::wallet::Wallet;
 pub use rosetta_core::{crypto, types, BlockchainConfig};
 
-mod client;
+/// Clients that communicates to different blockchains
+pub mod client;
 mod mnemonic;
 mod signer;
 mod tx_builder;

rosetta-client/src/wallet.rs

@@ -17,7 +17,8 @@ use std::path::Path;
 
 /// The wallet provides the main entry point to this crate.
 pub struct Wallet {
-    client: GenericClient,
+    /// GenericClient instance
+    pub client: GenericClient,
     account: AccountIdentifier,
     secret_key: DerivedSecretKey,
     public_key: PublicKey,
@@ -116,6 +117,13 @@ impl Wallet {
         })
     }
 
+    /// Return a stream of events, return None if the blockchain doesn't support events.
+    pub async fn listen(
+        &self,
+    ) -> Result<Option<<GenericClient as BlockchainClient>::EventStream<'_>>> {
+        self.client.listen().await
+    }
+
     /// Returns block data
     /// Takes PartialBlockIdentifier
     pub async fn block(&self, data: PartialBlockIdentifier) -> Result<Block> {