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

pub mod configuration;
mod network;
pub mod mutable_cipher;
pub mod client_base;

pub extern crate async_trait;
pub extern crate tokio;
pub extern crate tcp_handler;
#[cfg(feature = "serde")]
pub extern crate serde;

use async_trait::async_trait;
use tcp_handler::common::{AesCipher, StarterError};
use tokio::net::{TcpStream, ToSocketAddrs};
use crate::network::start_client;

pub use network::{send, recv};

#[async_trait]
pub trait ClientFactory<C: From<(TcpStream, AesCipher)>> {
    fn get_identifier(&self) -> &'static str;

    /// # Example
    /// ```ignore
    /// env!("CARGO_PKG_VERSION")
    /// ```
    fn get_version(&self) -> &'static str;

    async fn connect<A: ToSocketAddrs + Send>(&self, addr: A) -> Result<C, StarterError> {
        start_client(self, addr).await.map(|c| c.into())
    }
}

pub async fn quickly_connect<A: ToSocketAddrs + Send, C: From<(TcpStream, AesCipher)>>(identifier: &'static str, version:&'static str, addr: A) -> Result<C, StarterError> {
    struct TempClient(&'static str, &'static str);
    impl ClientFactory<(TcpStream, AesCipher)> for TempClient {
        fn get_identifier(&self) -> &'static str {
            self.0
        }

        fn get_version(&self) -> &'static str {
            self.1
        }
    }
    let client = TempClient(identifier, version);
    client.connect(addr).await.map(|c| c.into())
}