rama-http-headers 0.3.0-rc1

use rama_core::telemetry::tracing;
use rama_http_types::{HeaderName, HeaderValue, Method};

use crate::{Error, HeaderDecode, HeaderEncode, TypedHeader};

/// `Access-Control-Request-Method` header, as defined on
/// [mdn](https://developer.mozilla.org/en-US/docs/Web/HTTP/Reference/Headers/Access-Control-Request-Method).
///
/// The `Access-Control-Request-Method` header indicates which method will be
/// used in the actual request as part of the preflight request.
/// # ABNF
///
/// ```text
/// Access-Control-Request-Method: \"Access-Control-Request-Method\" \":\" Method
/// ```
///
/// # Example values
/// * `GET`
///
/// # Examples
///
/// ```
/// use rama_http_headers::AccessControlRequestMethod;
/// use rama_http_types::Method;
///
/// let req_method = AccessControlRequestMethod::from(Method::GET);
/// ```
#[derive(Clone, Debug, PartialEq, Eq, Hash)]
pub struct AccessControlRequestMethod(pub Method);

impl TypedHeader for AccessControlRequestMethod {
    fn name() -> &'static HeaderName {
        &::rama_http_types::header::ACCESS_CONTROL_REQUEST_METHOD
    }
}

impl HeaderDecode for AccessControlRequestMethod {
    fn decode<'i, I: Iterator<Item = &'i HeaderValue>>(values: &mut I) -> Result<Self, Error> {
        values
            .next()
            .and_then(|value| Method::from_bytes(value.as_bytes()).ok())
            .map(AccessControlRequestMethod)
            .ok_or_else(Error::invalid)
    }
}

impl HeaderEncode for AccessControlRequestMethod {
    fn encode<E: Extend<HeaderValue>>(&self, values: &mut E) {
        // For the more common methods, try to use a static string.
        let s = match self.0 {
            Method::GET => "GET",
            Method::POST => "POST",
            Method::PUT => "PUT",
            Method::DELETE => "DELETE",
            Method::HEAD => "HEAD",
            Method::OPTIONS => "OPTIONS",
            Method::CONNECT => "CONNECT",
            Method::PATCH => "PATCH",
            Method::TRACE => "TRACE",
            _ => {
                match HeaderValue::from_str(self.0.as_ref()) {
                    Ok(value) => values.extend(::std::iter::once(value)),
                    Err(err) => {
                        tracing::debug!(
                            "failed to encode access-control-request-method value as header: {err}"
                        );
                    }
                }
                return;
            }
        };

        values.extend(::std::iter::once(HeaderValue::from_static(s)));
    }
}

impl From<Method> for AccessControlRequestMethod {
    fn from(method: Method) -> Self {
        Self(method)
    }
}

impl From<AccessControlRequestMethod> for Method {
    fn from(method: AccessControlRequestMethod) -> Self {
        method.0
    }
}