ircv3_parse 4.0.0

Zero-copy parser for IRCv3 messages
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
83
84
85
86
87
88
89
90
91
92
93
94

use std::fmt;

use proptest::prelude::{prop_compose, prop_oneof, Strategy};

use super::host::rfc1123_strategy;

#[derive(Debug, Clone, PartialEq)]
pub enum SourceType {
    ServerName(String),
    Full(String, String, String),
    NickUser(String, String),
    NickHost(String, String),
    Nick(String),
}

impl fmt::Display for SourceType {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::ServerName(server) => write!(f, "{}", server),
            Self::Full(nick, user, host) => write!(f, "{}!{}@{}", nick, user, host),
            Self::NickUser(nick, user) => write!(f, "{}!{}", nick, user),
            Self::NickHost(nick, host) => write!(f, "{}@{}", nick, host),
            Self::Nick(nick) => write!(f, "{}", nick),
        }
    }
}

impl SourceType {
    pub fn expected_name(&self) -> &str {
        match self {
            Self::ServerName(server) => server,
            Self::Full(nick, _, _) => nick,
            Self::NickUser(nick, _) => nick,
            Self::NickHost(nick, _) => nick,
            Self::Nick(nick) => nick,
        }
    }

    pub fn expected_user(&self) -> Option<&str> {
        match self {
            Self::ServerName(_) => None,
            Self::Full(_, user, _) => Some(user),
            Self::NickUser(_, user) => Some(user),
            Self::NickHost(_, _) => None,
            Self::Nick(_) => None,
        }
    }

    pub fn expected_host(&self) -> Option<&str> {
        match self {
            Self::ServerName(_) => None,
            Self::Full(_, _, host) => Some(host),
            Self::NickUser(_, _) => None,
            Self::NickHost(_, host) => Some(host),
            Self::Nick(_) => None,
        }
    }
}

prop_compose! {
    pub fn source_strategy()(
        source in prop_oneof![
            rfc1123_strategy().prop_map(SourceType::ServerName),

            (nickname_strategy(), username_with_at_strategy(), rfc1123_strategy())
                .prop_map(|(n, u, h)|SourceType::Full(n, u, h)),

            (nickname_strategy(), username_without_at_strategy())
                .prop_map(|(n, u)| SourceType::NickUser(n, u)),

            (nickname_strategy(), rfc1123_strategy())
                .prop_map(|(n, h)| SourceType::NickHost(n, h)) ,

            nickname_strategy().prop_map(SourceType::Nick),
        ]) -> SourceType { source }
}

prop_compose! {
    pub fn nickname_strategy()(
        nickname in "[a-zA-Z][a-zA-Z0-9\\[\\]`\\^{}\\\\]*",
    ) -> String { nickname }
}

prop_compose! {
    pub fn username_with_at_strategy()(
        username in "[^ \r\n\0]+",
    ) -> String { username }
}

prop_compose! {
    pub fn username_without_at_strategy()(
        username in "[^ \r\n\0@]+",
    ) -> String { username }
}