2019-07-03 19:37:09 +02:00
|
|
|
use indexmap::IndexMap;
|
|
|
|
use nu::{
|
2019-08-03 04:17:28 +02:00
|
|
|
serve_plugin, CallInfo, Plugin, Primitive, ReturnSuccess, ReturnValue, ShellError, Signature,
|
2019-08-09 06:51:21 +02:00
|
|
|
Tagged, Value,
|
2019-07-03 19:37:09 +02:00
|
|
|
};
|
|
|
|
|
2019-07-24 09:44:12 +02:00
|
|
|
struct Skip {
|
2019-07-03 19:37:09 +02:00
|
|
|
skip_amount: i64,
|
|
|
|
}
|
2019-07-24 09:44:12 +02:00
|
|
|
impl Skip {
|
|
|
|
fn new() -> Skip {
|
|
|
|
Skip { skip_amount: 0 }
|
2019-07-03 19:37:09 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-07-24 09:44:12 +02:00
|
|
|
impl Plugin for Skip {
|
2019-08-02 21:15:07 +02:00
|
|
|
fn config(&mut self) -> Result<Signature, ShellError> {
|
|
|
|
Ok(Signature {
|
2019-07-03 19:37:09 +02:00
|
|
|
name: "skip".to_string(),
|
2019-07-13 04:07:06 +02:00
|
|
|
positional: vec![],
|
2019-07-03 19:37:09 +02:00
|
|
|
is_filter: true,
|
|
|
|
named: IndexMap::new(),
|
|
|
|
rest_positional: true,
|
|
|
|
})
|
|
|
|
}
|
2019-07-27 09:45:00 +02:00
|
|
|
fn begin_filter(&mut self, call_info: CallInfo) -> Result<Vec<ReturnValue>, ShellError> {
|
2019-07-20 04:27:10 +02:00
|
|
|
if let Some(args) = call_info.args.positional {
|
2019-07-03 19:37:09 +02:00
|
|
|
for arg in args {
|
|
|
|
match arg {
|
2019-08-01 03:58:42 +02:00
|
|
|
Tagged {
|
2019-07-03 19:37:09 +02:00
|
|
|
item: Value::Primitive(Primitive::Int(i)),
|
|
|
|
..
|
|
|
|
} => {
|
|
|
|
self.skip_amount = i;
|
|
|
|
}
|
2019-07-20 04:27:10 +02:00
|
|
|
_ => {
|
|
|
|
return Err(ShellError::labeled_error(
|
|
|
|
"Unrecognized type in params",
|
|
|
|
"expected an integer",
|
2019-08-01 03:58:42 +02:00
|
|
|
arg.span(),
|
2019-07-20 04:27:10 +02:00
|
|
|
))
|
|
|
|
}
|
2019-07-03 19:37:09 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-07-27 09:45:00 +02:00
|
|
|
Ok(vec![])
|
2019-07-03 19:37:09 +02:00
|
|
|
}
|
|
|
|
|
2019-08-01 03:58:42 +02:00
|
|
|
fn filter(&mut self, input: Tagged<Value>) -> Result<Vec<ReturnValue>, ShellError> {
|
2019-07-03 19:37:09 +02:00
|
|
|
if self.skip_amount == 0 {
|
2019-07-13 04:07:06 +02:00
|
|
|
Ok(vec![ReturnSuccess::value(input)])
|
2019-07-03 19:37:09 +02:00
|
|
|
} else {
|
|
|
|
self.skip_amount -= 1;
|
|
|
|
Ok(vec![])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn main() {
|
2019-07-24 09:44:12 +02:00
|
|
|
serve_plugin(&mut Skip::new());
|
2019-07-03 19:37:09 +02:00
|
|
|
}
|