use crate::{
client::Client,
error::Error,
request::{self, AuditLogReason, Nullable, Request, TryIntoRequest},
response::{Response, ResponseFuture},
routing::Route,
};
use serde::Serialize;
use std::future::IntoFuture;
use twilight_model::{
channel::{thread::AutoArchiveDuration, Channel},
id::{
marker::{ChannelMarker, TagMarker},
Id,
},
};
use twilight_validate::{
channel::{
name as validate_name, rate_limit_per_user as validate_rate_limit_per_user,
ChannelValidationError,
},
request::{audit_reason as validate_audit_reason, ValidationError},
};
#[derive(Serialize)]
struct UpdateThreadFields<'a> {
#[serde(skip_serializing_if = "Option::is_none")]
applied_tags: Option<Nullable<&'a [Id<TagMarker>]>>,
#[serde(skip_serializing_if = "Option::is_none")]
archived: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
auto_archive_duration: Option<AutoArchiveDuration>,
#[serde(skip_serializing_if = "Option::is_none")]
invitable: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
locked: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
name: Option<&'a str>,
#[serde(skip_serializing_if = "Option::is_none")]
rate_limit_per_user: Option<u16>,
}
#[must_use = "requests must be configured and executed"]
pub struct UpdateThread<'a> {
channel_id: Id<ChannelMarker>,
fields: UpdateThreadFields<'a>,
http: &'a Client,
reason: Option<&'a str>,
}
impl<'a> UpdateThread<'a> {
pub(crate) const fn new(http: &'a Client, channel_id: Id<ChannelMarker>) -> Self {
Self {
channel_id,
fields: UpdateThreadFields {
applied_tags: None,
archived: None,
auto_archive_duration: None,
invitable: None,
locked: None,
name: None,
rate_limit_per_user: None,
},
http,
reason: None,
}
}
pub const fn applied_tags(mut self, applied_tags: Option<&'a [Id<TagMarker>]>) -> Self {
self.fields.applied_tags = Some(Nullable(applied_tags));
self
}
pub const fn archived(mut self, archived: bool) -> Self {
self.fields.archived = Some(archived);
self
}
pub const fn auto_archive_duration(
mut self,
auto_archive_duration: AutoArchiveDuration,
) -> Self {
self.fields.auto_archive_duration = Some(auto_archive_duration);
self
}
pub const fn invitable(mut self, invitable: bool) -> Self {
self.fields.invitable = Some(invitable);
self
}
pub const fn locked(mut self, locked: bool) -> Self {
self.fields.locked = Some(locked);
self
}
pub fn name(mut self, name: &'a str) -> Result<Self, ChannelValidationError> {
validate_name(name)?;
self.fields.name = Some(name);
Ok(self)
}
pub const fn rate_limit_per_user(
mut self,
rate_limit_per_user: u16,
) -> Result<Self, ChannelValidationError> {
#[allow(clippy::question_mark)]
if let Err(source) = validate_rate_limit_per_user(rate_limit_per_user) {
return Err(source);
}
self.fields.rate_limit_per_user = Some(rate_limit_per_user);
Ok(self)
}
#[deprecated(since = "0.14.0", note = "use `.await` or `into_future` instead")]
pub fn exec(self) -> ResponseFuture<Channel> {
self.into_future()
}
}
impl<'a> AuditLogReason<'a> for UpdateThread<'a> {
fn reason(mut self, reason: &'a str) -> Result<Self, ValidationError> {
validate_audit_reason(reason)?;
self.reason.replace(reason);
Ok(self)
}
}
impl IntoFuture for UpdateThread<'_> {
type Output = Result<Response<Channel>, Error>;
type IntoFuture = ResponseFuture<Channel>;
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 UpdateThread<'_> {
fn try_into_request(self) -> Result<Request, Error> {
let mut request = Request::builder(&Route::UpdateChannel {
channel_id: self.channel_id.get(),
})
.json(&self.fields)?;
if let Some(reason) = &self.reason {
request = request.headers(request::audit_header(reason)?);
}
Ok(request.build())
}
}
#[cfg(test)]
mod tests {
use super::{UpdateThread, UpdateThreadFields};
use crate::{
request::{Request, TryIntoRequest},
routing::Route,
Client,
};
use std::error::Error;
use twilight_model::id::Id;
#[test]
fn request() -> Result<(), Box<dyn Error>> {
let client = Client::new("token".to_string());
let channel_id = Id::new(123);
let actual = UpdateThread::new(&client, channel_id)
.rate_limit_per_user(60)?
.try_into_request()?;
let expected = Request::builder(&Route::UpdateChannel {
channel_id: channel_id.get(),
})
.json(&UpdateThreadFields {
applied_tags: None,
archived: None,
auto_archive_duration: None,
invitable: None,
locked: None,
name: None,
rate_limit_per_user: Some(60),
})?
.build();
assert_eq!(expected.body(), actual.body());
assert_eq!(expected.path(), actual.path());
assert_eq!(expected.ratelimit_path(), actual.ratelimit_path());
Ok(())
}
}