forked from extern/nushell
Add support for defining known externals with their own custom completions (#4425)
* WIP for known externals * Now completions can work from scripts * Add support for definiing externs * finish cleaning up old proof-of-concept
This commit is contained in:
82
crates/nu-parser/src/known_external.rs
Normal file
82
crates/nu-parser/src/known_external.rs
Normal file
@ -0,0 +1,82 @@
|
||||
use nu_protocol::ast::Expr;
|
||||
use nu_protocol::engine::{EngineState, Stack, StateWorkingSet};
|
||||
use nu_protocol::PipelineData;
|
||||
use nu_protocol::{ast::Call, engine::Command, ShellError, Signature};
|
||||
|
||||
#[derive(Clone)]
|
||||
pub struct KnownExternal {
|
||||
pub name: String,
|
||||
pub signature: Box<Signature>,
|
||||
pub usage: String,
|
||||
}
|
||||
|
||||
impl Command for KnownExternal {
|
||||
fn name(&self) -> &str {
|
||||
&self.name
|
||||
}
|
||||
|
||||
fn signature(&self) -> Signature {
|
||||
*self.signature.clone()
|
||||
}
|
||||
|
||||
fn usage(&self) -> &str {
|
||||
&self.usage
|
||||
}
|
||||
|
||||
fn is_known_external(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
engine_state: &EngineState,
|
||||
stack: &mut Stack,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
// FIXME: This is a bit of a hack, and it'd be nice for the parser/AST to be able to handle the original
|
||||
// order of the parameters. Until then, we need to recover the original order.
|
||||
let call_span = call.span();
|
||||
let contents = engine_state.get_span_contents(&call_span);
|
||||
|
||||
let (lexed, _) = crate::lex(contents, call_span.start, &[], &[], true);
|
||||
|
||||
let spans: Vec<_> = lexed.into_iter().map(|x| x.span).collect();
|
||||
let mut working_set = StateWorkingSet::new(engine_state);
|
||||
let (external_call, _) = crate::parse_external_call(&mut working_set, &spans);
|
||||
|
||||
match external_call.expr {
|
||||
Expr::ExternalCall(head, args) => {
|
||||
let decl_id = engine_state
|
||||
.find_decl("run_external".as_bytes())
|
||||
.ok_or(ShellError::ExternalNotSupported(head.span))?;
|
||||
|
||||
let command = engine_state.get_decl(decl_id);
|
||||
|
||||
let mut call = Call::new(head.span);
|
||||
|
||||
call.positional.push(*head);
|
||||
|
||||
for arg in args {
|
||||
call.positional.push(arg.clone())
|
||||
}
|
||||
|
||||
// if last_expression {
|
||||
// call.named.push((
|
||||
// Spanned {
|
||||
// item: "last_expression".into(),
|
||||
// span: head.span,
|
||||
// },
|
||||
// None,
|
||||
// ))
|
||||
// }
|
||||
|
||||
command.run(engine_state, stack, &call, input)
|
||||
}
|
||||
x => {
|
||||
println!("{:?}", x);
|
||||
panic!("internal error: known external not actually external")
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user