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
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161

use crate::client::{Client, ClientType, CryptoClient};
use crate::types::{Amount, Currency, Symbol, WalletBalance};
use crate::types::{ClientOptions, SyncItem, SyncStats};
use crate::Result;
use async_trait::async_trait;
use num_bigint::BigInt;
use rpc_json_client::{ClientBuilder, RpcClient};
use serde_json::json;
use std::sync::Arc;

#[derive(Clone)]
pub struct EthereumRpcClient {
    client: Arc<RpcClient>,
}

pub fn get_client(client_type: ClientType, options: ClientOptions) -> crate::Result<Box<Client>> {
    match client_type {
        //@todo need url to come as well.
        ClientType::RPC => Ok(Box::new(EthereumRpcClient::new(options))),
        _ => Err(crate::error::ClientError::UnsupportedType(client_type)),
    }
}

impl EthereumRpcClient {
    pub fn new(options: ClientOptions) -> Self {
        let client = ClientBuilder::new(&options.url).with_retry().build();

        EthereumRpcClient {
            client: Arc::new(client),
        }
    }
}

#[async_trait]
impl CryptoClient for EthereumRpcClient {
    async fn sync_stats(&self) -> SyncStats {
        let result: serde_json::Value = match self.client.execute("eth_syncing", &[]).await {
            Ok(r) => r,
            //@todo we should just move this entire error handling outside of here once we have results
            //in this lib.
            Err(_) => {
                // info!("Node appeared to not be ready yet. Sleeping for 60 seconds");
                async_std::task::sleep(std::time::Duration::from_secs(60)).await;

                return SyncStats {
                    current_block: 0,
                    syncing: true,
                    sync_item: SyncItem::Block,
                    estimated_sync_item_remaining: 0.0,
                };
            }
        };

        //@todo Eth is so fucked that I don't know if this will return false even if there is more
        //state tries to sync.
        //
        //We need to keep up on this issue as it looks like Eth might at some point introduce an
        //RPC to facilitate with this process.
        //
        //
        //@todo actually looks like they now include "known states and pulled states". Not sure if
        //that can be used to accurately represent this shit, but let's invetigate.
        if let Some(syncing) = result.as_bool() {
            let result: serde_json::Value =
                self.client.execute("eth_blockNumber", &[]).await.unwrap();

            let block =
                u64::from_str_radix(result.as_str().unwrap().trim_start_matches("0x"), 16).unwrap();

            if syncing == false {
                return SyncStats {
                    current_block: block,
                    syncing: false,
                    sync_item: SyncItem::Block,
                    estimated_sync_item_remaining: 0.0,
                };
            }
        }

        //@todo use this to debug the Known and pulled states shit.
        dbg!(&result);

        let current_block = u64::from_str_radix(
            result
                .as_object()
                .unwrap()
                .get("currentBlock")
                .unwrap()
                .as_str()
                .unwrap()
                .trim_start_matches("0x"),
            16,
        )
        .unwrap();

        let tip = u64::from_str_radix(
            result
                .as_object()
                .unwrap()
                .get("highestBlock")
                .unwrap()
                .as_str()
                .unwrap()
                .trim_start_matches("0x"),
            16,
        )
        .unwrap();

        SyncStats {
            current_block,
            sync_item: SyncItem::Block,
            estimated_sync_item_remaining: (tip - current_block) as f64,
            syncing: true,
        }
    }

    async fn wallet_balance(&self, identifier: &str) -> Result<WalletBalance> {
        let balance_hex: String = self
            .client
            .execute("eth_getBalance", &[json!(identifier), json!("latest")])
            .await?;

        //@todo not sure if we want bigint or biguint here...
        let confirmed_balance =
            BigInt::parse_bytes(balance_hex.trim_start_matches("0x").as_bytes(), 16).unwrap();

        let unconfirmed_balance_hex: String = self
            .client
            .execute("eth_getBalance", &[json!(identifier), json!("pending")])
            .await?;

        //@todo not sure if we want bigint or biguint here...
        let unconfirmed_balance = BigInt::parse_bytes(
            unconfirmed_balance_hex.trim_start_matches("0x").as_bytes(),
            16,
        )
        .unwrap();

        Ok(WalletBalance {
            //We need bigints here actually.
            confirmed_balance: Amount {
                value: confirmed_balance,
                currency: Currency {
                    symbol: Symbol::ETH,
                    decimals: 18,
                },
            },
            unconfirmed_balance: Amount {
                value: unconfirmed_balance,
                currency: Currency {
                    symbol: Symbol::ETH,
                    decimals: 18,
                },
            },
        })
    }

    fn client_type(&self) -> ClientType {
        ClientType::RPC
    }
}