tonlib-client 0.23.1

Thin wrapper for tonlibjson
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

use std::sync::Arc;

use super::TonConnectionCallback;
use crate::client::{
    error, ConnectionCheck, MultiConnectionCallback, RetryStrategy, TonClient, TonConnectionParams,
    LOGGING_CONNECTION_CALLBACK, NOOP_CONNECTION_CALLBACK,
};

pub struct TonClientBuilder {
    pool_size: usize,
    connection_params: TonConnectionParams,
    retry_strategy: RetryStrategy,
    callback: Arc<dyn TonConnectionCallback>,
    connection_check: ConnectionCheck,
}

impl TonClientBuilder {
    pub fn new() -> Self {
        TonClientBuilder {
            pool_size: 1,
            connection_params: TonConnectionParams::default(),
            retry_strategy: RetryStrategy::default(),
            callback: LOGGING_CONNECTION_CALLBACK.clone(),
            connection_check: ConnectionCheck::None,
        }
    }

    pub fn with_pool_size(&mut self, pool_size: usize) -> &mut Self {
        self.pool_size = pool_size;
        self
    }

    pub fn with_connection_params(&mut self, connection_params: &TonConnectionParams) -> &mut Self {
        self.connection_params = connection_params.clone();
        self
    }

    pub fn with_config(&mut self, config: &str) -> &mut Self {
        self.connection_params.config = config.to_string();
        self
    }

    pub fn with_retry_strategy(&mut self, retry_strategy: &RetryStrategy) -> &mut Self {
        self.retry_strategy = retry_strategy.clone();
        self
    }

    pub fn with_callback(&mut self, callback: Arc<dyn TonConnectionCallback>) -> &mut Self {
        self.callback = callback;
        self
    }

    pub fn with_callbacks(&mut self, callbacks: Vec<Arc<dyn TonConnectionCallback>>) -> &mut Self {
        self.callback = Arc::new(MultiConnectionCallback::new(callbacks));
        self
    }

    pub fn without_callback(&mut self) -> &mut Self {
        self.callback = NOOP_CONNECTION_CALLBACK.clone();
        self
    }

    pub fn with_logging_callback(&mut self) -> &mut Self {
        self.callback = LOGGING_CONNECTION_CALLBACK.clone();
        self
    }

    pub fn with_keystore_dir(&mut self, keystore_dir: String) -> &mut Self {
        self.connection_params.keystore_dir = Some(keystore_dir);
        self
    }

    pub fn without_keystore(&mut self) -> &mut Self {
        self.connection_params.keystore_dir = None;
        self
    }

    pub fn with_connection_check(&mut self, connection_check: ConnectionCheck) -> &mut Self {
        self.connection_check = connection_check;
        self
    }

    pub async fn build(&self) -> Result<TonClient, error::TonClientError> {
        TonClient::new(
            self.pool_size,
            &self.connection_params,
            &self.retry_strategy,
            self.callback.clone(),
            self.connection_check.clone(),
        )
        .await
    }
}

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