2021-11-14 00:02:54 +01:00
|
|
|
use nu_engine::{eval_block, CallExt};
|
|
|
|
use nu_protocol::ast::{Call, CellPath};
|
2022-01-12 05:06:56 +01:00
|
|
|
use nu_protocol::engine::{CaptureBlock, Command, EngineState, Stack};
|
2021-11-14 00:02:54 +01:00
|
|
|
use nu_protocol::{
|
2022-01-12 05:06:56 +01:00
|
|
|
Category, Example, FromValue, IntoPipelineData, PipelineData, ShellError, Signature, Span,
|
|
|
|
SyntaxShape, Value,
|
2021-11-14 00:02:54 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
#[derive(Clone)]
|
|
|
|
pub struct Update;
|
|
|
|
|
|
|
|
impl Command for Update {
|
|
|
|
fn name(&self) -> &str {
|
|
|
|
"update"
|
|
|
|
}
|
|
|
|
|
|
|
|
fn signature(&self) -> Signature {
|
|
|
|
Signature::build("update")
|
|
|
|
.required(
|
|
|
|
"field",
|
|
|
|
SyntaxShape::CellPath,
|
2022-03-14 21:32:33 +01:00
|
|
|
"the name of the column to update or create",
|
2021-11-14 00:02:54 +01:00
|
|
|
)
|
|
|
|
.required(
|
|
|
|
"replacement value",
|
|
|
|
SyntaxShape::Any,
|
|
|
|
"the new value to give the cell(s)",
|
|
|
|
)
|
2021-11-17 05:22:37 +01:00
|
|
|
.category(Category::Filters)
|
2021-11-14 00:02:54 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
fn usage(&self) -> &str {
|
2022-03-14 21:32:33 +01:00
|
|
|
"Update an existing column to have a new value, or create a new column."
|
2021-11-14 00:02:54 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
fn run(
|
|
|
|
&self,
|
|
|
|
engine_state: &EngineState,
|
|
|
|
stack: &mut Stack,
|
|
|
|
call: &Call,
|
|
|
|
input: PipelineData,
|
|
|
|
) -> Result<nu_protocol::PipelineData, nu_protocol::ShellError> {
|
|
|
|
update(engine_state, stack, call, input)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn examples(&self) -> Vec<Example> {
|
|
|
|
vec![Example {
|
|
|
|
description: "Update a column value",
|
|
|
|
example: "echo {'name': 'nu', 'stars': 5} | update name 'Nushell'",
|
2021-12-19 08:46:13 +01:00
|
|
|
result: Some(Value::Record { cols: vec!["name".into(), "stars".into()], vals: vec![Value::test_string("Nushell"), Value::test_int(5)], span: Span::test_data()}),
|
2022-03-14 21:32:33 +01:00
|
|
|
}, Example {
|
|
|
|
description: "Add a new column",
|
|
|
|
example: "echo {'name': 'nu', 'stars': 5} | update language 'Rust'",
|
|
|
|
result: Some(Value::Record { cols: vec!["name".into(), "stars".into(), "language".into()], vals: vec![Value::test_string("nu"), Value::test_int(5), Value::test_string("Rust")], span: Span::test_data()}),
|
2022-02-18 14:06:52 +01:00
|
|
|
}, Example {
|
|
|
|
description: "Use in block form for more involved updating logic",
|
|
|
|
example: "echo [[count fruit]; [1 'apple']] | update count {|f| $f.count + 1}",
|
|
|
|
result: Some(Value::List { vals: vec![Value::Record { cols: vec!["count".into(), "fruit".into()], vals: vec![Value::test_int(2), Value::test_string("apple")], span: Span::test_data()}], span: Span::test_data()}),
|
2021-11-14 00:02:54 +01:00
|
|
|
}, Example {
|
|
|
|
description: "Use in block form for more involved updating logic",
|
2022-03-14 21:32:33 +01:00
|
|
|
example: "echo [[project, authors]; ['nu', ['Andrés', 'JT', 'Yehuda']]] | update authors {|a| $a.authors | str collect ','}",
|
2021-12-19 08:46:13 +01:00
|
|
|
result: Some(Value::List { vals: vec![Value::Record { cols: vec!["project".into(), "authors".into()], vals: vec![Value::test_string("nu"), Value::test_string("Andrés,JT,Yehuda")], span: Span::test_data()}], span: Span::test_data()}),
|
2021-11-14 00:02:54 +01:00
|
|
|
}]
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn update(
|
|
|
|
engine_state: &EngineState,
|
|
|
|
stack: &mut Stack,
|
|
|
|
call: &Call,
|
|
|
|
input: PipelineData,
|
|
|
|
) -> Result<PipelineData, ShellError> {
|
|
|
|
let span = call.head;
|
|
|
|
|
|
|
|
let cell_path: CellPath = call.req(engine_state, stack, 0)?;
|
|
|
|
let replacement: Value = call.req(engine_state, stack, 1)?;
|
2022-02-21 23:22:21 +01:00
|
|
|
|
|
|
|
let redirect_stdout = call.redirect_stdout;
|
|
|
|
let redirect_stderr = call.redirect_stderr;
|
|
|
|
|
2021-11-14 00:02:54 +01:00
|
|
|
let engine_state = engine_state.clone();
|
|
|
|
let ctrlc = engine_state.ctrlc.clone();
|
|
|
|
|
|
|
|
// Replace is a block, so set it up and run it instead of using it as the replacement
|
2022-01-12 05:06:56 +01:00
|
|
|
if replacement.as_block().is_ok() {
|
|
|
|
let capture_block: CaptureBlock = FromValue::from_value(&replacement)?;
|
|
|
|
let block = engine_state.get_block(capture_block.block_id).clone();
|
2021-11-14 00:02:54 +01:00
|
|
|
|
2022-01-12 05:06:56 +01:00
|
|
|
let mut stack = stack.captures_to_stack(&capture_block.captures);
|
2022-01-04 23:30:34 +01:00
|
|
|
let orig_env_vars = stack.env_vars.clone();
|
|
|
|
let orig_env_hidden = stack.env_hidden.clone();
|
2021-11-14 00:02:54 +01:00
|
|
|
|
|
|
|
input.map(
|
|
|
|
move |mut input| {
|
2022-01-04 23:30:34 +01:00
|
|
|
stack.with_env(&orig_env_vars, &orig_env_hidden);
|
|
|
|
|
2021-11-14 00:02:54 +01:00
|
|
|
if let Some(var) = block.signature.get_positional(0) {
|
|
|
|
if let Some(var_id) = &var.var_id {
|
|
|
|
stack.add_var(*var_id, input.clone())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
let output = eval_block(
|
|
|
|
&engine_state,
|
|
|
|
&mut stack,
|
|
|
|
&block,
|
|
|
|
input.clone().into_pipeline_data(),
|
2022-02-21 23:22:21 +01:00
|
|
|
redirect_stdout,
|
|
|
|
redirect_stderr,
|
2021-11-14 00:02:54 +01:00
|
|
|
);
|
|
|
|
|
|
|
|
match output {
|
|
|
|
Ok(pd) => {
|
|
|
|
if let Err(e) =
|
|
|
|
input.replace_data_at_cell_path(&cell_path.members, pd.into_value(span))
|
|
|
|
{
|
|
|
|
return Value::Error { error: e };
|
|
|
|
}
|
|
|
|
|
|
|
|
input
|
|
|
|
}
|
|
|
|
Err(e) => Value::Error { error: e },
|
|
|
|
}
|
|
|
|
},
|
|
|
|
ctrlc,
|
|
|
|
)
|
|
|
|
} else {
|
|
|
|
input.map(
|
|
|
|
move |mut input| {
|
|
|
|
let replacement = replacement.clone();
|
|
|
|
|
|
|
|
if let Err(e) = input.replace_data_at_cell_path(&cell_path.members, replacement) {
|
|
|
|
return Value::Error { error: e };
|
|
|
|
}
|
|
|
|
|
|
|
|
input
|
|
|
|
},
|
|
|
|
ctrlc,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
}
|
2022-03-14 21:32:33 +01:00
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod test {
|
|
|
|
use super::*;
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_examples() {
|
|
|
|
use crate::test_examples;
|
|
|
|
|
|
|
|
test_examples(Update {})
|
|
|
|
}
|
|
|
|
}
|