summaryrefslogtreecommitdiff
path: root/libs/shared/src/payloads.rs
blob: 6e51b2d602b4a4a54e35db037db52aef39c1873b (plain)
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
use std::fmt::Debug;

use serde::de::DeserializeSeed;
use serde::Deserializer;
use serde::{Deserialize, Serialize};
use serde_json::Value;
use tracing::trace_span;
use twilight_model::gateway::event::{DispatchEvent, DispatchEventWithTypeDeserializer};

#[derive(Debug, Clone)]
#[repr(transparent)]
pub struct DispatchEventTagged {
    pub data: DispatchEvent,
}

#[derive(Serialize, Deserialize)]
struct DispatchEventTaggedSerialized {
    #[serde(rename = "d")]
    pub data: Value,
    #[serde(rename = "t")]
    pub kind: String,
}

// todo(MatthieuCoder): Remove the use of the Value
impl<'de> Deserialize<'de> for DispatchEventTagged {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        let _s = trace_span!("deserializing DispatchEventTagged");
        let tagged = DispatchEventTaggedSerialized::deserialize(deserializer)?;
        let deserializer_seed = DispatchEventWithTypeDeserializer::new(&tagged.kind);
        let dispatch_event = deserializer_seed.deserialize(tagged.data).unwrap();
        Ok(DispatchEventTagged {
            data: dispatch_event,
        })
    }
}

impl Serialize for DispatchEventTagged {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        let _s = trace_span!("serializing DispatchEventTagged");
        let kind = self.data.kind().name().unwrap();
        DispatchEventTaggedSerialized {
            data: serde_json::to_value(&self.data).unwrap(),
            kind: kind.to_string(),
        }
        .serialize(serializer)
    }
}

/// Payload send to the nova cache queues
#[derive(Serialize, Deserialize, Debug, Clone)]
pub struct CachePayload {
    #[serde(flatten)]
    pub data: DispatchEventTagged,
}