grweb 0.1.0

use std::collections::HashMap;
use std::sync::Arc;
use crate::{Method, Context, Response, Middleware, MiddlewareChain};
use log::debug;

pub type Handler = Arc<dyn Fn(Context) -> Response + Send + Sync>;

/// 路由节点（支持路径参数）
pub struct RouterNode {
    handlers: HashMap<Method, Handler>,
    children: HashMap<String, RouterNode>,
    param_child: Option<Box<RouterNode>>,
    param_name: Option<String>,
}

impl RouterNode {
    pub fn new() -> Self {
        Self {
            handlers: HashMap::new(),
            children: HashMap::new(),
            param_child: None,
            param_name: None,
        }
    }
    
    fn insert(&mut self, method: Method, path: &str, handler: Handler) {
        let parts: Vec<&str> = path.split('/')
            .filter(|p| !p.is_empty())
            .collect();
        self.insert_parts(method, &parts, handler);
    }
    
    fn insert_parts(&mut self, method: Method, parts: &[&str], handler: Handler) {
        if parts.is_empty() {
            self.handlers.insert(method, handler);
            return;
        }
        
        let part = parts[0];
        let remaining = &parts[1..];
        
        if part.starts_with(":") {
            let param_name = part[1..].to_string();
            if self.param_child.is_none() {
                self.param_child = Some(Box::new(RouterNode::new()));
                self.param_name = Some(param_name.clone());
            }
            if let Some(ref mut child) = self.param_child {
                child.insert_parts(method, remaining, handler);
            }
        } else {
            let child = self.children
                .entry(part.to_string())
                .or_insert_with(RouterNode::new);
            child.insert_parts(method, remaining, handler);
        }
    }
    
    fn find(&self, method: &Method, path: &str) -> Option<(Handler, HashMap<String, String>)> {
        let parts: Vec<&str> = path.split('/')
            .filter(|p| !p.is_empty())
            .collect();
        let mut params = HashMap::new();
        if let Some(handler) = self.find_parts(method, &parts, &mut params) {
            Some((handler, params))
        } else {
            None
        }
    }
    
    fn find_parts(&self, method: &Method, parts: &[&str], params: &mut HashMap<String, String>) -> Option<Handler> {
        if parts.is_empty() {
            if let Some(handler) = self.handlers.get(method) {
                return Some(handler.clone());
            }
            return None;
        }
        
        let part = parts[0];
        let remaining = &parts[1..];
        
        // 尝试精确匹配
        if let Some(child) = self.children.get(part) {
            if let Some(handler) = child.find_parts(method, remaining, params) {
                return Some(handler);
            }
        }
        
        // 尝试参数匹配
        if let Some(ref param_child) = self.param_child {
            if let Some(ref param_name) = self.param_name {
                params.insert(param_name.clone(), part.to_string());
                return param_child.find_parts(method, remaining, params);
            }
        }
        
        None
    }
}

/// 路由器
pub struct Router {
    root: RouterNode,
    global_middlewares: Arc<Vec<Arc<dyn Middleware>>>,
}

impl Router {
    pub fn new() -> Self {
        Self {
            root: RouterNode::new(),
            global_middlewares: Arc::new(Vec::new()),
        }
    }
    
    pub fn use_middleware<M: Middleware + 'static>(&mut self, middleware: M) {
        Arc::make_mut(&mut self.global_middlewares).push(Arc::new(middleware));
    }
    
    pub fn add_route<F>(&mut self, method: Method, path: &str, handler: F) 
    where 
        F: Fn(Context) -> Response + Send + Sync + 'static 
    {
        let handler = Arc::new(handler);
        debug!("Adding route: {} {}", method.as_str(), path);
        self.root.insert(method, path, handler);
    }
    
    pub fn get<F, R>(&mut self, path: &str, handler: F) 
    where 
        F: Fn(Context) -> R + Send + Sync + 'static,
        R: Into<Response>
    {
        self.add_route(Method::GET, path, move |ctx| handler(ctx).into());
    }
    
    pub fn post<F, R>(&mut self, path: &str, handler: F) 
    where 
        F: Fn(Context) -> R + Send + Sync + 'static,
        R: Into<Response>
    {
        self.add_route(Method::POST, path, move |ctx| handler(ctx).into());
    }
    
    pub fn put<F, R>(&mut self, path: &str, handler: F) 
    where 
        F: Fn(Context) -> R + Send + Sync + 'static,
        R: Into<Response>
    {
        self.add_route(Method::PUT, path, move |ctx| handler(ctx).into());
    }
    
    pub fn delete<F, R>(&mut self, path: &str, handler: F) 
    where 
        F: Fn(Context) -> R + Send + Sync + 'static,
        R: Into<Response>
    {
        self.add_route(Method::DELETE, path, move |ctx| handler(ctx).into());
    }
    
    pub fn handle_request(&self, method: Method, path: String, req_data: Vec<u8>) -> Response {
        if let Some((handler, params)) = self.root.find(&method, &path) {
            let ctx = Context::new(method, path, params, req_data);
            MiddlewareChain::process(&self.global_middlewares, &handler, ctx)
        } else {
            Response::not_found()
        }
    }
}

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