potato/

lib.rs

#[cfg(feature = "acme")]
pub mod acme;
pub mod client;
pub mod global_config;
pub mod server;
pub mod utils;
#[cfg(feature = "webrtc")]
pub mod webrtc;

pub use client::*;
pub use global_config::*;
pub use hipstr;
pub use inventory;
pub use potato_macro::*;
pub use regex;
pub use rust_embed;
pub use serde_json;
pub use server::CorsConfig;
pub use server::*;
pub use utils::ai::*;
pub use utils::refstr::Headers;

#[cfg(all(feature = "jemalloc", not(target_os = "windows")))]
pub use utils::jemalloc_helper::*;

#[cfg(feature = "webrtc")]
pub use webrtc::*;

use anyhow::anyhow;
use chrono::Utc;
use core::str;
use hipstr::{LocalHipByt, LocalHipStr};
use http::Uri;
use http::uri::Scheme;
use rust_embed::Embed;
use sha1::{Digest, Sha1};
use std::any::{Any, TypeId};
use std::borrow::Cow;
use std::cell::RefCell;
use std::fmt;
use std::fs::{File, Metadata};
use std::io::Read;
use std::net::SocketAddr;
use std::path::Path;
use std::str::FromStr;
use std::sync::{Arc, LazyLock};
use std::time::UNIX_EPOCH;
use std::{collections::HashMap, collections::HashSet, future::Future, pin::Pin};
use strum::Display;
use thread_local::ThreadLocal;
use tokio::sync::Mutex;
use tokio::sync::mpsc::Receiver;
use utils::bytes::CompressExt;
use utils::enums::{HttpConnection, HttpContentType};
use utils::number::HttpCodeExt;
use utils::refstr::{HeaderItem, HeaderOrHipStr};
use utils::string::StringExt;
use utils::tcp_stream::{HttpStream, VecU8Ext};

/// 一次性缓存,用于单次请求的前处理、后处理及handler方法间传递参数
#[derive(Debug)]
pub struct OnceCache {
    data: HashMap<(String, TypeId), Box<dyn Any + Send + Sync>>,
}

impl Default for OnceCache {
    fn default() -> Self {
        Self::new()
    }
}

impl OnceCache {
    pub fn new() -> Self {
        Self {
            data: HashMap::new(),
        }
    }

    /// 获取不可变引用
    pub fn get<T: Any + Send + Sync + 'static>(&self, name: &str) -> Option<&T> {
        let key = (name.to_string(), TypeId::of::<T>());
        self.data
            .get(&key)
            .and_then(|boxed| boxed.downcast_ref::<T>())
    }

    /// 获取值或返回默认值（需要类型实现Clone）
    pub fn get_or_default<T: Any + Send + Sync + Clone + 'static>(
        &self,
        name: &str,
        default: T,
    ) -> T {
        self.get::<T>(name).cloned().unwrap_or(default)
    }

    /// 获取可变引用
    pub fn get_mut<T: Any + Send + Sync + 'static>(&mut self, name: &str) -> Option<&mut T> {
        let key = (name.to_string(), TypeId::of::<T>());
        self.data
            .get_mut(&key)
            .and_then(|boxed| boxed.downcast_mut::<T>())
    }

    /// 插入或更新值
    pub fn set<T: Any + Send + Sync + 'static>(&mut self, name: &str, value: T) {
        let key = (name.to_string(), TypeId::of::<T>());
        self.data.insert(key, Box::new(value));
    }

    /// 移除并返回值
    pub fn remove<T: Any + Send + Sync + 'static>(&mut self, name: &str) -> Option<T> {
        let key = (name.to_string(), TypeId::of::<T>());
        self.data
            .remove(&key)
            .and_then(|boxed| boxed.downcast::<T>().ok())
            .map(|boxed| *boxed)
    }

    /// 检查是否包含指定的键
    pub fn contains_key<T: Any + Send + Sync + 'static>(&self, name: &str) -> bool {
        let key = (name.to_string(), TypeId::of::<T>());
        self.data.contains_key(&key)
    }

    /// 清空所有缓存
    pub fn clear(&mut self) {
        self.data.clear();
    }

    /// 获取缓存项数量
    pub fn len(&self) -> usize {
        self.data.len()
    }

    /// 判断缓存是否为空
    pub fn is_empty(&self) -> bool {
        self.data.is_empty()
    }
}

use dashmap::DashMap;
use std::sync::RwLock;
use std::time::{Duration, Instant};

/// SessionCache 错误类型
#[derive(Debug)]
pub enum SessionCacheError {
    /// Token 解析失败（格式错误、签名验证失败等）
    InvalidToken(String),
    /// Token 已过期
    TokenExpired,
    /// Session 已过期
    SessionExpired,
    /// 缺少 Authorization header
    MissingAuthHeader,
    /// 内部错误
    InternalError(String),
}

impl std::fmt::Display for SessionCacheError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            SessionCacheError::InvalidToken(msg) => write!(f, "Invalid token: {msg}"),
            SessionCacheError::TokenExpired => write!(f, "Token has expired"),
            SessionCacheError::SessionExpired => write!(f, "Session has expired"),
            SessionCacheError::MissingAuthHeader => write!(f, "Missing Authorization header"),
            SessionCacheError::InternalError(msg) => write!(f, "Internal error: {msg}"),
        }
    }
}

impl std::error::Error for SessionCacheError {}

// SessionCache 的内部管理器(私有)
// 注意：JWT 密钥现在统一使用 global_config 中的 SERVER_JWT_SECRET
static SESSION_CACHE_MANAGER: std::sync::LazyLock<DashMap<i64, (SessionCache, Instant)>> =
    std::sync::LazyLock::new(DashMap::new);

/// Cookie属性配置
#[derive(Debug, Clone)]
pub struct CookieBuilder {
    /// Cookie名称
    name: String,
    /// Cookie值
    value: String,
    /// 路径（默认"/"）
    path: Option<String>,
    /// 域名
    domain: Option<String>,
    /// 过期时间（UTC时间戳，秒）
    expires: Option<i64>,
    /// 最大存活时间（秒）
    max_age: Option<i64>,
    /// 是否仅HTTPS传输
    secure: bool,
    /// 是否禁止JavaScript访问
    http_only: bool,
    /// SameSite策略: Strict, Lax, None
    same_site: Option<String>,
}

impl CookieBuilder {
    /// 创建新的Cookie
    pub fn new(name: &str, value: &str) -> Self {
        Self {
            name: name.to_string(),
            value: value.to_string(),
            path: Some("/".to_string()),
            domain: None,
            expires: None,
            max_age: None,
            secure: false,
            http_only: false,
            same_site: None,
        }
    }

    /// 设置路径
    pub fn path(mut self, path: &str) -> Self {
        self.path = Some(path.to_string());
        self
    }

    /// 设置域名
    pub fn domain(mut self, domain: &str) -> Self {
        self.domain = Some(domain.to_string());
        self
    }

    /// 设置过期时间（Unix时间戳，秒）
    pub fn expires(mut self, timestamp: i64) -> Self {
        self.expires = Some(timestamp);
        self
    }

    /// 设置最大存活时间（秒）
    pub fn max_age(mut self, seconds: i64) -> Self {
        self.max_age = Some(seconds);
        self
    }

    /// 设置Secure标志（仅HTTPS传输）
    pub fn secure(mut self, secure: bool) -> Self {
        self.secure = secure;
        self
    }

    /// 设置HttpOnly标志（禁止JavaScript访问）
    pub fn http_only(mut self, http_only: bool) -> Self {
        self.http_only = http_only;
        self
    }

    /// 设置SameSite策略 ("Strict", "Lax", "None")
    pub fn same_site(mut self, policy: &str) -> Self {
        self.same_site = Some(policy.to_string());
        self
    }

    /// 生成Set-Cookie header值
    pub fn to_set_cookie_string(&self) -> String {
        let mut parts = vec![format!("{}={}", self.name, self.value)];

        if let Some(ref path) = self.path {
            parts.push(format!("Path={}", path));
        }

        if let Some(ref domain) = self.domain {
            parts.push(format!("Domain={}", domain));
        }

        if let Some(expires) = self.expires {
            // 将Unix时间戳转换为HTTP日期格式
            let datetime = chrono::DateTime::<chrono::Utc>::from_timestamp(expires, 0);
            if let Some(dt) = datetime {
                parts.push(format!(
                    "Expires={}",
                    dt.format("%a, %d %b %Y %H:%M:%S GMT")
                ));
            }
        }

        if let Some(max_age) = self.max_age {
            parts.push(format!("Max-Age={}", max_age));
        }

        if self.secure {
            parts.push("Secure".to_string());
        }

        if self.http_only {
            parts.push("HttpOnly".to_string());
        }

        if let Some(ref same_site) = self.same_site {
            parts.push(format!("SameSite={}", same_site));
        }

        parts.join("; ")
    }

    /// 生成删除cookie的Set-Cookie header值
    pub fn to_delete_cookie_string(&self) -> String {
        let mut parts = vec![
            format!("{}=", self.name),
            "Path=/".to_string(),
            "Expires=Thu, 01 Jan 1970 00:00:00 GMT".to_string(),
        ];

        if let Some(ref domain) = self.domain {
            parts.push(format!("Domain={}", domain));
        }

        parts.join("; ")
    }
}

/// SessionCache数据类型别名
type SessionCacheData = Arc<RwLock<HashMap<(String, TypeId), Box<dyn Any + Send + Sync>>>>;

/// 会话级缓存,用于同一用户的不同请求间传递参数
/// 基于Bearer token中的id区分不同Session
#[derive(Debug, Clone)]
pub struct SessionCache {
    data: SessionCacheData,
    /// 存储从请求中读取的cookies
    request_cookies: Arc<RwLock<HashMap<String, String>>>,
    /// 存储需要设置到响应的cookies（包含完整属性）
    response_cookies: Arc<RwLock<Vec<CookieBuilder>>>,
}

impl Default for SessionCache {
    fn default() -> Self {
        Self::new()
    }
}

impl SessionCache {
    pub fn new() -> Self {
        Self {
            data: Arc::new(RwLock::new(HashMap::new())),
            request_cookies: Arc::new(RwLock::new(HashMap::new())),
            response_cookies: Arc::new(RwLock::new(Vec::new())),
        }
    }

    /// 从请求的Cookie header中解析cookies
    pub fn parse_request_cookies(&mut self, cookie_header: &str) {
        if let Ok(mut cookies) = self.request_cookies.write() {
            for pair in cookie_header.split(';') {
                let pair = pair.trim();
                if let Some((key, value)) = pair.split_once('=') {
                    cookies.insert(key.trim().to_string(), value.trim().to_string());
                }
            }
        }
    }

    /// 获取请求中的cookie值
    pub fn get_cookie(&self, name: &str) -> Option<String> {
        let cookies = self.request_cookies.read().ok()?;
        cookies.get(name).cloned()
    }

    /// 设置响应cookie（简单版本，仅设置名称和值）
    pub fn set_cookie(&self, name: &str, value: &str) {
        if let Ok(mut cookies) = self.response_cookies.write() {
            cookies.push(CookieBuilder::new(name, value));
        }
    }

    /// 设置响应cookie（完整配置版本）
    pub fn set_cookie_with_builder(&self, cookie: CookieBuilder) {
        if let Ok(mut cookies) = self.response_cookies.write() {
            cookies.push(cookie);
        }
    }

    /// 移除响应cookie（设置过期时间为过去）
    pub fn remove_cookie(&self, name: &str) {
        if let Ok(mut cookies) = self.response_cookies.write() {
            cookies.push(CookieBuilder::new(name, ""));
        }
    }

    /// 移除响应cookie（带域名配置）
    pub fn remove_cookie_with_domain(&self, name: &str, domain: &str) {
        if let Ok(mut cookies) = self.response_cookies.write() {
            cookies.push(CookieBuilder::new(name, "").domain(domain));
        }
    }

    /// 将所有待设置的cookies应用到HttpResponse
    pub fn apply_cookies(&self, response: &mut HttpResponse) {
        if let Ok(cookies) = self.response_cookies.read() {
            for cookie in cookies.iter() {
                let cookie_str = if cookie.value.is_empty() {
                    // 移除cookie
                    cookie.to_delete_cookie_string()
                } else {
                    // 设置cookie
                    cookie.to_set_cookie_string()
                };
                response.add_header(Cow::Borrowed("Set-Cookie"), Cow::Owned(cookie_str));
            }
        }
    }

    /// 获取值(需要类型实现Clone)
    pub fn get<T: Any + Send + Sync + Clone + 'static>(&self, name: &str) -> Option<T> {
        let key = (name.to_string(), TypeId::of::<T>());
        let data = self.data.read().ok()?;
        let value = data.get(&key).and_then(|boxed| boxed.downcast_ref::<T>())?;
        Some(value.clone())
    }

    /// 获取值并应用函数
    pub fn with_get<T: Any + Send + Sync + 'static, F, R>(&self, name: &str, f: F) -> Option<R>
    where
        F: FnOnce(&T) -> R,
    {
        let key = (name.to_string(), TypeId::of::<T>());
        let data = self.data.read().ok()?;
        let value = data.get(&key).and_then(|boxed| boxed.downcast_ref::<T>())?;
        Some(f(value))
    }

    /// 获取可变引用并应用函数
    pub fn with_mut<T: Any + Send + Sync + 'static, F, R>(&self, name: &str, f: F) -> Option<R>
    where
        F: FnOnce(&mut T) -> R,
    {
        let key = (name.to_string(), TypeId::of::<T>());
        let mut data = self.data.write().ok()?;
        let value = data
            .get_mut(&key)
            .and_then(|boxed| boxed.downcast_mut::<T>())?;
        Some(f(value))
    }

    /// 插入或更新值
    pub fn set<T: Any + Send + Sync + 'static>(&self, name: &str, value: T) {
        let key = (name.to_string(), TypeId::of::<T>());
        if let Ok(mut data) = self.data.write() {
            data.insert(key, Box::new(value));
        }
    }

    /// 移除并返回值
    pub fn remove<T: Any + Send + Sync + 'static>(&self, name: &str) -> Option<T> {
        let key = (name.to_string(), TypeId::of::<T>());
        let mut data = self.data.write().ok()?;
        data.remove(&key)
            .and_then(|boxed| boxed.downcast::<T>().ok())
            .map(|boxed| *boxed)
    }

    // ==================== 静态方法：Session管理 ====================

    /// 设置JWT签发秘钥（与ServerConfig共享同一个密钥）
    pub async fn set_jwt_secret(secret: &[u8]) {
        let secret_str = String::from_utf8_lossy(secret);
        crate::ServerConfig::set_jwt_secret(secret_str).await;
    }

    /// 获取JWT签发秘钥
    async fn get_jwt_secret() -> Vec<u8> {
        crate::ServerConfig::get_jwt_secret().await.into_bytes()
    }

    /// 签发JWT token
    /// 参数:
    /// - user_id: 用户ID
    /// - ttl: token有效期
    ///
    /// 返回: JWT token字符串
    pub async fn generate_token(
        user_id: i64,
        ttl: std::time::Duration,
    ) -> Result<String, anyhow::Error> {
        use jsonwebtoken::{EncodingKey, Header, encode};
        use serde::{Deserialize, Serialize};

        #[derive(Debug, Serialize, Deserialize)]
        struct SessionClaims {
            sub: i64,        // user_id
            exp: usize,      // token过期时间戳
            iat: usize,      // 签发时间戳
            sess_exp: usize, // session过期时间戳（固定，不随访问更新）
        }

        let now = std::time::SystemTime::now()
            .duration_since(std::time::UNIX_EPOCH)?
            .as_secs() as usize;

        let claims = SessionClaims {
            sub: user_id,
            exp: now + ttl.as_secs() as usize,
            iat: now,
            sess_exp: now + ttl.as_secs() as usize,
        };

        let secret = Self::get_jwt_secret().await;
        let token = encode(
            &Header::default(),
            &claims,
            &EncodingKey::from_secret(&secret),
        )?;

        Ok(token)
    }

    /// 解析JWT token
    /// 返回: (user_id, session_exp_duration)
    pub async fn parse_token(token: &str) -> Result<(i64, Duration), SessionCacheError> {
        use jsonwebtoken::{DecodingKey, Validation, decode};
        use serde::{Deserialize, Serialize};

        #[derive(Debug, Serialize, Deserialize)]
        struct SessionClaims {
            sub: i64,
            exp: usize,
            iat: usize,
            sess_exp: usize,
        }

        let secret = Self::get_jwt_secret().await;
        let token_data = decode::<SessionClaims>(
            token,
            &DecodingKey::from_secret(&secret),
            &Validation::default(),
        )
        .map_err(|e| SessionCacheError::InvalidToken(format!("Token decode failed: {e}")))?;

        let now = std::time::SystemTime::now()
            .duration_since(std::time::UNIX_EPOCH)
            .ok()
            .ok_or_else(|| {
                SessionCacheError::InternalError("Failed to get system time".to_string())
            })?
            .as_secs() as usize;

        if token_data.claims.exp < now {
            return Err(SessionCacheError::TokenExpired);
        }

        // 使用固定的session过期时间，不随访问更新
        let session_exp = token_data.claims.sess_exp;
        if session_exp < now {
            return Err(SessionCacheError::SessionExpired);
        }

        let remaining_secs = session_exp - now;
        Ok((
            token_data.claims.sub,
            Duration::from_secs(remaining_secs as u64),
        ))
    }

    // ==================== 内部缓存管理器 ====================

    /// 获取或创建Session缓存
    /// 参数:
    /// - token: JWT token
    ///
    /// 返回: SessionCache实例
    pub async fn from_token(token: &str) -> Result<Self, SessionCacheError> {
        let (user_id, ttl) = Self::parse_token(token).await?;

        let now = Instant::now();
        let expires_at = now + ttl;

        // 使用 DashMap 的 entry API 一次性完成检查和创建，避免竞态条件
        let mut entry = SESSION_CACHE_MANAGER
            .entry(user_id)
            .or_insert_with(|| (SessionCache::new(), expires_at));

        // 检查是否过期，如果过期则重新创建
        if entry.value().1 < now {
            // 已过期,重新创建
            *entry.value_mut() = (SessionCache::new(), expires_at);
        }

        // 返回session的克隆
        Ok(entry.value().0.clone())
    }

    /// 使指定用户的session失效（用于登出等场景）
    pub fn invalidate(user_id: i64) {
        SESSION_CACHE_MANAGER.remove(&user_id);
    }

    /// 内部方法:清理过期的session缓存（已废弃，使用后台清理任务替代）
    #[allow(dead_code)]
    fn cleanup_expired_internal() {
        use std::sync::atomic::{AtomicU64, Ordering};
        static CALL_COUNTER: AtomicU64 = AtomicU64::new(0);

        // 每100次调用触发一次清理
        if CALL_COUNTER.fetch_add(1, Ordering::Relaxed) % 100 == 0 {
            let now = Instant::now();
            SESSION_CACHE_MANAGER.retain(|_, (_, expires_at)| *expires_at > now);
        }
    }

    /// 内部方法:提供对session manager的访问，用于后台清理任务
    /// 注意：这是一个完全私有的内部方法，仅供服务器清理任务使用
    pub(crate) fn cleanup_expired_sessions() {
        let now = Instant::now();
        SESSION_CACHE_MANAGER.retain(|_, (_, expires_at)| *expires_at > now);
    }
}

/// HTTP conditional preflight result
#[derive(Debug, PartialEq)]
pub enum PreflightResult {
    /// Pass preflight check, can continue processing
    Proceed,
    /// Return 304 Not Modified
    NotModified,
    /// Return 412 Precondition Failed
    PreconditionFailed,
}

#[derive(Debug)]
pub enum HttpRequestParseError {
    BadRequest(String),
    NotImplemented(String),
    ExpectationFailed(String),
    RequestHeaderFieldsTooLarge(String),
    PayloadTooLarge(String),
}

impl fmt::Display for HttpRequestParseError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            HttpRequestParseError::BadRequest(msg) => write!(f, "{msg}"),
            HttpRequestParseError::NotImplemented(msg) => write!(f, "{msg}"),
            HttpRequestParseError::ExpectationFailed(msg) => write!(f, "{msg}"),
            HttpRequestParseError::RequestHeaderFieldsTooLarge(msg) => write!(f, "{msg}"),
            HttpRequestParseError::PayloadTooLarge(msg) => write!(f, "{msg}"),
        }
    }
}

impl std::error::Error for HttpRequestParseError {}

fn parse_declared_trailer_names(raw: Option<&str>) -> HashSet<String> {
    raw.map(|value| {
        value
            .split(',')
            .map(|name| name.trim().to_ascii_lowercase())
            .filter(|name| !name.is_empty())
            .collect::<HashSet<_>>()
    })
    .unwrap_or_default()
}

fn is_forbidden_trailer_field(name: &str) -> bool {
    // RFC 9110/9112: trailers must not carry framing or hop-by-hop control fields.
    matches!(
        name,
        "transfer-encoding"
            | "content-length"
            | "trailer"
            | "host"
            | "connection"
            | "keep-alive"
            | "te"
            | "upgrade"
            | "proxy-authenticate"
            | "proxy-authorization"
    )
}

fn parse_trailer_line(line: &[u8]) -> anyhow::Result<(String, String)> {
    let line = str::from_utf8(line)?.trim();
    let (name, value) = line
        .split_once(':')
        .ok_or_else(|| anyhow!("invalid trailer field line"))?;
    let name = name.trim();
    if name.is_empty() {
        Err(anyhow!("empty trailer field name"))?;
    }
    Ok((name.to_string(), value.trim().to_string()))
}

fn parse_transfer_encoding_tokens(raw: &str) -> anyhow::Result<Vec<String>> {
    let codings = raw
        .split(',')
        .map(|part| part.trim().to_ascii_lowercase())
        .filter(|part| !part.is_empty())
        .collect::<Vec<_>>();
    if codings.is_empty() {
        Err(anyhow!("empty Transfer-Encoding header"))?;
    }
    Ok(codings)
}

/// HTTP date parsing error
#[derive(Debug)]
pub struct HttpDateParseError;

impl std::fmt::Display for HttpDateParseError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "failed to parse HTTP date")
    }
}

impl std::error::Error for HttpDateParseError {}

/// Parse HTTP date format to Unix timestamp
/// Supports RFC 7231 standard HTTP date formats:
/// - RFC 1123: "Mon, 06 Nov 1994 08:49:37 GMT"
/// - RFC 850: "Monday, 06-Nov-94 08:49:37 GMT"
/// - ANSI C asctime(): "Mon Nov  6 08:49:37 1994"
pub fn parse_http_date(date_str: &str) -> Result<u64, HttpDateParseError> {
    // Use simple manual parsing method to handle RFC 1123 format
    // Format: "Fri, 12 Sep 2025 00:00:00 GMT"
    static DATE_REGEX: std::sync::LazyLock<Result<regex::Regex, regex::Error>> =
        std::sync::LazyLock::new(|| {
            regex::Regex::new(r"^\w+, (\d{1,2}) (\w+) (\d{4}) (\d{2}):(\d{2}):(\d{2}) GMT$")
        });

    if let Some(caps) = DATE_REGEX
        .as_ref()
        .ok()
        .and_then(|re| re.captures(date_str))
    {
        let day: u32 = caps[1].parse().map_err(|_| HttpDateParseError)?;
        let month_str = &caps[2];
        let year: i32 = caps[3].parse().map_err(|_| HttpDateParseError)?;
        let hour: u32 = caps[4].parse().map_err(|_| HttpDateParseError)?;
        let minute: u32 = caps[5].parse().map_err(|_| HttpDateParseError)?;
        let second: u32 = caps[6].parse().map_err(|_| HttpDateParseError)?;

        let month = match month_str {
            "Jan" => 1,
            "Feb" => 2,
            "Mar" => 3,
            "Apr" => 4,
            "May" => 5,
            "Jun" => 6,
            "Jul" => 7,
            "Aug" => 8,
            "Sep" => 9,
            "Oct" => 10,
            "Nov" => 11,
            "Dec" => 12,
            _ => return Err(HttpDateParseError),
        };

        if let Some(dt) = chrono::NaiveDate::from_ymd_opt(year, month, day)
            .and_then(|d| d.and_hms_opt(hour, minute, second))
        {
            let timestamp = dt.and_utc().timestamp() as u64;
            return Ok(timestamp);
        }
    }

    // Try RFC 1123 format
    if let Ok(dt) = chrono::DateTime::parse_from_str(date_str, "%a, %d %b %Y %H:%M:%S GMT") {
        let timestamp = dt.timestamp() as u64;
        return Ok(timestamp);
    }

    // Try RFC 850 format
    if let Ok(dt) = chrono::DateTime::parse_from_str(date_str, "%A, %d-%b-%y %H:%M:%S GMT") {
        let timestamp = dt.timestamp() as u64;
        return Ok(timestamp);
    }

    // Try ANSI C asctime() format
    if let Ok(dt) = chrono::NaiveDateTime::parse_from_str(date_str, "%a %b %e %H:%M:%S %Y") {
        let timestamp = dt.and_utc().timestamp() as u64;
        return Ok(timestamp);
    }

    Err(HttpDateParseError)
}

static SERVER_STR: LazyLock<String> =
    LazyLock::new(|| format!("{} {}", env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION")));

type AsyncHttpHandler =
    fn(&mut HttpRequest) -> Pin<Box<dyn Future<Output = HttpResponse> + Send + '_>>;
type AsyncNoSendHttpHandler =
    fn(&mut HttpRequest) -> Pin<Box<dyn Future<Output = HttpResponse> + '_>>;
type SyncHttpHandler = fn(&mut HttpRequest) -> HttpResponse;

#[derive(Clone, Copy)]
pub enum HttpHandler {
    Async(AsyncHttpHandler),
    AsyncNoSend(AsyncNoSendHttpHandler),
    Sync(SyncHttpHandler),
}

pub struct RequestHandlerFlagDoc {
    pub show: bool,
    pub auth: bool,
    pub summary: &'static str,
    pub desp: &'static str,
    pub args: &'static str,
    pub tag: &'static str, // Controller 名称，用于 Swagger 分组
}

impl RequestHandlerFlagDoc {
    pub const fn new(
        show: bool,
        auth: bool,
        summary: &'static str,
        desp: &'static str,
        args: &'static str,
        tag: &'static str,
    ) -> Self {
        RequestHandlerFlagDoc {
            show,
            auth,
            summary,
            desp,
            args,
            tag,
        }
    }
}

pub struct RequestHandlerFlag {
    pub method: HttpMethod,
    pub path: &'static str,
    pub handler: HttpHandler,
    pub doc: RequestHandlerFlagDoc,
}

impl RequestHandlerFlag {
    pub const fn new(
        method: HttpMethod,
        path: &'static str,
        handler: HttpHandler,
        doc: RequestHandlerFlagDoc,
    ) -> Self {
        RequestHandlerFlag {
            method,
            path,
            handler,
            doc,
        }
    }
}

inventory::collect!(RequestHandlerFlag);

/// 异步错误处理器类型
pub type AsyncErrorHandler =
    fn(&mut HttpRequest, anyhow::Error) -> Pin<Box<dyn Future<Output = HttpResponse> + Send + '_>>;

/// 同步错误处理器类型
pub type SyncErrorHandler = fn(&mut HttpRequest, anyhow::Error) -> HttpResponse;

/// 错误处理器枚举，支持异步和同步
#[derive(Clone)]
pub enum ErrorHandler {
    Async(AsyncErrorHandler),
    Sync(SyncErrorHandler),
}

/// 错误处理器注册标志
pub struct ErrorHandlerFlag {
    pub handler: ErrorHandler,
}

impl ErrorHandlerFlag {
    pub const fn new(handler: ErrorHandler) -> Self {
        ErrorHandlerFlag { handler }
    }
}

inventory::collect!(ErrorHandlerFlag);

/// Controller 结构体字段信息
pub struct ControllerStructFieldInfo {
    pub has_once_cache: bool,
    pub has_session_cache: bool,
}

/// Controller 结构体标志，用于在 impl 宏中传递字段信息
pub struct ControllerStructFlag {
    pub struct_name: &'static str,
    pub field_info: ControllerStructFieldInfo,
}

impl ControllerStructFlag {
    pub const fn new(struct_name: &'static str, field_info: ControllerStructFieldInfo) -> Self {
        ControllerStructFlag {
            struct_name,
            field_info,
        }
    }
}

inventory::collect!(ControllerStructFlag);

#[derive(Clone, Copy, Debug, Display, Eq, Hash, PartialEq)]
pub enum HttpMethod {
    GET,
    PUT,
    COPY,
    HEAD,
    LOCK,
    MOVE,
    POST,
    MKCOL,
    PATCH,
    TRACE,
    DELETE,
    UNLOCK,
    CONNECT,
    OPTIONS,
    PROPFIND,
    PROPPATCH,
}

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum HttpRequestTargetForm {
    Origin,
    Absolute,
    Authority,
    Asterisk,
}

#[derive(Debug, Eq, PartialEq)]
pub enum CompressMode {
    None,
    Gzip,
}

pub struct Websocket {
    stream: Arc<Mutex<HttpStream>>,
}

impl Websocket {
    pub async fn connect(url: &str, args: Vec<Headers>) -> anyhow::Result<Self> {
        let mut sess = Session::new();
        let mut req = sess.new_request(HttpMethod::GET, url).await?;
        for arg in args.into_iter() {
            req.apply_header(arg);
        }
        req.apply_header(Headers::Connection("Upgrade".to_string()));
        req.apply_header(Headers::Upgrade("Websocket".to_string()));
        req.apply_header(Headers::Sec_WebSocket_Version("13".to_string()));
        req.apply_header(Headers::Sec_WebSocket_Key("VerySecurity".to_string()));
        let res = sess.do_request(req).await?;
        if res.http_code != 101 {
            let body_str = match &res.body {
                HttpResponseBody::Data(data) => str::from_utf8(&data[..])?,
                HttpResponseBody::Stream(_) => "stream response",
            };
            Err(anyhow!("Server return code[{}]: {body_str}", res.http_code))?;
        }
        let stream = sess
            .sess_impl
            .ok_or_else(|| anyhow!("session impl is null"))?
            .stream;
        let stream = Arc::new(Mutex::new(stream));
        Ok(Self { stream })
    }

    async fn recv_impl(&mut self) -> anyhow::Result<WsFrameImpl> {
        let mut stream = self.stream.lock().await;
        let buf = {
            let mut buf = [0u8; 2];
            stream.read_exact(&mut buf).await?;
            buf
        };
        //let fin = buf[0] & 0b1000_0000 != 0;
        let opcode = buf[0] & 0b0000_1111;
        let payload_len = {
            let payload_len = buf[1] & 0b0111_1111;
            match payload_len {
                126 => {
                    let mut buf = [0u8; 2];
                    stream.read_exact(&mut buf).await?;
                    u16::from_be_bytes(buf) as usize
                }
                127 => {
                    let mut buf = [0u8; 8];
                    stream.read_exact(&mut buf).await?;
                    u64::from_be_bytes(buf) as usize
                }
                _ => payload_len as usize,
            }
        };
        let omask_key = match buf[1] & 0b1000_0000 != 0 {
            true => {
                let mut mask_key = [0u8; 4];
                stream.read_exact(&mut mask_key).await?;
                Some(mask_key)
            }
            false => None,
        };
        let mut payload = vec![0u8; payload_len];
        stream.read_exact(&mut payload).await?;
        if let Some(mask_key) = omask_key {
            for i in 0..payload.len() {
                payload[i] ^= mask_key[i % 4];
            }
        }
        match opcode {
            0x0 => Ok(WsFrameImpl::PartData(payload)),
            0x1 => Ok(WsFrameImpl::Text(payload)),
            0x2 => Ok(WsFrameImpl::Binary(payload)),
            0x8 => Ok(WsFrameImpl::Close),
            0x9 => Ok(WsFrameImpl::Ping),
            0xA => Ok(WsFrameImpl::Pong),
            _ => Err(anyhow::Error::msg("unsupported opcode")),
        }
    }

    pub async fn recv(&mut self) -> anyhow::Result<WsFrame> {
        let mut tmp = vec![];
        loop {
            let timeout = ServerConfig::get_ws_ping_duration().await;
            match tokio::time::timeout(timeout, self.recv_impl()).await {
                Ok(ws_frame) => match ws_frame? {
                    WsFrameImpl::Close => return Err(anyhow::Error::msg("close frame")),
                    WsFrameImpl::Ping => self.send_impl(WsFrameImpl::Pong).await?,
                    WsFrameImpl::Pong => (),
                    WsFrameImpl::Binary(data) => {
                        tmp.extend(data);
                        return Ok(WsFrame::Binary(tmp));
                    }
                    WsFrameImpl::Text(data) => {
                        tmp.extend(data);
                        let ret_str = String::from_utf8(tmp).unwrap_or("".to_string());
                        return Ok(WsFrame::Text(ret_str));
                    }
                    WsFrameImpl::PartData(data) => tmp.extend(data),
                },
                Err(_) => self.send_impl(WsFrameImpl::Ping).await?,
            }
        }
    }

    async fn send_impl(&mut self, frame: WsFrameImpl) -> anyhow::Result<()> {
        let (fin, opcode, payload) = match frame {
            WsFrameImpl::Close => (true, 0x8, vec![]),
            WsFrameImpl::Ping => (true, 0x9, vec![]),
            WsFrameImpl::Pong => (true, 0xA, vec![]),
            WsFrameImpl::Binary(data) => (true, 0x2, data),
            WsFrameImpl::Text(data) => (true, 0x1, data),
            WsFrameImpl::PartData(data) => (false, 0x0, data),
        };
        let payload_len = payload.len();
        let mut buf = vec![];
        buf.push((fin as u8) << 7 | opcode);
        if payload_len < 126 {
            buf.push(payload_len as u8);
        } else if payload_len < 65536 {
            buf.push(126);
            buf.extend((payload_len as u16).to_be_bytes().iter());
        } else {
            buf.push(127);
            buf.extend((payload_len as u64).to_be_bytes().iter());
        }
        let mut stream = self.stream.lock().await;
        stream.write_all(&buf).await?;
        stream.write_all(&payload).await?;
        Ok(())
    }

    pub async fn send_ping(&mut self) -> anyhow::Result<()> {
        self.send_impl(WsFrameImpl::Ping).await
    }

    pub async fn send(&mut self, frame: WsFrame) -> anyhow::Result<()> {
        match frame {
            WsFrame::Binary(data) => self.send_impl(WsFrameImpl::Binary(data)).await,
            WsFrame::Text(text) => {
                self.send_impl(WsFrameImpl::Text(text.as_bytes().to_vec()))
                    .await
            }
        }
    }

    pub async fn send_binary(&mut self, data: Vec<u8>) -> anyhow::Result<()> {
        self.send_impl(WsFrameImpl::Binary(data)).await
    }

    pub async fn send_text(&mut self, data: &str) -> anyhow::Result<()> {
        self.send_impl(WsFrameImpl::Text(data.as_bytes().to_vec()))
            .await
    }
}

#[derive(Debug)]
pub enum WsFrame {
    Binary(Vec<u8>),
    Text(String),
}

pub enum WsFrameImpl {
    Close,
    Ping,
    Pong,
    Binary(Vec<u8>),
    Text(Vec<u8>),
    PartData(Vec<u8>),
}

#[derive(Clone, Debug)]
pub struct PostFile {
    pub filename: LocalHipStr<'static>,
    pub data: LocalHipByt<'static>,
}

unsafe impl Send for PostFile {}

#[derive(Debug)]
pub struct HttpRequest {
    pub method: HttpMethod,
    pub target_form: HttpRequestTargetForm,
    pub url_path: LocalHipStr<'static>,
    pub url_query: HashMap<LocalHipStr<'static>, LocalHipStr<'static>>,
    pub version: u8,
    pub headers: HashMap<HeaderOrHipStr, LocalHipStr<'static>>,
    pub trailers: HashMap<HeaderOrHipStr, LocalHipStr<'static>>,
    pub body: LocalHipByt<'static>,
    pub body_pairs: HashMap<LocalHipStr<'static>, LocalHipStr<'static>>,
    pub body_files: HashMap<LocalHipStr<'static>, PostFile>,
    pub client_addr: Option<SocketAddr>,
    pub exts: HashMap<TypeId, Arc<dyn Any + Send + Sync + 'static>>,
}

unsafe impl Send for HttpRequest {}
unsafe impl Sync for HttpRequest {}

impl Default for HttpRequest {
    fn default() -> Self {
        Self::new()
    }
}

impl HttpRequest {
    fn bad_request(msg: impl Into<String>) -> anyhow::Error {
        anyhow::Error::new(HttpRequestParseError::BadRequest(msg.into()))
    }

    fn expectation_failed(msg: impl Into<String>) -> anyhow::Error {
        anyhow::Error::new(HttpRequestParseError::ExpectationFailed(msg.into()))
    }

    fn not_implemented(msg: impl Into<String>) -> anyhow::Error {
        anyhow::Error::new(HttpRequestParseError::NotImplemented(msg.into()))
    }

    fn request_header_fields_too_large(msg: impl Into<String>) -> anyhow::Error {
        anyhow::Error::new(HttpRequestParseError::RequestHeaderFieldsTooLarge(
            msg.into(),
        ))
    }

    pub fn bad_request_message(err: &anyhow::Error) -> Option<&str> {
        err.downcast_ref::<HttpRequestParseError>()
            .and_then(|parse_err| match parse_err {
                HttpRequestParseError::BadRequest(msg) => Some(msg.as_str()),
                HttpRequestParseError::NotImplemented(_) => None,
                HttpRequestParseError::ExpectationFailed(_) => None,
                HttpRequestParseError::RequestHeaderFieldsTooLarge(_) => None,
                HttpRequestParseError::PayloadTooLarge(_) => None,
            })
    }

    pub fn parse_error_response(err: &anyhow::Error) -> Option<HttpResponse> {
        err.downcast_ref::<HttpRequestParseError>()
            .map(|parse_err| {
                let (status, msg) = match parse_err {
                    HttpRequestParseError::BadRequest(msg) => (400, msg.as_str()),
                    HttpRequestParseError::NotImplemented(msg) => (501, msg.as_str()),
                    HttpRequestParseError::ExpectationFailed(msg) => (417, msg.as_str()),
                    HttpRequestParseError::RequestHeaderFieldsTooLarge(msg) => (431, msg.as_str()),
                    HttpRequestParseError::PayloadTooLarge(msg) => (413, msg.as_str()),
                };
                let mut res = HttpResponse::text(msg.to_string());
                res.http_code = status;
                res.add_header("Connection".into(), "close".into());
                res
            })
    }

    fn ensure_header_section_size(buf: &[u8], max_header_bytes: usize) -> anyhow::Result<()> {
        let header_len = match buf.windows(4).position(|w| w == b"\r\n\r\n") {
            Some(pos) => pos + 4,
            None => buf.len(),
        };
        if header_len > max_header_bytes {
            Err(Self::request_header_fields_too_large(format!(
                "request headers too large: {header_len} bytes exceeds {max_header_bytes} bytes"
            )))?;
        }
        Ok(())
    }

    async fn process_expect_header(
        &self,
        stream: &mut HttpStream,
        has_request_body: bool,
    ) -> anyhow::Result<()> {
        let Some(expect) = self.get_header_key(HeaderItem::Expect) else {
            return Ok(());
        };
        if self.version < 11 {
            Err(Self::expectation_failed(
                "Expect header is not supported for HTTP versions below 1.1",
            ))?;
        }
        let mut has_100_continue = false;
        for token in expect.split(',').map(|token| token.trim()) {
            if token.is_empty() {
                Err(Self::expectation_failed("empty Expect header"))?;
            }
            if token.eq_ignore_ascii_case("100-continue") {
                has_100_continue = true;
                continue;
            }
            Err(Self::expectation_failed(format!(
                "unsupported Expect header: {token}"
            )))?;
        }
        if has_100_continue && has_request_body {
            stream.write_all(b"HTTP/1.1 100 Continue\r\n\r\n").await?;
        }
        Ok(())
    }

    pub fn new() -> Self {
        Self {
            method: HttpMethod::GET,
            target_form: HttpRequestTargetForm::Origin,
            url_path: LocalHipStr::from("/"),
            url_query: HashMap::with_capacity(16),
            version: 11,
            headers: HashMap::with_capacity(16),
            trailers: HashMap::with_capacity(4),
            body: LocalHipByt::new(),
            body_pairs: HashMap::with_capacity(16),
            body_files: HashMap::with_capacity(4),
            client_addr: None,
            exts: HashMap::with_capacity(2),
        }
    }

    pub fn query_string(&self) -> String {
        let mut q = "?".to_string();
        for (k, v) in self.url_query.iter() {
            q.push_str(k);
            q.push('=');
            q.push_str(v);
            q.push('&');
        }
        q.pop();
        q
    }

    fn add_ext<T: Any + Send + Sync + 'static>(&mut self, item: Arc<T>) {
        let type_id = TypeId::of::<T>();
        self.exts.insert(type_id, item);
    }

    fn get_ext<T: Any + Send + Sync + 'static>(&self) -> Option<Arc<T>> {
        self.exts
            .get(&TypeId::of::<T>())
            .and_then(|any| any.clone().downcast().ok())
    }

    fn remove_ext<T: Any + Send + Sync + 'static>(&mut self) -> Option<Arc<T>> {
        self.exts
            .remove(&TypeId::of::<T>())
            .and_then(|any| any.clone().downcast().ok())
    }

    fn parse_path_and_query(&mut self, target: &str) {
        self.url_query.clear();
        match target.find('?') {
            Some(p) => {
                self.url_path = LocalHipStr::from(&target[..p]);
                self.url_query = target[p + 1..]
                    .split('&')
                    .map(|s| s.split_once('=').unwrap_or((s, "")))
                    .map(|(a, b)| (LocalHipStr::from(a), LocalHipStr::from(b)))
                    .collect();
            }
            None => {
                self.url_path = LocalHipStr::from(target);
            }
        }
    }

    fn request_target(&self) -> String {
        if matches!(
            self.target_form,
            HttpRequestTargetForm::Asterisk | HttpRequestTargetForm::Authority
        ) {
            return self.url_path.to_string();
        }
        let mut target = self.url_path.to_string();
        if !self.url_query.is_empty() {
            target.push_str(&self.query_string());
        }
        target
    }

    fn request_version_line(&self) -> String {
        match self.version {
            10 => "HTTP/1.0".to_string(),
            11 => "HTTP/1.1".to_string(),
            // Request-line serialization is HTTP/1.x only. Non-H1 internal markers
            // (e.g. 20/30 for H2/H3) must not leak into an HTTP/1 request line.
            _ => "HTTP/1.1".to_string(),
        }
    }

    fn parse_request_target(&mut self, target: &str) -> anyhow::Result<()> {
        if target == "*" {
            if self.method != HttpMethod::OPTIONS {
                Err(Self::bad_request(
                    "asterisk-form request-target requires OPTIONS",
                ))?;
            }
            self.target_form = HttpRequestTargetForm::Asterisk;
            self.url_query.clear();
            self.url_path = LocalHipStr::from("*");
            return Ok(());
        }

        if target.starts_with('/') {
            if self.method == HttpMethod::CONNECT {
                Err(Self::bad_request(
                    "CONNECT requires authority-form request-target",
                ))?;
            }
            self.target_form = HttpRequestTargetForm::Origin;
            self.parse_path_and_query(target);
            return Ok(());
        }

        if target.contains("://") {
            if self.method == HttpMethod::CONNECT {
                Err(Self::bad_request(
                    "CONNECT requires authority-form, absolute-form is invalid",
                ))?;
            }
            let uri = target
                .parse::<Uri>()
                .map_err(|_| Self::bad_request("invalid absolute-form request-target"))?;
            if uri.scheme().is_none() || uri.authority().is_none() {
                Err(Self::bad_request(
                    "absolute-form request-target must include scheme and authority",
                ))?;
            }
            self.target_form = HttpRequestTargetForm::Absolute;
            let path_and_query = uri.path_and_query().map(|v| v.as_str()).unwrap_or("/");
            self.parse_path_and_query(path_and_query);
            return Ok(());
        }

        if http::uri::Authority::from_str(target).is_ok() {
            if self.method != HttpMethod::CONNECT {
                Err(Self::bad_request(
                    "authority-form request-target is only valid for CONNECT",
                ))?;
            }
            self.target_form = HttpRequestTargetForm::Authority;
            self.url_query.clear();
            self.url_path = LocalHipStr::from(target);
            return Ok(());
        }

        Err(Self::bad_request("unsupported request-target form"))
    }

    pub fn get_uri(&self, is_https: bool) -> anyhow::Result<http::Uri> {
        let mut q = self.url_path.to_string();
        let mut is_first = true;
        for (k, v) in self.url_query.iter() {
            match is_first {
                true => {
                    is_first = false;
                    q.push('?');
                }
                false => q.push('&'),
            }
            q.push_str(k);
            q.push('=');
            q.push_str(v);
        }
        Ok(http::uri::Builder::new()
            .scheme(if is_https { "https" } else { "http" })
            .path_and_query(q)
            .build()?)
    }

    pub fn from_url(url: &str, method: HttpMethod) -> anyhow::Result<(Self, bool, u16)> {
        let uri = url.parse::<Uri>()?;
        let mut req = Self::new();
        req.method = method;
        req.target_form = HttpRequestTargetForm::Origin;
        req.parse_path_and_query(uri.path_and_query().map(|v| v.as_str()).unwrap_or("/"));
        req.headers.insert(
            HeaderOrHipStr::from_str("Host"),
            uri.authority()
                .map(|authority| authority.as_str())
                .unwrap_or("localhost")
                .into(),
        );
        let use_ssl = uri.scheme() == Some(&Scheme::HTTPS);
        let port = uri.port_u16().unwrap_or(if use_ssl { 443 } else { 80 });
        Ok((req, use_ssl, port))
    }

    pub fn set_header(
        &mut self,
        key: impl Into<HeaderOrHipStr>,
        value: impl Into<LocalHipStr<'static>>,
    ) {
        self.headers.insert(key.into(), value.into());
    }

    pub fn get_header(&self, key: &str) -> Option<&str> {
        if let Some(header_item) = HeaderItem::try_from_str(key) {
            if let Some(value) = self.headers.get(&HeaderOrHipStr::HeaderItem(header_item)) {
                return Some(&value[..]);
            }
        }
        self.headers
            .get(&HeaderOrHipStr::HipStr(LocalHipStr::from(key)))
            .map(|a| &a[..])
    }

    pub fn get_header_key(&self, key: HeaderItem) -> Option<&str> {
        self.headers.get(&key.into()).map(|a| &a[..])
    }

    pub fn set_trailer(
        &mut self,
        key: impl Into<HeaderOrHipStr>,
        value: impl Into<LocalHipStr<'static>>,
    ) {
        self.trailers.insert(key.into(), value.into());
    }

    pub fn get_trailer(&self, key: &str) -> Option<&str> {
        if let Some(header_item) = HeaderItem::try_from_str(key) {
            if let Some(value) = self.trailers.get(&HeaderOrHipStr::HeaderItem(header_item)) {
                return Some(&value[..]);
            }
        }
        self.trailers
            .get(&HeaderOrHipStr::HipStr(LocalHipStr::from(key)))
            .map(|a| &a[..])
    }

    pub fn get_header_accept_encoding(&self) -> CompressMode {
        Self::negotiate_accept_encoding(
            self.get_header_key(HeaderItem::Accept_Encoding)
                .unwrap_or(""),
        )
    }

    fn negotiate_accept_encoding(header: &str) -> CompressMode {
        let mut explicit_gzip_q: Option<u16> = None;
        let mut wildcard_q: Option<u16> = None;

        for item in header.split(',') {
            let trimmed = item.trim();
            if trimmed.is_empty() {
                continue;
            }

            let mut parts = trimmed.split(';');
            let coding = parts.next().unwrap_or("").trim().to_ascii_lowercase();
            if coding.is_empty() {
                continue;
            }

            let mut quality = 1000u16;
            let mut malformed_q = false;
            for param in parts {
                let param = param.trim();
                if param.is_empty() {
                    continue;
                }
                let mut key_val = param.splitn(2, '=');
                let key = key_val.next().unwrap_or("").trim().to_ascii_lowercase();
                if key != "q" {
                    continue;
                }
                let val = key_val.next().unwrap_or("").trim();
                if let Some(parsed_q) = Self::parse_qvalue_thousandths(val) {
                    quality = parsed_q;
                } else {
                    malformed_q = true;
                }
                break;
            }

            if malformed_q {
                continue;
            }

            match coding.as_str() {
                "gzip" => {
                    explicit_gzip_q =
                        Some(explicit_gzip_q.map_or(quality, |prev| prev.max(quality)));
                }
                "*" => {
                    wildcard_q = Some(wildcard_q.map_or(quality, |prev| prev.max(quality)));
                }
                _ => {}
            }
        }

        let selected_q = explicit_gzip_q.or(wildcard_q).unwrap_or(0);
        if selected_q > 0 {
            CompressMode::Gzip
        } else {
            CompressMode::None
        }
    }

    fn parse_qvalue_thousandths(raw: &str) -> Option<u16> {
        let val = raw.trim();
        if val == "1" || val == "1.0" || val == "1.00" || val == "1.000" {
            return Some(1000);
        }
        if val == "0" {
            return Some(0);
        }
        let frac = val.strip_prefix("0.")?;
        if frac.is_empty() || frac.len() > 3 || !frac.chars().all(|ch| ch.is_ascii_digit()) {
            return None;
        }
        let mut digits = frac.to_string();
        while digits.len() < 3 {
            digits.push('0');
        }
        digits.parse::<u16>().ok()
    }

    pub fn get_header_host(&self) -> Option<&str> {
        self.get_header_key(HeaderItem::Host)
    }

    pub fn get_header_connection(&self) -> HttpConnection {
        if let Some(conn) = self.get_header_key(HeaderItem::Connection) {
            HttpConnection::from_str(conn).unwrap_or(HttpConnection::Close)
        } else if self.version >= 11 {
            HttpConnection::KeepAlive
        } else {
            HttpConnection::Close
        }
    }

    pub fn get_header_content_length(&self) -> usize {
        self.get_header_key(HeaderItem::Content_Length)
            .map_or(0, |val| val.parse::<usize>().unwrap_or(0))
    }

    fn parse_header_content_length(&self) -> anyhow::Result<Option<usize>> {
        let Some(raw_val) = self.get_header_key(HeaderItem::Content_Length) else {
            return Ok(None);
        };
        let value = raw_val.trim();
        if value.is_empty() {
            Err(anyhow!("empty Content-Length header"))?
        }
        Ok(Some(value.parse::<usize>()?))
    }

    fn has_chunked_transfer_encoding(&self) -> anyhow::Result<bool> {
        let Some(raw_val) = self.get_header_key(HeaderItem::Transfer_Encoding) else {
            return Ok(false);
        };
        let codings: Vec<String> = raw_val
            .split(',')
            .map(|part| part.trim().to_ascii_lowercase())
            .filter(|part| !part.is_empty())
            .collect();
        if codings.is_empty() {
            Err(Self::bad_request("empty Transfer-Encoding header"))?
        }
        if codings.len() == 1 && codings[0] == "chunked" {
            return Ok(true);
        }
        Err(Self::not_implemented(format!(
            "unsupported Transfer-Encoding: {raw_val}"
        )))
    }

    async fn read_chunked_body(
        buf: &mut Vec<u8>,
        stream: &mut HttpStream,
        hdr_len: usize,
        allowed_trailers: &HashSet<String>,
    ) -> anyhow::Result<(
        LocalHipByt<'static>,
        HashMap<HeaderOrHipStr, LocalHipStr<'static>>,
        usize,
    )> {
        let mut cursor = hdr_len;
        let mut body = Vec::new();
        let mut trailers = HashMap::with_capacity(4);
        let mut tmp_buf = [0u8; 4096];

        loop {
            let line_end = loop {
                if let Some(pos) = buf[cursor..].windows(2).position(|part| part == b"\r\n") {
                    break cursor + pos;
                }
                let n = stream.read(&mut tmp_buf).await?;
                if n == 0 {
                    Err(anyhow::Error::msg("connection closed"))?;
                }
                buf.extend(&tmp_buf[..n]);
            };

            let chunk_size = {
                let size_line = str::from_utf8(&buf[cursor..line_end])?;
                let size_token = size_line
                    .split_once(';')
                    .map_or(size_line, |(size, _)| size)
                    .trim();
                if size_token.is_empty() {
                    Err(anyhow!("invalid chunk size"))?;
                }
                usize::from_str_radix(size_token, 16)?
            };
            cursor = line_end + 2;

            if chunk_size == 0 {
                let trailer_end = loop {
                    let line_start = cursor;
                    let line_end = loop {
                        if let Some(pos) = buf[cursor..].windows(2).position(|part| part == b"\r\n")
                        {
                            break cursor + pos;
                        }
                        let n = stream.read(&mut tmp_buf).await?;
                        if n == 0 {
                            Err(anyhow::Error::msg("connection closed"))?;
                        }
                        buf.extend(&tmp_buf[..n]);
                    };
                    cursor = line_end + 2;
                    if line_end == line_start {
                        break cursor;
                    }

                    let (name, value) = parse_trailer_line(&buf[line_start..line_end])?;
                    let name_lower = name.to_ascii_lowercase();
                    if is_forbidden_trailer_field(&name_lower) {
                        Err(anyhow!("forbidden trailer field: {name}"))?;
                    }
                    if !allowed_trailers.contains(&name_lower) {
                        Err(anyhow!("unexpected trailer field: {name}"))?;
                    }
                    trailers.insert(HeaderOrHipStr::from_str(&name), value.into());
                };
                cursor = trailer_end;
                break;
            }

            while buf.len() < cursor + chunk_size + 2 {
                let n = stream.read(&mut tmp_buf).await?;
                if n == 0 {
                    Err(anyhow::Error::msg("connection closed"))?;
                }
                buf.extend(&tmp_buf[..n]);
            }
            body.extend_from_slice(&buf[cursor..cursor + chunk_size]);
            if &buf[cursor + chunk_size..cursor + chunk_size + 2] != b"\r\n" {
                Err(anyhow!("invalid chunk terminator"))?;
            }
            cursor += chunk_size + 2;
        }

        Ok((LocalHipByt::from(body), trailers, cursor - hdr_len))
    }

    pub fn get_header_content_type<'a>(&'a self) -> Option<HttpContentType<'a>> {
        HttpContentType::from_str(self.get_header_key(HeaderItem::Content_Type).unwrap_or(""))
    }

    pub fn is_websocket(&self) -> bool {
        if self.method != HttpMethod::GET {
            return false;
        }
        if self.get_header_connection() != HttpConnection::Upgrade {
            return false;
        }
        if self
            .get_header_key(HeaderItem::Upgrade)
            .is_some_and(|val| val.to_lowercase() != "websocket")
        {
            return false;
        }
        if self
            .get_header("Sec-WebSocket-Version")
            .is_some_and(|val| val != "13")
        {
            return false;
        }
        if self
            .get_header("Sec-WebSocket-Key")
            .is_some_and(|val| val.is_empty())
        {
            return false;
        }
        true
    }

    pub async fn upgrade_websocket(&mut self) -> anyhow::Result<Websocket> {
        if !self.is_websocket() {
            Err(anyhow!("it is not a websocket request"))?;
        }
        let ws_key = self
            .get_header("Sec-WebSocket-Key")
            .unwrap_or("")
            .to_string();
        // let ws_ext = req.get_header("Sec-WebSocket-Extensions").unwrap_or("".to_string());
        let stream = match self.remove_ext::<Mutex<HttpStream>>() {
            Some(stream) => stream,
            None => Err(anyhow!("connot get stream"))?,
        };
        {
            let mut stream = stream.lock().await;
            let res = HttpResponse::from_websocket(&ws_key);
            stream.write_all(&res.as_bytes(CompressMode::None)).await?;
        }
        Ok(Websocket { stream })
    }

    pub async fn get_client_addr(&self) -> anyhow::Result<SocketAddr> {
        if let Some(addr) = self.client_addr {
            return Ok(addr);
        }
        match self.get_ext::<SocketAddr>() {
            Some(addr) => Ok(*addr),
            None => Err(anyhow!("no addr info")),
        }
    }

    async fn from_stream_impl(
        buf: &mut Vec<u8>,
        stream: &mut HttpStream,
    ) -> anyhow::Result<(Self, usize)> {
        let mut tmp_buf = [0u8; 4096];
        let (mut req, hdr_len) = loop {
            match HttpRequest::from_headers_part(&buf[..])? {
                Some((req, hdr_len)) => break (req, hdr_len),
                None => {
                    let n = stream.read(&mut tmp_buf).await?;
                    if n == 0 {
                        return Err(anyhow::Error::msg("connection closed"));
                    }
                    buf.extend(&tmp_buf[0..n]);
                }
            }
        };
        let has_chunked_transfer_encoding = req.has_chunked_transfer_encoding()?;
        let mut content_length = 0usize;
        if has_chunked_transfer_encoding {
            if req.get_header_key(HeaderItem::Content_Length).is_some() {
                Err(Self::bad_request(
                    "conflicting headers: Transfer-Encoding and Content-Length",
                ))?;
            }
        } else {
            content_length = req
                .parse_header_content_length()
                .map_err(|err| Self::bad_request(err.to_string()))?
                .unwrap_or(0);
        }

        let has_request_body = has_chunked_transfer_encoding || content_length > 0;
        req.process_expect_header(stream, has_request_body).await?;

        let bdy_len;
        if has_chunked_transfer_encoding {
            let allowed_trailers =
                parse_declared_trailer_names(req.get_header_key(HeaderItem::Trailer));
            let (body, trailers, consumed_len) =
                Self::read_chunked_body(buf, stream, hdr_len, &allowed_trailers)
                    .await
                    .map_err(|err| Self::bad_request(err.to_string()))?;
            req.body = body;
            req.trailers = trailers;
            bdy_len = consumed_len;
        } else {
            while hdr_len + content_length > buf.len() {
                let t = stream.read(&mut tmp_buf).await?;
                if t == 0 {
                    return Err(anyhow::Error::msg("connection closed"));
                }
                buf.extend(&tmp_buf[0..t]);
            }
            if content_length > 0 {
                req.body = LocalHipByt::from(&buf[hdr_len..hdr_len + content_length]);
            }
            bdy_len = content_length;
        }

        // 先获取Content-Type的字符串值，避免借用冲突
        let content_type_str = {
            req.get_header_key(HeaderItem::Content_Type)
                .map(|s| s.to_string())
        };

        // 根据内容类型字符串处理请求体
        if let Some(content_type_str) = content_type_str {
            // 解析内容类型
            let content_type_parsed = HttpContentType::from_str(&content_type_str);

            match content_type_parsed {
                Some(HttpContentType::ApplicationJson) => {
                    if let Ok(body_str) = std::str::from_utf8(&req.body) {
                        if let Ok(serde_json::Value::Object(obj)) =
                            serde_json::from_str::<serde_json::Value>(body_str)
                        {
                            for (k, v) in obj {
                                req.body_pairs
                                    .insert(LocalHipStr::from(k), LocalHipStr::from(v.to_string()));
                            }
                        }
                    }
                }
                Some(HttpContentType::ApplicationXWwwFormUrlencoded) => {
                    if let Ok(body_str) = std::str::from_utf8(&req.body) {
                        body_str.split('&').for_each(|s| {
                            if let Some((a, b)) = s.split_once('=') {
                                req.body_pairs
                                    .insert(a.url_decode().into(), b.url_decode().into());
                            }
                        });
                    }
                }
                Some(HttpContentType::MultipartFormData(boundary)) => {
                    if let Ok(body_str) = std::str::from_utf8(&req.body) {
                        let split_str = ssformat!(64, "--{boundary}");
                        for mut s in body_str.split(split_str.as_str()) {
                            if s == "--" {
                                break;
                            }
                            if s.ends_with("\r\n") {
                                s = &s[..s.len() - 2];
                            }
                            if let Some((key_str, content)) = s.split_once("\r\n\r\n") {
                                let keys: Vec<&str> = key_str
                                    .split("\r\n")
                                    .map(|p| p.split(";").collect::<Vec<_>>())
                                    .collect::<Vec<_>>()
                                    .into_iter()
                                    .flatten()
                                    .collect();
                                let mut name = None;
                                let mut filename = None;
                                for key in keys.into_iter() {
                                    if let Some((k, v)) = key.trim().split_once('=') {
                                        if k == "name" {
                                            name = Some(LocalHipStr::from(&v[1..v.len() - 1]));
                                        } else if k == "filename" {
                                            filename = Some(LocalHipStr::from(&v[1..v.len() - 1]));
                                        }
                                    }
                                }
                                if let Some(name) = name {
                                    if let Some(filename) = filename {
                                        let data = LocalHipByt::from(content.as_bytes());
                                        req.body_files.insert(name, PostFile { filename, data });
                                    } else {
                                        req.body_pairs.insert(name, LocalHipStr::from(content));
                                    }
                                }
                            }
                        }
                    }
                }
                None => {}
            }
        }
        Ok((req, hdr_len + bdy_len))
    }

    pub async fn from_stream(
        buf: &mut Vec<u8>,
        stream: Arc<Mutex<HttpStream>>,
    ) -> anyhow::Result<(Self, usize)> {
        let mut stream = stream.lock().await;
        Self::from_stream_impl(buf, &mut stream).await
    }

    pub fn from_headers_part(buf: &[u8]) -> anyhow::Result<Option<(Self, usize)>> {
        let max_header_count = ServerConfig::get_max_header_count();
        let max_header_bytes = ServerConfig::get_max_header_bytes();
        Self::ensure_header_section_size(buf, max_header_bytes)?;

        let mut headers = vec![httparse::EMPTY_HEADER; max_header_count];
        let (rreq, n) = {
            let mut req = httparse::Request::new(&mut headers);
            let n = match httparse::ParserConfig::default().parse_request(&mut req, buf) {
                Ok(httparse::Status::Complete(n)) => n,
                Ok(httparse::Status::Partial) => return Ok(None),
                Err(httparse::Error::TooManyHeaders) => {
                    Err(Self::request_header_fields_too_large(format!(
                        "too many request headers: exceeds configured limit {max_header_count}"
                    )))?
                }
                Err(err) => Err(anyhow!(err))?,
            };
            (req, n)
        };
        let parsed_header_count = rreq.headers.iter().filter(|h| !h.name.is_empty()).count();
        if parsed_header_count > max_header_count {
            Err(Self::request_header_fields_too_large(format!(
                "too many request headers: {parsed_header_count} exceeds {max_header_count}"
            )))?;
        }
        let mut req = HttpRequest::new();
        let mut content_length_seen: Option<String> = None;
        let mut host_header_count = 0usize;
        let mut has_valid_host = false;
        req.method = {
            let method = rreq
                .method
                .ok_or_else(|| anyhow!("Missing HTTP method in request"))?;
            match method.len() {
                3 if method == "GET" => HttpMethod::GET,
                3 if method == "PUT" => HttpMethod::PUT,
                4 if method == "COPY" => HttpMethod::COPY,
                4 if method == "HEAD" => HttpMethod::HEAD,
                4 if method == "LOCK" => HttpMethod::LOCK,
                4 if method == "MOVE" => HttpMethod::MOVE,
                4 if method == "POST" => HttpMethod::POST,
                5 if method == "MKCOL" => HttpMethod::MKCOL,
                5 if method == "PATCH" => HttpMethod::PATCH,
                5 if method == "TRACE" => HttpMethod::TRACE,
                6 if method == "DELETE" => HttpMethod::DELETE,
                6 if method == "UNLOCK" => HttpMethod::UNLOCK,
                7 if method == "OPTIONS" => HttpMethod::OPTIONS,
                7 if method == "CONNECT" => HttpMethod::CONNECT,
                8 if method == "PROPFIND" => HttpMethod::PROPFIND,
                9 if method == "PROPPATCH" => HttpMethod::PROPPATCH,
                _ => Err(Self::not_implemented(format!(
                    "unsupported method: {method}"
                )))?,
            }
        };
        let target = rreq
            .path
            .ok_or_else(|| anyhow!("Missing HTTP path in request"))?;
        req.parse_request_target(target)?;
        req.version = rreq.version.unwrap_or(1) + 10;
        for h in rreq.headers.iter() {
            if h.name.is_empty() {
                break;
            }
            let header_value = str::from_utf8(h.value)?;
            let normalized_header_value = header_value.trim();
            if h.name.eq_ignore_ascii_case("Content-Length") {
                let cl = normalized_header_value;
                if cl.is_empty() {
                    Err(anyhow!("empty Content-Length header"))?;
                }
                if let Some(prev) = &content_length_seen {
                    if prev != cl {
                        Err(anyhow!("conflicting duplicate Content-Length headers"))?;
                    }
                } else {
                    content_length_seen = Some(cl.to_string());
                }
            }
            if h.name.eq_ignore_ascii_case("Host") {
                host_header_count += 1;
                if host_header_count > 1 {
                    Err(Self::bad_request("multiple Host headers are not allowed"))?;
                }
                if normalized_header_value.is_empty() {
                    Err(Self::bad_request("empty Host header"))?;
                }
                if http::uri::Authority::from_str(normalized_header_value).is_err() {
                    Err(Self::bad_request("invalid Host header"))?;
                }
                has_valid_host = true;
            }
            if h.name.eq_ignore_ascii_case("Expect") {
                let expect_key: HeaderOrHipStr = HeaderItem::Expect.into();
                if let Some(existing) = req.headers.get(&expect_key) {
                    req.headers.insert(
                        expect_key,
                        LocalHipStr::from(format!(
                            "{}, {normalized_header_value}",
                            existing.as_str()
                        )),
                    );
                } else {
                    req.headers
                        .insert(expect_key, LocalHipStr::from(normalized_header_value));
                }
            } else {
                req.headers.insert(
                    HeaderOrHipStr::from_str(h.name),
                    LocalHipStr::from(normalized_header_value),
                );
            }
        }
        if req.version >= 11 && !has_valid_host {
            Err(Self::bad_request("missing required Host header"))?;
        }
        Ok(Some((req, n)))
    }

    /// Check HTTP conditional preflight headers to determine if special status codes should be returned
    ///
    /// This method handles the following HTTP conditional headers:
    /// - If-Modified-Since: Return 304 if resource is not modified
    /// - If-None-Match: Return 304 if ETag matches
    /// - If-Match: Return 412 if ETag doesn't match
    /// - If-Unmodified-Since: Return 412 if resource is modified
    ///
    /// # Parameters
    /// - `meta`: File metadata (optional)
    /// - `etag`: Resource's ETag value (optional)
    ///
    /// # Return Values
    /// - `PreflightResult::Proceed`: Pass preflight check, can continue processing
    /// - `PreflightResult::NotModified`: Should return 304 status code
    /// - `PreflightResult::PreconditionFailed`: Should return 412 status code
    pub fn check_precondition_headers(
        &self,
        meta: Option<&Metadata>,
        etag: Option<&str>,
    ) -> PreflightResult {
        use crate::utils::refstr::HeaderItem;
        use std::time::UNIX_EPOCH;

        let is_get_or_head = matches!(self.method, HttpMethod::GET | HttpMethod::HEAD);

        // Get file's last modified time (Unix timestamp in seconds)
        let last_modified_timestamp = meta
            .and_then(|m| m.modified().ok())
            .and_then(|t| t.duration_since(UNIX_EPOCH).ok())
            .map(|d| d.as_secs());

        // Check If-Match header (if exists and doesn't match, return 412)
        if let Some(if_match) = self.get_header_key(HeaderItem::If_Match) {
            if if_match != "*" {
                if let Some(current_etag) = etag {
                    // Parse ETag list in If-Match
                    let match_found = if_match
                        .split(',')
                        .map(|s| s.trim())
                        .any(|expected_etag| expected_etag == current_etag);

                    if !match_found {
                        return PreflightResult::PreconditionFailed;
                    }
                } else {
                    // No ETag but client requires match, return 412
                    return PreflightResult::PreconditionFailed;
                }
            }
        }

        // Check If-Unmodified-Since header (if resource is modified, return 412)
        if let Some(if_unmodified_since) = self.get_header_key(HeaderItem::If_Unmodified_Since) {
            if let Some(last_modified) = last_modified_timestamp {
                if let Ok(since_timestamp) = parse_http_date(if_unmodified_since) {
                    if last_modified > since_timestamp {
                        return PreflightResult::PreconditionFailed;
                    }
                }
            }
        }

        // Check If-None-Match header
        if let Some(if_none_match) = self.get_header_key(HeaderItem::If_None_Match) {
            if if_none_match == "*" {
                // If resource exists, GET/HEAD -> 304, others -> 412.
                if etag.is_some() || meta.is_some() {
                    return if is_get_or_head {
                        PreflightResult::NotModified
                    } else {
                        PreflightResult::PreconditionFailed
                    };
                }
            } else if let Some(current_etag) = etag {
                // Check if ETag is in If-None-Match list.
                let match_found = if_none_match
                    .split(',')
                    .map(|s| s.trim())
                    .any(|expected_etag| expected_etag == current_etag);

                if match_found {
                    return if is_get_or_head {
                        PreflightResult::NotModified
                    } else {
                        PreflightResult::PreconditionFailed
                    };
                }
            }
        }

        // Check If-Modified-Since header (if resource is not modified, return 304)
        // Note: only applies to GET/HEAD and only when there's no If-None-Match header.
        if is_get_or_head && self.get_header_key(HeaderItem::If_None_Match).is_none() {
            if let Some(if_modified_since) = self.get_header_key(HeaderItem::If_Modified_Since) {
                if let Some(last_modified) = last_modified_timestamp {
                    if let Ok(since_timestamp) = parse_http_date(if_modified_since) {
                        if last_modified <= since_timestamp {
                            return PreflightResult::NotModified;
                        }
                    }
                }
            }
        }

        PreflightResult::Proceed
    }

    pub fn as_bytes(&self) -> Vec<u8> {
        let use_chunked = self
            .get_header_key(HeaderItem::Transfer_Encoding)
            .map(|encodings| {
                encodings
                    .split(',')
                    .map(|coding| coding.trim())
                    .any(|coding| coding.eq_ignore_ascii_case("chunked"))
            })
            .unwrap_or(false);

        let declared_trailer_names =
            parse_declared_trailer_names(self.get_header_key(HeaderItem::Trailer));
        let mut outbound_trailers: Vec<(String, String)> = Vec::with_capacity(self.trailers.len());
        for (key, value) in self.trailers.iter() {
            let key_str = key.to_str();
            let lower = key_str.to_ascii_lowercase();
            if is_forbidden_trailer_field(&lower) {
                continue;
            }
            if !declared_trailer_names.is_empty() && !declared_trailer_names.contains(&lower) {
                continue;
            }
            outbound_trailers.push((key_str.to_string(), value.to_string()));
        }

        let mut req_str = format!(
            "{} {} {}\r\n",
            self.method,
            self.request_target(),
            self.request_version_line()
        );
        for (k, v) in self.headers.iter() {
            if let HeaderOrHipStr::HeaderItem(HeaderItem::Content_Length) = k {
                continue;
            }
            req_str.push_str(&format!("{}: {v}\r\n", k.to_str()));
        }
        if use_chunked {
            if declared_trailer_names.is_empty() && !outbound_trailers.is_empty() {
                let trailer_names = outbound_trailers
                    .iter()
                    .map(|(name, _)| name.as_str())
                    .collect::<Vec<_>>()
                    .join(", ");
                req_str.push_str(&format!("Trailer: {trailer_names}\r\n"));
            }
            req_str.push_str("\r\n");
            let mut ret = req_str.as_bytes().to_vec();
            if self.body.is_empty() {
                if outbound_trailers.is_empty() {
                    ret.extend_from_slice(b"0\r\n\r\n");
                } else {
                    ret.extend_from_slice(b"0\r\n");
                    for (name, value) in outbound_trailers.iter() {
                        ret.extend_from_slice(format!("{name}: {value}\r\n").as_bytes());
                    }
                    ret.extend_from_slice(b"\r\n");
                }
            } else {
                ret.extend_from_slice(format!("{:x}\r\n", self.body.len()).as_bytes());
                ret.extend(&self.body[..]);
                if outbound_trailers.is_empty() {
                    ret.extend_from_slice(b"\r\n0\r\n\r\n");
                } else {
                    ret.extend_from_slice(b"\r\n0\r\n");
                    for (name, value) in outbound_trailers.iter() {
                        ret.extend_from_slice(format!("{name}: {value}\r\n").as_bytes());
                    }
                    ret.extend_from_slice(b"\r\n");
                }
            }
            ret
        } else {
            req_str.push_str(&format!(
                "{}: {}\r\n",
                HeaderItem::Content_Length.to_str(),
                self.body.len()
            ));
            req_str.push_str("\r\n");
            let mut ret = req_str.as_bytes().to_vec();
            ret.extend(&self.body[..]);
            ret
        }
    }
}

#[derive(Debug)]
pub enum HttpResponseBody {
    Data(Vec<u8>),
    Stream(Receiver<Vec<u8>>),
}

pub struct HttpResponseBodyStream<'a> {
    body: &'a mut HttpResponseBody,
    data_consumed: bool,
}

impl HttpResponseBody {
    pub async fn data(&mut self) -> &[u8] {
        // 先检查是否是 Stream 类型
        let is_stream = matches!(self, HttpResponseBody::Stream(_));

        if is_stream {
            // 从 Stream 中替换出 rx，接收所有数据
            if let HttpResponseBody::Stream(mut rx) =
                std::mem::replace(self, HttpResponseBody::Data(vec![]))
            {
                let mut data = Vec::with_capacity(1024);
                while let Some(chunk) = rx.recv().await {
                    data.extend_from_slice(&chunk);
                }
                *self = HttpResponseBody::Data(data);
            }
        }

        match self {
            HttpResponseBody::Data(data) => data.as_slice(),
            HttpResponseBody::Stream(_) => &[], // Should not reach here
        }
    }

    pub fn stream_data(&mut self) -> HttpResponseBodyStream<'_> {
        HttpResponseBodyStream {
            body: self,
            data_consumed: false,
        }
    }
}

impl HttpResponseBodyStream<'_> {
    pub async fn next(&mut self) -> Option<Vec<u8>> {
        match self.body {
            HttpResponseBody::Data(data) => {
                if self.data_consumed {
                    None
                } else {
                    self.data_consumed = true;
                    Some(data.clone())
                }
            }
            HttpResponseBody::Stream(rx) => rx.recv().await,
        }
    }
}

#[derive(Debug)]
pub struct HttpResponse {
    pub version: String,
    pub http_code: u16,
    pub headers: HashMap<Cow<'static, str>, Cow<'static, str>>,
    pub trailers: HashMap<Cow<'static, str>, Cow<'static, str>>,
    pub body: HttpResponseBody,
}
unsafe impl Send for HttpResponse {}
unsafe impl Sync for HttpResponse {}
impl Clone for HttpResponse {
    fn clone(&self) -> Self {
        Self {
            version: self.version.clone(),
            http_code: self.http_code,
            headers: self.headers.clone(),
            trailers: self.trailers.clone(),
            body: match &self.body {
                HttpResponseBody::Data(data) => HttpResponseBody::Data(data.clone()),
                HttpResponseBody::Stream(_) => HttpResponseBody::Data(vec![]), // Clone stream as empty data
            },
        }
    }
}

macro_rules! make_resp_by_text {
    ($fn_name:ident, $cnt_type:expr) => {
        pub fn $fn_name(body: impl Into<String>) -> Self {
            let body = body.into();
            Self {
                version: "HTTP/1.1".into(),
                http_code: 200,
                headers: Self::default_headers($cnt_type),
                trailers: HashMap::with_capacity(4),
                body: HttpResponseBody::Data(body.as_bytes().to_vec()),
            }
        }
    };
}

macro_rules! make_resp_by_binary {
    ($fn_name:ident, $cnt_type:expr) => {
        pub fn $fn_name(body: &[u8]) -> Self {
            Self {
                version: "HTTP/1.1".into(),
                http_code: 200,
                headers: Self::default_headers($cnt_type),
                trailers: HashMap::with_capacity(4),
                body: HttpResponseBody::Data(body.to_vec()),
            }
        }
    };
}

impl Default for HttpResponse {
    fn default() -> Self {
        Self::new()
    }
}

impl HttpResponse {
    make_resp_by_text!(html, "text/html");
    make_resp_by_text!(css, "text/css");
    make_resp_by_text!(csv, "text/csv");
    make_resp_by_text!(js, "text/javascript");
    make_resp_by_text!(text, "text/plain");
    make_resp_by_text!(json, "application/json");
    make_resp_by_text!(xml, "application/xml");
    make_resp_by_binary!(png, "image/png");

    fn default_headers(
        cnt_type: impl Into<String>,
    ) -> HashMap<Cow<'static, str>, Cow<'static, str>> {
        let now = Utc::now();
        let current_ts = now.timestamp();

        static TL_TIMESTAMP: ThreadLocal<RefCell<(i64, Cow<'static, str>)>> = ThreadLocal::new();
        let mut tl_timestamp = TL_TIMESTAMP.get_or_default().borrow_mut();
        let date_str = if current_ts == tl_timestamp.0 {
            tl_timestamp.1.clone()
        } else {
            let new_date: Cow<'_, str> = now.format("%a, %d %b %Y %H:%M:%S GMT").to_string().into();
            *tl_timestamp = (current_ts, new_date.clone());
            new_date
        };

        [
            ("Date".into(), date_str),
            ("Server".into(), SERVER_STR.clone().into()),
            ("Connection".into(), "keep-alive".into()),
            ("Content-Type".into(), cnt_type.into().into()),
            ("Pragma".into(), "no-cache".into()),
            ("Cache-Control".into(), "no-cache".into()),
        ]
        .into()
    }

    pub fn new() -> Self {
        Self {
            version: "".into(),
            http_code: 0,
            headers: HashMap::with_capacity(16),
            trailers: HashMap::with_capacity(4),
            body: HttpResponseBody::Data(vec![]),
        }
    }

    pub fn add_header(&mut self, key: Cow<'static, str>, value: Cow<'static, str>) {
        self.headers.insert(key, value);
    }

    pub fn not_found() -> Self {
        let mut ret = Self::html("404 not found");
        ret.http_code = 404;
        ret
    }

    pub fn error(payload: impl Into<String>) -> Self {
        let mut ret = Self::html(payload);
        ret.http_code = 500;
        ret
    }

    pub fn empty() -> Self {
        Self::html("")
    }

    pub fn chunked(rx: Receiver<Vec<u8>>) -> Self {
        Self {
            version: "HTTP/1.1".into(),
            http_code: 200,
            headers: [
                ("Transfer-Encoding".into(), "chunked".into()),
                ("Content-Type".into(), "application/octet-stream".into()),
                ("Cache-Control".into(), "no-cache".into()),
                ("Connection".into(), "keep-alive".into()),
            ]
            .into(),
            trailers: HashMap::with_capacity(4),
            body: HttpResponseBody::Stream(rx),
        }
    }

    /// Create a SSE response with chunked transfer encoding
    pub fn sse(rx: Receiver<Vec<u8>>) -> Self {
        let mut res = Self::chunked(rx);
        res.add_header("Content-Type".into(), "text/event-stream".into());
        res
    }

    pub fn from_file(path: &str, download: bool, meta: Option<Metadata>) -> Self {
        let mut buffer = vec![];
        if let Ok(mut file) = File::open(path) {
            _ = file.read_to_end(&mut buffer);
        }
        Self::from_mem_file(path, buffer, download, meta)
    }

    pub fn from_mem_file(
        path: &str,
        data: Vec<u8>,
        download: bool,
        meta: Option<Metadata>,
    ) -> Self {
        if let Some(meta) = meta {
            let mut ret = Self::from_mem_file(path, data, download, None);
            // Add Last-Modified header
            if let Ok(modified) = meta.modified() {
                if let Ok(duration) = modified.duration_since(UNIX_EPOCH) {
                    let modified_time =
                        chrono::DateTime::<chrono::Utc>::from(UNIX_EPOCH + duration);
                    ret.add_header(
                        "Last-Modified".into(),
                        modified_time
                            .format("%a, %d %b %Y %H:%M:%S GMT")
                            .to_string()
                            .into(),
                    );
                }
            }

            // Add ETag header (format: "hex-file-modified-time-hex-file-size")
            if let Ok(modified) = meta.modified() {
                if let Ok(duration) = modified.duration_since(UNIX_EPOCH) {
                    let modified_secs = duration.as_secs();
                    let file_size = meta.len();
                    let etag = format!("\"{:x}-{:x}\"", modified_secs, file_size);
                    ret.add_header("ETag".into(), etag.into());
                }
            }
            ret
        } else {
            let mut ret = Self::empty();
            let mime_type = match path.split('.').next_back() {
                Some("css") => "text/css",
                Some("csv") => "text/csv",
                Some("htm") => "text/html",
                Some("html") => "text/html",
                Some("js") => "application/javascript",
                Some("json") => "application/json",
                Some("pdf") => "application/pdf",
                Some("xml") => "application/xml",
                _ if path.ends_with('/') => "text/html",
                _ => "application/octet-stream",
            };
            ret.add_header("Content-Type".into(), mime_type.into());
            if download {
                let file = match path.rfind('/') {
                    Some(p) => &path[p + 1..],
                    None => path,
                };
                if !file.is_empty() {
                    ret.add_header(
                        "Content-Disposition".into(),
                        format!("attachment; filename={file}").into(),
                    );
                }
            }
            ret.body = HttpResponseBody::Data(data);
            ret
        }
    }

    pub fn from_websocket(ws_key: &str) -> Self {
        #[allow(deprecated)]
        let ws_accept = {
            let mut sha1 = Sha1::default();
            sha1.update(ws_key);
            sha1.update(&b"258EAFA5-E914-47DA-95CA-C5AB0DC85B11"[..]);
            base64::encode(sha1.finalize())
        };
        Self {
            version: "HTTP/1.1".into(),
            http_code: 101,
            headers: [
                (
                    "Date".into(),
                    Utc::now()
                        .format("%a, %d %b %Y %H:%M:%S GMT")
                        .to_string()
                        .into(),
                ),
                ("Server".into(), SERVER_STR.clone().into()),
                ("Connection".into(), "Upgrade".into()),
                ("Upgrade".into(), "websocket".into()),
                ("Sec-WebSocket-Accept".into(), ws_accept.into()),
            ]
            .into(),
            trailers: HashMap::with_capacity(4),
            body: HttpResponseBody::Data(vec![]),
        }
    }

    pub fn add_trailer(&mut self, key: Cow<'static, str>, value: Cow<'static, str>) {
        self.trailers.insert(key, value);
    }

    pub fn get_trailer(&self, key: &str) -> Option<&str> {
        self.trailers.get(key).map(|v| v.as_ref())
    }

    fn status_disallows_response_body(status: u16) -> bool {
        (100..200).contains(&status) || status == 204 || status == 304
    }

    fn method_disallows_response_body(request_method: Option<HttpMethod>) -> bool {
        request_method == Some(HttpMethod::HEAD)
    }

    fn transfer_encoding_has_chunked(raw: &str) -> anyhow::Result<bool> {
        let codings = parse_transfer_encoding_tokens(raw)?;
        if codings.iter().any(|coding| coding == "chunked") {
            if codings.last().is_some_and(|coding| coding == "chunked") {
                return Ok(true);
            }
            Err(anyhow!(
                "invalid Transfer-Encoding order: chunked must be final coding"
            ))?;
        }
        Err(anyhow!("unsupported Transfer-Encoding: {raw}"))
    }

    async fn read_chunked_body(
        buf: &mut Vec<u8>,
        stream: &mut HttpStream,
        hdr_len: usize,
        allowed_trailers: &HashSet<String>,
    ) -> anyhow::Result<(
        Vec<u8>,
        HashMap<Cow<'static, str>, Cow<'static, str>>,
        usize,
    )> {
        let mut cursor = hdr_len;
        let mut body = Vec::new();
        let mut trailers = HashMap::with_capacity(4);
        let mut tmp_buf = [0u8; 4096];

        loop {
            let line_end = loop {
                if let Some(pos) = buf[cursor..].windows(2).position(|part| part == b"\r\n") {
                    break cursor + pos;
                }
                let n = stream.read(&mut tmp_buf).await?;
                if n == 0 {
                    Err(anyhow::Error::msg("connection closed"))?;
                }
                buf.extend(&tmp_buf[..n]);
            };

            let chunk_size = {
                let size_line = str::from_utf8(&buf[cursor..line_end])?;
                let size_token = size_line
                    .split_once(';')
                    .map_or(size_line, |(size, _)| size)
                    .trim();
                if size_token.is_empty() {
                    Err(anyhow!("invalid chunk size"))?;
                }
                usize::from_str_radix(size_token, 16)?
            };
            cursor = line_end + 2;

            if chunk_size == 0 {
                loop {
                    let line_start = cursor;
                    let line_end = loop {
                        if let Some(pos) = buf[cursor..].windows(2).position(|part| part == b"\r\n")
                        {
                            break cursor + pos;
                        }
                        let n = stream.read(&mut tmp_buf).await?;
                        if n == 0 {
                            Err(anyhow::Error::msg("connection closed"))?;
                        }
                        buf.extend(&tmp_buf[..n]);
                    };
                    cursor = line_end + 2;
                    if line_end == line_start {
                        break;
                    }

                    let (name, value) = parse_trailer_line(&buf[line_start..line_end])?;
                    let name_lower = name.to_ascii_lowercase();
                    if is_forbidden_trailer_field(&name_lower) {
                        Err(anyhow!("forbidden trailer field: {name}"))?;
                    }
                    if !allowed_trailers.contains(&name_lower) {
                        Err(anyhow!("unexpected trailer field: {name}"))?;
                    }
                    trailers.insert(name.into(), value.into());
                }
                break;
            }

            while buf.len() < cursor + chunk_size + 2 {
                let n = stream.read(&mut tmp_buf).await?;
                if n == 0 {
                    Err(anyhow::Error::msg("connection closed"))?;
                }
                buf.extend(&tmp_buf[..n]);
            }
            body.extend_from_slice(&buf[cursor..cursor + chunk_size]);
            if &buf[cursor + chunk_size..cursor + chunk_size + 2] != b"\r\n" {
                Err(anyhow!("invalid chunk terminator"))?;
            }
            cursor += chunk_size + 2;
        }

        Ok((body, trailers, cursor - hdr_len))
    }

    pub fn as_bytes(&self, mut cmode: CompressMode) -> Vec<u8> {
        match &self.body {
            HttpResponseBody::Data(data) => {
                let suppress_body = Self::status_disallows_response_body(self.http_code);
                let mut payload_tmp: Vec<u8> = vec![];
                if cmode == CompressMode::Gzip
                    && data.len() >= 32
                    && self.get_header("Content-Encoding").is_none()
                    && !suppress_body
                {
                    if let Ok(compressed_data) = data.compress() {
                        payload_tmp = compressed_data;
                    }
                }
                let mut payload_ref = if payload_tmp.is_empty() {
                    cmode = CompressMode::None;
                    data.as_slice()
                } else {
                    payload_tmp.as_slice()
                };
                if suppress_body {
                    cmode = CompressMode::None;
                    payload_ref = &[];
                }
                //
                let mut ret = smallstr::SmallString::<[u8; 4096]>::new();
                let status_str = self.http_code.http_code_to_desp();
                ret.push_str(&ssformat!(
                    64,
                    "{} {} {status_str}\r\n",
                    self.version,
                    self.http_code
                ));
                for (key, value) in self.headers.iter() {
                    if key == "Content-Length" || key.eq_ignore_ascii_case("Transfer-Encoding") {
                        continue;
                    }
                    ret.push_str(&ssformat!(512, "{key}: {value}\r\n"));
                }
                if !suppress_body {
                    ret.push_str(&ssformat!(64, "Content-Length: {}\r\n", payload_ref.len()));
                }
                if cmode == CompressMode::Gzip && !suppress_body {
                    ret.push_str("Content-Encoding: gzip\r\n");
                }
                ret.push_str("\r\n");
                let mut ret: Vec<u8> = ret.as_bytes().to_vec();
                ret.extend(payload_ref);
                ret
            }
            HttpResponseBody::Stream(_) => vec![], // Stream responses are handled separately
        }
    }

    /// Write response to stream, handling both Data and Stream body types
    pub async fn write_to_stream(
        &mut self,
        stream: &mut crate::utils::tcp_stream::HttpStream,
        cmode: CompressMode,
        request_method: Option<HttpMethod>,
    ) -> anyhow::Result<()> {
        let suppress_body_by_status = Self::status_disallows_response_body(self.http_code);
        let suppress_body_by_method = Self::method_disallows_response_body(request_method);
        let suppress_body = suppress_body_by_status || suppress_body_by_method;
        let no_content_encoding = self.get_header("Content-Encoding").is_none();
        let declared_trailer_names = parse_declared_trailer_names(self.get_header("Trailer"));
        let mut outbound_stream_trailers: Vec<(String, String)> =
            Vec::with_capacity(self.trailers.len());
        for (key, value) in self.trailers.iter() {
            let lower = key.to_ascii_lowercase();
            if is_forbidden_trailer_field(&lower) {
                continue;
            }
            if !declared_trailer_names.is_empty() && !declared_trailer_names.contains(&lower) {
                continue;
            }
            outbound_stream_trailers.push((key.to_string(), value.to_string()));
        }
        match &mut self.body {
            HttpResponseBody::Data(data) => {
                let mut payload_tmp: Vec<u8> = vec![];
                let mut cmode = cmode;
                if cmode == CompressMode::Gzip
                    && data.len() >= 32
                    && no_content_encoding
                    && !suppress_body_by_status
                {
                    if let Ok(compressed_data) = data.compress() {
                        payload_tmp = compressed_data;
                    }
                }
                let mut payload_ref = if payload_tmp.is_empty() {
                    cmode = CompressMode::None;
                    data.as_slice()
                } else {
                    payload_tmp.as_slice()
                };
                if suppress_body_by_status {
                    cmode = CompressMode::None;
                    payload_ref = &[];
                }

                let mut ret = smallstr::SmallString::<[u8; 4096]>::new();
                let status_str = self.http_code.http_code_to_desp();
                ret.push_str(&ssformat!(
                    64,
                    "{} {} {status_str}\r\n",
                    self.version,
                    self.http_code
                ));
                if self.headers.len() == 6 {
                    if let (
                        Some(date),
                        Some(server),
                        Some(connection),
                        Some(content_type),
                        Some(pragma),
                        Some(cache_control),
                    ) = (
                        self.headers.get("Date"),
                        self.headers.get("Server"),
                        self.headers.get("Connection"),
                        self.headers.get("Content-Type"),
                        self.headers.get("Pragma"),
                        self.headers.get("Cache-Control"),
                    ) {
                        ret.push_str(&ssformat!(512, "Date: {date}\r\n"));
                        ret.push_str(&ssformat!(512, "Server: {server}\r\n"));
                        ret.push_str(&ssformat!(512, "Connection: {connection}\r\n"));
                        ret.push_str(&ssformat!(512, "Content-Type: {content_type}\r\n"));
                        ret.push_str(&ssformat!(512, "Pragma: {pragma}\r\n"));
                        ret.push_str(&ssformat!(512, "Cache-Control: {cache_control}\r\n"));
                    } else {
                        for (key, value) in self.headers.iter() {
                            if key == "Content-Length"
                                || key.eq_ignore_ascii_case("Transfer-Encoding")
                            {
                                continue;
                            }
                            ret.push_str(&ssformat!(512, "{key}: {value}\r\n"));
                        }
                    }
                } else {
                    for (key, value) in self.headers.iter() {
                        if key == "Content-Length" || key.eq_ignore_ascii_case("Transfer-Encoding")
                        {
                            continue;
                        }
                        ret.push_str(&ssformat!(512, "{key}: {value}\r\n"));
                    }
                }
                if !suppress_body_by_status {
                    ret.push_str(&ssformat!(64, "Content-Length: {}\r\n", payload_ref.len()));
                }
                if cmode == CompressMode::Gzip && !suppress_body_by_status {
                    ret.push_str("Content-Encoding: gzip\r\n");
                }
                ret.push_str("\r\n");

                if suppress_body || payload_ref.is_empty() {
                    stream.write_all(ret.as_bytes()).await?;
                } else {
                    stream
                        .write_all_vectored2(ret.as_bytes(), payload_ref)
                        .await?;
                }
            }
            HttpResponseBody::Stream(rx) => {
                // For Stream body, send headers first, then chunks
                let mut ret = smallstr::SmallString::<[u8; 4096]>::new();
                let status_str = self.http_code.http_code_to_desp();
                ret.push_str(&ssformat!(
                    64,
                    "{} {} {status_str}\r\n",
                    self.version,
                    self.http_code
                ));
                for (key, value) in self.headers.iter() {
                    if key == "Content-Length"
                        || (suppress_body && key.eq_ignore_ascii_case("Transfer-Encoding"))
                    {
                        continue;
                    }
                    ret.push_str(&ssformat!(512, "{key}: {value}\r\n"));
                }
                if !suppress_body {
                    ret.push_str("Transfer-Encoding: chunked\r\n");
                }
                if declared_trailer_names.is_empty() && !outbound_stream_trailers.is_empty() {
                    let trailer_names = outbound_stream_trailers
                        .iter()
                        .map(|(name, _)| name.as_str())
                        .collect::<Vec<_>>()
                        .join(", ");
                    ret.push_str(&ssformat!(512, "Trailer: {trailer_names}\r\n"));
                }
                ret.push_str("\r\n");
                let header_bytes: Vec<u8> = ret.as_bytes().to_vec();
                stream.write_all(&header_bytes).await?;

                if suppress_body {
                    return Ok(());
                }

                // Send chunks
                while let Some(chunk) = rx.recv().await {
                    if chunk.is_empty() {
                        break;
                    }
                    // Write chunked encoding: length in hex, \r\n, data, \r\n
                    let chunk_len_hex = format!("{:x}\r\n", chunk.len());
                    stream.write_all(chunk_len_hex.as_bytes()).await?;
                    stream.write_all(&chunk).await?;
                    stream.write_all(b"\r\n").await?;
                }
                // Send final chunk (length 0)
                if outbound_stream_trailers.is_empty() {
                    stream.write_all(b"0\r\n\r\n").await?;
                } else {
                    stream.write_all(b"0\r\n").await?;
                    for (name, value) in outbound_stream_trailers.iter() {
                        stream
                            .write_all(format!("{name}: {value}\r\n").as_bytes())
                            .await?;
                    }
                    stream.write_all(b"\r\n").await?;
                }
            }
        }
        Ok(())
    }

    pub async fn from_stream(
        buf: &mut Vec<u8>,
        stream: &mut HttpStream,
    ) -> anyhow::Result<(Self, usize)> {
        Self::from_stream_with_request_method(buf, stream, None).await
    }

    pub async fn from_stream_with_request_method(
        buf: &mut Vec<u8>,
        stream: &mut HttpStream,
        request_method: Option<HttpMethod>,
    ) -> anyhow::Result<(Self, usize)> {
        let (mut res, hdr_len) = loop {
            match HttpResponse::from_headers_part(&buf[..])? {
                Some((res, hdr_len)) => break (res, hdr_len),
                None => {
                    buf.extend_by_streams(stream).await?;
                }
            }
        };
        let mut bdy_len = 0;
        let skip_body = request_method == Some(HttpMethod::HEAD)
            || Self::status_disallows_response_body(res.http_code);
        if skip_body {
            return Ok((res, hdr_len));
        }
        let has_chunked_transfer_encoding =
            if let Some(raw_te) = res.headers.get("Transfer-Encoding") {
                Self::transfer_encoding_has_chunked(raw_te)?
            } else {
                false
            };
        if has_chunked_transfer_encoding && res.headers.contains_key("Content-Length") {
            Err(anyhow!(
                "conflicting headers: Transfer-Encoding and Content-Length"
            ))?;
        }
        if has_chunked_transfer_encoding {
            let allowed_trailers = parse_declared_trailer_names(res.get_header("Trailer"));
            let (chunked_body, trailers, consumed_len) =
                Self::read_chunked_body(buf, stream, hdr_len, &allowed_trailers).await?;
            bdy_len = consumed_len;
            res.trailers = trailers;
            res.body = HttpResponseBody::Data(chunked_body);
        } else if let Some(cnt_len) = res.headers.get("Content-Length") {
            bdy_len = cnt_len.parse::<usize>().unwrap_or(0);
            while hdr_len + bdy_len > buf.len() {
                buf.extend_by_streams(stream).await?;
            }
            res.body = HttpResponseBody::Data(buf[hdr_len..hdr_len + bdy_len].to_vec());
        }

        Ok((res, hdr_len + bdy_len))
    }

    pub fn from_headers_part(buf: &[u8]) -> anyhow::Result<Option<(Self, usize)>> {
        let mut headers = [httparse::EMPTY_HEADER; 96];
        let (rres, n) = {
            let mut res = httparse::Response::new(&mut headers);
            let n = match httparse::ParserConfig::default().parse_response(&mut res, buf)? {
                httparse::Status::Complete(n) => n,
                httparse::Status::Partial => return Ok(None),
            };
            (res, n)
        };

        let mut req = HttpResponse::new();
        let mut content_length_seen: Option<String> = None;
        let mut transfer_encoding_seen = false;
        req.version = format!("HTTP/1.{}", rres.version.unwrap_or(0));
        req.http_code = rres.code.unwrap_or(0);
        for h in rres.headers.iter() {
            if h.name.is_empty() {
                break;
            }
            let header_value = str::from_utf8(h.value)?.trim();
            if h.name.eq_ignore_ascii_case("Content-Length") {
                if header_value.is_empty() {
                    Err(anyhow!("empty Content-Length header"))?;
                }
                if let Some(prev) = &content_length_seen {
                    if prev != header_value {
                        Err(anyhow!("conflicting duplicate Content-Length headers"))?;
                    }
                } else {
                    content_length_seen = Some(header_value.to_string());
                    req.headers
                        .insert("Content-Length".into(), header_value.to_string().into());
                }
                continue;
            }
            if h.name.eq_ignore_ascii_case("Transfer-Encoding") {
                if header_value.is_empty() {
                    Err(anyhow!("empty Transfer-Encoding header"))?;
                }
                transfer_encoding_seen = true;
                if let Some(existing) = req.headers.get_mut("Transfer-Encoding") {
                    *existing = format!("{}, {header_value}", existing.as_ref()).into();
                } else {
                    req.headers
                        .insert("Transfer-Encoding".into(), header_value.to_string().into());
                }
                continue;
            }
            req.headers.insert(
                h.name.http_std_case().into(),
                header_value.to_string().into(),
            );
        }
        if transfer_encoding_seen && content_length_seen.is_some() {
            Err(anyhow!(
                "conflicting headers: Transfer-Encoding and Content-Length"
            ))?;
        }
        Ok(Some((req, n)))
    }

    pub fn get_header(&self, key: &str) -> Option<&str> {
        let header_key = key.http_std_case();
        self.headers.get(header_key.as_str()).map(|a| a.as_ref())
    }
}

pub fn load_embed<T: Embed>() -> HashMap<String, Cow<'static, [u8]>> {
    let mut ret = HashMap::with_capacity(16);
    for name in T::iter() {
        if let Some(file) = T::get(&name) {
            if name.ends_with("index.htm") || name.ends_with("index.html") {
                if let Some(path) = Path::new(&name[..]).parent() {
                    if let Some(path) = path.to_str() {
                        ret.insert(path.to_string(), file.data.clone());
                    }
                }
            }
            ret.insert(name.to_string(), file.data);
        }
    }
    ret
}

/// WebTransport 客户端连接宏
///
/// 用于快速连接到 WebTransport 服务器。
///
/// # 示例
///
/// ```rust,no_run
/// #[tokio::main]
/// async fn main() -> Result<(), Box<dyn std::error::Error>> {
///     // 基本连接
///     let mut wt = potato::webtransport!("https://server.com/wt").await?;
///
///     // 带自定义头连接
///     let mut wt = potato::webtransport!(
///         "https://server.com/wt",
///         Authorization = "Bearer token"
///     ).await?;
///     Ok(())
/// }
/// ```
#[cfg(feature = "http3")]
#[macro_export]
macro_rules! webtransport {
    ($url:expr) => {
        $crate::WebTransport::connect($url, vec![])
    };
    ($url:expr, $($key:ident = $value:expr),+ $(,)?) => {
        $crate::WebTransport::connect($url, vec![
            $(
                $crate::Headers::Custom((stringify!($key).to_string(), $value.to_string())),
            )+
        ])
    };
}

#[cfg(test)]
mod tests {
    use super::{CompressMode, HttpMethod, HttpRequest, HttpResponse};

    #[test]
    fn accept_encoding_supports_simple_gzip_token() {
        let mut req = HttpRequest::new();
        req.set_header("Accept-Encoding", "gzip");
        assert_eq!(req.get_header_accept_encoding(), CompressMode::Gzip);
    }

    #[test]
    fn accept_encoding_supports_qvalue_for_gzip() {
        let mut req = HttpRequest::new();
        req.set_header("Accept-Encoding", "br;q=1, gzip;q=0.3");
        assert_eq!(req.get_header_accept_encoding(), CompressMode::Gzip);
    }

    #[test]
    fn accept_encoding_uses_wildcard_when_gzip_not_listed() {
        let mut req = HttpRequest::new();
        req.set_header("Accept-Encoding", "br;q=1, *;q=0.8");
        assert_eq!(req.get_header_accept_encoding(), CompressMode::Gzip);
    }

    #[test]
    fn accept_encoding_respects_explicit_gzip_zero_over_wildcard() {
        let mut req = HttpRequest::new();
        req.set_header("Accept-Encoding", "gzip;q=0, *;q=1");
        assert_eq!(req.get_header_accept_encoding(), CompressMode::None);
    }

    #[test]
    fn accept_encoding_ignores_invalid_qvalue() {
        let mut req = HttpRequest::new();
        req.set_header("Accept-Encoding", "gzip;q=xyz");
        assert_eq!(req.get_header_accept_encoding(), CompressMode::None);
    }

    #[test]
    fn request_parser_returns_431_for_too_many_headers() {
        let mut raw = String::from("GET / HTTP/1.1\r\nHost: example.com\r\n");
        for i in 0..64 {
            raw.push_str(&format!("X-Header-{i}: value\r\n"));
        }
        raw.push_str("\r\n");

        let err = HttpRequest::from_headers_part(raw.as_bytes()).unwrap_err();
        let res = HttpRequest::parse_error_response(&err).unwrap();
        assert_eq!(res.http_code, 431);
    }

    #[test]
    fn request_parser_returns_431_for_oversized_header_section() {
        let oversized = "a".repeat(20 * 1024);
        let raw = format!("GET / HTTP/1.1\r\nHost: example.com\r\nX-Large: {oversized}\r\n\r\n");

        let err = HttpRequest::from_headers_part(raw.as_bytes()).unwrap_err();
        let res = HttpRequest::parse_error_response(&err).unwrap();
        assert_eq!(res.http_code, 431);
    }

    #[test]
    fn request_serialization_includes_query_and_version() {
        let (mut req, _, _) =
            HttpRequest::from_url("http://example.com/search?q=rust", HttpMethod::GET).unwrap();
        req.version = 10;
        let serialized = String::from_utf8(req.as_bytes()).unwrap();

        assert!(serialized.starts_with("GET /search?q=rust HTTP/1.0\r\n"));
    }

    #[test]
    fn request_serialization_falls_back_to_http11_for_non_h1_versions() {
        let (mut req, _, _) =
            HttpRequest::from_url("http://example.com/", HttpMethod::GET).unwrap();
        req.version = 20;
        let serialized = String::from_utf8(req.as_bytes()).unwrap();

        assert!(serialized.starts_with("GET / HTTP/1.1\r\n"));
    }

    #[test]
    fn from_url_host_header_keeps_non_default_port() {
        let (req, _, _) =
            HttpRequest::from_url("http://example.com:8080/demo", HttpMethod::GET).unwrap();
        assert_eq!(req.get_header("Host"), Some("example.com:8080"));
    }

    #[test]
    fn response_parser_rejects_conflicting_duplicate_content_length() {
        let raw = b"HTTP/1.1 200 OK\r\nContent-Length: 5\r\nContent-Length: 6\r\n\r\n";
        assert!(HttpResponse::from_headers_part(raw).is_err());
    }

    #[test]
    fn response_parser_rejects_transfer_encoding_content_length_conflict() {
        let raw = b"HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\nContent-Length: 5\r\n\r\n";
        assert!(HttpResponse::from_headers_part(raw).is_err());
    }

    #[test]
    fn response_transfer_encoding_supports_case_insensitive_lists() {
        assert!(HttpResponse::transfer_encoding_has_chunked("gzip, Chunked").unwrap());
    }

    #[test]
    fn response_data_serialization_omits_transfer_encoding() {
        let mut res = HttpResponse::text("hello");
        res.add_header("Transfer-Encoding".into(), "chunked".into());

        let serialized = String::from_utf8(res.as_bytes(CompressMode::None)).unwrap();
        assert!(serialized.contains("Content-Length: 5\r\n"));
        assert!(!serialized.contains("Transfer-Encoding:"));
    }
}