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

use crate::Driver;

#[derive(Debug)]
pub enum WebsocketMessageDriver {
    Message(String),
    Connection { callback_id: u32 },
    Close,
}

/// Websocket message type on which a websocket handler operates.
pub enum WebsocketMessage {
    Message(String),
    Connection(WebsocketConnection),
    Close,
}

/// Represents websocket connection.
#[derive(PartialEq)]
pub struct WebsocketConnection {
    callback_id: u32,
    driver: Driver,
}

impl WebsocketConnection {
    pub fn new(callback_id: u32, driver: Driver) -> WebsocketConnection {
        WebsocketConnection { callback_id, driver }
    }

    pub fn send(&self, message: impl Into<String>) {
        let message = message.into();
        self.driver.websocket_send_message(self.callback_id, message);
    }
}