mirror of
https://github.com/nushell/nushell
synced 2025-01-09 19:59:02 +00: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`).
108 lines
3.3 KiB
Rust
108 lines
3.3 KiB
Rust
use nu::{serve_plugin, Plugin};
|
|
use nu_errors::ShellError;
|
|
use nu_protocol::{
|
|
CallInfo, Primitive, ReturnSuccess, ReturnValue, Signature, SyntaxShape, UntaggedValue, Value,
|
|
};
|
|
|
|
use regex::Regex;
|
|
|
|
struct Match {
|
|
column: String,
|
|
regex: Regex,
|
|
}
|
|
|
|
impl Match {
|
|
fn new() -> Self {
|
|
Match {
|
|
column: String::new(),
|
|
regex: Regex::new("").unwrap(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Plugin for Match {
|
|
fn config(&mut self) -> Result<Signature, ShellError> {
|
|
Ok(Signature::build("match")
|
|
.desc("filter rows by regex")
|
|
.required("member", SyntaxShape::Member, "the column name to match")
|
|
.required("regex", SyntaxShape::String, "the regex to match with")
|
|
.filter())
|
|
}
|
|
fn begin_filter(&mut self, call_info: CallInfo) -> Result<Vec<ReturnValue>, ShellError> {
|
|
if let Some(args) = call_info.args.positional {
|
|
match &args[0] {
|
|
Value {
|
|
value: UntaggedValue::Primitive(Primitive::String(s)),
|
|
..
|
|
} => {
|
|
self.column = s.clone();
|
|
}
|
|
Value { tag, .. } => {
|
|
return Err(ShellError::labeled_error(
|
|
"Unrecognized type in params",
|
|
"value",
|
|
tag,
|
|
));
|
|
}
|
|
}
|
|
match &args[1] {
|
|
Value {
|
|
value: UntaggedValue::Primitive(Primitive::String(s)),
|
|
..
|
|
} => {
|
|
self.regex = Regex::new(s).unwrap();
|
|
}
|
|
Value { tag, .. } => {
|
|
return Err(ShellError::labeled_error(
|
|
"Unrecognized type in params",
|
|
"value",
|
|
tag,
|
|
));
|
|
}
|
|
}
|
|
}
|
|
Ok(vec![])
|
|
}
|
|
|
|
fn filter(&mut self, input: Value) -> Result<Vec<ReturnValue>, ShellError> {
|
|
let flag: bool;
|
|
match &input {
|
|
Value {
|
|
value: UntaggedValue::Row(dict),
|
|
tag,
|
|
} => {
|
|
if let Some(val) = dict.entries.get(&self.column) {
|
|
match val {
|
|
Value {
|
|
value: UntaggedValue::Primitive(Primitive::String(s)),
|
|
..
|
|
} => {
|
|
flag = self.regex.is_match(s);
|
|
}
|
|
Value { tag, .. } => {
|
|
return Err(ShellError::labeled_error("expected string", "value", tag));
|
|
}
|
|
}
|
|
} else {
|
|
return Err(ShellError::labeled_error(
|
|
format!("column not in row! {:?} {:?}", &self.column, dict),
|
|
"row",
|
|
tag,
|
|
));
|
|
}
|
|
}
|
|
Value { tag, .. } => {
|
|
return Err(ShellError::labeled_error("Expected row", "value", tag));
|
|
}
|
|
}
|
|
if flag {
|
|
Ok(vec![Ok(ReturnSuccess::Value(input))])
|
|
} else {
|
|
Ok(vec![])
|
|
}
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
serve_plugin(&mut Match::new());
|
|
}
|