//! 客户端

use log::{trace, warn};
use std::sync::{Arc, Mutex};
use std::thread;
use unmp::link::{Driver, Link, Recver};
use unmp::net;
use websocket::OwnedMessage;

/// 客户端链路驱动接口
struct Client {
    socket: Mutex<websocket::sender::Writer<std::net::TcpStream>>,
}
impl Driver for Client {
    fn send(&self, buf: &[u8]) {
        trace!("websocket_client send: {:02X?}.", buf);
        let mut socket = self.socket.lock().unwrap();
        socket
            .send_message(&OwnedMessage::Binary(buf.into()))
            .expect("websocket_client send error");
    }
}

/// 创建一个链路实例，并连接到指定地址端口
pub fn start(url: &str) -> Link {
    let client = websocket::client::ClientBuilder::new(url)
        .unwrap()
        .add_protocol("unmp")
        .connect_insecure()
        .unwrap();
    let (mut receiver, sender) = client.split().unwrap();
    let driver = Arc::new(Client {
        socket: Mutex::new(sender),
    });
    let link = Link::new(format!("WebSocket_Client({})", url), driver.clone());

    let driver_tmp = driver.clone();
    let link_tmp = link.clone();
    thread::spawn(move || {
        let mut recver: Recver = Recver::new();
        for message in receiver.incoming_messages() {
            match message {
                Ok(m) => match m {
                    OwnedMessage::Binary(buf) => {
                        let mut buf: &[u8] = &buf;
                        trace!("websocket_client recv: {:02X?}.", buf);

                        // 接收数据帧
                        while let Some(load) = {
                            let load = recver.recv(buf);
                            buf = &[];
                            load
                        } {
                            net::recv_handle(link_tmp.clone(), &load);
                        }
                    }
                    _ => {}
                },
                Err(e) => {
                    warn!("websocket_client recv error: {:?}", e);
                    let mut socket = driver_tmp.socket.lock().unwrap();
                    let _ = socket.send_message(&OwnedMessage::Close(None));
                    return;
                }
            };
        }
    });
    return link;
}