/src/server/extractors/headers/cp_connecting_ip.rs

cp_connecting_ip.rs

use std::net::IpAddr;

use axum::http::HeaderMap;

#[derive(Debug, thiserror::Error)]
pub enum IpHeaderError {
    #[error("`CF-Connecting-IP` header is missing")]
    MissingHeader,

    #[error("`CF-Connecting-IP` header contains invalid characters")]
    InvalidChars,

    #[error("Could not parse string into `std::net::IpAddr`")]
    ParseFailure,
}

/// Extracts the value from the "CF-Connecting-IP" header.
pub fn cf_connecting_ip(headers: &HeaderMap) -> Result<IpAddr, IpHeaderError> {
    headers
        .get("CF-Connecting-IP")
        .ok_or(IpHeaderError::MissingHeader)?
        .to_str()
        .map_err(|_| IpHeaderError::InvalidChars)?
        .parse()
        .map_err(|_| IpHeaderError::ParseFailure)
}