123 lines
4.0 KiB
Rust
123 lines
4.0 KiB
Rust
use crate::commands::WholeStreamCommand;
|
|
use crate::prelude::*;
|
|
use nu_errors::ShellError;
|
|
use nu_protocol::hir::Operator;
|
|
use nu_protocol::{
|
|
Primitive, RangeInclusion, ReturnSuccess, Signature, SyntaxShape, UntaggedValue, Value,
|
|
};
|
|
|
|
pub struct Echo;
|
|
|
|
#[derive(Deserialize)]
|
|
pub struct EchoArgs {
|
|
pub rest: Vec<Value>,
|
|
}
|
|
|
|
#[async_trait]
|
|
impl WholeStreamCommand for Echo {
|
|
fn name(&self) -> &str {
|
|
"echo"
|
|
}
|
|
|
|
fn signature(&self) -> Signature {
|
|
Signature::build("echo").rest(SyntaxShape::Any, "the values to echo")
|
|
}
|
|
|
|
fn usage(&self) -> &str {
|
|
"Echo the arguments back to the user."
|
|
}
|
|
|
|
async fn run(
|
|
&self,
|
|
args: CommandArgs,
|
|
registry: &CommandRegistry,
|
|
) -> Result<OutputStream, ShellError> {
|
|
echo(args, registry).await
|
|
}
|
|
|
|
fn examples(&self) -> Vec<Example> {
|
|
vec![
|
|
Example {
|
|
description: "Put a hello message in the pipeline",
|
|
example: "echo 'hello'",
|
|
result: Some(vec![Value::from("hello")]),
|
|
},
|
|
Example {
|
|
description: "Print the value of the special '$nu' variable",
|
|
example: "echo $nu",
|
|
result: None,
|
|
},
|
|
]
|
|
}
|
|
}
|
|
|
|
async fn echo(args: CommandArgs, registry: &CommandRegistry) -> Result<OutputStream, ShellError> {
|
|
let registry = registry.clone();
|
|
let (args, _): (EchoArgs, _) = args.process(®istry).await?;
|
|
|
|
let stream = args.rest.into_iter().map(|i| {
|
|
match i.as_string() {
|
|
Ok(s) => {
|
|
OutputStream::one(Ok(ReturnSuccess::Value(
|
|
UntaggedValue::string(s).into_value(i.tag.clone()),
|
|
)))
|
|
}
|
|
_ => match i {
|
|
Value {
|
|
value: UntaggedValue::Table(table),
|
|
..
|
|
} => {
|
|
futures::stream::iter(table.into_iter().map(ReturnSuccess::value)).to_output_stream()
|
|
}
|
|
Value {
|
|
value: UntaggedValue::Primitive(Primitive::Range(range)),
|
|
tag
|
|
} => {
|
|
let mut output_vec = vec![];
|
|
|
|
let mut current = range.from.0.item;
|
|
while current != range.to.0.item {
|
|
output_vec.push(Ok(ReturnSuccess::Value(UntaggedValue::Primitive(current.clone()).into_value(&tag))));
|
|
current = match crate::data::value::compute_values(Operator::Plus, &UntaggedValue::Primitive(current), &UntaggedValue::int(1)) {
|
|
Ok(result) => match result {
|
|
UntaggedValue::Primitive(p) => p,
|
|
_ => {
|
|
return OutputStream::one(Err(ShellError::unimplemented("Internal error: expected a primitive result from increment")));
|
|
}
|
|
},
|
|
Err((left_type, right_type)) => {
|
|
return OutputStream::one(Err(ShellError::coerce_error(
|
|
left_type.spanned(tag.span),
|
|
right_type.spanned(tag.span),
|
|
)));
|
|
}
|
|
}
|
|
}
|
|
if let RangeInclusion::Inclusive = range.to.1 {
|
|
output_vec.push(Ok(ReturnSuccess::Value(UntaggedValue::Primitive(current).into_value(&tag))));
|
|
}
|
|
|
|
futures::stream::iter(output_vec.into_iter()).to_output_stream()
|
|
}
|
|
_ => {
|
|
OutputStream::one(Ok(ReturnSuccess::Value(i.clone())))
|
|
}
|
|
},
|
|
}
|
|
});
|
|
|
|
Ok(futures::stream::iter(stream).flatten().to_output_stream())
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::Echo;
|
|
|
|
#[test]
|
|
fn examples_work_as_expected() {
|
|
use crate::examples::test as test_examples;
|
|
|
|
test_examples(Echo {})
|
|
}
|
|
}
|