nushell/crates/nu-command/src/commands/termsize.rs

use crate::prelude::*;
use indexmap::IndexMap;
use nu_engine::WholeStreamCommand;
use nu_errors::ShellError;
use nu_protocol::{Dictionary, Signature, UntaggedValue};

pub struct TermSize;

#[derive(Deserialize, Clone)]
pub struct TermSizeArgs {
    wide: bool,
    tall: bool,
}

impl WholeStreamCommand for TermSize {
    fn name(&self) -> &str {
        "term size"
    }

    fn signature(&self) -> Signature {
        Signature::build("term size")
            .switch("wide", "Report only the width of the terminal", Some('w'))
            .switch("tall", "Report only the height of the terminal", Some('t'))
    }

    fn usage(&self) -> &str {
        "Returns the terminal size as W H"
    }

    fn run_with_actions(&self, args: CommandArgs) -> Result<ActionStream, ShellError> {
        let tag = args.call_info.name_tag.clone();

        let wide = args.has_flag("wide");
        let tall = args.has_flag("tall");

        let size = term_size::dimensions();
        match size {
            Some((w, h)) => {
                if wide && !tall {
                    Ok(ActionStream::one(
                        UntaggedValue::int(w as i64).into_value(tag),
                    ))
                } else if !wide && tall {
                    Ok(ActionStream::one(
                        UntaggedValue::int(h as i64).into_value(tag),
                    ))
                } else {
                    let mut indexmap = IndexMap::with_capacity(2);
                    indexmap.insert(
                        "width".to_string(),
                        UntaggedValue::int(w as i64).into_value(&tag),
                    );
                    indexmap.insert(
                        "height".to_string(),
                        UntaggedValue::int(h as i64).into_value(&tag),
                    );
                    let value = UntaggedValue::Row(Dictionary::from(indexmap)).into_value(&tag);
                    Ok(ActionStream::one(value))
                }
            }
            _ => Ok(ActionStream::one(
                UntaggedValue::string("0 0".to_string()).into_value(tag),
            )),
        }
    }

    fn examples(&self) -> Vec<Example> {
        vec![
            Example {
                description: "Return the width height of the terminal",
                example: "term size",
                result: None,
            },
            Example {
                description: "Return the width of the terminal",
                example: "term size -w",
                result: None,
            },
            Example {
                description: "Return the height (t for tall) of the terminal",
                example: "term size -t",
                result: None,
            },
        ]
    }
}