feat(hd_wallet): evm hd wallet and trezor

mm2src/coins/Cargo.toml

@@ -17,6 +17,7 @@ enable-solana = [
 ]
 default = []
 run-docker-tests = []
+for-tests = []
 
 [lib]
 name = "coins"

mm2src/coins/eth/eth_balance_events.rs

@@ -1,6 +1,7 @@
 use async_trait::async_trait;
 use common::{executor::{AbortSettings, SpawnAbortable, Timer},
              log, Future01CompatExt};
+use ethereum_types::Address;
 use futures::{channel::oneshot::{self, Receiver, Sender},
               stream::FuturesUnordered,
               StreamExt};
@@ -10,65 +11,104 @@ use mm2_err_handle::prelude::MmError;
 use mm2_event_stream::{behaviour::{EventBehaviour, EventInitStatus},
                        Event, EventStreamConfiguration};
 use mm2_number::BigDecimal;
-use std::collections::HashMap;
+use std::collections::{HashMap, HashSet};
 
 use super::EthCoin;
 use crate::{eth::{u256_to_big_decimal, Erc20TokenInfo},
             BalanceError, MmCoin};
+struct BalanceData {
+    ticker: String,
+    address: String,
+    balance: BigDecimal,
+}
+
+struct BalanceFetchError {
+    ticker: String,
+    address: String,
+    error: MmError<BalanceError>,
+}
+
+type BalanceResult = Result<BalanceData, BalanceFetchError>;
 
 /// This implementation differs from others, as they immediately return
 /// an error if any of the requests fails. This one completes all futures
 /// and returns their results individually.
-async fn get_all_balance_results_concurrently(
-    coin: &EthCoin,
-) -> Vec<Result<(String, BigDecimal), (String, MmError<BalanceError>)>> {
+async fn get_all_balance_results_concurrently(coin: &EthCoin, addresses: HashSet<Address>) -> Vec<BalanceResult> {
     let mut tokens = coin.get_erc_tokens_infos();
-
     // Workaround for performance purposes.
     //
     // Unlike tokens, the platform coin length is constant (=1). Instead of creating a generic
     // type and mapping the platform coin and the entire token list (which can grow at any time), we map
     // the platform coin to Erc20TokenInfo so that we can use the token list right away without
     // additional mapping.
     tokens.insert(coin.ticker.clone(), Erc20TokenInfo {
-        token_address: coin.my_address,
+        // This is a dummy value, since there is no token address for the platform coin.
+        // In the fetch_balance function, we check if the token_ticker is equal to this
+        // coin's ticker to avoid using token_address to fetch the balance
+        // and to use address_balance instead.
+        token_address: Address::default(),
         decimals: coin.decimals,
     });
+    drop_mutability!(tokens);
+
+    let mut all_jobs = FuturesUnordered::new();
 
-    let jobs = tokens
-        .into_iter()
-        .map(|(token_ticker, info)| async move { fetch_balance(coin, token_ticker, &info).await })
-        .collect::<FuturesUnordered<_>>();
+    for address in addresses {
+        let jobs = tokens.iter().map(|(token_ticker, info)| {
+            let coin = coin.clone();
+            let token_ticker = token_ticker.clone();
+            let info = info.clone();
+            async move { fetch_balance(&coin, address, token_ticker, &info).await }
+        });
 
-    jobs.collect().await
+        all_jobs.extend(jobs);
+    }
+
+    all_jobs.collect().await
 }
 
 async fn fetch_balance(
     coin: &EthCoin,
+    address: Address,
     token_ticker: String,
     info: &Erc20TokenInfo,
-) -> Result<(String, BigDecimal), (String, MmError<BalanceError>)> {
+) -> Result<BalanceData, BalanceFetchError> {
     let (balance_as_u256, decimals) = if token_ticker == coin.ticker {
         (
-            coin.address_balance(coin.my_address)
+            coin.address_balance(address)
                 .compat()
                 .await
-                .map_err(|e| (token_ticker.clone(), e))?,
+                .map_err(|error| BalanceFetchError {
+                    ticker: token_ticker.clone(),
+                    address: address.to_string(),
+                    error,
+                })?,
             coin.decimals,
         )
     } else {
         (
-            coin.get_token_balance_by_address(info.token_address)
+            coin.get_token_balance(info.token_address)
                 .await
-                .map_err(|e| (token_ticker.clone(), e))?,
+                .map_err(|error| BalanceFetchError {
+                    ticker: token_ticker.clone(),
+                    address: address.to_string(),
+                    error,
+                })?,
             info.decimals,
         )
     };
 
-    let balance_as_big_decimal =
-        u256_to_big_decimal(balance_as_u256, decimals).map_err(|e| (token_ticker.clone(), e.into()))?;
-
-    Ok((token_ticker, balance_as_big_decimal))
+    let balance_as_big_decimal = u256_to_big_decimal(balance_as_u256, decimals).map_err(|e| BalanceFetchError {
+        ticker: token_ticker.clone(),
+        address: address.to_string(),
+        error: e.into(),
+    })?;
+
+    Ok(BalanceData {
+        ticker: token_ticker,
+        address: address.to_string(),
+        balance: balance_as_big_decimal,
+    })
 }
 
 #[async_trait]
@@ -80,31 +120,69 @@ impl EventBehaviour for EthCoin {
         const RECEIVER_DROPPED_MSG: &str = "Receiver is dropped, which should never happen.";
 
         async fn start_polling(coin: EthCoin, ctx: MmArc, interval: f64) {
-            let mut cache: HashMap<String, BigDecimal> = HashMap::new();
+            async fn sleep_remaining_time(interval: f64, now: Instant) {
+                // If the interval is x seconds,
+                // our goal is to broadcast changed balances every x seconds.
+                // To achieve this, we need to subtract the time complexity of each iteration.
+                // Given that an iteration already takes 80% of the interval,
+                // this will lead to inconsistency in the events.
+                let remaining_time = interval - now.elapsed().as_secs_f64();
+                // Not worth to make a call for less than `0.1` durations
+                if remaining_time >= 0.1 {
+                    Timer::sleep(remaining_time).await;
+                }
+            }
+
+            let mut cache: HashMap<String, HashMap<String, BigDecimal>> = HashMap::new();
 
             loop {
                 let now = Instant::now();
 
+                let addresses = match coin.my_addresses().await {
+                    Ok(addresses) => addresses,
+                    Err(e) => {
+                        log::error!("Failed getting addresses for {}. Error: {}", coin.ticker, e);
+                        let e = serde_json::to_value(e).expect("Serialization shouldn't fail.");
+                        ctx.stream_channel_controller
+                            .broadcast(Event::new(
+                                format!("{}:{}", EthCoin::ERROR_EVENT_NAME, coin.ticker),
+                                e.to_string(),
+                            ))
+                            .await;
+                        sleep_remaining_time(interval, now).await;
+                        continue;
+                    },
+                };
+
                 let mut balance_updates = vec![];
-                for result in get_all_balance_results_concurrently(&coin).await {
+                for result in get_all_balance_results_concurrently(&coin, addresses).await {
                     match result {
-                        Ok((ticker, balance)) => {
-                            if Some(&balance) == cache.get(&ticker) {
+                        Ok(res) => {
+                            if Some(&res.balance) == cache.get(&res.ticker).and_then(|map| map.get(&res.address)) {
                                 continue;
                             }
 
                             balance_updates.push(json!({
-                                "ticker": ticker,
-                                "balance": { "spendable": balance, "unspendable": BigDecimal::default() }
+                                "ticker": res.ticker,
+                                "address": res.address,
+                                "balance": { "spendable": res.balance, "unspendable": BigDecimal::default() }
                             }));
-                            cache.insert(ticker.to_owned(), balance);
+                            cache
+                                .entry(res.ticker.clone())
+                                .or_insert_with(HashMap::new)
+                                .insert(res.address, res.balance);
                         },
-                        Err((ticker, e)) => {
-                            log::error!("Failed getting balance for '{ticker}' with {interval} interval. Error: {e}");
-                            let e = serde_json::to_value(e).expect("Serialization should't fail.");
+                        Err(err) => {
+                            log::error!(
+                                "Failed getting balance for '{}:{}' with {interval} interval. Error: {}",
+                                err.ticker,
+                                err.address,
+                                err.error
+                            );
+                            let e = serde_json::to_value(err.error).expect("Serialization shouldn't fail.");
                             ctx.stream_channel_controller
                                 .broadcast(Event::new(
-                                    format!("{}:{}", EthCoin::ERROR_EVENT_NAME, ticker),
+                                    format!("{}:{}:{}", EthCoin::ERROR_EVENT_NAME, err.ticker, err.address),
                                     e.to_string(),
                                 ))
                                 .await;
@@ -121,15 +199,7 @@ impl EventBehaviour for EthCoin {
                         .await;
                 }
 
-                // If the interval is x seconds, our goal is to broadcast changed balances every x seconds.
-                // To achieve this, we need to subtract the time complexity of each iteration.
-                // Given that an iteration already takes 80% of the interval, this will lead to inconsistency
-                // in the events.
-                let remaining_time = interval - now.elapsed().as_secs_f64();
-                // Not worth to make a call for less than `0.1` durations
-                if remaining_time >= 0.1 {
-                    Timer::sleep(remaining_time).await;
-                }
+                sleep_remaining_time(interval, now).await;
             }
         }