nushell/crates/nu-command/src/filters/shuffle.rs

36 lines
935 B
Rust
Raw Normal View History

use nu_protocol::ast::Call;
use nu_protocol::engine::{Command, EngineState, Stack};
use nu_protocol::{Category, IntoInterruptiblePipelineData, PipelineData, ShellError, Signature};
use rand::prelude::SliceRandom;
use rand::thread_rng;
#[derive(Clone)]
pub struct Shuffle;
impl Command for Shuffle {
fn name(&self) -> &str {
"shuffle"
}
fn signature(&self) -> nu_protocol::Signature {
Signature::build("shuffle").category(Category::Filters)
}
fn usage(&self) -> &str {
"Shuffle rows randomly."
}
fn run(
&self,
engine_state: &EngineState,
_stack: &mut Stack,
_call: &Call,
input: PipelineData,
) -> Result<PipelineData, ShellError> {
let mut v: Vec<_> = input.into_iter().collect();
v.shuffle(&mut thread_rng());
let iter = v.into_iter();
Ok(iter.into_pipeline_data(engine_state.ctrlc.clone()))
}
}