pueue-lib 0.31.0

The shared library to work with the Pueue client and daemon.
Documentation 
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

use rustls::pki_types::ServerName;
use tokio::net::TcpStream;

use super::{ConnectionSettings, GenericStream, get_tls_connector};
use crate::error::Error;

/// Get a new stream for the client.
/// This can either be a UnixStream or a Tls encrypted TCPStream, depending on the parameters.
pub async fn get_client_stream(settings: ConnectionSettings<'_>) -> Result<GenericStream, Error> {
    match settings {
        ConnectionSettings::TlsTcpSocket {
            host,
            port,
            certificate,
        } => {
            // Connect to the daemon via TCP
            let address = format!("{host}:{port}");
            let tcp_stream = TcpStream::connect(&address).await.map_err(|_| {
                Error::Connection(format!(
                    "Failed to connect to the daemon on {address}. Did you start it?"
                ))
            })?;

            // Get the configured rustls TlsConnector
            let tls_connector = get_tls_connector(certificate).await.map_err(|err| {
                Error::Connection(format!("Failed to initialize tls connector:\n{err}."))
            })?;

            // Initialize the TLS layer
            let stream = tls_connector
                .connect(ServerName::try_from("pueue.local").unwrap(), tcp_stream)
                .await
                .map_err(|err| Error::Connection(format!("Failed to initialize tls {err}.")))?;

            Ok(Box::new(stream))
        }
    }
}