nushell/src/commands/reject.rs

34 lines
855 B
Rust
Raw Normal View History

use crate::errors::ShellError;
2019-05-22 09:12:03 +02:00
use crate::object::base::reject_fields;
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 {
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-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-08-09 06:51:21 +02:00
let stream = input
.values
.map(move |item| reject_fields(&item, &fields, item.tag()).into_tagged_value());
Ok(stream.from_input_stream())
}