Implement GetChatId for teloxide-core::types::*

This commit is contained in:
Henriquelay 2024-02-06 01:25:41 -03:00
parent df59ceee24
commit 24310df9b3
No known key found for this signature in database
GPG key ID: 02843DDA217C9D81
2 changed files with 98 additions and 8 deletions

View file

@ -44,6 +44,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
- `filter_video_chat_ended` - `filter_video_chat_ended`
- `filter_video_chat_participants_invited` - `filter_video_chat_participants_invited`
- `filter_web_app_data` - `filter_web_app_data`
- Implement `GetChatId` for `teloxide_core::types::{BotCommandScope, Chat, ChatJoinRequest, ChatMemberUpdated, MessageCommon, Recipient, ResponseParameters, TargetMessage, UpdateKind}`.
### Fixed ### Fixed

View file

@ -1,25 +1,114 @@
use crate::types::{CallbackQuery, ChatId, Message, Update}; use crate::types::{
BotCommandScope, CallbackQuery, Chat, ChatId, ChatJoinRequest, ChatMemberUpdated, Message,
MessageCommon, Recipient, ResponseParameters, TargetMessage, Update, UpdateKind,
};
/// Something that may has a chat ID. /// Something that may have a chat ID.
pub trait GetChatId { pub trait GetChatId {
#[must_use] #[must_use]
fn chat_id(&self) -> Option<ChatId>; fn chat_id(&self) -> Option<ChatId>;
} }
impl GetChatId for Message {
fn chat_id(&self) -> Option<ChatId> {
Some(self.chat.id)
}
}
impl GetChatId for CallbackQuery { impl GetChatId for CallbackQuery {
fn chat_id(&self) -> Option<ChatId> { fn chat_id(&self) -> Option<ChatId> {
self.message.as_ref().map(|mes| mes.chat.id) self.message.as_ref().map(|mes| mes.chat.id)
} }
} }
impl GetChatId for MessageCommon {
fn chat_id(&self) -> Option<ChatId> {
self.sender_chat.as_ref().map(|chat| chat.id)
}
}
impl GetChatId for Update { impl GetChatId for Update {
fn chat_id(&self) -> Option<ChatId> { fn chat_id(&self) -> Option<ChatId> {
self.chat().map(|chat| chat.id) self.chat().map(|chat| chat.id)
} }
} }
impl GetChatId for Recipient {
fn chat_id(&self) -> Option<ChatId> {
match self {
Recipient::Id(chat_id) => Some(*chat_id),
Recipient::ChannelUsername(_) => None,
}
}
}
impl GetChatId for BotCommandScope {
fn chat_id(&self) -> Option<ChatId> {
match self {
BotCommandScope::Default
| BotCommandScope::AllPrivateChats
| BotCommandScope::AllGroupChats
| BotCommandScope::AllChatAdministrators => None,
BotCommandScope::Chat { chat_id: recipient }
| BotCommandScope::ChatAdministrators { chat_id: recipient }
| BotCommandScope::ChatMember { chat_id: recipient, .. } => recipient.chat_id(),
}
}
}
impl GetChatId for Chat {
fn chat_id(&self) -> Option<ChatId> {
Some(self.id)
}
}
impl GetChatId for UpdateKind {
fn chat_id(&self) -> Option<ChatId> {
match self {
UpdateKind::Message(message)
| UpdateKind::EditedMessage(message)
| UpdateKind::ChannelPost(message)
| UpdateKind::EditedChannelPost(message) => GetChatId::chat_id(message),
UpdateKind::CallbackQuery(callback_query) => callback_query.chat_id(),
UpdateKind::MyChatMember(chat_member_updated) => chat_member_updated.chat_id(),
UpdateKind::ChatMember(chat_member_updated) => chat_member_updated.chat_id(),
UpdateKind::ChatJoinRequest(chat_join_request) => chat_join_request.chat_id(),
UpdateKind::InlineQuery(_)
| UpdateKind::ChosenInlineResult(_)
| UpdateKind::ShippingQuery(_)
| UpdateKind::PreCheckoutQuery(_)
| UpdateKind::Poll(_)
| UpdateKind::PollAnswer(_)
| UpdateKind::Error(_) => None,
}
}
}
impl GetChatId for ResponseParameters {
fn chat_id(&self) -> Option<ChatId> {
match self {
ResponseParameters::MigrateToChatId(chat_id) => Some(*chat_id),
ResponseParameters::RetryAfter(_) => None,
}
}
}
impl GetChatId for TargetMessage {
fn chat_id(&self) -> Option<ChatId> {
match self {
TargetMessage::Common { chat_id: recipient, .. } => recipient.chat_id(),
TargetMessage::Inline { .. } => None,
}
}
}
/// Implements [`GetChatId`] for all types passed in, as long as they have a
/// `chat` field with a `Chat` type
macro_rules! impl_GetChatId_for_chat_field {
// Comma-separated list of types bound to `t`
($($t:ty),* $(,)?) => {
$(
impl GetChatId for $t {
fn chat_id(&self) -> Option<ChatId> {
Some(self.chat.id)
}
}
)*
};
}
impl_GetChatId_for_chat_field!(Message, ChatMemberUpdated, ChatJoinRequest);