shitbox/src/cmd/hostname/mod.rs

use super::Cmd;
use clap::{Arg, ArgAction, ArgMatches, Command};
use std::{error::Error, io};

#[derive(Debug, Default)]
pub struct Hostname;

impl Cmd for Hostname {
    fn cli(&self) -> clap::Command {
        Command::new("hostname")
            .author("Nathan Fisher")
            .about("Prints the name of the current host. The super-user can set the host name by supplying an argument.")
            .args([
                Arg::new("NAME")
                    .help("name to set"),
                Arg::new("STRIP")
                    .help("Removes any domain information from the printed name.")
                    .short('s')
                    .long("strip")
                    .action(ArgAction::SetTrue)
            ])
    }

    fn run(&self, matches: Option<&ArgMatches>) -> Result<(), Box<dyn Error>> {
        let matches = matches.unwrap();
        if let Some(name) = matches.get_one::<String>("NAME") {
            hostname::set(name)?;
            Ok(())
        } else {
            let hostname = hostname::get()?;
            if matches.get_flag("STRIP") {
                println!(
                    "{}",
                    if let Some(s) = hostname.to_string_lossy().split('.').next() {
                        s
                    } else {
                        return Err(io::Error::new(
                            io::ErrorKind::Other,
                            "hostname: missing operand",
                        )
                        .into());
                    }
                );
            } else {
                println!("{}", hostname.to_string_lossy());
            }
            Ok(())
        }
    }

    fn path(&self) -> Option<crate::Path> {
        Some(crate::Path::Bin)
    }
}