forked from extern/nushell
f5fad393d0
* Remove completion code from help/value shells * Tweak the `Completer` trait for nushell. Previously, this trait was built around rustyline's completion traits, and for `Shell` instances. Now it is built for individual completers inside of nushell that will complete a specific location based on a partial string. For example, for completing a partially typed command in the command position.
36 lines
1.0 KiB
Rust
36 lines
1.0 KiB
Rust
use crate::completion::{Completer, Context, Suggestion};
|
|
use crate::context;
|
|
|
|
pub struct FlagCompleter {
|
|
pub(crate) cmd: String,
|
|
}
|
|
|
|
impl Completer for FlagCompleter {
|
|
fn complete(&self, ctx: &Context<'_>, partial: &str) -> Vec<Suggestion> {
|
|
let context: &context::Context = ctx.as_ref();
|
|
|
|
if let Some(cmd) = context.registry.get_command(&self.cmd) {
|
|
let sig = cmd.signature();
|
|
let mut suggestions = Vec::new();
|
|
for (name, (named_type, _desc)) in sig.named.iter() {
|
|
suggestions.push(format!("--{}", name));
|
|
|
|
if let Some(c) = named_type.get_short() {
|
|
suggestions.push(format!("-{}", c));
|
|
}
|
|
}
|
|
|
|
suggestions
|
|
.into_iter()
|
|
.filter(|v| v.starts_with(partial))
|
|
.map(|v| Suggestion {
|
|
replacement: format!("{} ", v),
|
|
display: v,
|
|
})
|
|
.collect()
|
|
} else {
|
|
Vec::new()
|
|
}
|
|
}
|
|
}
|