walle-core 0.2.0

Onebot Lib in Rust
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

use crate::{ExtendedMap, MessageSegment};

use super::Message;

pub trait IntoMessage {
    fn into_message(self) -> Message;
}

impl IntoMessage for String {
    fn into_message(self) -> Message {
        vec![MessageSegment::text(self)]
    }
}

impl IntoMessage for &str {
    fn into_message(self) -> Message {
        vec![MessageSegment::text(self.to_string())]
    }
}

impl IntoMessage for Message {
    fn into_message(self) -> Message {
        self
    }
}

macro_rules! impl_self_build {
    ($fname0: ident, $fname1: ident,$sub: tt) => {
        fn $fname0(mut self, extend: ExtendedMap) -> Self {
            self.push(MessageSegment::$sub { extend });
            self
        }
        fn $fname1(mut self) -> Self {
            self.push(MessageSegment::$sub { extend: ExtendedMap::new() });
            self
        }
    };
    ($fname0: ident, $fname1: ident,$sub: tt, $($key: ident: $key_ty: ty),*) => {
        fn $fname0(mut self, $($key: $key_ty),*, extend: ExtendedMap) -> Self {
            self.push(MessageSegment::$sub { $($key ,)* extend, });
            self
        }
        fn $fname1(mut self, $($key: $key_ty),*) -> Self {
            self.push(MessageSegment::$sub { $($key ,)* extend: ExtendedMap::new(), });
            self
        }
    };
}

/// Message 构建 trait
pub trait MessageBuild {
    fn text_with_extend(self, text: String, extend: ExtendedMap) -> Self;
    fn mention_with_extend(self, user_id: String, extend: ExtendedMap) -> Self;
    fn mention_all_with_extend(self, extend: ExtendedMap) -> Self;
    fn image_with_extend(self, file_id: String, extend: ExtendedMap) -> Self;
    fn voice_with_extend(self, file_id: String, extend: ExtendedMap) -> Self;
    fn audio_with_extend(self, file_id: String, extend: ExtendedMap) -> Self;
    fn video_with_extend(self, file_id: String, extend: ExtendedMap) -> Self;
    fn file_with_extend(self, file_id: String, extend: ExtendedMap) -> Self;
    fn location_with_extend(
        self,
        latitude: f64,
        longitude: f64,
        title: String,
        content: String,
        extend: ExtendedMap,
    ) -> Self;
    fn reply_with_extend(self, message_id: String, user_id: String, extend: ExtendedMap) -> Self;

    fn text(self, text: String) -> Self;
    fn mention(self, user_id: String) -> Self;
    fn mention_all(self) -> Self;
    fn image(self, file_id: String) -> Self;
    fn voice(self, file_id: String) -> Self;
    fn audio(self, file_id: String) -> Self;
    fn video(self, file_id: String) -> Self;
    fn file(self, file_id: String) -> Self;
    fn location(self, latitude: f64, longitude: f64, title: String, content: String) -> Self;
    fn reply(self, message_id: String, user_id: String) -> Self;
    fn custom(self, ty: String, data: ExtendedMap) -> Self;
}

impl MessageBuild for Message {
    impl_self_build!(text_with_extend, text, Text, text: String);
    impl_self_build!(mention_with_extend, mention, Mention, user_id: String);
    impl_self_build!(mention_all_with_extend, mention_all, MentionAll);
    impl_self_build!(image_with_extend, image, Image, file_id: String);
    impl_self_build!(voice_with_extend, voice, Voice, file_id: String);
    impl_self_build!(audio_with_extend, audio, Audio, file_id: String);
    impl_self_build!(video_with_extend, video, Video, file_id: String);
    impl_self_build!(file_with_extend, file, File, file_id: String);
    impl_self_build!(
        location_with_extend,
        location,
        Location,
        latitude: f64,
        longitude: f64,
        title: String,
        content: String
    );
    impl_self_build!(
        reply_with_extend,
        reply,
        Reply,
        message_id: String,
        user_id: String
    );
    fn custom(mut self, ty: String, data: ExtendedMap) -> Self {
        self.push(MessageSegment::Custom { ty, data });
        self
    }
}