mirror of
https://github.com/nushell/nushell.git
synced 2024-12-15 11:42:45 +01:00
e4226def16
This commit extracts five new crates:
- nu-source, which contains the core source-code handling logic in Nu,
including Text, Span, and also the pretty.rs-based debug logic
- nu-parser, which is the parser and expander logic
- nu-protocol, which is the bulk of the types and basic conveniences
used by plugins
- nu-errors, which contains ShellError, ParseError and error handling
conveniences
- nu-textview, which is the textview plugin extracted into a crate
One of the major consequences of this refactor is that it's no longer
possible to `impl X for Spanned<Y>` outside of the `nu-source` crate, so
a lot of types became more concrete (Value became a concrete type
instead of Spanned<Value>, for example).
This also turned a number of inherent methods in the main nu crate into
plain functions (impl Value {} became a bunch of functions in the
`value` namespace in `crate::data::value`).
103 lines
3.1 KiB
Rust
103 lines
3.1 KiB
Rust
use nu::{serve_plugin, Plugin, ValueExt};
|
|
use nu_errors::ShellError;
|
|
use nu_protocol::{
|
|
CallInfo, ColumnPath, Primitive, ReturnSuccess, ReturnValue, Signature, SpannedTypeName,
|
|
SyntaxShape, UntaggedValue, Value,
|
|
};
|
|
use nu_source::Tagged;
|
|
|
|
struct Edit {
|
|
field: Option<Tagged<ColumnPath>>,
|
|
value: Option<UntaggedValue>,
|
|
}
|
|
impl Edit {
|
|
fn new() -> Edit {
|
|
Edit {
|
|
field: None,
|
|
value: None,
|
|
}
|
|
}
|
|
|
|
fn edit(&self, value: Value) -> Result<Value, ShellError> {
|
|
let value_tag = value.tag();
|
|
match (value, self.value.clone()) {
|
|
(
|
|
obj @ Value {
|
|
value: UntaggedValue::Row(_),
|
|
..
|
|
},
|
|
Some(v),
|
|
) => match &self.field {
|
|
Some(f) => {
|
|
match obj.replace_data_at_column_path(&f, v.clone().into_untagged_value()) {
|
|
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,
|
|
"the name of the column to edit",
|
|
)
|
|
.required(
|
|
"Value",
|
|
SyntaxShape::String,
|
|
"the new value to give the cell(s)",
|
|
)
|
|
.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 @ Value {
|
|
value: UntaggedValue::Primitive(Primitive::ColumnPath(_)),
|
|
..
|
|
} => {
|
|
self.field = Some(table.as_column_path()?);
|
|
}
|
|
value => return Err(ShellError::type_error("table", value.spanned_type_name())),
|
|
}
|
|
|
|
match &args[1] {
|
|
Value { value: v, .. } => {
|
|
self.value = Some(v.clone());
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(vec![])
|
|
}
|
|
|
|
fn filter(&mut self, input: Value) -> Result<Vec<ReturnValue>, ShellError> {
|
|
Ok(vec![ReturnSuccess::value(self.edit(input)?)])
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
serve_plugin(&mut Edit::new());
|
|
}
|