harmony_rust_sdk 0.8.0

Rust library to work with the Harmony chat protocol.
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

pub use crate::api::chat::{
    content::{AttachmentContent, EmbedContent, PhotoContent, TextContent},
    format as text_format, DeleteMessageRequest, Format as TextFormat, FormattedText,
    GetMessageRequest, SendMessageRequest, UpdateMessageTextRequest,
};

use super::{harmonytypes::Metadata, *};

/// Trait that implements convenience methods for [`Message`] type.
pub trait MessageExt {
    /// Get the text content of the message if it has one.
    fn text(&self) -> Option<&str>;
    /// Get the embed content of the message if it has one.
    fn embeds(&self) -> Option<&[Embed]>;
    /// Get the file content of the message if it has one.
    fn files(&self) -> Option<&[Attachment]>;
}

impl MessageExt for Message {
    fn text(&self) -> Option<&str> {
        match self.content.as_ref()?.content.as_ref()? {
            content::Content::TextMessage(text) => text.content.as_ref().map(|f| f.text.as_str()),
            _ => None,
        }
    }

    fn embeds(&self) -> Option<&[Embed]> {
        match self.content.as_ref()?.content.as_ref()? {
            content::Content::EmbedMessage(embeds) => Some(&embeds.embeds),
            _ => None,
        }
    }

    fn files(&self) -> Option<&[Attachment]> {
        match self.content.as_ref()?.content.as_ref()? {
            content::Content::AttachmentMessage(files) => Some(&files.files),
            _ => None,
        }
    }
}

/// Convenience type to create a valid [`SendMessageRequest`].
#[impl_call_action(chat)]
#[into_request("SendMessageRequest")]
#[derive(new, Debug, Clone, self_builder)]
pub struct SendMessage {
    guild_id: u64,
    channel_id: u64,
    #[new(default)]
    content: Content,
    #[new(default)]
    #[builder(setter(strip_option))]
    echo_id: Option<u64>,
    #[new(default)]
    in_reply_to: Option<u64>,
    #[new(default)]
    #[builder(setter(strip_option))]
    overrides: Option<Overrides>,
    #[new(default)]
    #[builder(setter(strip_option))]
    metadata: Option<Metadata>,
}

impl SendMessage {
    pub fn text(mut self, text: impl std::fmt::Display) -> Self {
        self.content.content = Some(content::Content::TextMessage(TextContent {
            content: Some(FormattedText {
                text: text.to_string(),
                format: Vec::new(),
            }),
        }));
        self
    }

    pub fn files(mut self, files: impl Into<Vec<Attachment>>) -> Self {
        self.content.content = Some(content::Content::AttachmentMessage(AttachmentContent {
            files: files.into(),
        }));
        self
    }

    pub fn embed(mut self, embed: impl Into<Vec<Embed>>) -> Self {
        self.content.content = Some(content::Content::EmbedMessage(EmbedContent {
            embeds: embed.into(),
        }));
        self
    }
}