tslime 0.1.2

//! Terminal capability detection.
//!
//! This module provides functionality to inspect the current terminal environment
//! and determine its capabilities, such as color support (TrueColor vs 256 colors),
//! mouse tracking support, and estimated refresh rate.

use std::env;

/// Detected color capabilities of the terminal.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ColorCapability {
    /// Terminal supports 24-bit TrueColor (16 million colors).
    TrueColor,
    /// Terminal supports 8-bit color (256 colors).
    Bits256,
}

/// Aggregated capabilities of the detected terminal environment.
#[derive(Debug, Clone)]
pub struct TerminalCapabilities {
    /// The best supported color mode.
    pub color_capability: ColorCapability,
    /// Estimated screen refresh rate (e.g., 60Hz), derived from environment or defaults.
    pub estimated_refresh_rate: f32,
    /// Whether the terminal likely supports mouse input tracking.
    pub supports_mouse_tracking: bool,
    /// The name of the terminal program, if detected (e.g., "iTerm.app").
    pub terminal_name: Option<String>,
}

impl TerminalCapabilities {
    /// Detect the capabilities of the current terminal environment.
    ///
    /// Inspects environment variables like `COLORTERM`, `TERM`, and `TERM_PROGRAM`.
    pub fn detect() -> Self {
        Self {
            color_capability: detect_truecolor(),
            estimated_refresh_rate: estimate_refresh_rate(),
            supports_mouse_tracking: detect_mouse_support(),
            terminal_name: detect_terminal_name(),
        }
    }

    /// Select the appropriate color mode based on capabilities and user request.
    ///
    /// If a specific mode is requested, it is returned. Otherwise, the best available
    /// detected mode is used.
    pub fn auto_select_color_mode(
        &self,
        requested: Option<super::super::cli::ColorMode>,
    ) -> super::super::cli::ColorMode {
        match requested {
            Some(mode) => mode,
            None => match self.color_capability {
                ColorCapability::TrueColor => super::super::cli::ColorMode::TrueColor,
                ColorCapability::Bits256 => super::super::cli::ColorMode::Bits256,
            },
        }
    }
}

fn detect_truecolor() -> ColorCapability {
    if env::var("COLORTERM")
        .ok()
        .map(|v| v.to_lowercase())
        .filter(|v| v.contains("truecolor") || v.contains("24bit"))
        .is_some()
    {
        return ColorCapability::TrueColor;
    }

    if let Ok(term) = env::var("TERM_PROGRAM") {
        let term_lower = term.to_lowercase();
        if term_lower.contains("iterm")
            || term_lower.contains("terminal.app")
            || term_lower.contains("apple_terminal")
            || term_lower.contains("warp")
            || term_lower.contains("vscode")
        {
            return ColorCapability::TrueColor;
        }
    }

    if let Ok(term) = env::var("TERM") {
        let term_lower = term.to_lowercase();
        if term_lower.contains("xterm-256color") || term_lower.contains("screen-256color") {
            return ColorCapability::Bits256;
        }
    }

    ColorCapability::Bits256
}

fn estimate_refresh_rate() -> f32 {
    if let Ok(fps_str) = env::var("TSLIME_REFRESH_RATE") {
        if let Ok(fps) = fps_str.parse::<f32>() {
            if (10.0..=144.0).contains(&fps) {
                return fps;
            }
        }
    }

    60.0
}

fn detect_mouse_support() -> bool {
    let term = env::var("TERM").unwrap_or_default().to_lowercase();
    let term_program = env::var("TERM_PROGRAM").unwrap_or_default().to_lowercase();

    !term.contains("dumb")
        && (term_program.contains("iterm")
            || term_program.contains("terminal.app")
            || term_program.contains("warp")
            || term.contains("xterm")
            || term.contains("screen")
            || term.contains("tmux"))
}

fn detect_terminal_name() -> Option<String> {
    env::var("TERM_PROGRAM").or_else(|_| env::var("TERM")).ok()
}

/// Log the detected terminal capabilities to standard error if verbose mode is enabled.
pub fn log_capabilities(caps: &TerminalCapabilities, verbose: bool) {
    if !verbose {
        return;
    }

    eprintln!(
        "[Terminal] Color: {:?}, Refresh: {:.0}Hz, Mouse: {}, Term: {:?}",
        caps.color_capability,
        caps.estimated_refresh_rate,
        caps.supports_mouse_tracking,
        caps.terminal_name
    );
}

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

    #[test]
    fn test_auto_select_color_mode() {
        let caps = TerminalCapabilities {
            color_capability: ColorCapability::TrueColor,
            estimated_refresh_rate: 60.0,
            supports_mouse_tracking: true,
            terminal_name: None,
        };
        assert_eq!(
            caps.auto_select_color_mode(None),
            crate::cli::ColorMode::TrueColor
        );
        assert_eq!(
            caps.auto_select_color_mode(Some(crate::cli::ColorMode::Bits256)),
            crate::cli::ColorMode::Bits256
        );

        let caps_8bit = TerminalCapabilities {
            color_capability: ColorCapability::Bits256,
            ..caps
        };
        assert_eq!(
            caps_8bit.auto_select_color_mode(None),
            crate::cli::ColorMode::Bits256
        );
    }

    #[test]
    fn test_detect_truecolor_env() {
        std::env::set_var("COLORTERM", "truecolor");
        assert_eq!(detect_truecolor(), ColorCapability::TrueColor);
        std::env::remove_var("COLORTERM");
    }

    #[test]
    fn test_estimate_refresh_rate_env() {
        std::env::set_var("TSLIME_REFRESH_RATE", "120");
        assert_eq!(estimate_refresh_rate(), 120.0);
        std::env::remove_var("TSLIME_REFRESH_RATE");
    }

    #[test]
    fn test_log_capabilities() {
        let caps = TerminalCapabilities::detect();
        log_capabilities(&caps, false); // Should not print
        log_capabilities(&caps, true); // Should print
    }
}