apimock 4.6.6

use console::style;
use http_body_util::BodyExt;
use hyper::header::ORIGIN;
use hyper::http::request::Parts;
use hyper::{body::Incoming, Version};
use serde_json::{to_string_pretty, Value};

use std::time::{SystemTime, UNIX_EPOCH};

use crate::core::{
    config::log_config::verbose_config::VerboseConfig,
    util::http::{content_type_is_application_json, normalize_url_path},
};

#[derive(Debug)]
pub struct ParsedRequest {
    pub url_path: String,
    pub component_parts: Parts,
    pub body_json: Option<Value>,
}

impl ParsedRequest {
    pub async fn from(request: hyper::Request<Incoming>) -> Result<Self, String> {
        let (component_parts, body) = request.into_parts();

        let body_bytes = match body.boxed().collect().await {
            Ok(x) => Some(x.to_bytes()),
            Err(err) => {
                log::warn!("failed to collect request incoming body: {}", err);
                None
            }
        };

        let has_body = body_bytes.is_some() && !body_bytes.as_ref().unwrap().is_empty();

        let mut body_json: Option<Value> = None;
        if has_body {
            let raw_body_json = serde_json::from_slice::<Option<Value>>(&body_bytes.unwrap());

            let _ = match content_type_is_application_json(&component_parts.headers) {
                // case application/json: get json body
                Some(x) if x && raw_body_json.is_err() => {
                    return Err(format!(
                        "failed to get json value from request body: {}",
                        raw_body_json.unwrap_err()
                    ));
                }
                // todo: support other types than json (such as form value) in the future
                Some(x) if !x => {
                    log::warn!("request has body but its content-type is not application/json")
                }
                None => log::warn!("request has body but doesn't have content-type"),
                _ => (),
            };

            if raw_body_json.is_ok() {
                body_json = raw_body_json.unwrap();
            }
        }

        let url_path = normalize_url_path(component_parts.uri.path(), None);

        Ok(ParsedRequest {
            url_path,
            component_parts,
            body_json,
        })
    }

    /// print out logs
    pub fn capture_in_log(&self, verbose: VerboseConfig) {
        // server log (timestamp)
        let now = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_secs();
        let hours = (now / 3600) % 24;
        let minutes = (now / 60) % 60;
        let seconds = now % 60;
        let timestamp = format!("{:02}:{:02}:{:02}", hours, minutes, seconds);

        // request info (url path, origin etc.)
        let version = match self.component_parts.version {
            Version::HTTP_3 => "HTTP/3",
            Version::HTTP_2 => "HTTP/2",
            Version::HTTP_11 => "HTTP/1.1",
            _ => "HTTP/1.0 or earlier, or HTTP/4 or later",
        };

        let origin = if let Some(origin) = self.component_parts.headers.get(ORIGIN) {
            if let Ok(origin) = origin.to_str() {
                Some(origin)
            } else {
                None
            }
        } else {
            None
        };

        // print
        // - server info and request info
        let mut printed = format!(
            "<- {}\n   [{}]",
            style(self.url_path.as_str()).yellow(),
            self.component_parts.method,
        );
        if let Some(origin) = origin {
            printed.push_str(&format!(" [ORIGIN {}]", origin));
        }
        printed.push_str(&format!(
            " [{}] request received (at {} UTC)",
            version, timestamp
        ));

        // - headers
        if verbose.header || verbose.body {
            printed.push_str("\n");
        }
        if verbose.header {
            let headers = self
                .component_parts
                .headers
                .iter()
                .map(|x| format!("\n{}: {}", x.0, x.1.to_str().unwrap()))
                .collect::<String>();
            let printed_headers = format!("{}", headers);
            printed.push_str(&format!(
                "   [request.headers]{}\n",
                style(printed_headers).magenta()
            ));
        }

        // - body (url query, json params)
        let mut is_verbose_body = false;
        if verbose.body {
            let query = self.component_parts.uri.query();
            if let Some(query) = query {
                printed.push_str(&format!("   [request.query] {}\n", query));
                is_verbose_body = true;
            }

            if let Some(request_body_json_value) = &self.body_json {
                printed.push_str("   [request.body.json]\n");

                let body_str = match to_string_pretty(request_body_json_value) {
                    Ok(x) => x,
                    Err(err) => {
                        log::warn!(
                            "failed to prettify JSON: {} ({})",
                            request_body_json_value,
                            err
                        );
                        request_body_json_value.to_string()
                    }
                };
                let styled_body_str = body_str
                    .split("\n")
                    .map(|s| style(s).green().to_string())
                    .collect::<Vec<String>>()
                    .join("\n");
                printed.push_str(styled_body_str.as_str());

                is_verbose_body = true;
            }
        }
        if verbose.header || is_verbose_body {
            printed.push_str("\n");
        }

        log::info!("{}", printed);
    }
}