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>>;
pub async fn connect(url: String) -> Result<WebsocketConnection, tungstenite::Error> {
let tuple = connect_async(url).await?;
Ok(tuple.0)
}
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)
}
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
}
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)
}
pub fn encode<T: Serialize>(message: T) -> anyhow::Result<tungstenite::Message> {
let message = bincode::serialize(&message)?;
Ok(Message::Binary(message))
}