kozabrada123/chorus
kozabrada123
/
chorus
mirror of https://github.com/polyphony-chat/chorus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: add handling for MFA Required response errors

This commit is contained in:
xystrive 2024-07-04 18:37:33 +01:00
parent 2c4e069269
commit a3aa4625f9
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/ratelimiter.rs
View File

@ -14,7 +14,7 @@ use serde_json::from_str;
use crate::{ use crate::{
errors::{ChorusError, ChorusResult}, errors::{ChorusError, ChorusResult},
instance::ChorusUser, instance::ChorusUser,
types::{types::subconfigs::limits::rates::RateLimits, Limit, LimitType, LimitsConfiguration}, types::{types::subconfigs::limits::rates::RateLimits, Limit, LimitType, LimitsConfiguration, MfaRequiredSchema},
}; };
/// Chorus' request struct. This struct is used to send rate-limited requests to the Spacebar server. /// Chorus' request struct. This struct is used to send rate-limited requests to the Spacebar server.
@ -266,7 +266,14 @@ impl ChorusRequest {
async fn interpret_error(response: reqwest::Response) -> ChorusError { async fn interpret_error(response: reqwest::Response) -> ChorusError {
match response.status().as_u16() { match response.status().as_u16() {
401..=403 | 407 => ChorusError::NoPermission, 401 => {
let response = response.text().await.unwrap();
match serde_json::from_str::<MfaRequiredSchema>(&response) {
Ok(response) => ChorusError::MfaRequired { error: response },
Err(_) => ChorusError::NoPermission,
}
}
402..=403 | 407 => ChorusError::NoPermission,
404 => ChorusError::NotFound { 404 => ChorusError::NotFound {
error: response.text().await.unwrap(), error: response.text().await.unwrap(),
}, },