twilight_http/request/guild/
update_guild_mfa.rs

1use crate::{
2    client::Client,
3    error::Error,
4    request::{self, AuditLogReason, Request, TryIntoRequest},
5    response::{Response, ResponseFuture},
6    routing::Route,
7};
8use serde::Serialize;
9use std::future::IntoFuture;
10use twilight_model::{
11    guild::MfaLevel,
12    id::{marker::GuildMarker, Id},
13};
14use twilight_validate::request::{audit_reason as validate_audit_reason, ValidationError};
15
16#[derive(Serialize)]
17struct UpdateGuildMfaFields {
18    level: MfaLevel,
19}
20
21/// Update a guild's MFA level.
22#[must_use = "requests must be configured and executed"]
23pub struct UpdateGuildMfa<'a> {
24    fields: UpdateGuildMfaFields,
25    guild_id: Id<GuildMarker>,
26    http: &'a Client,
27    reason: Result<Option<&'a str>, ValidationError>,
28}
29
30impl<'a> UpdateGuildMfa<'a> {
31    pub(crate) const fn new(http: &'a Client, guild_id: Id<GuildMarker>, level: MfaLevel) -> Self {
32        Self {
33            fields: UpdateGuildMfaFields { level },
34            guild_id,
35            http,
36            reason: Ok(None),
37        }
38    }
39}
40
41impl IntoFuture for UpdateGuildMfa<'_> {
42    type Output = Result<Response<MfaLevel>, Error>;
43
44    type IntoFuture = ResponseFuture<MfaLevel>;
45
46    fn into_future(self) -> Self::IntoFuture {
47        let http = self.http;
48
49        match self.try_into_request() {
50            Ok(request) => http.request(request),
51            Err(source) => ResponseFuture::error(source),
52        }
53    }
54}
55
56impl<'a> AuditLogReason<'a> for UpdateGuildMfa<'a> {
57    fn reason(mut self, reason: &'a str) -> Self {
58        self.reason = validate_audit_reason(reason).and(Ok(Some(reason)));
59
60        self
61    }
62}
63
64impl TryIntoRequest for UpdateGuildMfa<'_> {
65    fn try_into_request(self) -> Result<Request, Error> {
66        let mut request = Request::builder(&Route::UpdateGuildMfa {
67            guild_id: self.guild_id.get(),
68        })
69        .json(&self.fields);
70
71        if let Some(reason) = self.reason.map_err(Error::validation)? {
72            request = request.headers(request::audit_header(reason)?);
73        }
74
75        request.build()
76    }
77}