lighthouse/common/warp_utils/src/json.rs

use bytes::Bytes;
use eth2::{CONTENT_TYPE_HEADER, SSZ_CONTENT_TYPE_HEADER};
use serde::de::DeserializeOwned;
use std::error::Error as StdError;
use warp::{Filter, Rejection};

use crate::reject;

struct Json;

type BoxError = Box<dyn StdError + Send + Sync>;

impl Json {
    fn decode<T: DeserializeOwned>(bytes: Bytes) -> Result<T, BoxError> {
        serde_json::from_slice(&bytes).map_err(Into::into)
    }
}

pub fn json<T: DeserializeOwned + Send>() -> impl Filter<Extract = (T,), Error = Rejection> + Copy {
    warp::header::optional::<String>(CONTENT_TYPE_HEADER)
        .and(warp::body::bytes())
        .and_then(|header: Option<String>, bytes: Bytes| async move {
            if let Some(header) = header
                && header == SSZ_CONTENT_TYPE_HEADER
            {
                return Err(reject::unsupported_media_type(
                    "The request's content-type is not supported".to_string(),
                ));
            }
            Json::decode(bytes)
                .map_err(|err| reject::custom_deserialize_error(format!("{:?}", err)))
        })
}

// Add a json_no_body function to handle the case when no body is provided in the HTTP request
pub fn json_no_body<T: DeserializeOwned + Default + Send>()
-> impl Filter<Extract = (T,), Error = Rejection> + Copy {
    warp::header::optional::<String>(CONTENT_TYPE_HEADER)
        .and(warp::body::bytes())
        .and_then(|header: Option<String>, bytes: Bytes| async move {
            if let Some(header) = header
                && header == SSZ_CONTENT_TYPE_HEADER
            {
                return Err(reject::unsupported_media_type(
                    "The request's content-type is not supported".to_string(),
                ));
            }

            // Handle the case when the HTTP request has no body, i.e., without the -d header
            if bytes.is_empty() {
                return Ok(T::default());
            }

            Json::decode(bytes)
                .map_err(|err| reject::custom_deserialize_error(format!("{:?}", err)))
        })
}