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-05-23 09:23:06 +02:00
|
|
|
pub fn take(args: CommandArgs) -> Result<OutputStream, ShellError> {
|
2019-05-22 09:12:03 +02:00
|
|
|
let amount = args.args[0].as_int()?;
|
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
|
|
|
}
|