forked from extern/nushell
34033afce4
Also simplify commands and reduce papercuts
73 lines
2.2 KiB
Rust
73 lines
2.2 KiB
Rust
use crate::object::{Dictionary, Primitive, Value};
|
|
use crate::prelude::*;
|
|
|
|
fn from_node_to_value<'a, 'd>(n: &roxmltree::Node<'a, 'd>) -> Value {
|
|
if n.is_element() {
|
|
let name = n.tag_name().name().trim().to_string();
|
|
|
|
let mut children_values = vec![];
|
|
for c in n.children() {
|
|
children_values.push(from_node_to_value(&c));
|
|
}
|
|
|
|
let children_values: Vec<Value> = children_values
|
|
.into_iter()
|
|
.filter(|x| match x {
|
|
Value::Primitive(Primitive::String(f)) => {
|
|
if f.trim() == "" {
|
|
false
|
|
} else {
|
|
true
|
|
}
|
|
}
|
|
_ => true,
|
|
})
|
|
.collect();
|
|
|
|
let mut collected = Dictionary::default();
|
|
collected.add(name.clone(), Value::List(children_values));
|
|
|
|
Value::Object(collected)
|
|
} else if n.is_comment() {
|
|
Value::string("<comment>")
|
|
} else if n.is_pi() {
|
|
Value::string("<processing_instruction>")
|
|
} else if n.is_text() {
|
|
Value::string(n.text().unwrap())
|
|
} else {
|
|
Value::string("<unknown>")
|
|
}
|
|
}
|
|
|
|
fn from_document_to_value(d: &roxmltree::Document) -> Value {
|
|
from_node_to_value(&d.root_element())
|
|
}
|
|
|
|
pub fn from_xml_string_to_value(s: String) -> Result<Value, Box<dyn std::error::Error>> {
|
|
let parsed = roxmltree::Document::parse(&s)?;
|
|
Ok(from_document_to_value(&parsed))
|
|
}
|
|
|
|
pub fn from_xml(args: CommandArgs) -> Result<OutputStream, ShellError> {
|
|
let out = args.input;
|
|
let span = args.name_span;
|
|
Ok(out
|
|
.values
|
|
.map(move |a| match a {
|
|
Value::Primitive(Primitive::String(s)) => match from_xml_string_to_value(s) {
|
|
Ok(x) => ReturnSuccess::value(x),
|
|
Err(_) => Err(ShellError::maybe_labeled_error(
|
|
"Could not parse as XML",
|
|
"piped data failed XML parse",
|
|
span,
|
|
)),
|
|
},
|
|
_ => Err(ShellError::maybe_labeled_error(
|
|
"Expected string values from pipeline",
|
|
"expects strings from pipeline",
|
|
span,
|
|
)),
|
|
})
|
|
.to_output_stream())
|
|
}
|