forked from extern/nushell
53 lines
1.6 KiB
Rust
53 lines
1.6 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")
|
||
|
.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;
|
||
|
|
||
|
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))?
|
||
|
.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,
|
||
|
)),
|
||
|
}
|
||
|
}
|
||
|
}
|