forked from extern/nushell
b35914bd17
* category option for signature * category option for signature * column description for $scope
36 lines
935 B
Rust
36 lines
935 B
Rust
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()))
|
|
}
|
|
}
|