ws-async 0.1.0

Asynchronous WebSocket Library
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
67
68
69
70
71
72
73
74
75
76
77
78
79

use crate::{handshake, WebSocket};
use hyper::client::conn::Builder;
use hyper::client::service::Connect;
use hyper::{header, Body, Request, Uri};
use std::future::Future;
use std::pin::Pin;
use std::task::{Context, Poll};
use tower_service::Service;

mod connect;
pub use connect::WsConnector;

#[derive(Debug)]
pub struct Client<P = Body> {
    http: Connect<WsConnector, P, Uri>,
}

impl Default for Client {
    fn default() -> Self {
        Self::from(Builder::new())
    }
}

impl Client {
    pub fn new() -> Self {
        Self::default()
    }
    pub async fn connect(&mut self, uri: Uri) -> hyper::Result<WebSocket> {
        self.call(uri).await
    }
}

impl<P> From<Builder> for Client<P> {
    fn from(builder: Builder) -> Self {
        let http = Connect::new(WsConnector::new(), builder);
        Self { http }
    }
}

impl Service<Uri> for Client {
    type Response = WebSocket;
    type Error = hyper::Error;
    type Future = Pin<Box<dyn Future<Output = hyper::Result<Self::Response>> + Send + 'static>>;

    fn poll_ready(&mut self, _cx: &mut Context) -> Poll<hyper::Result<()>> {
        Ok(()).into()
    }

    fn call(&mut self, uri: Uri) -> Self::Future {
        let svc_fut = self.http.call(uri);

        Box::pin(async move {
            let mut svc = svc_fut.await?;

            // TODO don't unwrap
            let key = handshake::generate().await.unwrap();
            let req = Request::builder()
                .header(header::CONNECTION, header::UPGRADE)
                .header(header::UPGRADE, "websocket")
                .header(handshake::SEC_WEBSOCKET_KEY, &key)
                .body(Body::empty())
                .unwrap();

            let res = svc.call(req).await?;
            if let Some(accept) = res.headers().get(handshake::SEC_WEBSOCKET_ACCEPT) {
                // TODO don't unwrap
                let clone = handshake::accept(&key).await.unwrap();

                if accept == &clone {
                    WebSocket::upgrade(res.into_body()).await
                } else {
                    unimplemented!()
                }
            } else {
                unimplemented!()
            }
        })
    }
}