redact-client 1.5.9

use crate::routes::error::{QueryParamValidationRejection, RelayRejection};
use crate::routes::{
    BadRequestRejection, IframeTokensDoNotMatchRejection, SessionTokenNotFoundRejection,
};
use serde::Serialize;
use std::convert::Infallible;
use warp::http::StatusCode;
use warp::{Rejection, Reply};

/// An API error serializable to JSON.
#[derive(Serialize)]
struct ErrorMessage {
    code: u16,
    message: String,
}

// This function receives a `Rejection` and tries to return a custom
// value, otherwise simply passes the rejection along.
pub async fn handle_rejection(err: Rejection) -> Result<impl Reply, Infallible> {
    let code;
    let message;

    println!("{:?}", err);

    if err.is_not_found() {
        code = StatusCode::NOT_FOUND;
        message = "NOT FOUND";
    } else if err.find::<SessionTokenNotFoundRejection>().is_some() {
        code = StatusCode::UNAUTHORIZED;
        message = "SESSION TOKEN NOT FOUND";
    } else if err.find::<IframeTokensDoNotMatchRejection>().is_some() {
        code = StatusCode::UNAUTHORIZED;
        message = "IFRAME TOKENS DO NOT MATCH";
    } else if err.find::<BadRequestRejection>().is_some() {
        code = StatusCode::BAD_REQUEST;
        message = "BAD REQUEST";
    } else if err.find::<RelayRejection>().is_some() {
        code = StatusCode::INTERNAL_SERVER_ERROR;
        message = "INTERNAL SERVER ERROR - Relay Error";
    } else if err.find::<QueryParamValidationRejection>().is_some() {
        code = StatusCode::BAD_REQUEST;
        message = "INVALID QUERY PARAMETERS";
    } else {
        code = StatusCode::INTERNAL_SERVER_ERROR;
        message = "INTERNAL SERVER ERROR";
    }

    let json = warp::reply::json(&ErrorMessage {
        code: code.as_u16(),
        message: message.into(),
    });

    Ok(warp::reply::with_status(json, code))
}