mirror of
https://github.com/nushell/nushell
synced 2024-12-28 14:03:09 +00:00
Create random binary
command (#13542)
# Description/User-Facing Changes Creates a new `random binary <LENGTH>` command, which returns random bytes. Resolve #13500
This commit is contained in:
parent
56ed532038
commit
1c37f4b958
3 changed files with 67 additions and 0 deletions
|
@ -397,6 +397,7 @@ pub fn add_shell_command_context(mut engine_state: EngineState) -> EngineState {
|
||||||
RandomFloat,
|
RandomFloat,
|
||||||
RandomInt,
|
RandomInt,
|
||||||
RandomUuid,
|
RandomUuid,
|
||||||
|
RandomBinary
|
||||||
};
|
};
|
||||||
|
|
||||||
// Generators
|
// Generators
|
||||||
|
|
64
crates/nu-command/src/random/binary.rs
Normal file
64
crates/nu-command/src/random/binary.rs
Normal file
|
@ -0,0 +1,64 @@
|
||||||
|
use nu_engine::command_prelude::*;
|
||||||
|
|
||||||
|
use rand::{thread_rng, RngCore};
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct SubCommand;
|
||||||
|
|
||||||
|
impl Command for SubCommand {
|
||||||
|
fn name(&self) -> &str {
|
||||||
|
"random binary"
|
||||||
|
}
|
||||||
|
|
||||||
|
fn signature(&self) -> Signature {
|
||||||
|
Signature::build("random binary")
|
||||||
|
.input_output_types(vec![(Type::Nothing, Type::Binary)])
|
||||||
|
.allow_variants_without_examples(true)
|
||||||
|
.required("length", SyntaxShape::Int, "Length of the output binary.")
|
||||||
|
.category(Category::Random)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn usage(&self) -> &str {
|
||||||
|
"Generate random bytes."
|
||||||
|
}
|
||||||
|
|
||||||
|
fn search_terms(&self) -> Vec<&str> {
|
||||||
|
vec!["generate", "bytes"]
|
||||||
|
}
|
||||||
|
|
||||||
|
fn run(
|
||||||
|
&self,
|
||||||
|
engine_state: &EngineState,
|
||||||
|
stack: &mut Stack,
|
||||||
|
call: &Call,
|
||||||
|
_input: PipelineData,
|
||||||
|
) -> Result<PipelineData, ShellError> {
|
||||||
|
let length = call.req(engine_state, stack, 0)?;
|
||||||
|
let mut rng = thread_rng();
|
||||||
|
|
||||||
|
let mut out = vec![0u8; length];
|
||||||
|
rng.fill_bytes(&mut out);
|
||||||
|
|
||||||
|
Ok(Value::binary(out, call.head).into_pipeline_data())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn examples(&self) -> Vec<Example> {
|
||||||
|
vec![Example {
|
||||||
|
description: "Generate 16 random bytes",
|
||||||
|
example: "random bytes 16",
|
||||||
|
result: None,
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod test {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_examples() {
|
||||||
|
use crate::test_examples;
|
||||||
|
|
||||||
|
test_examples(SubCommand {})
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,3 +1,4 @@
|
||||||
|
mod binary;
|
||||||
mod bool;
|
mod bool;
|
||||||
mod chars;
|
mod chars;
|
||||||
mod dice;
|
mod dice;
|
||||||
|
@ -6,6 +7,7 @@ mod int;
|
||||||
mod random_;
|
mod random_;
|
||||||
mod uuid;
|
mod uuid;
|
||||||
|
|
||||||
|
pub use self::binary::SubCommand as RandomBinary;
|
||||||
pub use self::bool::SubCommand as RandomBool;
|
pub use self::bool::SubCommand as RandomBool;
|
||||||
pub use self::chars::SubCommand as RandomChars;
|
pub use self::chars::SubCommand as RandomChars;
|
||||||
pub use self::dice::SubCommand as RandomDice;
|
pub use self::dice::SubCommand as RandomDice;
|
||||||
|
|
Loading…
Reference in a new issue