nushell/src/commands/where_.rs

56 lines
1.7 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;
2019-07-24 00:22:11 +02:00
use crate::parser::hir::SyntaxType;
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 {
2019-08-03 04:17:28 +02:00
Signature::build("where").required("condition", SyntaxType::Block)
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,
_shell_manager: &ShellManager,
2019-08-14 19:02:39 +02:00
input: Tagged<Value>,
) -> Result<VecDeque<ReturnValue>, ShellError> {
let input_clone = input.clone();
2019-08-15 07:02:02 +02:00
let condition = call_info.args.expect_nth(0)?;
2019-08-14 19:02:39 +02:00
match condition {
Tagged {
item: Value::Block(block),
tag,
} => {
let result = block.invoke(&input_clone);
match result {
Ok(v) => {
if v.is_true() {
Ok(VecDeque::from(vec![Ok(ReturnSuccess::Value(input_clone))]))
} else {
Ok(VecDeque::new())
}
}
Err(e) => Err(ShellError::labeled_error(
format!("Could not evaluate ({})", e.to_string()),
"could not evaluate",
tag.span,
)),
}
}
Tagged { tag, .. } => Err(ShellError::labeled_error(
"Expected a condition",
"where needs a condition",
tag.span,
)),
}
2019-08-02 21:15:07 +02:00
}
}