stoatchat/crates/core/result/src/axum.rs

use axum::{
    http::{header, StatusCode},
    response::IntoResponse,
    Json,
};

use crate::{Error, ErrorType};

/// HTTP response builder for Error enum
impl IntoResponse for Error {
    fn into_response(self) -> axum::response::Response {
        let status = match self.error_type {
            ErrorType::LabelMe => StatusCode::INTERNAL_SERVER_ERROR,

            ErrorType::AlreadyOnboarded => StatusCode::FORBIDDEN,

            ErrorType::UnknownUser => StatusCode::NOT_FOUND,
            ErrorType::InvalidUsername => StatusCode::BAD_REQUEST,
            ErrorType::UsernameTaken => StatusCode::CONFLICT,
            ErrorType::DiscriminatorChangeRatelimited => StatusCode::TOO_MANY_REQUESTS,
            ErrorType::AlreadyFriends => StatusCode::CONFLICT,
            ErrorType::AlreadySentRequest => StatusCode::CONFLICT,
            ErrorType::Blocked => StatusCode::CONFLICT,
            ErrorType::BlockedByOther => StatusCode::FORBIDDEN,
            ErrorType::NotFriends => StatusCode::FORBIDDEN,
            ErrorType::TooManyPendingFriendRequests { .. } => StatusCode::BAD_REQUEST,

            ErrorType::UnknownChannel => StatusCode::NOT_FOUND,
            ErrorType::UnknownMessage => StatusCode::NOT_FOUND,
            ErrorType::UnknownAttachment => StatusCode::BAD_REQUEST,
            ErrorType::CannotDeleteMessage => StatusCode::FORBIDDEN,
            ErrorType::CannotEditMessage => StatusCode::FORBIDDEN,
            ErrorType::CannotJoinCall => StatusCode::BAD_REQUEST,
            ErrorType::TooManyAttachments { .. } => StatusCode::BAD_REQUEST,
            ErrorType::TooManyReplies { .. } => StatusCode::BAD_REQUEST,
            ErrorType::EmptyMessage => StatusCode::UNPROCESSABLE_ENTITY,
            ErrorType::PayloadTooLarge => StatusCode::UNPROCESSABLE_ENTITY,
            ErrorType::CannotRemoveYourself => StatusCode::BAD_REQUEST,
            ErrorType::GroupTooLarge { .. } => StatusCode::FORBIDDEN,
            ErrorType::AlreadyInGroup => StatusCode::CONFLICT,
            ErrorType::NotInGroup => StatusCode::NOT_FOUND,
            ErrorType::AlreadyPinned => StatusCode::BAD_REQUEST,
            ErrorType::NotPinned => StatusCode::BAD_REQUEST,
            ErrorType::InSlowmode { retry_after: _ } => StatusCode::TOO_MANY_REQUESTS,

            ErrorType::CantCreateServers => StatusCode::FORBIDDEN,
            ErrorType::UnknownServer => StatusCode::NOT_FOUND,
            ErrorType::InvalidRole => StatusCode::NOT_FOUND,
            ErrorType::Banned => StatusCode::FORBIDDEN,
            ErrorType::AlreadyInServer => StatusCode::CONFLICT,
            ErrorType::CannotTimeoutYourself => StatusCode::BAD_REQUEST,

            ErrorType::TooManyServers { .. } => StatusCode::BAD_REQUEST,
            ErrorType::TooManyEmbeds { .. } => StatusCode::BAD_REQUEST,
            ErrorType::TooManyEmoji { .. } => StatusCode::BAD_REQUEST,
            ErrorType::TooManyChannels { .. } => StatusCode::BAD_REQUEST,
            ErrorType::TooManyRoles { .. } => StatusCode::BAD_REQUEST,

            ErrorType::ReachedMaximumBots => StatusCode::BAD_REQUEST,
            ErrorType::IsBot => StatusCode::BAD_REQUEST,
            ErrorType::IsNotBot => StatusCode::BAD_REQUEST,
            ErrorType::BotIsPrivate => StatusCode::FORBIDDEN,

            ErrorType::CannotReportYourself => StatusCode::BAD_REQUEST,

            ErrorType::MissingPermission { .. } => StatusCode::FORBIDDEN,
            ErrorType::MissingUserPermission { .. } => StatusCode::FORBIDDEN,
            ErrorType::NotElevated => StatusCode::FORBIDDEN,
            ErrorType::NotPrivileged => StatusCode::FORBIDDEN,
            ErrorType::CannotGiveMissingPermissions => StatusCode::FORBIDDEN,
            ErrorType::NotOwner => StatusCode::FORBIDDEN,
            ErrorType::IsElevated => StatusCode::FORBIDDEN,

            ErrorType::DatabaseError { .. } => StatusCode::INTERNAL_SERVER_ERROR,
            ErrorType::InternalError => StatusCode::INTERNAL_SERVER_ERROR,
            ErrorType::InvalidOperation => StatusCode::BAD_REQUEST,
            ErrorType::InvalidCredentials => StatusCode::UNAUTHORIZED,
            ErrorType::InvalidProperty => StatusCode::BAD_REQUEST,
            ErrorType::InvalidSession => StatusCode::UNAUTHORIZED,
            ErrorType::NotAuthenticated => StatusCode::UNAUTHORIZED,
            ErrorType::DuplicateNonce => StatusCode::CONFLICT,
            ErrorType::VosoUnavailable => StatusCode::BAD_REQUEST,
            ErrorType::NotFound => StatusCode::NOT_FOUND,
            ErrorType::NoEffect => StatusCode::BAD_REQUEST,
            ErrorType::FailedValidation { .. } => StatusCode::BAD_REQUEST,
            ErrorType::LiveKitUnavailable => StatusCode::BAD_REQUEST,
            ErrorType::NotConnected => StatusCode::BAD_REQUEST,
            ErrorType::NotAVoiceChannel => StatusCode::BAD_REQUEST,
            ErrorType::AlreadyConnected => StatusCode::BAD_REQUEST,
            ErrorType::UnknownNode => StatusCode::BAD_REQUEST,
            ErrorType::InvalidFlagValue => StatusCode::BAD_REQUEST,
            ErrorType::FeatureDisabled { .. } => StatusCode::BAD_REQUEST,

            ErrorType::ProxyError => StatusCode::BAD_REQUEST,
            ErrorType::FileTooSmall => StatusCode::UNPROCESSABLE_ENTITY,
            ErrorType::FileTooLarge { .. } => StatusCode::UNPROCESSABLE_ENTITY,
            ErrorType::FileTypeNotAllowed => StatusCode::BAD_REQUEST,
            ErrorType::ImageProcessingFailed => StatusCode::INTERNAL_SERVER_ERROR,
            ErrorType::NoEmbedData => StatusCode::BAD_REQUEST,
            ErrorType::RenderFail => StatusCode::INTERNAL_SERVER_ERROR,
            ErrorType::MissingHeaders => StatusCode::BAD_REQUEST,
            ErrorType::CaptchaFailed => StatusCode::BAD_REQUEST,
            ErrorType::BlockedByShield => StatusCode::BAD_REQUEST,
            ErrorType::UnverifiedAccount => StatusCode::FORBIDDEN,
            ErrorType::EmailFailed => StatusCode::INTERNAL_SERVER_ERROR,
            ErrorType::InvalidToken => StatusCode::UNAUTHORIZED,
            ErrorType::MissingInvite => StatusCode::BAD_REQUEST,
            ErrorType::InvalidInvite => StatusCode::BAD_REQUEST,
            ErrorType::CompromisedPassword => StatusCode::BAD_REQUEST,
            ErrorType::ShortPassword => StatusCode::BAD_REQUEST,
            ErrorType::Blacklisted => {
                return (
                    StatusCode::UNAUTHORIZED,
                    [(header::CONTENT_TYPE, "application/json")],
                    "{\"type\":\"DisallowedContactSupport\", \"note\":\"If you see this messages right here, you're probably doing something you shouldn't be.\"}"
                ).into_response()
            }
            ErrorType::LockedOut => StatusCode::FORBIDDEN,
            ErrorType::TotpAlreadyEnabled => StatusCode::BAD_REQUEST,
            ErrorType::DisallowedMFAMethod => StatusCode::BAD_REQUEST,
            ErrorType::OperationFailed => StatusCode::INTERNAL_SERVER_ERROR,
            ErrorType::IncorrectData { .. } => StatusCode::BAD_REQUEST,
        };

        (status, Json(&self)).into_response()
    }
}