mirror of
https://github.com/nushell/nushell
synced 2025-01-15 14:44:14 +00:00
Port all?
command (#365)
* Implement `From<bool>` for `Value` * Add `All` command * Change `IntoPipelineData` and `IntoInterruptiblePipelineData` bounds * Refactor `PipelineIterator` impls * Add `PipelineData::into_interruptible_iter` * Use `into_interruptible_iter` instead of `all` helper * Merge imports * Refactor `PipelineData::{filter, map}` * Change comment pronoun * Treat `RowCondition` as a block * Remove unnecessary braces * Address cluppy warning
This commit is contained in:
parent
0ba0daa2c4
commit
63c3d19c67
5 changed files with 142 additions and 24 deletions
|
@ -20,6 +20,7 @@ pub fn create_default_context() -> EngineState {
|
||||||
// TODO: sort default context items categorically
|
// TODO: sort default context items categorically
|
||||||
bind_command!(
|
bind_command!(
|
||||||
Alias,
|
Alias,
|
||||||
|
All,
|
||||||
Append,
|
Append,
|
||||||
Benchmark,
|
Benchmark,
|
||||||
BuildString,
|
BuildString,
|
||||||
|
|
98
crates/nu-command/src/filters/all.rs
Normal file
98
crates/nu-command/src/filters/all.rs
Normal file
|
@ -0,0 +1,98 @@
|
||||||
|
use nu_engine::eval_block;
|
||||||
|
use nu_protocol::{
|
||||||
|
ast::{Call, Expr, Expression},
|
||||||
|
engine::{Command, EngineState, Stack},
|
||||||
|
Category, Example, IntoPipelineData, PipelineData, ShellError, Signature, SyntaxShape,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct All;
|
||||||
|
|
||||||
|
impl Command for All {
|
||||||
|
fn name(&self) -> &str {
|
||||||
|
"all?"
|
||||||
|
}
|
||||||
|
|
||||||
|
fn signature(&self) -> Signature {
|
||||||
|
Signature::build(self.name())
|
||||||
|
.required(
|
||||||
|
"predicate",
|
||||||
|
SyntaxShape::RowCondition,
|
||||||
|
"the predicate that must match",
|
||||||
|
)
|
||||||
|
.category(Category::Filters)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn usage(&self) -> &str {
|
||||||
|
"Test if every element of the input matches a predicate."
|
||||||
|
}
|
||||||
|
|
||||||
|
fn examples(&self) -> Vec<Example> {
|
||||||
|
use nu_protocol::Value;
|
||||||
|
|
||||||
|
vec![
|
||||||
|
Example {
|
||||||
|
description: "Find if services are running",
|
||||||
|
example: "echo [[status]; [UP] [UP]] | all? status == UP",
|
||||||
|
result: Some(Value::from(true)),
|
||||||
|
},
|
||||||
|
Example {
|
||||||
|
description: "Check that all values are even",
|
||||||
|
example: "echo [2 4 6 8] | all? ($it mod 2) == 0",
|
||||||
|
result: Some(Value::from(true)),
|
||||||
|
},
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
fn run(
|
||||||
|
&self,
|
||||||
|
engine_state: &EngineState,
|
||||||
|
stack: &mut Stack,
|
||||||
|
call: &Call,
|
||||||
|
input: PipelineData,
|
||||||
|
) -> Result<PipelineData, ShellError> {
|
||||||
|
let predicate = &call.positional[0];
|
||||||
|
let block_id = match predicate {
|
||||||
|
Expression {
|
||||||
|
expr: Expr::RowCondition(block_id),
|
||||||
|
..
|
||||||
|
} => *block_id,
|
||||||
|
_ => return Err(ShellError::InternalError("Expected row condition".into())),
|
||||||
|
};
|
||||||
|
|
||||||
|
let span = call.head;
|
||||||
|
|
||||||
|
let block = engine_state.get_block(block_id);
|
||||||
|
let var_id = block.signature.get_positional(0).and_then(|arg| arg.var_id);
|
||||||
|
let mut stack = stack.collect_captures(&block.captures);
|
||||||
|
|
||||||
|
let ctrlc = engine_state.ctrlc.clone();
|
||||||
|
let engine_state = engine_state.clone();
|
||||||
|
|
||||||
|
Ok(input
|
||||||
|
.into_interruptible_iter(ctrlc)
|
||||||
|
.all(move |value| {
|
||||||
|
if let Some(var_id) = var_id {
|
||||||
|
stack.add_var(var_id, value);
|
||||||
|
}
|
||||||
|
|
||||||
|
eval_block(&engine_state, &mut stack, block, PipelineData::new(span))
|
||||||
|
.map_or(false, |pipeline_data| {
|
||||||
|
pipeline_data.into_value(span).is_true()
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.into_pipeline_data())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_examples() {
|
||||||
|
use crate::test_examples;
|
||||||
|
|
||||||
|
test_examples(All)
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,3 +1,4 @@
|
||||||
|
mod all;
|
||||||
mod append;
|
mod append;
|
||||||
mod collect;
|
mod collect;
|
||||||
mod drop;
|
mod drop;
|
||||||
|
@ -17,6 +18,7 @@ mod where_;
|
||||||
mod wrap;
|
mod wrap;
|
||||||
mod zip;
|
mod zip;
|
||||||
|
|
||||||
|
pub use all::All;
|
||||||
pub use append::Append;
|
pub use append::Append;
|
||||||
pub use collect::Collect;
|
pub use collect::Collect;
|
||||||
pub use drop::*;
|
pub use drop::*;
|
||||||
|
|
|
@ -51,6 +51,16 @@ impl PipelineData {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn into_interruptible_iter(self, ctrlc: Option<Arc<AtomicBool>>) -> PipelineIterator {
|
||||||
|
let mut iter = self.into_iter();
|
||||||
|
|
||||||
|
if let PipelineIterator(PipelineData::Stream(s)) = &mut iter {
|
||||||
|
s.ctrlc = ctrlc;
|
||||||
|
}
|
||||||
|
|
||||||
|
iter
|
||||||
|
}
|
||||||
|
|
||||||
pub fn collect_string(self, separator: &str, config: &Config) -> String {
|
pub fn collect_string(self, separator: &str, config: &Config) -> String {
|
||||||
match self {
|
match self {
|
||||||
PipelineData::Value(v) => v.into_string(separator, config),
|
PipelineData::Value(v) => v.into_string(separator, config),
|
||||||
|
@ -104,13 +114,10 @@ impl PipelineData {
|
||||||
PipelineData::Value(Value::Range { val, .. }) => {
|
PipelineData::Value(Value::Range { val, .. }) => {
|
||||||
Ok(val.into_range_iter()?.map(f).into_pipeline_data(ctrlc))
|
Ok(val.into_range_iter()?.map(f).into_pipeline_data(ctrlc))
|
||||||
}
|
}
|
||||||
PipelineData::Value(v) => {
|
PipelineData::Value(v) => match f(v) {
|
||||||
let output = f(v);
|
Value::Error { error } => Err(error),
|
||||||
match output {
|
v => Ok(v.into_pipeline_data()),
|
||||||
Value::Error { error } => Err(error),
|
},
|
||||||
v => Ok(v.into_pipeline_data()),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -153,10 +160,9 @@ impl PipelineData {
|
||||||
Ok(vals.into_iter().filter(f).into_pipeline_data(ctrlc))
|
Ok(vals.into_iter().filter(f).into_pipeline_data(ctrlc))
|
||||||
}
|
}
|
||||||
PipelineData::Stream(stream) => Ok(stream.filter(f).into_pipeline_data(ctrlc)),
|
PipelineData::Stream(stream) => Ok(stream.filter(f).into_pipeline_data(ctrlc)),
|
||||||
PipelineData::Value(Value::Range { val, .. }) => match val.into_range_iter() {
|
PipelineData::Value(Value::Range { val, .. }) => {
|
||||||
Ok(iter) => Ok(iter.filter(f).into_pipeline_data(ctrlc)),
|
Ok(val.into_range_iter()?.filter(f).into_pipeline_data(ctrlc))
|
||||||
Err(error) => Err(error),
|
}
|
||||||
},
|
|
||||||
PipelineData::Value(v) => {
|
PipelineData::Value(v) => {
|
||||||
if f(&v) {
|
if f(&v) {
|
||||||
Ok(v.into_pipeline_data())
|
Ok(v.into_pipeline_data())
|
||||||
|
@ -190,12 +196,12 @@ impl IntoIterator for PipelineData {
|
||||||
}))
|
}))
|
||||||
}
|
}
|
||||||
PipelineData::Value(Value::Range { val, .. }) => match val.into_range_iter() {
|
PipelineData::Value(Value::Range { val, .. }) => match val.into_range_iter() {
|
||||||
Ok(val) => PipelineIterator(PipelineData::Stream(ValueStream {
|
Ok(iter) => PipelineIterator(PipelineData::Stream(ValueStream {
|
||||||
stream: Box::new(val),
|
stream: Box::new(iter),
|
||||||
ctrlc: None,
|
ctrlc: None,
|
||||||
})),
|
})),
|
||||||
Err(e) => PipelineIterator(PipelineData::Stream(ValueStream {
|
Err(error) => PipelineIterator(PipelineData::Stream(ValueStream {
|
||||||
stream: Box::new(vec![Value::Error { error: e }].into_iter()),
|
stream: Box::new(std::iter::once(Value::Error { error })),
|
||||||
ctrlc: None,
|
ctrlc: None,
|
||||||
})),
|
})),
|
||||||
},
|
},
|
||||||
|
@ -210,10 +216,7 @@ impl Iterator for PipelineIterator {
|
||||||
fn next(&mut self) -> Option<Self::Item> {
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
match &mut self.0 {
|
match &mut self.0 {
|
||||||
PipelineData::Value(Value::Nothing { .. }) => None,
|
PipelineData::Value(Value::Nothing { .. }) => None,
|
||||||
PipelineData::Value(v) => {
|
PipelineData::Value(v) => Some(std::mem::take(v)),
|
||||||
let prev = std::mem::take(v);
|
|
||||||
Some(prev)
|
|
||||||
}
|
|
||||||
PipelineData::Stream(stream) => stream.next(),
|
PipelineData::Stream(stream) => stream.next(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -223,9 +226,12 @@ pub trait IntoPipelineData {
|
||||||
fn into_pipeline_data(self) -> PipelineData;
|
fn into_pipeline_data(self) -> PipelineData;
|
||||||
}
|
}
|
||||||
|
|
||||||
impl IntoPipelineData for Value {
|
impl<V> IntoPipelineData for V
|
||||||
|
where
|
||||||
|
V: Into<Value>,
|
||||||
|
{
|
||||||
fn into_pipeline_data(self) -> PipelineData {
|
fn into_pipeline_data(self) -> PipelineData {
|
||||||
PipelineData::Value(self)
|
PipelineData::Value(self.into())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -233,13 +239,15 @@ pub trait IntoInterruptiblePipelineData {
|
||||||
fn into_pipeline_data(self, ctrlc: Option<Arc<AtomicBool>>) -> PipelineData;
|
fn into_pipeline_data(self, ctrlc: Option<Arc<AtomicBool>>) -> PipelineData;
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<T> IntoInterruptiblePipelineData for T
|
impl<I> IntoInterruptiblePipelineData for I
|
||||||
where
|
where
|
||||||
T: Iterator<Item = Value> + Send + 'static,
|
I: IntoIterator + Send + 'static,
|
||||||
|
I::IntoIter: Send + 'static,
|
||||||
|
<I::IntoIter as Iterator>::Item: Into<Value>,
|
||||||
{
|
{
|
||||||
fn into_pipeline_data(self, ctrlc: Option<Arc<AtomicBool>>) -> PipelineData {
|
fn into_pipeline_data(self, ctrlc: Option<Arc<AtomicBool>>) -> PipelineData {
|
||||||
PipelineData::Stream(ValueStream {
|
PipelineData::Stream(ValueStream {
|
||||||
stream: Box::new(self),
|
stream: Box::new(self.into_iter().map(Into::into)),
|
||||||
ctrlc,
|
ctrlc,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,5 +1,14 @@
|
||||||
use crate::{ShellError, Span, Value};
|
use crate::{ShellError, Span, Value};
|
||||||
|
|
||||||
|
impl From<bool> for Value {
|
||||||
|
fn from(val: bool) -> Self {
|
||||||
|
Value::Bool {
|
||||||
|
val,
|
||||||
|
span: Span::unknown(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl From<u8> for Value {
|
impl From<u8> for Value {
|
||||||
fn from(val: u8) -> Self {
|
fn from(val: u8) -> Self {
|
||||||
Value::Int {
|
Value::Int {
|
||||||
|
|
Loading…
Reference in a new issue