solana-ws 0.2.0

/// WebSocket client used for testing against the Autobahn Test Suite
extern crate solana_ws as ws;

use std::cell::Cell;
use std::rc::Rc;
use ws::{connect, CloseCode, Message, Result};

#[cfg(feature = "permessage-deflate")]
use ws::deflate::DeflateHandler;

const AGENT: &str = "WS-RS";

#[cfg(not(feature = "permessage-deflate"))]
fn main() {
    let total = get_case_count().unwrap();
    let mut case_id = 1;

    while case_id <= total {
        let case_url = format!(
            "ws://127.0.0.1:9001/runCase?case={}&agent={}",
            case_id, AGENT
        );

        connect(case_url, |out| move |msg| out.send(msg)).unwrap();

        case_id += 1
    }

    update_reports().unwrap();
}

#[cfg(feature = "permessage-deflate")]
fn main() {
    let total = get_case_count().unwrap();
    let mut case_id = 1;

    while case_id <= total {
        let case_url = format!(
            "ws://127.0.0.1:9001/runCase?case={}&agent={}",
            case_id, AGENT
        );

        connect(case_url, |out| {
            DeflateHandler::new(move |msg| out.send(msg))
        }).unwrap();

        case_id += 1
    }

    update_reports().unwrap();
}

fn get_case_count() -> Result<u32> {
    // sadly we need to use a Cell because we need to set the total, and RC is immutable
    let total = Rc::new(Cell::new(0));

    connect("ws://127.0.0.1:9001/getCaseCount", |out| {
        let my_total = total.clone();

        move |msg: Message| {
            let count = msg.as_text()?;

            my_total.set(count.parse::<u32>().unwrap());

            out.close(CloseCode::Normal)
        }
    })?;

    Ok(total.get())
}

fn update_reports() -> Result<()> {
    let report_url = format!("ws://127.0.0.1:9001/updateReports?agent={}", AGENT);

    connect(report_url, |out| move |_| out.close(CloseCode::Normal))
}