use std::sync::Arc;

use session_rs::ws::{Frame, WebSocket};

#[tokio::main(flavor = "current_thread")]
async fn main() -> session_rs::Result<()> {
    let session = Arc::new(WebSocket::connect("127.0.0.1:8080", "/").await?);

    // Spawn read loop
    let read_session = Arc::clone(&session);
    tokio::spawn(async move {
        loop {
            println!("{:?}", read_session.read().await);
            // match read_session.read().await {
            //     Ok(Frame::Text(text)) => {
            //         println!("Server says: {}", text);
            //     }
            //     Ok(_) => {}
            //     Err(_) => break,
            // }
        }
    });

    // Send a few messages
    for i in 0..5 {
        println!("sending");
        let msg = serde_json::json!({ "hello": i });
        session.send(&msg.to_string()).await?;
        tokio::time::sleep(std::time::Duration::from_secs(1)).await;
    }

    session.close().await?;
    Ok(())
}