grip-control 0.8.0

A Rust crate for grip control.
Documentation 
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

/// Returns a WebSocket-over-HTTP formatted control message
pub fn ws_control(kind: &str) -> Vec<u8> {
    format!("{kind}\r\n").into_bytes()
}

pub fn ws_text(msg: &str) -> Vec<u8> {
    format!("TEXT {:x}\r\n{}\r\n", msg.len(), msg).into_bytes()
}

/// Returns a TEXT-wrapped GRIP subscription command for the given channel
pub fn ws_sub(ch: &str) -> Vec<u8> {
    let control_msg = format!("c:{{\"type\":\"subscribe\",\"channel\":\"{}\"}}", ch);
    ws_text(&control_msg)
}

/// Returns a TEXT-wrapped GRIP unsubscribe command for the given channel
pub fn ws_unsub(ch: &str) -> Vec<u8> {
    let control_msg = format!("c:{{\"type\":\"unsubscribe\",\"channel\":\"{}\"}}", ch);
    ws_text(&control_msg)
}

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    fn test_ws_control() {
        let frame = ws_control("OPEN");
        assert_eq!(frame, b"OPEN\r\n");

        let frame = ws_control("PING");
        assert_eq!(frame, b"PING\r\n");

        let frame = ws_control("DISCONNECT");
        assert_eq!(frame, b"DISCONNECT\r\n");
    }

    #[test]
    fn test_ws_text() {
        let frame = ws_text("hello");
        assert_eq!(frame, b"TEXT 5\r\nhello\r\n");

        let frame = ws_text("hello world");
        assert_eq!(frame, b"TEXT b\r\nhello world\r\n");

        let frame = ws_text("");
        assert_eq!(frame, b"TEXT 0\r\n\r\n");
    }

    #[test]
    fn test_ws_sub() {
        let frame = ws_sub("test-channel");
        let expected = b"TEXT 2f\r\nc:{\"type\":\"subscribe\",\"channel\":\"test-channel\"}\r\n";
        assert_eq!(frame, expected);

        let frame = ws_sub("my-room");
        let expected = b"TEXT 2a\r\nc:{\"type\":\"subscribe\",\"channel\":\"my-room\"}\r\n";
        assert_eq!(frame, expected);
    }

    #[test]
    fn test_ws_unsub() {
        let frame = ws_unsub("test-channel");
        let expected = b"TEXT 31\r\nc:{\"type\":\"unsubscribe\",\"channel\":\"test-channel\"}\r\n";
        assert_eq!(frame, expected);

        let frame = ws_unsub("my-room");
        let expected = b"TEXT 2c\r\nc:{\"type\":\"unsubscribe\",\"channel\":\"my-room\"}\r\n";
        assert_eq!(frame, expected);
    }
}