mirror of
https://github.com/nushell/nushell
synced 2025-01-25 11:25:21 +00:00
193b00764b
* Moves off of draining between filters. Instead, the sink will pull on the stream, and will drain element-wise. This moves the whole stream to being lazy. * Adds ctrl-c support and connects it into some of the key points where we pull on the stream. If a ctrl-c is detect, we immediately halt pulling on the stream and return to the prompt. * Moves away from having a SourceMap where anchor locations are stored. Now AnchorLocation is kept directly in the Tag. * To make this possible, split tag and span. Span is largely used in the parser and is copyable. Tag is now no longer copyable.
84 lines
2.6 KiB
Rust
84 lines
2.6 KiB
Rust
use nu::{
|
|
serve_plugin, CallInfo, Plugin, ReturnSuccess, ReturnValue, ShellError, Signature, SyntaxShape,
|
|
Tagged, Value,
|
|
};
|
|
|
|
pub type ColumnPath = Tagged<Vec<Tagged<String>>>;
|
|
|
|
struct Edit {
|
|
field: Option<ColumnPath>,
|
|
value: Option<Value>,
|
|
}
|
|
impl Edit {
|
|
fn new() -> Edit {
|
|
Edit {
|
|
field: None,
|
|
value: None,
|
|
}
|
|
}
|
|
|
|
fn edit(&self, value: Tagged<Value>) -> Result<Tagged<Value>, ShellError> {
|
|
let value_tag = value.tag();
|
|
match (value.item, self.value.clone()) {
|
|
(obj @ Value::Row(_), Some(v)) => match &self.field {
|
|
Some(f) => match obj.replace_data_at_column_path(value_tag, &f, v) {
|
|
Some(v) => return Ok(v),
|
|
None => {
|
|
return Err(ShellError::labeled_error(
|
|
"edit could not find place to insert column",
|
|
"column name",
|
|
&f.tag,
|
|
))
|
|
}
|
|
},
|
|
None => Err(ShellError::untagged_runtime_error(
|
|
"edit needs a column when changing a value in a table",
|
|
)),
|
|
},
|
|
_ => Err(ShellError::labeled_error(
|
|
"Unrecognized type in stream",
|
|
"original value",
|
|
value_tag,
|
|
)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Plugin for Edit {
|
|
fn config(&mut self) -> Result<Signature, ShellError> {
|
|
Ok(Signature::build("edit")
|
|
.desc("Edit an existing column to have a new value.")
|
|
.required("Field", SyntaxShape::ColumnPath)
|
|
.required("Value", SyntaxShape::String)
|
|
.filter())
|
|
}
|
|
|
|
fn begin_filter(&mut self, call_info: CallInfo) -> Result<Vec<ReturnValue>, ShellError> {
|
|
if let Some(args) = call_info.args.positional {
|
|
match &args[0] {
|
|
table @ Tagged {
|
|
item: Value::Table(_),
|
|
..
|
|
} => {
|
|
self.field = Some(table.as_column_path()?);
|
|
}
|
|
value => return Err(ShellError::type_error("table", value.tagged_type_name())),
|
|
}
|
|
match &args[1] {
|
|
Tagged { item: v, .. } => {
|
|
self.value = Some(v.clone());
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(vec![])
|
|
}
|
|
|
|
fn filter(&mut self, input: Tagged<Value>) -> Result<Vec<ReturnValue>, ShellError> {
|
|
Ok(vec![ReturnSuccess::value(self.edit(input)?)])
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
serve_plugin(&mut Edit::new());
|
|
}
|