Files
bandwhich/src/cli.rs
Yashodhan Joshi 99cdda6cb6 replace derivative with derive_more crate
Signed-off-by: Yashodhan Joshi <yjdoc2@gmail.com>
2024-10-16 15:25:34 +05:30

79 lines
2.0 KiB
Rust

use std::{net::Ipv4Addr, path::PathBuf};
use clap::{Args, Parser, ValueEnum, ValueHint};
use clap_verbosity_flag::{InfoLevel, Verbosity};
use derive_more::Debug;
use strum::EnumIter;
#[derive(Clone, Debug, Parser, Default)]
#[command(name = "bandwhich", version)]
pub struct Opt {
#[arg(short, long)]
/// The network interface to listen on, eg. eth0
pub interface: Option<String>,
#[arg(short, long)]
/// Machine friendlier output
pub raw: bool,
#[arg(short, long)]
/// Do not attempt to resolve IPs to their hostnames
pub no_resolve: bool,
#[arg(short, long)]
/// Show DNS queries
pub show_dns: bool,
#[arg(short, long)]
/// A dns server ip to use instead of the system default
pub dns_server: Option<Ipv4Addr>,
#[arg(long, value_hint = ValueHint::FilePath)]
/// Enable debug logging to a file
pub log_to: Option<PathBuf>,
#[command(flatten)]
pub verbosity: Verbosity<InfoLevel>,
#[command(flatten)]
pub render_opts: RenderOpts,
}
#[derive(Copy, Clone, Debug, Default, Args)]
pub struct RenderOpts {
#[arg(short, long)]
/// Show processes table only
pub processes: bool,
#[arg(short, long)]
/// Show connections table only
pub connections: bool,
#[arg(short, long)]
/// Show remote addresses table only
pub addresses: bool,
#[arg(short, long, value_enum, default_value_t)]
/// Choose a specific family of units
pub unit_family: UnitFamily,
#[arg(short, long)]
/// Show total (cumulative) usages
pub total_utilization: bool,
}
// IMPRV: it would be nice if we can `#[cfg_attr(not(build), derive(strum::EnumIter))]` this
// unfortunately there is no configuration option for build script detection
#[derive(Copy, Clone, Debug, Default, Eq, PartialEq, ValueEnum, EnumIter)]
pub enum UnitFamily {
#[default]
/// bytes, in powers of 2^10
BinBytes,
/// bits, in powers of 2^10
BinBits,
/// bytes, in powers of 10^3
SiBytes,
/// bits, in powers of 10^3
SiBits,
}