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

use crate::match_rule::MatchRuleError;
use std::{
    convert::TryFrom,
    fmt::{Display, Formatter, Result as FmtResult},
};

/// An enum representing the [message type].
///
/// [message type]: https://dbus.freedesktop.org/doc/dbus-specification.html#message-protocol-types
#[repr(u8)]
#[derive(Debug, PartialOrd, PartialEq, Ord, Eq, Clone)]
pub enum MessageType {
    /// The message is a [`METHOD_CALL`].
    ///
    /// [`METHOD_CALL`]: https://dbus.freedesktop.org/doc/dbus-specification.html#message-protocol-types-method
    MethodCall = 1,
    /// The message is a [`METHOD_RETURN`].
    ///
    /// [`METHOD_RETURN`]: https://dbus.freedesktop.org/doc/dbus-specification.html#message-protocol-types-method
    MethodReturn = 2,
    /// The message is a [`ERROR`].
    ///
    /// [`ERROR`]: https://dbus.freedesktop.org/doc/dbus-specification.html#message-protocol-types-errors
    Error = 3,
    /// The message is a [`SIGNAL`].
    ///
    /// [`SIGNAL`]: https://dbus.freedesktop.org/doc/dbus-specification.html#message-protocol-types-signal
    Signal = 4,
}

impl TryFrom<u8> for MessageType {
    type Error = u8;

    fn try_from(message_type: u8) -> Result<MessageType, u8> {
        match message_type {
            1 => Ok(MessageType::MethodCall),
            2 => Ok(MessageType::MethodReturn),
            3 => Ok(MessageType::Error),
            4 => Ok(MessageType::Signal),
            message_type => Err(message_type),
        }
    }
}

impl Display for MessageType {
    fn fmt(&self, f: &mut Formatter<'_>) -> FmtResult {
        match self {
            MessageType::MethodCall => write!(f, "method_call"),
            MessageType::MethodReturn => write!(f, "method_return"),
            MessageType::Error => write!(f, "error"),
            MessageType::Signal => write!(f, "signal"),
        }
    }
}

impl TryFrom<&str> for MessageType {
    type Error = MatchRuleError;

    fn try_from(r#type: &str) -> Result<Self, MatchRuleError> {
        match r#type {
            "method_call" => Ok(MessageType::MethodCall),
            "method_return" => Ok(MessageType::MethodReturn),
            "error" => Ok(MessageType::Error),
            "signal" => Ok(MessageType::Signal),
            _ => Err(MatchRuleError::TypeUnknown),
        }
    }
}