mirror of
https://github.com/nushell/nushell.git
synced 2024-11-26 02:13:47 +01:00
Move format/parse to core commands
This commit is contained in:
parent
251c3e103d
commit
e98ed1b43d
@ -1,149 +0,0 @@
|
|||||||
use nu::{serve_plugin, Plugin, TaggedDictBuilder};
|
|
||||||
use nu_errors::ShellError;
|
|
||||||
use nu_protocol::{
|
|
||||||
CallInfo, Primitive, ReturnSuccess, ReturnValue, Signature, SyntaxShape, UntaggedValue, Value,
|
|
||||||
};
|
|
||||||
|
|
||||||
use nom::{
|
|
||||||
bytes::complete::{tag, take_while},
|
|
||||||
IResult,
|
|
||||||
};
|
|
||||||
use regex::Regex;
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
enum ParseCommand {
|
|
||||||
Text(String),
|
|
||||||
Column(String),
|
|
||||||
}
|
|
||||||
|
|
||||||
fn parse(input: &str) -> IResult<&str, Vec<ParseCommand>> {
|
|
||||||
let mut output = vec![];
|
|
||||||
|
|
||||||
let mut loop_input = input;
|
|
||||||
loop {
|
|
||||||
let (input, before) = take_while(|c| c != '{')(loop_input)?;
|
|
||||||
if !before.is_empty() {
|
|
||||||
output.push(ParseCommand::Text(before.to_string()));
|
|
||||||
}
|
|
||||||
if input != "" {
|
|
||||||
// Look for column as we're now at one
|
|
||||||
let (input, _) = tag("{")(input)?;
|
|
||||||
let (input, column) = take_while(|c| c != '}')(input)?;
|
|
||||||
let (input, _) = tag("}")(input)?;
|
|
||||||
|
|
||||||
output.push(ParseCommand::Column(column.to_string()));
|
|
||||||
loop_input = input;
|
|
||||||
} else {
|
|
||||||
loop_input = input;
|
|
||||||
}
|
|
||||||
if loop_input == "" {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok((loop_input, output))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn column_names(commands: &[ParseCommand]) -> Vec<String> {
|
|
||||||
let mut output = vec![];
|
|
||||||
|
|
||||||
for command in commands {
|
|
||||||
if let ParseCommand::Column(c) = command {
|
|
||||||
output.push(c.clone());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
output
|
|
||||||
}
|
|
||||||
|
|
||||||
fn build_regex(commands: &[ParseCommand]) -> String {
|
|
||||||
let mut output = String::new();
|
|
||||||
|
|
||||||
for command in commands {
|
|
||||||
match command {
|
|
||||||
ParseCommand::Text(s) => {
|
|
||||||
output.push_str(&s.replace("(", "\\("));
|
|
||||||
}
|
|
||||||
ParseCommand::Column(_) => {
|
|
||||||
output.push_str("(.*)");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
output
|
|
||||||
}
|
|
||||||
struct Parse {
|
|
||||||
regex: Regex,
|
|
||||||
column_names: Vec<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Parse {
|
|
||||||
fn new() -> Self {
|
|
||||||
Parse {
|
|
||||||
regex: Regex::new("").unwrap(),
|
|
||||||
column_names: vec![],
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Plugin for Parse {
|
|
||||||
fn config(&mut self) -> Result<Signature, ShellError> {
|
|
||||||
Ok(Signature::build("parse")
|
|
||||||
.desc("Parse columns from string data using a simple pattern")
|
|
||||||
.required(
|
|
||||||
"pattern",
|
|
||||||
SyntaxShape::Any,
|
|
||||||
"the pattern to match. Eg) \"{foo}: {bar}\"",
|
|
||||||
)
|
|
||||||
.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(pattern)),
|
|
||||||
..
|
|
||||||
} => {
|
|
||||||
//self.pattern = s.clone();
|
|
||||||
let parse_pattern = parse(&pattern).unwrap();
|
|
||||||
let parse_regex = build_regex(&parse_pattern.1);
|
|
||||||
|
|
||||||
self.column_names = column_names(&parse_pattern.1);
|
|
||||||
|
|
||||||
self.regex = Regex::new(&parse_regex).unwrap();
|
|
||||||
}
|
|
||||||
Value { tag, .. } => {
|
|
||||||
return Err(ShellError::labeled_error(
|
|
||||||
"Unrecognized type in params",
|
|
||||||
"expected a string",
|
|
||||||
tag,
|
|
||||||
));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Ok(vec![])
|
|
||||||
}
|
|
||||||
|
|
||||||
fn filter(&mut self, input: Value) -> Result<Vec<ReturnValue>, ShellError> {
|
|
||||||
let mut results = vec![];
|
|
||||||
if let Ok(s) = input.as_string() {
|
|
||||||
for cap in self.regex.captures_iter(&s) {
|
|
||||||
let mut dict = TaggedDictBuilder::new(input.tag());
|
|
||||||
|
|
||||||
for (idx, column_name) in self.column_names.iter().enumerate() {
|
|
||||||
dict.insert_untagged(
|
|
||||||
column_name,
|
|
||||||
UntaggedValue::string(&cap[idx + 1].to_string()),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
results.push(ReturnSuccess::value(dict.into_value()));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Ok(results)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn main() {
|
|
||||||
serve_plugin(&mut Parse::new());
|
|
||||||
}
|
|
Loading…
Reference in New Issue
Block a user