whois_cli/

cli.rs

use clap::Parser;

#[derive(Parser)]
#[command(
    author = "Pysio",
    version = env!("CARGO_PKG_VERSION"),
    about = "A simple WHOIS query tool with advanced features"
)]
pub struct Cli {
    /// Domain name or IP address to query
    pub domain: String,

    /// WHOIS server to use (bypasses IANA lookup)
    #[arg(short, long)]
    pub server: Option<String>,

    /// Port number to use
    #[arg(short, long, default_value_t = 43)]
    pub port: u16,

    /// Display verbose output
    #[arg(short, long)]
    pub verbose: bool,
    
    /// Query DN42 information from lantian.pub
    #[arg(long = "42")]
    pub dn42: bool,
    
    /// Query from bgp.tools
    #[arg(long)]
    pub bgptools: bool,
    
    /// Disable colored output
    #[arg(long)]
    pub no_color: bool,
    
    /// Easter egg: MTF flag colors (hidden option)
    #[arg(long, hide = true)]
    pub mtf: bool,

    /// Disable hyperlinks in terminal output (hyperlinks are enabled by default)
    #[arg(long, help = "Disable clickable hyperlinks for RIR database results")]
    pub no_hyperlinks: bool,

    /// Disable server-side coloring protocol (server-side rendering is default)
    #[arg(long, help = "Disable server-side coloring and use client-side only")]
    pub no_server_color: bool,

    /// Enable Markdown formatting from server
    #[arg(long, help = "Request Markdown-formatted output from server")]
    pub markdown: bool,

    /// Enable image display in terminal
    #[arg(long, help = "Enable inline image display in terminal")]
    pub images: bool,
}

impl Cli {
    /// Check if colored output should be used
    pub fn use_color(&self) -> bool {
        !self.no_color
    }

    /// Check if MTF colors should be used
    pub fn use_mtf_colors(&self) -> bool {
        self.mtf
    }

    /// Check if DN42 mode should be used
    pub fn use_dn42(&self) -> bool {
        self.dn42 || self.domain.to_uppercase().starts_with("AS42424")
    }

    /// Check if BGP tools mode should be used
    pub fn use_bgptools(&self) -> bool {
        self.bgptools
    }

    /// Check if hyperlinks should be used
    pub fn use_hyperlinks(&self) -> bool {
        !self.no_hyperlinks
    }

    /// Check if server-side coloring should be used (default: true)
    pub fn use_server_color(&self) -> bool {
        !self.no_server_color
    }

    /// Check if Markdown formatting should be requested
    pub fn use_markdown(&self) -> bool {
        self.markdown
    }

    /// Check if image display should be enabled
    pub fn use_images(&self) -> bool {
        self.images
    }
}

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

    fn create_test_cli(domain: &str) -> Cli {
        Cli {
            domain: domain.to_string(),
            server: None,
            port: 43,
            verbose: false,
            dn42: false,
            bgptools: false,
            no_color: false,
            mtf: false,
            no_hyperlinks: false,
            no_server_color: false,
            markdown: false,
            images: false,
        }
    }

    #[test]
    fn test_use_color_default() {
        let cli = create_test_cli("example.com");
        assert!(cli.use_color());
    }

    #[test]
    fn test_use_color_disabled() {
        let mut cli = create_test_cli("example.com");
        cli.no_color = true;
        assert!(!cli.use_color());
    }

    #[test]
    fn test_use_mtf_colors() {
        let mut cli = create_test_cli("example.com");
        assert!(!cli.use_mtf_colors());
        
        cli.mtf = true;
        assert!(cli.use_mtf_colors());
    }

    #[test]
    fn test_use_dn42_explicit_flag() {
        let mut cli = create_test_cli("example.com");
        assert!(!cli.use_dn42());
        
        cli.dn42 = true;
        assert!(cli.use_dn42());
    }

    #[test]
    fn test_use_dn42_auto_detection() {
        let cli = create_test_cli("AS4242420000");
        assert!(cli.use_dn42());
        
        let cli = create_test_cli("as4242420000");
        assert!(cli.use_dn42());
        
        let cli = create_test_cli("AS4242421234");
        assert!(cli.use_dn42());
    }

    #[test]
    fn test_use_dn42_not_triggered() {
        let cli = create_test_cli("AS15169");
        assert!(!cli.use_dn42());
        
        let cli = create_test_cli("example.com");
        assert!(!cli.use_dn42());
    }

    #[test]
    fn test_use_bgptools() {
        let mut cli = create_test_cli("AS15169");
        assert!(!cli.use_bgptools());
        
        cli.bgptools = true;
        assert!(cli.use_bgptools());
    }

    #[test]
    fn test_use_hyperlinks_default() {
        let cli = create_test_cli("example.com");
        assert!(cli.use_hyperlinks());
    }

    #[test]
    fn test_use_hyperlinks_disabled() {
        let mut cli = create_test_cli("example.com");
        cli.no_hyperlinks = true;
        assert!(!cli.use_hyperlinks());
    }

    #[test]
    fn test_port_default() {
        let cli = create_test_cli("example.com");
        assert_eq!(cli.port, 43);
    }

    #[test]
    fn test_domain_assignment() {
        let cli = create_test_cli("test.example.com");
        assert_eq!(cli.domain, "test.example.com");
    }

    #[test]
    fn test_use_server_color() {
        let mut cli = create_test_cli("example.com");
        assert!(cli.use_server_color()); // Default is true
        
        cli.no_server_color = true;
        assert!(!cli.use_server_color());
    }

    #[test]
    fn test_use_markdown() {
        let mut cli = create_test_cli("example.com");
        assert!(!cli.use_markdown()); // Default is false
        
        cli.markdown = true;
        assert!(cli.use_markdown());
    }

    #[test]
    fn test_use_images() {
        let mut cli = create_test_cli("example.com");
        assert!(!cli.use_images()); // Default is false
        
        cli.images = true;
        assert!(cli.use_images());
    }

    #[test]
    fn test_all_flags_together() {
        let mut cli = create_test_cli("AS4242420000");
        cli.dn42 = true;
        cli.bgptools = true;
        cli.no_color = true;
        cli.mtf = true;
        cli.no_hyperlinks = true;
        cli.verbose = true;
        cli.no_server_color = true;
        cli.markdown = true;
        cli.images = true;
        
        // DN42 should be true due to both flag and auto-detection
        assert!(cli.use_dn42());
        assert!(cli.use_bgptools());
        assert!(!cli.use_color());
        assert!(cli.use_mtf_colors());
        assert!(!cli.use_hyperlinks());
        assert!(!cli.use_server_color());
        assert!(cli.use_markdown());
        assert!(cli.use_images());
        assert!(cli.verbose);
    }
}