forked from extern/nushell
91 lines
2.5 KiB
Rust
91 lines
2.5 KiB
Rust
use crate::command_registry::CommandRegistry;
|
|
use crate::commands::WholeStreamCommand;
|
|
use crate::prelude::*;
|
|
use nu_errors::ShellError;
|
|
use nu_protocol::{ReturnSuccess, Signature, SyntaxShape, UntaggedValue, Value};
|
|
|
|
#[derive(Deserialize)]
|
|
struct Arguments {
|
|
value: Value,
|
|
}
|
|
|
|
pub struct Command;
|
|
|
|
#[async_trait]
|
|
impl WholeStreamCommand for Command {
|
|
fn name(&self) -> &str {
|
|
"append"
|
|
}
|
|
|
|
fn signature(&self) -> Signature {
|
|
Signature::build("append").required(
|
|
"row value",
|
|
SyntaxShape::Any,
|
|
"the value of the row to append to the table",
|
|
)
|
|
}
|
|
|
|
fn usage(&self) -> &str {
|
|
"Append a row to the table"
|
|
}
|
|
|
|
async fn run(
|
|
&self,
|
|
args: CommandArgs,
|
|
registry: &CommandRegistry,
|
|
) -> Result<OutputStream, ShellError> {
|
|
let (Arguments { mut value }, input) = args.process(registry).await?;
|
|
|
|
let input: Vec<Value> = input.collect().await;
|
|
|
|
if let Some(first) = input.get(0) {
|
|
value.tag = first.tag();
|
|
}
|
|
|
|
// Checks if we are trying to append a row literal
|
|
if let Value {
|
|
value: UntaggedValue::Table(values),
|
|
tag,
|
|
} = &value
|
|
{
|
|
if values.len() == 1 && values[0].is_row() {
|
|
value = values[0].value.clone().into_value(tag);
|
|
}
|
|
}
|
|
|
|
Ok(futures::stream::iter(
|
|
input
|
|
.into_iter()
|
|
.chain(vec![value])
|
|
.map(ReturnSuccess::value),
|
|
)
|
|
.to_output_stream())
|
|
}
|
|
|
|
fn examples(&self) -> Vec<Example> {
|
|
use nu_protocol::row;
|
|
|
|
vec![
|
|
Example {
|
|
description: "Add values to the end of the table",
|
|
example: "echo [1 2 3] | append 4",
|
|
result: Some(vec![
|
|
UntaggedValue::int(1).into(),
|
|
UntaggedValue::int(2).into(),
|
|
UntaggedValue::int(3).into(),
|
|
UntaggedValue::int(4).into(),
|
|
]),
|
|
},
|
|
Example {
|
|
description: "Add row value to the end of the table",
|
|
example: "echo [[country]; [Ecuador] ['New Zealand']] | append [[country]; [USA]]",
|
|
result: Some(vec![
|
|
row! { "country".into() => Value::from("Ecuador")},
|
|
row! { "country".into() => Value::from("New Zealand")},
|
|
row! { "country".into() => Value::from("USA")},
|
|
]),
|
|
},
|
|
]
|
|
}
|
|
}
|