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 { 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()) } }