mirror of
https://github.com/nushell/nushell.git
synced 2025-06-30 14:40:06 +02:00
enable cargo build --features=extra
to build plugins (#448)
This commit is contained in:
94
crates/nu_plugin_example/src/example.rs
Normal file
94
crates/nu_plugin_example/src/example.rs
Normal file
@ -0,0 +1,94 @@
|
||||
use nu_plugin::{EvaluatedCall, LabeledError};
|
||||
use nu_protocol::Value;
|
||||
pub struct Example;
|
||||
|
||||
impl Example {
|
||||
fn print_values(
|
||||
&self,
|
||||
index: u32,
|
||||
call: &EvaluatedCall,
|
||||
input: &Value,
|
||||
) -> Result<(), LabeledError> {
|
||||
// Note. When debugging your plugin, you may want to print something to the console
|
||||
// Use the eprintln macro to print your messages. Trying to print to stdout will
|
||||
// cause a decoding error for your message
|
||||
eprintln!("Calling test {} signature", index);
|
||||
eprintln!("value received {:?}", input);
|
||||
|
||||
// To extract the arguments from the Call object you can use the functions req, has_flag,
|
||||
// opt, rest, and get_flag
|
||||
//
|
||||
// Note that plugin calls only accept simple arguments, this means that you can
|
||||
// pass to the plug in Int and String. This should be improved when the plugin has
|
||||
// the ability to call back to NuShell to extract more information
|
||||
// Keep this in mind when designing your plugin signatures
|
||||
let a: i64 = call.req(0)?;
|
||||
let b: String = call.req(1)?;
|
||||
let flag = call.has_flag("flag");
|
||||
let opt: Option<i64> = call.opt(2)?;
|
||||
let named: Option<String> = call.get_flag("named")?;
|
||||
let rest: Vec<String> = call.rest(3)?;
|
||||
|
||||
eprintln!("Required values");
|
||||
eprintln!("a: {:}", a);
|
||||
eprintln!("b: {:}", b);
|
||||
eprintln!("flag: {:}", flag);
|
||||
eprintln!("rest: {:?}", rest);
|
||||
|
||||
match opt {
|
||||
Some(v) => eprintln!("Found optional value opt: {:}", v),
|
||||
None => eprintln!("No optional value found"),
|
||||
}
|
||||
|
||||
match named {
|
||||
Some(v) => eprintln!("Named value: {:?}", v),
|
||||
None => eprintln!("No named value found"),
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn test1(&self, call: &EvaluatedCall, input: &Value) -> Result<Value, LabeledError> {
|
||||
self.print_values(1, call, input)?;
|
||||
|
||||
Ok(Value::Nothing { span: call.head })
|
||||
}
|
||||
|
||||
pub fn test2(&self, call: &EvaluatedCall, input: &Value) -> Result<Value, LabeledError> {
|
||||
self.print_values(2, call, input)?;
|
||||
|
||||
let cols = vec!["one".to_string(), "two".to_string(), "three".to_string()];
|
||||
|
||||
let vals = (0..10i64)
|
||||
.map(|i| {
|
||||
let vals = (0..3)
|
||||
.map(|v| Value::Int {
|
||||
val: v * i,
|
||||
span: call.head,
|
||||
})
|
||||
.collect::<Vec<Value>>();
|
||||
|
||||
Value::Record {
|
||||
cols: cols.clone(),
|
||||
vals,
|
||||
span: call.head,
|
||||
}
|
||||
})
|
||||
.collect::<Vec<Value>>();
|
||||
|
||||
Ok(Value::List {
|
||||
vals,
|
||||
span: call.head,
|
||||
})
|
||||
}
|
||||
|
||||
pub fn test3(&self, call: &EvaluatedCall, input: &Value) -> Result<Value, LabeledError> {
|
||||
self.print_values(3, call, input)?;
|
||||
|
||||
Err(LabeledError {
|
||||
label: "ERROR from plugin".into(),
|
||||
msg: "error message pointing to call head span".into(),
|
||||
span: Some(call.head),
|
||||
})
|
||||
}
|
||||
}
|
4
crates/nu_plugin_example/src/lib.rs
Normal file
4
crates/nu_plugin_example/src/lib.rs
Normal file
@ -0,0 +1,4 @@
|
||||
mod example;
|
||||
mod nu;
|
||||
|
||||
pub use example::Example;
|
@ -1,148 +1,6 @@
|
||||
use nu_plugin::{serve_plugin, EvaluatedCall, LabeledError, Plugin};
|
||||
use nu_protocol::{Category, Signature, SyntaxShape, Value};
|
||||
use nu_plugin::serve_plugin;
|
||||
use nu_plugin_example::Example;
|
||||
|
||||
fn main() {
|
||||
serve_plugin(&mut Example {})
|
||||
}
|
||||
|
||||
struct Example {}
|
||||
|
||||
impl Plugin for Example {
|
||||
fn signature(&self) -> Vec<Signature> {
|
||||
// It is possible to declare multiple signature in a plugin
|
||||
// Each signature will be converted to a command declaration once the
|
||||
// plugin is registered to nushell
|
||||
vec![
|
||||
Signature::build("test-1")
|
||||
.desc("Signature test 1 for plugin. Returns Value::Nothing")
|
||||
.required("a", SyntaxShape::Int, "required integer value")
|
||||
.required("b", SyntaxShape::String, "required string value")
|
||||
.switch("flag", "a flag for the signature", Some('f'))
|
||||
.optional("opt", SyntaxShape::Int, "Optional number")
|
||||
.named("named", SyntaxShape::String, "named string", Some('n'))
|
||||
.rest("rest", SyntaxShape::String, "rest value string")
|
||||
.category(Category::Experimental),
|
||||
Signature::build("test-2")
|
||||
.desc("Signature test 2 for plugin. Returns list of records")
|
||||
.required("a", SyntaxShape::Int, "required integer value")
|
||||
.required("b", SyntaxShape::String, "required string value")
|
||||
.switch("flag", "a flag for the signature", Some('f'))
|
||||
.optional("opt", SyntaxShape::Int, "Optional number")
|
||||
.named("named", SyntaxShape::String, "named string", Some('n'))
|
||||
.rest("rest", SyntaxShape::String, "rest value string")
|
||||
.category(Category::Experimental),
|
||||
Signature::build("test-3")
|
||||
.desc("Signature test 3 for plugin. Returns labeled error")
|
||||
.required("a", SyntaxShape::Int, "required integer value")
|
||||
.required("b", SyntaxShape::String, "required string value")
|
||||
.switch("flag", "a flag for the signature", Some('f'))
|
||||
.optional("opt", SyntaxShape::Int, "Optional number")
|
||||
.named("named", SyntaxShape::String, "named string", Some('n'))
|
||||
.rest("rest", SyntaxShape::String, "rest value string")
|
||||
.category(Category::Experimental),
|
||||
]
|
||||
}
|
||||
|
||||
fn run(
|
||||
&mut self,
|
||||
name: &str,
|
||||
call: &EvaluatedCall,
|
||||
input: &Value,
|
||||
) -> Result<Value, LabeledError> {
|
||||
// You can use the name to identify what plugin signature was called
|
||||
match name {
|
||||
"test-1" => test1(call, input),
|
||||
"test-2" => test2(call, input),
|
||||
"test-3" => test3(call, input),
|
||||
_ => Err(LabeledError {
|
||||
label: "Plugin call with wrong name signature".into(),
|
||||
msg: "using the wrong signature".into(),
|
||||
span: Some(call.head),
|
||||
}),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn print_values(index: u32, call: &EvaluatedCall, input: &Value) -> Result<(), LabeledError> {
|
||||
// Note. When debugging your plugin, you may want to print something to the console
|
||||
// Use the eprintln macro to print your messages. Trying to print to stdout will
|
||||
// cause a decoding error for your message
|
||||
eprintln!("Calling test {} signature", index);
|
||||
eprintln!("value received {:?}", input);
|
||||
|
||||
// To extract the arguments from the Call object you can use the functions req, has_flag,
|
||||
// opt, rest, and get_flag
|
||||
//
|
||||
// Note that plugin calls only accept simple arguments, this means that you can
|
||||
// pass to the plug in Int and String. This should be improved when the plugin has
|
||||
// the ability to call back to NuShell to extract more information
|
||||
// Keep this in mind when designing your plugin signatures
|
||||
let a: i64 = call.req(0)?;
|
||||
let b: String = call.req(1)?;
|
||||
let flag = call.has_flag("flag");
|
||||
let opt: Option<i64> = call.opt(2)?;
|
||||
let named: Option<String> = call.get_flag("named")?;
|
||||
let rest: Vec<String> = call.rest(3)?;
|
||||
|
||||
eprintln!("Required values");
|
||||
eprintln!("a: {:}", a);
|
||||
eprintln!("b: {:}", b);
|
||||
eprintln!("flag: {:}", flag);
|
||||
eprintln!("rest: {:?}", rest);
|
||||
|
||||
match opt {
|
||||
Some(v) => eprintln!("Found optional value opt: {:}", v),
|
||||
None => eprintln!("No optional value found"),
|
||||
}
|
||||
|
||||
match named {
|
||||
Some(v) => eprintln!("Named value: {:?}", v),
|
||||
None => eprintln!("No named value found"),
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn test1(call: &EvaluatedCall, input: &Value) -> Result<Value, LabeledError> {
|
||||
print_values(1, call, input)?;
|
||||
|
||||
Ok(Value::Nothing { span: call.head })
|
||||
}
|
||||
|
||||
fn test2(call: &EvaluatedCall, input: &Value) -> Result<Value, LabeledError> {
|
||||
print_values(2, call, input)?;
|
||||
|
||||
let cols = vec!["one".to_string(), "two".to_string(), "three".to_string()];
|
||||
|
||||
let vals = (0..10i64)
|
||||
.map(|i| {
|
||||
let vals = (0..3)
|
||||
.map(|v| Value::Int {
|
||||
val: v * i,
|
||||
span: call.head,
|
||||
})
|
||||
.collect::<Vec<Value>>();
|
||||
|
||||
Value::Record {
|
||||
cols: cols.clone(),
|
||||
vals,
|
||||
span: call.head,
|
||||
}
|
||||
})
|
||||
.collect::<Vec<Value>>();
|
||||
|
||||
Ok(Value::List {
|
||||
vals,
|
||||
span: call.head,
|
||||
})
|
||||
}
|
||||
|
||||
fn test3(call: &EvaluatedCall, input: &Value) -> Result<Value, LabeledError> {
|
||||
print_values(3, call, input)?;
|
||||
|
||||
Err(LabeledError {
|
||||
label: "ERROR from plugin".into(),
|
||||
msg: "error message pointing to call head span".into(),
|
||||
span: Some(call.head),
|
||||
})
|
||||
}
|
||||
|
59
crates/nu_plugin_example/src/nu/mod.rs
Normal file
59
crates/nu_plugin_example/src/nu/mod.rs
Normal file
@ -0,0 +1,59 @@
|
||||
use crate::Example;
|
||||
use nu_plugin::{EvaluatedCall, LabeledError, Plugin};
|
||||
use nu_protocol::{Category, Signature, SyntaxShape, Value};
|
||||
|
||||
impl Plugin for Example {
|
||||
fn signature(&self) -> Vec<Signature> {
|
||||
// It is possible to declare multiple signature in a plugin
|
||||
// Each signature will be converted to a command declaration once the
|
||||
// plugin is registered to nushell
|
||||
vec![
|
||||
Signature::build("test-1")
|
||||
.desc("Signature test 1 for plugin. Returns Value::Nothing")
|
||||
.required("a", SyntaxShape::Int, "required integer value")
|
||||
.required("b", SyntaxShape::String, "required string value")
|
||||
.switch("flag", "a flag for the signature", Some('f'))
|
||||
.optional("opt", SyntaxShape::Int, "Optional number")
|
||||
.named("named", SyntaxShape::String, "named string", Some('n'))
|
||||
.rest("rest", SyntaxShape::String, "rest value string")
|
||||
.category(Category::Experimental),
|
||||
Signature::build("test-2")
|
||||
.desc("Signature test 2 for plugin. Returns list of records")
|
||||
.required("a", SyntaxShape::Int, "required integer value")
|
||||
.required("b", SyntaxShape::String, "required string value")
|
||||
.switch("flag", "a flag for the signature", Some('f'))
|
||||
.optional("opt", SyntaxShape::Int, "Optional number")
|
||||
.named("named", SyntaxShape::String, "named string", Some('n'))
|
||||
.rest("rest", SyntaxShape::String, "rest value string")
|
||||
.category(Category::Experimental),
|
||||
Signature::build("test-3")
|
||||
.desc("Signature test 3 for plugin. Returns labeled error")
|
||||
.required("a", SyntaxShape::Int, "required integer value")
|
||||
.required("b", SyntaxShape::String, "required string value")
|
||||
.switch("flag", "a flag for the signature", Some('f'))
|
||||
.optional("opt", SyntaxShape::Int, "Optional number")
|
||||
.named("named", SyntaxShape::String, "named string", Some('n'))
|
||||
.rest("rest", SyntaxShape::String, "rest value string")
|
||||
.category(Category::Experimental),
|
||||
]
|
||||
}
|
||||
|
||||
fn run(
|
||||
&mut self,
|
||||
name: &str,
|
||||
call: &EvaluatedCall,
|
||||
input: &Value,
|
||||
) -> Result<Value, LabeledError> {
|
||||
// You can use the name to identify what plugin signature was called
|
||||
match name {
|
||||
"test-1" => self.test1(call, input),
|
||||
"test-2" => self.test2(call, input),
|
||||
"test-3" => self.test3(call, input),
|
||||
_ => Err(LabeledError {
|
||||
label: "Plugin call with wrong name signature".into(),
|
||||
msg: "using the wrong signature".into(),
|
||||
span: Some(call.head),
|
||||
}),
|
||||
}
|
||||
}
|
||||
}
|
@ -1,12 +1,13 @@
|
||||
use crate::GStat;
|
||||
use nu_plugin::{EvaluatedCall, LabeledError, Plugin};
|
||||
use nu_protocol::{Signature, Span, Spanned, SyntaxShape, Value};
|
||||
use nu_protocol::{Category, Signature, Span, Spanned, SyntaxShape, Value};
|
||||
|
||||
impl Plugin for GStat {
|
||||
fn signature(&self) -> Vec<Signature> {
|
||||
vec![Signature::build("gstat")
|
||||
.desc("Get the git status of a repo")
|
||||
.optional("path", SyntaxShape::String, "path to repo")]
|
||||
.optional("path", SyntaxShape::String, "path to repo")
|
||||
.category(Category::Custom("Prompt".to_string()))]
|
||||
}
|
||||
|
||||
fn run(
|
||||
|
Reference in New Issue
Block a user