edgee-proxy 1.3.0

use crate::tools::real_ip::Realip;
use anyhow::Context;
use http::header::COOKIE;
use http::{header::HOST, uri::PathAndQuery, HeaderMap};
use hyper::body::Incoming;
use std::{net::SocketAddr, str::FromStr};

pub struct IncomingContext {
    pub body: Incoming,
    pub parts: http::request::Parts,
    pub request: RequestHandle,
}

impl IncomingContext {
    pub fn new(request: http::Request<Incoming>, remote_addr: SocketAddr, proto: &str) -> Self {
        let (parts, body) = request.into_parts();
        let parts_cloned = parts.clone();

        let root_path = PathAndQuery::from_str("/").expect("'/' should be a valid path");
        let path = parts.uri.path_and_query().unwrap_or(&root_path).to_owned();

        // Compute if HTTPS is used using the x-forwarded-proto header is set
        let is_https = if let Some(forwarded_proto) = parts
            .headers
            .get("x-forwarded-proto")
            .and_then(|value| value.to_str().ok())
        {
            forwarded_proto == "https"
        } else {
            proto == "https"
        };

        // debug mode
        let all_cookies = parts.headers.get_all(COOKIE);
        let is_debug_mode = all_cookies
            .iter()
            .filter_map(|cookie| cookie.to_str().ok())
            .any(|cookie| cookie.contains("_edgeedebug=true"));

        let client_ip = Realip::new().get_from_request(&remote_addr, &parts.headers);

        let req = RequestHandle {
            host: match (parts.headers.get(HOST), parts.uri.host()) {
                (None, Some(value)) => Some(String::from(value)),
                (Some(value), _) => Some(value.to_str().unwrap().to_string()),
                (None, None) => None,
            }
            .and_then(|host| host.split(':').next().map(|s| s.to_string()))
            .context("extracting hostname from request")
            .unwrap(),
            path: path.path().to_string(),
            query: path.query().unwrap_or("").to_string(),
            method: parts.method,
            path_and_query: path,
            headers: parts.headers,
            is_https,
            proto: if is_https {
                "https".to_string()
            } else {
                "http".to_string()
            },
            is_debug_mode,
            client_ip,
        };

        Self {
            body,
            parts: parts_cloned,
            request: req,
        }
    }

    pub fn get_request(&self) -> &RequestHandle {
        &self.request
    }
}

#[derive(Debug, Clone)]
pub struct RequestHandle {
    headers: HeaderMap,
    method: http::Method,
    is_https: bool,
    proto: String,
    is_debug_mode: bool,
    client_ip: String,
    host: String,
    path: String,
    query: String,
    path_and_query: PathAndQuery,
}

impl Default for RequestHandle {
    fn default() -> Self {
        Self {
            headers: HeaderMap::new(),
            method: http::Method::GET,
            is_https: false,
            proto: String::new(),
            is_debug_mode: false,
            client_ip: String::new(),
            host: String::new(),
            path: "".to_string(),
            query: "".to_string(),
            path_and_query: PathAndQuery::from_str("/").unwrap(),
        }
    }
}

impl RequestHandle {
    pub fn get_headers(&self) -> &HeaderMap {
        &self.headers
    }

    #[cfg(test)]
    pub fn default_with_headers(headers: HeaderMap) -> Self {
        Self {
            headers,
            method: http::Method::GET,
            is_https: false,
            proto: String::new(),
            is_debug_mode: false,
            client_ip: String::new(),
            host: String::new(),
            path: "".to_string(),
            query: "".to_string(),
            path_and_query: PathAndQuery::from_str("/").unwrap(),
        }
    }

    pub fn get_header(&self, key: impl http::header::AsHeaderName) -> Option<String> {
        self.headers
            .get(key)
            .and_then(|h| h.to_str().ok())
            .map(|s| s.to_string())
    }

    pub fn get_method(&self) -> &http::Method {
        &self.method
    }

    pub fn is_https(&self) -> bool {
        self.is_https
    }

    pub fn get_proto(&self) -> &String {
        &self.proto
    }

    pub fn is_debug_mode(&self) -> bool {
        self.is_debug_mode
    }

    pub fn get_client_ip(&self) -> &String {
        &self.client_ip
    }

    pub fn get_host(&self) -> &String {
        &self.host
    }
    pub fn get_path(&self) -> &String {
        &self.path
    }

    pub fn get_query(&self) -> &String {
        &self.query
    }

    pub fn get_path_and_query(&self) -> &PathAndQuery {
        &self.path_and_query
    }

    pub fn get_content_type(&self) -> String {
        self.get_header(http::header::CONTENT_TYPE)
            .unwrap_or_default()
    }
}