57 lines
1.8 KiB
Rust
57 lines
1.8 KiB
Rust
use nu_errors::ShellError;
|
|
use nu_plugin::Plugin;
|
|
use nu_protocol::{
|
|
CallInfo, Primitive, ReturnSuccess, ReturnValue, Signature, SyntaxShape, UntaggedValue, Value,
|
|
};
|
|
use nu_source::TaggedItem;
|
|
|
|
use crate::{selector::begin_selector_query, Selector};
|
|
|
|
impl Plugin for Selector {
|
|
fn config(&mut self) -> Result<Signature, ShellError> {
|
|
Ok(Signature::build("selector")
|
|
.desc("execute selector query on html/web")
|
|
.required("query", SyntaxShape::String, "selector query")
|
|
.switch("as_html", "return the query output as html", Some('a'))
|
|
.filter())
|
|
}
|
|
|
|
fn begin_filter(&mut self, call_info: CallInfo) -> Result<Vec<ReturnValue>, ShellError> {
|
|
let tag = call_info.name_tag;
|
|
let query = call_info.args.nth(0).ok_or_else(|| {
|
|
ShellError::labeled_error(
|
|
"selector query not passed",
|
|
"selector query not passed",
|
|
&tag,
|
|
)
|
|
})?;
|
|
|
|
self.query = query.as_string()?;
|
|
self.tag = tag;
|
|
self.as_html = call_info.args.has("as_html");
|
|
|
|
Ok(vec![])
|
|
}
|
|
|
|
fn filter(&mut self, input: Value) -> Result<Vec<ReturnValue>, ShellError> {
|
|
match input {
|
|
Value {
|
|
value: UntaggedValue::Primitive(Primitive::String(s)),
|
|
..
|
|
} => Ok(
|
|
begin_selector_query(s, (*self.query).tagged(&self.tag), self.as_html)?
|
|
.into_iter()
|
|
.map(ReturnSuccess::value)
|
|
.collect(),
|
|
),
|
|
Value { tag, .. } => Err(ShellError::labeled_error_with_secondary(
|
|
"Expected text from pipeline",
|
|
"requires text input",
|
|
&self.tag,
|
|
"value originates from here",
|
|
tag,
|
|
)),
|
|
}
|
|
}
|
|
}
|