sc2-proxy 0.1.0

StarCraft II bot API management layer
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

//! Proxy WebSocket receiver

use crossbeam::channel::Sender;
use log::debug;
use std::net::ToSocketAddrs;

use websocket::client::sync::Client as GenericClient;
use websocket::server::sync::Server as GenericServer;
use websocket::server::NoTlsAcceptor;
use websocket::stream::sync::TcpStream;

/// Server socket
pub type Server = GenericServer<NoTlsAcceptor>;
/// Client socket
pub type Client = GenericClient<TcpStream>;

/// Accept a new connection
fn get_connection(server: &mut Server) -> Option<Client> {
    Some(server.accept().ok()?.accept().expect("Unable to accept"))
}

/// Run the proxy server
pub fn run<A: ToSocketAddrs>(addr: A, channel_out: Sender<Client>) -> ! {
    let mut server = Server::bind(addr).expect("Unable to bind");

    loop {
        debug!("Waiting for connection");
        if let Some(conn) = get_connection(&mut server) {
            debug!("Connection accepted: {:?}", conn.peer_addr().unwrap());
            channel_out.send(conn).expect("Send failed");
        }
    }
}