netutils-cli 0.3.2

mod cli;
mod connections;
mod connectivity;
mod diag;
mod diagnose;
mod dns;
mod i18n;
mod icmp;
mod info;
mod output;
mod ping;
mod portscan;
mod table;
mod traceroute;
mod util;

use std::time::Duration;

use clap::Parser;
use cli::{Cli, Commands};
use output::OutputMode;

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let cli = Cli::parse();

    // 初始化 i18n
    i18n::init(cli.lang);

    // 确定输出模式
    let mode = if cli.json {
        OutputMode::Json
    } else {
        OutputMode::Table
    };

    match cli.command {
        None | Some(Commands::All) => info::print_all(mode),
        Some(Commands::Iface) => info::print_interfaces(mode),
        Some(Commands::Egress) => info::print_egress(mode),
        Some(Commands::Route) => info::print_routes(mode),
        Some(Commands::Proxy) => info::print_proxy(mode),
        Some(Commands::Ping { host, count, timeout, interval }) => {
            ping::run(&host, count, Duration::from_secs(timeout), Duration::from_secs(interval), mode).await
        }
        Some(Commands::Dns { domain, r#type, server }) => {
            dns::run(&domain, r#type, server, mode).await
        }
        Some(Commands::Trace { host, max_hops }) => {
            traceroute::run(&host, max_hops, mode).await
        }
        Some(Commands::Scan { host, ports, concurrency }) => {
            let port_list = ports.as_ref().map(|s| util::parse_ports(s));
            let port_ref = port_list
                .as_ref()
                .filter(|v| !v.is_empty())
                .map(|v| v.as_slice());
            portscan::run(&host, port_ref, concurrency, mode).await
        }
        Some(Commands::Check { target, count, timeout, timing, proxy, no_proxy }) => {
            connectivity::run(&target, count, Duration::from_secs(timeout), timing, proxy, no_proxy, mode).await
        }
        Some(Commands::Connections { state, port, process, proto }) => {
            let filter = connections::ConnFilter {
                state,
                port,
                process,
                proto,
            };
            connections::run(filter, mode)
        }
        Some(Commands::Diag) => diag::run(mode).await,
        Some(Commands::Diagnose { host }) => diagnose::run(&host, mode).await,
    }

    Ok(())
}