pub mod connection;
pub mod error;
pub use connection::{Connection, Request, Response};
pub use error::Error;
use gio::{Cancellable, SocketClient, SocketProtocol, TlsCertificate, prelude::SocketClientExt};
use glib::Priority;
pub const DEFAULT_TIMEOUT: u32 = 30;
pub const DEFAULT_SESSION_RESUMPTION: bool = false;
pub struct Client {
is_session_resumption: bool,
pub socket: SocketClient,
}
impl Default for Client {
fn default() -> Self {
Self::new()
}
}
impl Client {
pub fn new() -> Self {
let socket = SocketClient::new();
socket.set_protocol(SocketProtocol::Tcp);
socket.set_timeout(DEFAULT_TIMEOUT);
Self {
is_session_resumption: DEFAULT_SESSION_RESUMPTION,
socket,
}
}
pub fn request_async(
&self,
request: Request,
priority: Priority,
cancellable: Cancellable,
client_certificate: Option<TlsCertificate>,
server_certificates: Option<Vec<TlsCertificate>>,
callback: impl FnOnce(Result<(Response, Connection), Error>) + 'static,
) {
match request.to_network_address(crate::DEFAULT_PORT) {
Ok(network_address) => {
self.socket
.connect_async(&network_address.clone(), Some(&cancellable.clone()), {
let is_session_resumption = self.is_session_resumption;
move |result| match result {
Ok(socket_connection) => {
match Connection::build(
socket_connection.clone(),
network_address,
client_certificate,
server_certificates,
is_session_resumption,
) {
Ok(connection) => connection.clone().request_async(
request,
priority,
cancellable,
move |result| {
callback(match result {
Ok(response) => Ok(response),
Err(e) => Err(Error::Request(connection, e)),
})
},
),
Err(e) => {
callback(Err(Error::Connection(socket_connection, e)))
}
}
}
Err(e) => callback(Err(Error::Connect(network_address, e))),
}
})
}
Err(e) => callback(Err(Error::NetworkAddress(e))),
}
}
pub fn set_session_resumption(&mut self, is_enabled: bool) {
self.is_session_resumption = is_enabled
}
}