lit_rust_sdk/client/
mod.rs

1use crate::config::LitNodeClientConfig;
2use alloy::providers::{DynProvider, Provider as ProviderTrait, ProviderBuilder};
3use dashmap::DashMap;
4use eyre::Result;
5use reqwest::Client;
6use std::sync::Arc;
7
8mod connect;
9mod execute;
10mod pkp;
11mod state;
12
13use crate::blockchain::staking::LibStakingStorage::Epoch;
14use crate::blockchain::Staking;
15
16pub struct LitNodeClient<P = DynProvider>
17where
18    P: ProviderTrait,
19{
20    pub(crate) config: LitNodeClientConfig,
21    pub(crate) http_client: Client,
22    pub(crate) connection_state: Arc<DashMap<String, crate::types::NodeConnectionInfo>>,
23    pub(crate) ready: bool,
24    pub(crate) subnet_pub_key: Option<String>,
25    pub(crate) network_pub_key: Option<String>,
26    pub(crate) network_pub_key_set: Option<String>,
27    pub(crate) hd_root_pubkeys: Option<Vec<String>>,
28    pub(crate) latest_blockhash: Option<String>,
29    pub(crate) staking: Staking::StakingInstance<P>,
30    pub(crate) min_node_count: Option<usize>,
31    pub(crate) epoch: Option<Epoch>,
32}
33
34impl LitNodeClient<DynProvider> {
35    pub async fn new(config: LitNodeClientConfig) -> Result<Self> {
36        let http_client = Client::builder().timeout(config.connect_timeout).build()?;
37
38        let rpc_url = config.lit_network.rpc_url();
39        let provider = ProviderBuilder::new().connect(&rpc_url).await?;
40        let staking_address = config.lit_network.staking_contract_address()?;
41
42        let staking = Staking::new(staking_address, provider.erased());
43
44        Ok(Self {
45            config,
46            http_client,
47            connection_state: Arc::new(DashMap::new()),
48            ready: false,
49            subnet_pub_key: None,
50            network_pub_key: None,
51            network_pub_key_set: None,
52            hd_root_pubkeys: None,
53            latest_blockhash: None,
54            staking,
55            min_node_count: None,
56            epoch: None,
57        })
58    }
59}