shitbox/src/cmd/hostname/mod.rs

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

#[derive(Debug)]
pub struct Hostname {
    name: &'static str,
    path: Option<Path>,
}

pub const HOSTNAME: Hostname = Hostname {
    name: "hostname",
    path: Some(Path::Bin),
};

impl Cmd for Hostname {
    fn name(&self) -> &str {
        self.name
    }

    fn cli(&self) -> clap::Command {
        Command::new(self.name)
            .author("The JeanG3nie <jeang3nie@hitchhiker-linux.org>")
            .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> {
        self.path
    }
}