elif_http/server/

server.rs

//! # Elif HTTP Server
//!
//! A NestJS-like HTTP server that provides a clean, intuitive API while using Axum under the hood.
//! Users interact only with framework types - Axum is completely abstracted away.

use super::lifecycle::{build_internal_router, start_server};
use crate::{
    config::HttpConfig,
    errors::{HttpError, HttpResult},
    middleware::v2::{Middleware, MiddlewarePipelineV2},
    routing::ElifRouter,
};
use elif_core::container::IocContainer;
use std::net::SocketAddr;
use std::sync::Arc;
use tracing::info;

/// The main HTTP server - NestJS-like experience
///
/// # Example
///
/// ```rust,no_run
/// use elif_http::{Server, HttpConfig};
/// use elif_core::container::{IocContainer, ServiceBinder};
/// use std::sync::Arc;
///
/// #[tokio::main]
/// async fn main() -> Result<(), Box<dyn std::error::Error>> {
///     let mut container = IocContainer::new();
///     container.build().unwrap();
///     let server = Server::new(container, HttpConfig::default())?;
///     server.listen("0.0.0.0:3000").await?;
///     Ok(())
/// }
/// ```
pub struct Server {
    container: Arc<IocContainer>,
    config: HttpConfig,
    router: Option<ElifRouter>,
    middleware: MiddlewarePipelineV2,
}

impl Server {
    /// Create a new server instance
    pub fn new(container: IocContainer, config: HttpConfig) -> HttpResult<Self> {
        Ok(Self {
            container: Arc::new(container),
            config,
            router: None,
            middleware: MiddlewarePipelineV2::new(),
        })
    }

    /// Create a new server with existing Arc<IocContainer>
    pub fn with_container(container: Arc<IocContainer>, config: HttpConfig) -> HttpResult<Self> {
        Ok(Self {
            container,
            config,
            router: None,
            middleware: MiddlewarePipelineV2::new(),
        })
    }

    /// Set custom routes using framework router
    ///
    /// # Example
    ///
    /// ```rust,no_run
    /// use elif_http::{Server, ElifRouter, HttpConfig, ElifRequest, HttpResult};
    /// use elif_core::container::{IocContainer, ServiceBinder};
    /// use std::sync::Arc;
    ///
    /// # async fn get_users(_req: ElifRequest) -> HttpResult<&'static str> { Ok("users") }
    /// # async fn create_user(_req: ElifRequest) -> HttpResult<&'static str> { Ok("created") }
    ///
    /// # fn main() -> Result<(), Box<dyn std::error::Error>> {
    /// let mut container = IocContainer::new();
    /// container.build().unwrap();
    /// let mut server = Server::new(container, HttpConfig::default())?;
    ///
    /// let router = ElifRouter::new()
    ///     .get("/users", get_users)
    ///     .post("/users", create_user);
    ///
    /// server.use_router(router);
    /// # Ok(())
    /// # }
    /// ```
    pub fn use_router(&mut self, router: ElifRouter) -> &mut Self {
        self.router = Some(router);
        self
    }

    /// Add middleware to the server
    ///
    /// # Example
    ///
    /// ```rust,no_run
    /// use elif_http::{Server, HttpConfig};
    /// use elif_core::container::{IocContainer, ServiceBinder};
    /// use std::sync::Arc;
    ///
    /// # #[derive(Debug)]
    /// # struct LoggingMiddleware;
    /// # impl LoggingMiddleware {
    /// #     fn default() -> Self { LoggingMiddleware }
    /// # }
    /// # impl elif_http::Middleware for LoggingMiddleware {
    /// #     fn handle(&self, request: elif_http::ElifRequest, next: elif_http::Next) -> std::pin::Pin<Box<dyn std::future::Future<Output = elif_http::ElifResponse> + Send + 'static>> {
    /// #         next.call(request)
    /// #     }
    /// # }
    ///
    /// # fn main() -> Result<(), Box<dyn std::error::Error>> {
    /// let mut container = IocContainer::new();
    /// container.build().unwrap();
    /// let mut server = Server::new(container, HttpConfig::default())?;
    /// server.use_middleware(LoggingMiddleware::default());
    /// # Ok(())
    /// # }
    /// ```
    pub fn use_middleware<M>(&mut self, middleware: M) -> &mut Self
    where
        M: Middleware + 'static,
    {
        self.middleware.add_mut(middleware);
        self
    }

    /// Enable debug mode for detailed middleware execution logs
    ///
    /// # Example
    ///
    /// ```rust,no_run
    /// use elif_http::{Server, HttpConfig};
    /// use elif_core::container::{IocContainer, ServiceBinder};
    ///
    /// # fn main() -> Result<(), Box<dyn std::error::Error>> {
    /// let mut container = IocContainer::new();
    /// container.build().unwrap();
    /// let mut server = Server::new(container, HttpConfig::default())?;
    /// server.debug_middleware(true);
    /// # Ok(())
    /// # }
    /// ```
    pub fn debug_middleware(&mut self, enable: bool) -> &mut Self {
        if enable {
            // Convert current pipeline to debug pipeline (for future use)
            let _debug_pipeline = self.middleware.clone().with_debug();
            println!("🐛 Middleware debug mode enabled");
            println!("   Middleware chain: {:?}", self.middleware.names());
        } else {
            println!("🔇 Middleware debug mode disabled");
        }
        self
    }

    /// Inspect all registered middleware and show execution order
    ///
    /// # Example
    ///
    /// ```rust,no_run
    /// use elif_http::{Server, HttpConfig};
    /// use elif_core::container::{IocContainer, ServiceBinder};
    ///
    /// # fn main() -> Result<(), Box<dyn std::error::Error>> {
    /// let mut container = IocContainer::new();
    /// container.build().unwrap();
    /// let server = Server::new(container, HttpConfig::default())?;
    /// server.inspect_middleware();
    /// # Ok(())
    /// # }
    /// ```
    pub fn inspect_middleware(&self) {
        let info = self.middleware.debug_info();

        println!("🔍 Middleware Pipeline Inspection");
        println!("   Total middleware: {}", info.middleware_count);

        if info.middleware_count == 0 {
            println!("   No middleware registered");
            return;
        }

        println!("   Execution order:");
        for (index, name) in info.middleware_names.iter().enumerate() {
            println!("     {}. {}", index + 1, name);
        }

        println!("\n💡 Tip: Use debug_middleware(true) for runtime execution logs");
    }

    /// Add profiler middleware to log timing for each request
    ///
    /// # Example
    ///
    /// ```rust,no_run
    /// use elif_http::{Server, HttpConfig};
    /// use elif_core::container::{IocContainer, ServiceBinder};
    ///
    /// # fn main() -> Result<(), Box<dyn std::error::Error>> {
    /// let mut container = IocContainer::new();
    /// container.build().unwrap();
    /// let mut server = Server::new(container, HttpConfig::default())?;
    /// server.use_profiler();
    /// # Ok(())
    /// # }
    /// ```
    pub fn use_profiler(&mut self) -> &mut Self {
        use crate::middleware::v2::ProfilerMiddleware;
        self.use_middleware(ProfilerMiddleware::new());
        println!("📊 Profiler middleware enabled - request timings will be logged");
        self
    }

    /// Start the server on the specified address
    ///
    /// # Example
    ///
    /// ```rust,no_run
    /// # use elif_http::{Server, HttpConfig};
    /// # use elif_core::container::IocContainer;
    /// # use std::sync::Arc;
    /// #
    /// # #[tokio::main]
    /// # async fn main() -> Result<(), Box<dyn std::error::Error>> {
    /// #     let mut container = IocContainer::new();
    /// #     container.build().unwrap();
    /// #     let server = Server::new(container, HttpConfig::default())?;
    /// server.listen("0.0.0.0:3000").await?;
    /// #     Ok(())
    /// # }
    /// ```
    pub async fn listen<A: Into<String>>(self, addr: A) -> HttpResult<()> {
        let addr_str = addr.into();
        let socket_addr: SocketAddr = addr_str
            .parse()
            .map_err(|e| HttpError::config(format!("Invalid address '{}': {}", addr_str, e)))?;

        self.listen_on(socket_addr).await
    }

    /// Start the server on the specified SocketAddr
    pub async fn listen_on(self, addr: SocketAddr) -> HttpResult<()> {
        info!("🚀 Starting Elif server on {}", addr);
        info!(
            "📋 Health check endpoint: {}",
            self.config.health_check_path
        );

        // Build the internal router
        let axum_router = build_internal_router(
            self.container.clone(),
            self.config.clone(),
            self.router,
            self.middleware,
        )
        .await?;

        // Start the server
        start_server(addr, axum_router).await?;

        info!("🛑 Server shut down gracefully");
        Ok(())
    }

    // Getter methods for testing and inspection
    pub fn container(&self) -> &Arc<IocContainer> {
        &self.container
    }

    pub fn config(&self) -> &HttpConfig {
        &self.config
    }

    pub fn router(&self) -> Option<&ElifRouter> {
        self.router.as_ref()
    }

    pub fn middleware(&self) -> &MiddlewarePipelineV2 {
        &self.middleware
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::testing::create_test_container;

    #[test]
    fn test_server_creation() {
        let container = create_test_container();
        let config = HttpConfig::default();

        let server = Server::with_container(container, config);
        assert!(server.is_ok());
    }

    #[test]
    fn test_server_with_arc_container() {
        let container = create_test_container();
        let config = HttpConfig::default();

        let server = Server::with_container(container, config);
        assert!(server.is_ok());
    }

    #[test]
    fn test_server_configuration() {
        let container = create_test_container();
        let config = HttpConfig::default();
        let server = Server::with_container(container, config).unwrap();

        assert_eq!(server.config().health_check_path, "/health");
    }

    #[test]
    fn test_middleware_debugging_tools() {
        let container = create_test_container();
        let config = HttpConfig::default();
        let mut server = Server::with_container(container, config).unwrap();

        // Add some middleware
        server
            .use_middleware(crate::middleware::v2::LoggingMiddleware)
            .use_middleware(crate::middleware::v2::factories::cors())
            .use_profiler();

        // Test inspect_middleware - should not panic
        server.inspect_middleware();

        // Test debug_middleware - should not panic
        server.debug_middleware(true);
        server.debug_middleware(false);

        // Verify middleware count
        assert_eq!(server.middleware().len(), 3); // LoggingMiddleware + CorsMiddleware + ProfilerMiddleware
    }
}