nushell/src/commands/first.rs

32 lines
787 B
Rust

use crate::errors::ShellError;
use crate::prelude::*;
// TODO: "Amount remaining" wrapper
pub fn first(args: CommandArgs) -> Result<OutputStream, ShellError> {
if args.len() == 0 {
return Err(ShellError::maybe_labeled_error(
"First requires an amount",
"needs parameter",
args.call_info.name_span,
));
}
let amount = args.expect_nth(0)?.as_i64();
let amount = match amount {
Ok(o) => o,
Err(_) => {
return Err(ShellError::labeled_error(
"Value is not a number",
"expected integer",
args.expect_nth(0)?.span,
))
}
};
let input = args.input;
Ok(OutputStream::from_input(input.values.take(amount as u64)))
}