b2c5af457e
This improves incremental build time when working on what was previously the root package. For example, previously all plugins would be rebuilt with a change to `src/commands/classified/external.rs`, but now only `nu-cli` will have to be rebuilt (and anything that depends on it).
67 lines
1.8 KiB
Rust
67 lines
1.8 KiB
Rust
use crate::commands::PerItemCommand;
|
|
use crate::context::CommandRegistry;
|
|
use crate::prelude::*;
|
|
use nu_errors::ShellError;
|
|
use nu_protocol::{CallInfo, ReturnSuccess, Signature, SyntaxShape, UntaggedValue, Value};
|
|
use nu_value_ext::ValueExt;
|
|
|
|
pub struct Insert;
|
|
|
|
impl PerItemCommand for Insert {
|
|
fn name(&self) -> &str {
|
|
"insert"
|
|
}
|
|
|
|
fn signature(&self) -> Signature {
|
|
Signature::build("insert")
|
|
.required(
|
|
"column",
|
|
SyntaxShape::ColumnPath,
|
|
"the column name to insert",
|
|
)
|
|
.required(
|
|
"value",
|
|
SyntaxShape::String,
|
|
"the value to give the cell(s)",
|
|
)
|
|
}
|
|
|
|
fn usage(&self) -> &str {
|
|
"Edit an existing column to have a new value."
|
|
}
|
|
|
|
fn run(
|
|
&self,
|
|
call_info: &CallInfo,
|
|
_registry: &CommandRegistry,
|
|
_raw_args: &RawCommandArgs,
|
|
value: Value,
|
|
) -> Result<OutputStream, ShellError> {
|
|
let value_tag = value.tag();
|
|
let field = call_info.args.expect_nth(0)?.as_column_path()?;
|
|
let replacement = call_info.args.expect_nth(1)?.tagged_unknown();
|
|
|
|
let stream = match value {
|
|
obj
|
|
@
|
|
Value {
|
|
value: UntaggedValue::Row(_),
|
|
..
|
|
} => match obj.insert_data_at_column_path(&field, replacement.item.clone()) {
|
|
Ok(v) => futures::stream::iter(vec![Ok(ReturnSuccess::Value(v))]),
|
|
Err(err) => return Err(err),
|
|
},
|
|
|
|
_ => {
|
|
return Err(ShellError::labeled_error(
|
|
"Unrecognized type in stream",
|
|
"original value",
|
|
value_tag,
|
|
))
|
|
}
|
|
};
|
|
|
|
Ok(stream.to_output_stream())
|
|
}
|
|
}
|