stream-ws 0.1.1

use std::io;
use stream_ws::{tungstenite::WsMessageHandler, WsMessageHandle};
use tokio::net::{TcpListener, TcpStream};
use tokio::{
    fs::File,
    io::{AsyncReadExt, AsyncWriteExt},
};
use tokio_tungstenite::accept_async;

#[tokio::main]
async fn main() -> io::Result<()> {
    let addr = std::env::args()
        .nth(1)
        .unwrap_or("127.0.0.1:8080".to_owned());
    let listener = TcpListener::bind(&addr).await?;
    while let Ok((stream, addr)) = listener.accept().await {
        println!("New connection: {}", &addr);
        tokio::spawn(async move {
            let ret = handler(stream).await;
            if let Err(e) = ret {
                eprintln!("Error: {}", e);
            }
            println!("Connection closed: {}", &addr);
        });
    }
    Ok(())
}

async fn handler(stream: TcpStream) -> io::Result<()> {
    let ws = accept_async(stream).await.unwrap();
    // let mut stream = stream_ws::tungstenite::WsByteStream::new(ws);
    let mut stream = WsMessageHandler::wrap_stream(ws);
    let mut buf = vec![];
    let mut file = File::open("examples/print_self_server.rs").await?;
    file.read_to_end(&mut buf).await?;
    drop(file);
    stream.write(&buf[..]).await?;
    stream.flush().await?;
    stream.shutdown().await?;
    Ok(())
}