botrs 0.12.2

A Rust QQ Bot framework based on QQ Guild Bot API
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
72
73
74
75
76
77
78
79
80
81
82

use crate::api::BotApi;
use crate::http::HttpClient;
use crate::models::gateway::*;
use serde::de::DeserializeOwned;

pub fn ParseData<T: DeserializeOwned>(message: &[u8]) -> crate::Result<T> {
    let value: serde_json::Value = serde_json::from_slice(message)?;
    serde_json::from_value(value.get("d").cloned().unwrap_or(serde_json::Value::Null))
        .map_err(Into::into)
}

pub trait PayloadData: Sized {
    fn parse_from_payload(payload: &WSPayload, message: &[u8]) -> crate::Result<Self>;
}

impl<T> PayloadData for T
where
    T: DeserializeOwned,
{
    fn parse_from_payload(_payload: &WSPayload, message: &[u8]) -> crate::Result<Self> {
        ParseData(message)
    }
}

fn payload_data(message: &[u8]) -> crate::Result<serde_json::Value> {
    let value: serde_json::Value = serde_json::from_slice(message)?;
    Ok(value.get("d").cloned().unwrap_or(serde_json::Value::Null))
}

fn payload_event_id(payload: &WSPayload) -> Option<String> {
    payload.base.event_id.clone()
}

fn event_api() -> BotApi {
    BotApi::new(HttpClient::new(crate::DEFAULT_TIMEOUT, false).expect("valid default api client"))
}

macro_rules! impl_constructed_payload_data {
    ($ty:ty, $ctor:expr) => {
        impl PayloadData for $ty {
            fn parse_from_payload(payload: &WSPayload, message: &[u8]) -> crate::Result<Self> {
                let data = payload_data(message)?;
                Ok($ctor(payload, data))
            }
        }
    };
}

impl PayloadData for crate::reaction::Reaction {
    fn parse_from_payload(payload: &WSPayload, message: &[u8]) -> crate::Result<Self> {
        let data = ParseData(message)?;
        Ok(crate::reaction::Reaction::from_message_reaction(
            event_api(),
            payload_event_id(payload),
            data,
        ))
    }
}
impl_constructed_payload_data!(
    crate::forum::Thread,
    |payload: &WSPayload, data: serde_json::Value| {
        crate::forum::Thread::new(event_api(), payload_event_id(payload), &data)
    }
);
impl_constructed_payload_data!(
    crate::forum::Post,
    |payload: &WSPayload, data: serde_json::Value| {
        crate::forum::Post::new(event_api(), payload_event_id(payload), &data)
    }
);
impl_constructed_payload_data!(
    crate::forum::Reply,
    |payload: &WSPayload, data: serde_json::Value| {
        crate::forum::Reply::new(event_api(), payload_event_id(payload), &data)
    }
);
impl_constructed_payload_data!(
    crate::interaction::Interaction,
    |payload: &WSPayload, data: serde_json::Value| {
        crate::interaction::Interaction::new(event_api(), payload_event_id(payload), &data)
    }
);