nushell/src/commands/where_.rs

60 lines
1.6 KiB
Rust
Raw Normal View History

2019-08-15 07:02:02 +02:00
use crate::commands::PerItemCommand;
2019-05-16 04:42:44 +02:00
use crate::errors::ShellError;
use crate::parser::hir::SyntaxShape;
2019-08-02 21:15:07 +02:00
use crate::parser::registry;
2019-05-16 04:42:44 +02:00
use crate::prelude::*;
2019-07-24 00:22:11 +02:00
pub struct Where;
2019-08-14 19:02:39 +02:00
impl PerItemCommand for Where {
2019-07-24 00:22:11 +02:00
fn name(&self) -> &str {
"where"
}
2019-05-28 08:45:18 +02:00
2019-08-02 21:15:07 +02:00
fn signature(&self) -> registry::Signature {
Signature::build("where").required("condition", SyntaxShape::Block)
}
fn usage(&self) -> &str {
"Filter table to match the condition."
2019-05-22 09:12:03 +02:00
}
2019-07-24 00:22:11 +02:00
2019-08-02 21:15:07 +02:00
fn run(
&self,
2019-08-15 07:02:02 +02:00
call_info: &CallInfo,
_registry: &registry::CommandRegistry,
2019-08-29 05:53:45 +02:00
_raw_args: &RawCommandArgs,
2019-08-14 19:02:39 +02:00
input: Tagged<Value>,
2019-08-24 21:36:19 +02:00
) -> Result<OutputStream, ShellError> {
2019-08-14 19:02:39 +02:00
let input_clone = input.clone();
2019-08-15 07:02:02 +02:00
let condition = call_info.args.expect_nth(0)?;
2019-08-24 21:36:19 +02:00
let stream = match condition {
2019-08-14 19:02:39 +02:00
Tagged {
item: Value::Block(block),
..
2019-08-14 19:02:39 +02:00
} => {
let result = block.invoke(&input_clone);
match result {
Ok(v) => {
if v.is_true() {
2019-08-24 21:36:19 +02:00
VecDeque::from(vec![Ok(ReturnSuccess::Value(input_clone))])
2019-08-14 19:02:39 +02:00
} else {
2019-08-24 21:36:19 +02:00
VecDeque::new()
2019-08-14 19:02:39 +02:00
}
}
2019-09-11 16:36:50 +02:00
Err(e) => return Err(e),
2019-08-14 19:02:39 +02:00
}
}
2019-08-24 21:36:19 +02:00
Tagged { tag, .. } => {
return Err(ShellError::labeled_error(
"Expected a condition",
"where needs a condition",
tag,
2019-08-24 21:36:19 +02:00
))
}
};
Ok(stream.to_output_stream())
2019-08-02 21:15:07 +02:00
}
}