nushell/crates/nu-command/src/examples/double_ls.rs

use crate::examples::sample::ls::file_listing;

use nu_engine::{CommandArgs, WholeStreamCommand};
use nu_errors::ShellError;
use nu_protocol::{ReturnSuccess, Signature, UntaggedValue, Value};
use nu_stream::{ActionStream, IntoActionStream};

pub struct Command;

impl WholeStreamCommand for Command {
    fn name(&self) -> &str {
        "ls"
    }

    fn signature(&self) -> Signature {
        Signature::build("ls")
    }

    fn usage(&self) -> &str {
        "Mock ls."
    }

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

        let mut base_value =
            UntaggedValue::string("Andrés N. Robalino in Portland").into_value(name_tag);
        let input: Vec<Value> = args.input.collect();

        if let Some(first) = input.get(0) {
            base_value = first.clone()
        }

        Ok((file_listing()
            .iter()
            .map(|row| Value {
                value: row.value.clone(),
                tag: base_value.tag.clone(),
            })
            .collect::<Vec<_>>()
            .into_iter()
            .map(ReturnSuccess::value))
        .into_action_stream())
    }
}