2019-05-15 18:12:38 +02:00
|
|
|
use crate::errors::ShellError;
|
|
|
|
use crate::prelude::*;
|
|
|
|
|
2019-05-16 02:21:46 +02:00
|
|
|
// TODO: "Amount remaining" wrapper
|
2019-05-15 18:12:38 +02:00
|
|
|
|
2019-06-02 20:53:30 +02:00
|
|
|
pub fn first(args: CommandArgs) -> Result<OutputStream, ShellError> {
|
2019-06-22 05:43:37 +02:00
|
|
|
if args.len() == 0 {
|
2019-06-15 20:36:17 +02:00
|
|
|
return Err(ShellError::maybe_labeled_error(
|
|
|
|
"First requires an amount",
|
|
|
|
"needs parameter",
|
|
|
|
args.name_span,
|
|
|
|
));
|
2019-06-08 00:35:07 +02:00
|
|
|
}
|
|
|
|
|
2019-06-22 05:43:37 +02:00
|
|
|
let amount = args.expect_nth(0)?.as_i64();
|
2019-06-08 00:35:07 +02:00
|
|
|
|
|
|
|
let amount = match amount {
|
|
|
|
Ok(o) => o,
|
|
|
|
Err(_) => {
|
|
|
|
return Err(ShellError::labeled_error(
|
|
|
|
"Value is not a number",
|
|
|
|
"expected integer",
|
2019-06-22 05:43:37 +02:00
|
|
|
args.expect_nth(0)?.span,
|
2019-06-08 00:35:07 +02:00
|
|
|
))
|
|
|
|
}
|
|
|
|
};
|
2019-05-16 02:21:46 +02:00
|
|
|
|
2019-05-23 09:23:06 +02:00
|
|
|
let input = args.input;
|
2019-05-15 18:12:38 +02:00
|
|
|
|
2019-05-23 09:23:06 +02:00
|
|
|
Ok(input
|
|
|
|
.take(amount as u64)
|
2019-05-22 09:12:03 +02:00
|
|
|
.map(|v| ReturnValue::Value(v))
|
2019-05-23 09:23:06 +02:00
|
|
|
.boxed())
|
2019-05-15 18:12:38 +02:00
|
|
|
}
|