2019-05-15 23:44:06 +02:00
|
|
|
use crate::errors::ShellError;
|
2019-05-22 09:12:03 +02:00
|
|
|
use crate::object::base::reject_fields;
|
2019-05-15 23:44:06 +02:00
|
|
|
use crate::prelude::*;
|
|
|
|
|
2019-07-24 00:22:11 +02:00
|
|
|
pub fn reject(args: CommandArgs, registry: &CommandRegistry) -> Result<OutputStream, ShellError> {
|
|
|
|
let args = args.evaluate_once(registry)?;
|
|
|
|
let len = args.len();
|
|
|
|
let span = args.name_span();
|
|
|
|
let (input, args) = args.parts();
|
2019-07-09 06:31:26 +02:00
|
|
|
|
2019-07-24 00:22:11 +02:00
|
|
|
if len == 0 {
|
2019-08-05 10:54:29 +02:00
|
|
|
return Err(ShellError::labeled_error(
|
2019-06-15 20:36:17 +02:00
|
|
|
"Reject requires fields",
|
|
|
|
"needs parameter",
|
2019-07-24 00:22:11 +02:00
|
|
|
span,
|
2019-06-15 20:36:17 +02:00
|
|
|
));
|
2019-05-22 09:12:03 +02:00
|
|
|
}
|
2019-05-15 23:44:06 +02:00
|
|
|
|
2019-07-24 00:22:11 +02:00
|
|
|
let fields: Result<Vec<String>, _> = args
|
|
|
|
.positional
|
|
|
|
.iter()
|
|
|
|
.flatten()
|
|
|
|
.map(|a| a.as_string())
|
|
|
|
.collect();
|
|
|
|
|
2019-05-22 09:12:03 +02:00
|
|
|
let fields = fields?;
|
2019-05-15 23:44:06 +02:00
|
|
|
|
2019-08-09 06:51:21 +02:00
|
|
|
let stream = input
|
2019-08-05 10:54:29 +02:00
|
|
|
.values
|
|
|
|
.map(move |item| reject_fields(&item, &fields, item.tag()).into_tagged_value());
|
2019-05-15 23:44:06 +02:00
|
|
|
|
2019-07-03 22:31:15 +02:00
|
|
|
Ok(stream.from_input_stream())
|
2019-05-15 23:44:06 +02:00
|
|
|
}
|