presentar-terminal 0.3.5

//! `ConnectionsPanel` widget for TCP/UDP connection monitoring.
//!
//! Displays active network connections with state and process mapping.

use presentar_core::{
    Brick, BrickAssertion, BrickBudget, BrickVerification, Canvas, Color, Constraints, Event,
    LayoutResult, Point, Rect, Size, TextStyle, TypeId, Widget,
};
use std::any::Any;
use std::time::Duration;

/// TCP connection state.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
pub enum TcpState {
    #[default]
    Established,
    Listen,
    TimeWait,
    CloseWait,
    SynSent,
    SynRecv,
    FinWait1,
    FinWait2,
    LastAck,
    Closing,
    Closed,
}

impl TcpState {
    /// Get short display string.
    pub fn short(&self) -> &'static str {
        match self {
            Self::Established => "EST",
            Self::Listen => "LSN",
            Self::TimeWait => "TW",
            Self::CloseWait => "CW",
            Self::SynSent => "SS",
            Self::SynRecv => "SR",
            Self::FinWait1 => "FW1",
            Self::FinWait2 => "FW2",
            Self::LastAck => "LA",
            Self::Closing => "CLG",
            Self::Closed => "CLD",
        }
    }

    /// Get state color.
    pub fn color(&self) -> Color {
        match self {
            Self::Established => Color::new(0.4, 0.9, 0.4, 1.0), // Green
            Self::Listen => Color::new(0.4, 0.6, 1.0, 1.0),      // Blue
            Self::TimeWait => Color::new(0.6, 0.6, 0.6, 1.0),    // Gray
            Self::CloseWait => Color::new(1.0, 0.8, 0.2, 1.0),   // Yellow
            _ => Color::new(0.7, 0.7, 0.7, 1.0),                 // Default gray
        }
    }
}

/// A network connection entry.
#[derive(Debug, Clone)]
pub struct ConnectionEntry {
    /// Protocol (tcp, udp).
    pub protocol: String,
    /// Local address.
    pub local_addr: String,
    /// Local port.
    pub local_port: u16,
    /// Remote address.
    pub remote_addr: String,
    /// Remote port.
    pub remote_port: u16,
    /// Connection state.
    pub state: TcpState,
    /// Process name (if available).
    pub process: Option<String>,
    /// Process ID.
    pub pid: Option<u32>,
}

impl ConnectionEntry {
    /// Create a new TCP connection.
    #[must_use]
    pub fn tcp(local_port: u16, remote_addr: impl Into<String>, remote_port: u16) -> Self {
        Self {
            protocol: "tcp".to_string(),
            local_addr: "0.0.0.0".to_string(),
            local_port,
            remote_addr: remote_addr.into(),
            remote_port,
            state: TcpState::Established,
            process: None,
            pid: None,
        }
    }

    /// Create a listening socket.
    #[must_use]
    pub fn listen(port: u16) -> Self {
        Self {
            protocol: "tcp".to_string(),
            local_addr: "0.0.0.0".to_string(),
            local_port: port,
            remote_addr: "*".to_string(),
            remote_port: 0,
            state: TcpState::Listen,
            process: None,
            pid: None,
        }
    }

    /// Set connection state.
    #[must_use]
    pub fn with_state(mut self, state: TcpState) -> Self {
        self.state = state;
        self
    }

    /// Set process info.
    #[must_use]
    pub fn with_process(mut self, name: impl Into<String>, pid: u32) -> Self {
        self.process = Some(name.into());
        self.pid = Some(pid);
        self
    }

    /// Set local address.
    #[must_use]
    pub fn with_local_addr(mut self, addr: impl Into<String>) -> Self {
        self.local_addr = addr.into();
        self
    }

    /// Get service name from port.
    pub fn service_name(&self) -> &str {
        match self.local_port {
            22 => "ssh",
            80 => "http",
            443 => "https",
            3306 => "mysql",
            5432 => "pgsql",
            6379 => "redis",
            8080 => "http-alt",
            27017 => "mongodb",
            _ => "",
        }
    }

    /// Format local endpoint.
    pub fn local_display(&self) -> String {
        format!(":{}", self.local_port)
    }

    /// Format remote endpoint.
    pub fn remote_display(&self) -> String {
        if self.remote_addr == "*" || self.remote_addr == "0.0.0.0" {
            "*".to_string()
        } else {
            format!("{}:{}", self.remote_addr, self.remote_port)
        }
    }
}

/// Connections panel displaying network connections.
#[derive(Debug, Clone)]
pub struct ConnectionsPanel {
    /// Connection entries.
    connections: Vec<ConnectionEntry>,
    /// Show listening sockets.
    show_listening: bool,
    /// Show established connections.
    show_established: bool,
    /// Max connections to show.
    max_connections: usize,
    /// Show column headers.
    show_headers: bool,
    /// Cached bounds.
    bounds: Rect,
}

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

impl ConnectionsPanel {
    /// Create a new connections panel.
    #[must_use]
    pub fn new() -> Self {
        Self {
            connections: Vec::new(),
            show_listening: true,
            show_established: true,
            max_connections: 10,
            show_headers: true,
            bounds: Rect::default(),
        }
    }

    /// Add a connection.
    pub fn add_connection(&mut self, connection: ConnectionEntry) {
        self.connections.push(connection);
    }

    /// Set all connections.
    #[must_use]
    pub fn with_connections(mut self, connections: Vec<ConnectionEntry>) -> Self {
        self.connections = connections;
        self
    }

    /// Toggle listening sockets.
    #[must_use]
    pub fn show_listening(mut self, show: bool) -> Self {
        self.show_listening = show;
        self
    }

    /// Toggle established connections.
    #[must_use]
    pub fn show_established(mut self, show: bool) -> Self {
        self.show_established = show;
        self
    }

    /// Set max connections.
    #[must_use]
    pub fn max_connections(mut self, max: usize) -> Self {
        self.max_connections = max;
        self
    }

    /// Toggle headers.
    #[must_use]
    pub fn show_headers(mut self, show: bool) -> Self {
        self.show_headers = show;
        self
    }

    /// Get established count.
    pub fn established_count(&self) -> usize {
        self.connections
            .iter()
            .filter(|c| c.state == TcpState::Established)
            .count()
    }

    /// Get listening count.
    pub fn listening_count(&self) -> usize {
        self.connections
            .iter()
            .filter(|c| c.state == TcpState::Listen)
            .count()
    }

    /// Get visible connections (filtered).
    fn visible_connections(&self) -> impl Iterator<Item = &ConnectionEntry> {
        self.connections
            .iter()
            .filter(|c| {
                (self.show_listening && c.state == TcpState::Listen)
                    || (self.show_established && c.state == TcpState::Established)
                    || (c.state != TcpState::Listen && c.state != TcpState::Established)
            })
            .take(self.max_connections)
    }

    /// Draw header row.
    fn draw_header(&self, canvas: &mut dyn Canvas, x: f32, y: f32) {
        let header = "SVC   LOCAL   → REMOTE         ST  PROC";
        canvas.draw_text(
            header,
            Point::new(x, y),
            &TextStyle {
                color: Color::new(0.6, 0.6, 0.6, 1.0),
                ..Default::default()
            },
        );
    }

    /// Draw a connection line.
    fn draw_connection(
        &self,
        canvas: &mut dyn Canvas,
        conn: &ConnectionEntry,
        x: f32,
        y: f32,
        width: f32,
    ) {
        // Service name or port
        let svc = {
            let name = conn.service_name();
            if name.is_empty() {
                format!("{:5}", conn.local_port)
            } else {
                format!("{name:5}")
            }
        };
        canvas.draw_text(
            &svc,
            Point::new(x, y),
            &TextStyle {
                color: Color::WHITE,
                ..Default::default()
            },
        );

        // Local port
        canvas.draw_text(
            &conn.local_display(),
            Point::new(x + 6.0, y),
            &TextStyle {
                color: Color::new(0.6, 0.8, 1.0, 1.0),
                ..Default::default()
            },
        );

        // Arrow
        canvas.draw_text(
            "→",
            Point::new(x + 12.0, y),
            &TextStyle {
                color: Color::new(0.5, 0.5, 0.5, 1.0),
                ..Default::default()
            },
        );

        // Remote (truncated)
        let remote = {
            let r = conn.remote_display();
            if r.len() > 14 {
                format!("{}...", &r[..11])
            } else {
                format!("{r:14}")
            }
        };
        canvas.draw_text(
            &remote,
            Point::new(x + 14.0, y),
            &TextStyle {
                color: Color::new(0.8, 0.8, 0.8, 1.0),
                ..Default::default()
            },
        );

        // State
        canvas.draw_text(
            conn.state.short(),
            Point::new(x + 29.0, y),
            &TextStyle {
                color: conn.state.color(),
                ..Default::default()
            },
        );

        // Process name (if available and fits)
        if let Some(ref proc) = conn.process {
            let proc_x = x + 33.0;
            if proc_x < x + width {
                let max_len = ((width - 33.0) as usize).min(10);
                let name = if proc.len() > max_len {
                    format!("{}...", &proc[..max_len.saturating_sub(3)])
                } else {
                    proc.clone()
                };
                canvas.draw_text(
                    &name,
                    Point::new(proc_x, y),
                    &TextStyle {
                        color: Color::new(0.6, 0.6, 0.6, 1.0),
                        ..Default::default()
                    },
                );
            }
        }
    }
}

impl Brick for ConnectionsPanel {
    fn brick_name(&self) -> &'static str {
        "connections_panel"
    }

    fn assertions(&self) -> &[BrickAssertion] {
        static ASSERTIONS: &[BrickAssertion] = &[BrickAssertion::max_latency_ms(8)];
        ASSERTIONS
    }

    fn budget(&self) -> BrickBudget {
        BrickBudget::uniform(8)
    }

    fn verify(&self) -> BrickVerification {
        BrickVerification {
            passed: vec![BrickAssertion::max_latency_ms(8)],
            failed: vec![],
            verification_time: Duration::from_micros(25),
        }
    }

    fn to_html(&self) -> String {
        String::new()
    }

    fn to_css(&self) -> String {
        String::new()
    }
}

impl Widget for ConnectionsPanel {
    fn type_id(&self) -> TypeId {
        TypeId::of::<Self>()
    }

    fn measure(&self, constraints: Constraints) -> Size {
        let header_lines = usize::from(self.show_headers);
        let visible = self.visible_connections().count();
        let height = ((header_lines + visible) as f32)
            .max(1.0)
            .min(constraints.max_height);
        Size::new(constraints.max_width, height)
    }

    fn layout(&mut self, bounds: Rect) -> LayoutResult {
        self.bounds = bounds;
        LayoutResult {
            size: Size::new(bounds.width, bounds.height),
        }
    }

    fn paint(&self, canvas: &mut dyn Canvas) {
        if self.bounds.width < 20.0 || self.bounds.height < 1.0 {
            return;
        }

        let mut y = self.bounds.y;
        let x = self.bounds.x;

        // Draw header
        if self.show_headers {
            self.draw_header(canvas, x, y);
            y += 1.0;
        }

        // Draw connections
        for conn in self.visible_connections() {
            if y >= self.bounds.y + self.bounds.height {
                break;
            }
            self.draw_connection(canvas, conn, x, y, self.bounds.width);
            y += 1.0;
        }

        // If no connections, show message
        if self.connections.is_empty() {
            canvas.draw_text(
                "No connections",
                Point::new(x, y),
                &TextStyle {
                    color: Color::new(0.5, 0.5, 0.5, 1.0),
                    ..Default::default()
                },
            );
        }
    }

    fn event(&mut self, _event: &Event) -> Option<Box<dyn Any + Send>> {
        None
    }

    fn children(&self) -> &[Box<dyn Widget>] {
        &[]
    }

    fn children_mut(&mut self) -> &mut [Box<dyn Widget>] {
        &mut []
    }
}

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

    #[test]
    fn test_tcp_state_short() {
        assert_eq!(TcpState::Established.short(), "EST");
        assert_eq!(TcpState::Listen.short(), "LSN");
        assert_eq!(TcpState::TimeWait.short(), "TW");
    }

    #[test]
    fn test_connection_entry_tcp() {
        let conn = ConnectionEntry::tcp(443, "1.2.3.4", 52341).with_process("nginx", 1234);

        assert_eq!(conn.local_port, 443);
        assert_eq!(conn.remote_port, 52341);
        assert_eq!(conn.process, Some("nginx".to_string()));
        assert_eq!(conn.service_name(), "https");
    }

    #[test]
    fn test_connection_entry_listen() {
        let conn = ConnectionEntry::listen(8080);
        assert_eq!(conn.state, TcpState::Listen);
        assert_eq!(conn.remote_display(), "*");
        assert_eq!(conn.service_name(), "http-alt");
    }

    #[test]
    fn test_panel_counts() {
        let mut panel = ConnectionsPanel::new();
        panel.add_connection(ConnectionEntry::listen(80));
        panel.add_connection(ConnectionEntry::listen(443));
        panel.add_connection(ConnectionEntry::tcp(443, "1.2.3.4", 52341));

        assert_eq!(panel.listening_count(), 2);
        assert_eq!(panel.established_count(), 1);
    }

    #[test]
    fn test_panel_builder() {
        let panel = ConnectionsPanel::new()
            .show_listening(false)
            .show_established(true)
            .max_connections(5)
            .show_headers(false);

        assert!(!panel.show_listening);
        assert!(panel.show_established);
        assert_eq!(panel.max_connections, 5);
        assert!(!panel.show_headers);
    }

    #[test]
    fn test_tcp_state_all_short() {
        assert_eq!(TcpState::CloseWait.short(), "CW");
        assert_eq!(TcpState::SynSent.short(), "SS");
        assert_eq!(TcpState::SynRecv.short(), "SR");
        assert_eq!(TcpState::FinWait1.short(), "FW1");
        assert_eq!(TcpState::FinWait2.short(), "FW2");
        assert_eq!(TcpState::LastAck.short(), "LA");
        assert_eq!(TcpState::Closing.short(), "CLG");
        assert_eq!(TcpState::Closed.short(), "CLD");
    }

    #[test]
    fn test_tcp_state_colors() {
        // Test all states return valid colors
        for state in [
            TcpState::Established,
            TcpState::Listen,
            TcpState::TimeWait,
            TcpState::CloseWait,
            TcpState::SynSent,
            TcpState::SynRecv,
            TcpState::FinWait1,
            TcpState::FinWait2,
            TcpState::LastAck,
            TcpState::Closing,
            TcpState::Closed,
        ] {
            let color = state.color();
            assert!(color.r >= 0.0 && color.r <= 1.0);
        }
    }

    #[test]
    fn test_connection_entry_service_names() {
        assert_eq!(ConnectionEntry::listen(22).service_name(), "ssh");
        assert_eq!(ConnectionEntry::listen(80).service_name(), "http");
        assert_eq!(ConnectionEntry::listen(443).service_name(), "https");
        assert_eq!(ConnectionEntry::listen(3306).service_name(), "mysql");
        assert_eq!(ConnectionEntry::listen(5432).service_name(), "pgsql");
        assert_eq!(ConnectionEntry::listen(6379).service_name(), "redis");
        assert_eq!(ConnectionEntry::listen(27017).service_name(), "mongodb");
        assert_eq!(ConnectionEntry::listen(9999).service_name(), "");
    }

    #[test]
    fn test_connection_entry_with_local_addr() {
        let conn = ConnectionEntry::listen(80).with_local_addr("127.0.0.1");
        assert_eq!(conn.local_addr, "127.0.0.1");
    }

    #[test]
    fn test_connection_entry_with_state() {
        let conn = ConnectionEntry::tcp(443, "1.2.3.4", 12345).with_state(TcpState::TimeWait);
        assert_eq!(conn.state, TcpState::TimeWait);
    }

    #[test]
    fn test_connection_entry_local_display() {
        let conn = ConnectionEntry::listen(8080);
        assert_eq!(conn.local_display(), ":8080");
    }

    #[test]
    fn test_connection_entry_remote_display_zero() {
        let conn = ConnectionEntry::tcp(80, "0.0.0.0", 0);
        assert_eq!(conn.remote_display(), "*");
    }

    #[test]
    fn test_connection_entry_remote_display_normal() {
        let conn = ConnectionEntry::tcp(443, "192.168.1.1", 54321);
        assert_eq!(conn.remote_display(), "192.168.1.1:54321");
    }

    #[test]
    fn test_connections_panel_with_connections() {
        let connections = vec![
            ConnectionEntry::listen(80),
            ConnectionEntry::tcp(443, "1.2.3.4", 12345),
        ];
        let panel = ConnectionsPanel::new().with_connections(connections);
        assert_eq!(panel.listening_count() + panel.established_count(), 2);
    }

    #[test]
    fn test_connections_panel_brick_traits() {
        let panel = ConnectionsPanel::new();
        assert_eq!(panel.brick_name(), "connections_panel");
        assert!(!panel.assertions().is_empty());
        assert!(panel.budget().paint_ms > 0);
        assert!(panel.verify().is_valid());
        assert!(panel.to_html().is_empty());
        assert!(panel.to_css().is_empty());
    }

    #[test]
    fn test_connections_panel_widget_traits() {
        let mut panel = ConnectionsPanel::new().with_connections(vec![ConnectionEntry::listen(80)]);

        // Measure
        let size = panel.measure(Constraints {
            min_width: 0.0,
            min_height: 0.0,
            max_width: 80.0,
            max_height: 20.0,
        });
        assert!(size.width > 0.0);
        assert!(size.height > 0.0);

        // Layout
        let result = panel.layout(Rect::new(0.0, 0.0, 80.0, 10.0));
        assert_eq!(result.size.width, 80.0);

        // Type ID
        assert_eq!(Widget::type_id(&panel), TypeId::of::<ConnectionsPanel>());

        // Event
        assert!(panel
            .event(&Event::KeyDown {
                key: presentar_core::Key::Enter
            })
            .is_none());

        // Children
        assert!(panel.children().is_empty());
        assert!(panel.children_mut().is_empty());
    }

    #[test]
    fn test_connections_panel_paint_with_header() {
        use crate::direct::{CellBuffer, DirectTerminalCanvas};

        let connections = vec![
            ConnectionEntry::listen(80).with_process("nginx", 1234),
            ConnectionEntry::tcp(443, "192.168.1.100", 54321).with_process("curl", 5678),
            ConnectionEntry::tcp(3306, "10.0.0.1", 12345).with_state(TcpState::CloseWait),
        ];

        let mut panel = ConnectionsPanel::new()
            .with_connections(connections)
            .show_headers(true);

        panel.layout(Rect::new(0.0, 0.0, 60.0, 10.0));

        let mut buffer = CellBuffer::new(60, 10);
        let mut canvas = DirectTerminalCanvas::new(&mut buffer);
        panel.paint(&mut canvas);
    }

    #[test]
    fn test_connections_panel_paint_without_header() {
        use crate::direct::{CellBuffer, DirectTerminalCanvas};

        let connections = vec![ConnectionEntry::listen(443)];

        let mut panel = ConnectionsPanel::new()
            .with_connections(connections)
            .show_headers(false);

        panel.layout(Rect::new(0.0, 0.0, 60.0, 10.0));

        let mut buffer = CellBuffer::new(60, 10);
        let mut canvas = DirectTerminalCanvas::new(&mut buffer);
        panel.paint(&mut canvas);
    }

    #[test]
    fn test_connections_panel_paint_empty() {
        use crate::direct::{CellBuffer, DirectTerminalCanvas};

        let mut panel = ConnectionsPanel::new();
        panel.layout(Rect::new(0.0, 0.0, 60.0, 10.0));

        let mut buffer = CellBuffer::new(60, 10);
        let mut canvas = DirectTerminalCanvas::new(&mut buffer);
        panel.paint(&mut canvas);
    }

    #[test]
    fn test_connections_panel_paint_small_bounds() {
        use crate::direct::{CellBuffer, DirectTerminalCanvas};

        let connections = vec![ConnectionEntry::listen(80)];
        let mut panel = ConnectionsPanel::new().with_connections(connections);
        panel.layout(Rect::new(0.0, 0.0, 10.0, 0.5)); // Too small

        let mut buffer = CellBuffer::new(10, 1);
        let mut canvas = DirectTerminalCanvas::new(&mut buffer);
        panel.paint(&mut canvas); // Should early return
    }

    #[test]
    fn test_connections_panel_long_process_name() {
        use crate::direct::{CellBuffer, DirectTerminalCanvas};

        let connections = vec![ConnectionEntry::tcp(443, "1.2.3.4", 12345)
            .with_process("very_long_process_name_here", 1234)];

        let mut panel = ConnectionsPanel::new().with_connections(connections);
        panel.layout(Rect::new(0.0, 0.0, 60.0, 10.0));

        let mut buffer = CellBuffer::new(60, 10);
        let mut canvas = DirectTerminalCanvas::new(&mut buffer);
        panel.paint(&mut canvas);
    }

    #[test]
    fn test_connections_panel_long_remote_address() {
        use crate::direct::{CellBuffer, DirectTerminalCanvas};

        let connections = vec![ConnectionEntry::tcp(
            443,
            "2001:0db8:85a3:0000:0000:8a2e:0370:7334",
            12345,
        )];

        let mut panel = ConnectionsPanel::new().with_connections(connections);
        panel.layout(Rect::new(0.0, 0.0, 60.0, 10.0));

        let mut buffer = CellBuffer::new(60, 10);
        let mut canvas = DirectTerminalCanvas::new(&mut buffer);
        panel.paint(&mut canvas);
    }

    #[test]
    fn test_connections_panel_filter_listening() {
        let connections = vec![
            ConnectionEntry::listen(80),
            ConnectionEntry::tcp(443, "1.2.3.4", 12345),
            ConnectionEntry::listen(8080),
        ];

        let panel = ConnectionsPanel::new()
            .with_connections(connections)
            .show_listening(false)
            .show_established(true);

        let visible: Vec<_> = panel.visible_connections().collect();
        assert_eq!(visible.len(), 1); // Only established
    }

    #[test]
    fn test_connections_panel_filter_established() {
        let connections = vec![
            ConnectionEntry::listen(80),
            ConnectionEntry::tcp(443, "1.2.3.4", 12345),
        ];

        let panel = ConnectionsPanel::new()
            .with_connections(connections)
            .show_listening(true)
            .show_established(false);

        let visible: Vec<_> = panel.visible_connections().collect();
        assert_eq!(visible.len(), 1); // Only listening
    }

    #[test]
    fn test_connections_panel_default() {
        let panel = ConnectionsPanel::default();
        assert!(panel.show_listening);
        assert!(panel.show_established);
        assert!(panel.show_headers);
        assert_eq!(panel.max_connections, 10);
    }

    #[test]
    fn test_tcp_state_default() {
        let state = TcpState::default();
        assert_eq!(state, TcpState::Established);
    }

    #[test]
    fn test_connections_panel_other_states_visible() {
        let connections = vec![
            ConnectionEntry::tcp(443, "1.2.3.4", 12345).with_state(TcpState::TimeWait),
            ConnectionEntry::tcp(443, "1.2.3.5", 12346).with_state(TcpState::CloseWait),
        ];

        // These are neither listening nor established, but should be visible
        let panel = ConnectionsPanel::new()
            .with_connections(connections)
            .show_listening(false)
            .show_established(false);

        let visible: Vec<_> = panel.visible_connections().collect();
        assert_eq!(visible.len(), 2); // Both TIME_WAIT and CLOSE_WAIT visible
    }

    #[test]
    fn test_connections_panel_unknown_port() {
        use crate::direct::{CellBuffer, DirectTerminalCanvas};

        // Port with no service name
        let connections = vec![ConnectionEntry::listen(12345)];

        let mut panel = ConnectionsPanel::new().with_connections(connections);
        panel.layout(Rect::new(0.0, 0.0, 60.0, 10.0));

        let mut buffer = CellBuffer::new(60, 10);
        let mut canvas = DirectTerminalCanvas::new(&mut buffer);
        panel.paint(&mut canvas);
    }
}