use super::GetChannelMessagesConfigured;
use crate::{
client::Client,
error::Error,
request::{Request, TryIntoRequest},
response::{Response, ResponseFuture, marker::ListBody},
routing::Route,
};
use std::future::IntoFuture;
use twilight_model::{
channel::Message,
id::{
Id,
marker::{ChannelMarker, MessageMarker},
},
};
use twilight_validate::request::{
ValidationError, get_channel_messages_limit as validate_get_channel_messages_limit,
};
struct GetChannelMessagesFields {
limit: Option<u16>,
}
#[must_use = "requests must be configured and executed"]
pub struct GetChannelMessages<'a> {
channel_id: Id<ChannelMarker>,
fields: Result<GetChannelMessagesFields, ValidationError>,
http: &'a Client,
}
impl<'a> GetChannelMessages<'a> {
pub(crate) const fn new(http: &'a Client, channel_id: Id<ChannelMarker>) -> Self {
Self {
channel_id,
fields: Ok(GetChannelMessagesFields { limit: None }),
http,
}
}
pub fn after(self, message_id: Id<MessageMarker>) -> GetChannelMessagesConfigured<'a> {
GetChannelMessagesConfigured::new(
self.http,
self.channel_id,
Some(message_id),
None,
None,
self.fields.map(|fields| fields.limit),
)
}
pub fn around(self, message_id: Id<MessageMarker>) -> GetChannelMessagesConfigured<'a> {
GetChannelMessagesConfigured::new(
self.http,
self.channel_id,
None,
Some(message_id),
None,
self.fields.map(|fields| fields.limit),
)
}
pub fn before(self, message_id: Id<MessageMarker>) -> GetChannelMessagesConfigured<'a> {
GetChannelMessagesConfigured::new(
self.http,
self.channel_id,
None,
None,
Some(message_id),
self.fields.map(|fields| fields.limit),
)
}
pub fn limit(mut self, limit: u16) -> Self {
self.fields = self.fields.and_then(|mut fields| {
validate_get_channel_messages_limit(limit)?;
fields.limit = Some(limit);
Ok(fields)
});
self
}
}
impl IntoFuture for GetChannelMessages<'_> {
type Output = Result<Response<ListBody<Message>>, Error>;
type IntoFuture = ResponseFuture<ListBody<Message>>;
fn into_future(self) -> Self::IntoFuture {
let http = self.http;
match self.try_into_request() {
Ok(request) => http.request(request),
Err(source) => ResponseFuture::error(source),
}
}
}
impl TryIntoRequest for GetChannelMessages<'_> {
fn try_into_request(self) -> Result<Request, Error> {
let fields = self.fields.map_err(Error::validation)?;
Ok(Request::from_route(&Route::GetMessages {
after: None,
around: None,
before: None,
channel_id: self.channel_id.get(),
limit: fields.limit,
}))
}
}