cord-nvim 2.0.0-beta.34

🚀 The most extensible Discord Rich Presence plugin for Neovim, powered by Rust.
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
83
84
85
86
87
88

#![allow(clippy::large_enum_variant)]

use crate::messages::events::event::{EventContext, OnEvent};
use crate::presence::activity::Activity;
use crate::protocol::msgpack::{Deserialize, MsgPack};
use crate::types::config::PluginConfig;

pub mod clear_activity;
pub mod connect;
pub mod disconnect;
pub mod initialize;
pub mod shutdown;
pub mod update_activity;

pub use clear_activity::ClearActivityEvent;
pub use connect::ConnectEvent;
pub use disconnect::DisconnectEvent;
pub use initialize::InitializeEvent;
pub use shutdown::ShutdownEvent;
pub use update_activity::UpdateActivityEvent;

#[derive(Debug)]
pub enum ClientEvent {
    Connect(ConnectEvent),
    Initialize(InitializeEvent),
    UpdateActivity(UpdateActivityEvent),
    ClearActivity(ClearActivityEvent),
    Disconnect(DisconnectEvent),
    Shutdown(ShutdownEvent),
}

/// Extracts the 'data' field from a map and returns an error if it is missing or invalid.
///
/// This macro is useful for extracting and validating the 'data' field from a map-like structure.
macro_rules! data {
    ($map:expr) => {
        $map.remove("data")
            .ok_or("Missing or invalid 'data' field")?
    };
    ($map:expr, $expr:expr) => {
        $map.remove("data")
            .and_then($expr)
            .ok_or("Missing or invalid 'data' field")?
    };
}

impl ClientEvent {
    // { type: string, data: any }
    pub fn deserialize(json: &[u8]) -> crate::Result<Self> {
        let mut map = MsgPack::deserialize(json)?
            .take_map()
            .ok_or("Invalid message")?;

        let ty = map
            .get("type")
            .and_then(|v| v.as_str())
            .ok_or("Missing 'type' field")?;

        Ok(match ty {
            "connect" => Self::Connect(ConnectEvent),
            "initialize" => Self::Initialize(InitializeEvent::new(
                PluginConfig::deserialize(data!(map))?,
            )),
            "update_activity" => Self::UpdateActivity(
                UpdateActivityEvent::new(Activity::deserialize(data!(map))?),
            ),
            "clear_activity" => Self::ClearActivity(
                ClearActivityEvent::deserialize(data!(map))?,
            ),
            "disconnect" => Self::Disconnect(DisconnectEvent),
            "shutdown" => Self::Shutdown(ShutdownEvent),
            _ => return Err(format!("Unknown message type: {}", ty).into()),
        })
    }
}

impl OnEvent for ClientEvent {
    fn on_event(self, ctx: &mut EventContext) -> crate::Result<()> {
        match self {
            Self::Initialize(e) => e.on_event(ctx),
            Self::Connect(e) => e.on_event(ctx),
            Self::Disconnect(e) => e.on_event(ctx),
            Self::UpdateActivity(e) => e.on_event(ctx),
            Self::ClearActivity(e) => e.on_event(ctx),
            Self::Shutdown(e) => e.on_event(ctx),
        }
    }
}