2019-08-19 07:16:39 +02:00
|
|
|
use crate::commands::WholeStreamCommand;
|
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-08-19 07:16:39 +02:00
|
|
|
pub struct Reject;
|
|
|
|
|
|
|
|
impl WholeStreamCommand for Reject {
|
|
|
|
fn run(
|
|
|
|
&self,
|
|
|
|
args: CommandArgs,
|
|
|
|
registry: &CommandRegistry,
|
|
|
|
) -> Result<OutputStream, ShellError> {
|
|
|
|
reject(args, registry)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn name(&self) -> &str {
|
|
|
|
"reject"
|
|
|
|
}
|
2019-07-09 06:31:26 +02:00
|
|
|
|
2019-08-19 07:16:39 +02:00
|
|
|
fn signature(&self) -> Signature {
|
|
|
|
Signature::build("reject").required("fields", SyntaxType::Any)
|
2019-05-22 09:12:03 +02:00
|
|
|
}
|
2019-08-19 07:16:39 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
fn reject(args: CommandArgs, registry: &CommandRegistry) -> Result<OutputStream, ShellError> {
|
|
|
|
let args = args.evaluate_once(registry)?;
|
|
|
|
let (input, args) = args.parts();
|
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
|
|
|
}
|