jsglue 0.1.0

A Rust framework for building frontends with JS/TS!
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
39
40
41
42
43
44
45
46
47

use std::net::SocketAddr;

use axum::{
    body::Body,
    debug_handler,
    extract::{ConnectInfo, Request},
    response::IntoResponse,
    Extension,
};
use axumite::upgrade::WebSocketUpgrade;
use hyper::{
    header::{SEC_WEBSOCKET_PROTOCOL, USER_AGENT},
    HeaderMap,
};
use tungstenite::http::HeaderValue;

use crate::{state::ProxyState, ws::handler::handle_websocket};

#[debug_handler]
pub async fn websocket_handler(
    Extension(state): Extension<ProxyState>,
    ws: WebSocketUpgrade,
    headers: HeaderMap,
    ConnectInfo(addr): ConnectInfo<SocketAddr>,
    req: Request<Body>,
) -> impl IntoResponse {
    let user_agent = headers
        .get(USER_AGENT)
        .map(|v| v.to_str().unwrap())
        .unwrap_or("Unknown browser");

    debug!("{} connected. (UA: {})", addr, user_agent);

    let cloned_state = state.clone();

    let mut res = ws
        .on_upgrade(move |socket| handle_websocket(cloned_state, req.uri().clone(), socket, addr));

    if let Some(proto) = state.framework.get_subprotocol() {
        res.headers_mut().insert(
            SEC_WEBSOCKET_PROTOCOL,
            HeaderValue::from_str(proto).unwrap(),
        );
    }

    res
}