use crate::{
client::Client,
error::Error,
request::{self, AuditLogReason, AuditLogReasonError, Pending, Request},
routing::Route,
};
use twilight_model::id::{MessageId, WebhookId};
pub struct DeleteWebhookMessage<'a> {
fut: Option<Pending<'a, ()>>,
http: &'a Client,
message_id: MessageId,
reason: Option<String>,
token: String,
webhook_id: WebhookId,
}
impl<'a> DeleteWebhookMessage<'a> {
pub(crate) fn new(
http: &'a Client,
webhook_id: WebhookId,
token: impl Into<String>,
message_id: MessageId,
) -> Self {
Self {
fut: None,
http,
message_id,
reason: None,
token: token.into(),
webhook_id,
}
}
fn request(&self) -> Result<Request, Error> {
let mut request = Request::builder(Route::DeleteWebhookMessage {
message_id: self.message_id.0,
token: self.token.clone(),
webhook_id: self.webhook_id.0,
})
.use_authorization_token(false);
if let Some(reason) = self.reason.as_ref() {
request = request.headers(request::audit_header(reason)?);
}
Ok(request.build())
}
fn start(&mut self) -> Result<(), Error> {
let request = self.request()?;
self.fut.replace(Box::pin(self.http.verify(request)));
Ok(())
}
}
impl<'a> AuditLogReason for DeleteWebhookMessage<'a> {
fn reason(mut self, reason: impl Into<String>) -> Result<Self, AuditLogReasonError> {
self.reason
.replace(AuditLogReasonError::validate(reason.into())?);
Ok(self)
}
}
poll_req!(DeleteWebhookMessage<'_>, ());
#[cfg(test)]
mod tests {
use super::DeleteWebhookMessage;
use crate::{client::Client, request::Request, routing::Route};
use twilight_model::id::{MessageId, WebhookId};
#[test]
fn test_request() {
let client = Client::new("token");
let builder = DeleteWebhookMessage::new(&client, WebhookId(1), "token", MessageId(2));
let actual = builder.request().expect("failed to create request");
let expected = Request::from_route(Route::DeleteWebhookMessage {
message_id: 2,
token: "token".to_owned(),
webhook_id: 1,
});
assert_eq!(expected.body, actual.body);
assert_eq!(expected.path, actual.path);
}
}