aquatic_ws_protocol 0.9.0

WebTorrent tracker protocol
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

use anyhow::Context;
use serde::{Deserialize, Serialize};

pub mod announce;
pub mod scrape;

pub use announce::*;
pub use scrape::*;

/// Message received by tracker
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
#[serde(untagged)]
pub enum InMessage {
    AnnounceRequest(AnnounceRequest),
    ScrapeRequest(ScrapeRequest),
}

#[cfg(feature = "tungstenite")]
impl InMessage {
    #[inline]
    pub fn to_ws_message(&self) -> ::tungstenite::Message {
        ::tungstenite::Message::from(::serde_json::to_string(&self).unwrap())
    }

    #[inline]
    pub fn from_ws_message(ws_message: tungstenite::Message) -> ::anyhow::Result<Self> {
        use tungstenite::Message;

        match ws_message {
            Message::Text(text) => {
                let mut text: Vec<u8> = text.into();

                ::simd_json::serde::from_slice(&mut text).context("deserialize with serde")
            }
            Message::Binary(mut bytes) => {
                ::simd_json::serde::from_slice(&mut bytes[..]).context("deserialize with serde")
            }
            _ => Err(anyhow::anyhow!("Message is neither text nor binary")),
        }
    }
}