forked from extern/nushell
36 lines
897 B
Rust
36 lines
897 B
Rust
|
use nu_protocol::ast::Call;
|
||
|
use nu_protocol::engine::{Command, EngineState, Stack};
|
||
|
use nu_protocol::{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")
|
||
|
}
|
||
|
|
||
|
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()))
|
||
|
}
|
||
|
}
|