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

use async_tungstenite::{
    tokio::{connect_async, TokioAdapter},
    WebSocketStream,
};
use serde::{de::DeserializeOwned, Serialize};

use tokio::net::TcpStream;

use auth::AuthHandler;

use message::listener::RequestMetadata;
use resource::listener::WebsocketAuthToken;
use tungstenite::{handshake::client::Request, Message};
mod auth;
pub mod bus;
mod common;
pub mod message;
pub mod resource;
pub mod service;

pub type WebsocketConnection = WebSocketStream<TokioAdapter<TcpStream>>;

/// Connects to the provided URL, with no authentication token
pub async fn connect(url: String) -> Result<WebsocketConnection, tungstenite::Error> {
    let tuple = connect_async(url).await?;
    Ok(tuple.0)
}

/// Connects to the provided URL, given an authentication token
pub async fn connect_authorized(
    url: String,
    token: String,
) -> Result<WebsocketConnection, tungstenite::Error> {
    let request = Request::builder()
        .uri(url)
        .header("Authorization", token.trim())
        .body(())?;

    let (stream, _resp) = connect_async(request).await?;
    Ok(stream)
}

/// Binds to the TCP stream as a server, requring the auth token, and capturing request metadata via a lifeline request.
pub async fn bind(
    tcp: TcpStream,
    auth_token: WebsocketAuthToken,
    request_metadata: lifeline::request::Request<(), RequestMetadata>,
) -> Result<WebsocketConnection, tungstenite::Error> {
    let auth = AuthHandler::with_metadata(auth_token, Some(request_metadata));
    async_tungstenite::tokio::accept_hdr_async(tcp, auth).await
}

/// Decodes the bincode-serialized message
pub fn decode<T: DeserializeOwned>(
    message: Result<tungstenite::Message, tungstenite::Error>,
) -> anyhow::Result<T> {
    let message = message?;
    let data = bincode::deserialize::<T>(message.into_data().as_slice())?;
    Ok(data)
}

/// Encodes the message into a bincode-serialized tungstenite message
pub fn encode<T: Serialize>(message: T) -> anyhow::Result<tungstenite::Message> {
    let message = bincode::serialize(&message)?;
    Ok(Message::Binary(message))
}