63 lines
2.1 KiB
Rust
63 lines
2.1 KiB
Rust
use actix_web::{
|
|
body::{BodySize, BoxBody, MessageBody},
|
|
dev::{ConnectionInfo, ServiceResponse},
|
|
error::{Error, JsonPayloadError},
|
|
http::StatusCode,
|
|
middleware::{ErrorHandlerResponse, ErrorHandlers},
|
|
web::{Form, Json},
|
|
Either,
|
|
HttpRequest,
|
|
};
|
|
use serde_json::json;
|
|
|
|
use mitra_utils::urls::guess_protocol;
|
|
|
|
use crate::errors::HttpError;
|
|
|
|
pub type FormOrJson<T> = Either<Form<T>, Json<T>>;
|
|
|
|
/// Error handler for 401 Unauthorized
|
|
pub fn create_auth_error_handler<B: MessageBody + 'static>() -> ErrorHandlers<B> {
|
|
// Creates and returns actix middleware
|
|
ErrorHandlers::new()
|
|
.handler(StatusCode::UNAUTHORIZED, |response: ServiceResponse<B>| {
|
|
let response_new = response.map_body(|_, body| {
|
|
if let BodySize::None | BodySize::Sized(0) = body.size() {
|
|
// Insert error description if response body is empty
|
|
// https://github.com/actix/actix-extras/issues/156
|
|
let error_data = json!({
|
|
"message": "auth header is not present",
|
|
});
|
|
return BoxBody::new(error_data.to_string());
|
|
};
|
|
body.boxed()
|
|
});
|
|
Ok(ErrorHandlerResponse::Response(response_new.map_into_right_body()))
|
|
})
|
|
}
|
|
|
|
/// Convert JSON payload deserialization errors into validation errors
|
|
pub fn json_error_handler(
|
|
error: JsonPayloadError,
|
|
_: &HttpRequest,
|
|
) -> Error {
|
|
match error {
|
|
JsonPayloadError::Deserialize(de_error) => {
|
|
HttpError::ValidationError(de_error.to_string()).into()
|
|
},
|
|
other_error => other_error.into(),
|
|
}
|
|
}
|
|
|
|
pub fn get_request_base_url(connection_info: ConnectionInfo) -> String {
|
|
// TODO: HTTP server should set X-Forwarded-Proto header
|
|
// let scheme = connection_info.scheme();
|
|
let host = connection_info.host();
|
|
let scheme = if let Some((hostname, _port)) = host.split_once(':') {
|
|
guess_protocol(hostname)
|
|
} else {
|
|
guess_protocol(host)
|
|
};
|
|
format!("{}://{}", scheme, host)
|
|
}
|