Skip to content

Commit

Permalink
feat(complete): Add dynamic-support shell enum
Browse files Browse the repository at this point in the history
  • Loading branch information
epage committed Jul 19, 2023
1 parent 278ae3e commit 830dd74
Show file tree
Hide file tree
Showing 3 changed files with 56 additions and 1 deletion.
4 changes: 3 additions & 1 deletion clap_complete/src/dynamic/mod.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
//! Complete commands within shells
pub mod bash;
mod completer;

pub mod bash;
pub mod shells;

pub use completer::*;
5 changes: 5 additions & 0 deletions clap_complete/src/dynamic/shells/mod.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
//! Shell support
mod shell;

pub use shell::*;
48 changes: 48 additions & 0 deletions clap_complete/src/dynamic/shells/shell.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
use std::fmt::Display;
use std::str::FromStr;

use clap::builder::PossibleValue;
use clap::ValueEnum;

/// Shell with auto-generated completion script available.
#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
#[non_exhaustive]
pub enum Shell {
/// Bourne Again SHell (bash)
Bash,
}

impl Display for Shell {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
self.to_possible_value()
.expect("no values are skipped")
.get_name()
.fmt(f)
}
}

impl FromStr for Shell {
type Err = String;

fn from_str(s: &str) -> Result<Self, Self::Err> {
for variant in Self::value_variants() {
if variant.to_possible_value().unwrap().matches(s, false) {
return Ok(*variant);
}
}
Err(format!("invalid variant: {s}"))
}
}

// Hand-rolled so it can work even when `derive` feature is disabled
impl ValueEnum for Shell {
fn value_variants<'a>() -> &'a [Self] {
&[Shell::Bash]
}

fn to_possible_value<'a>(&self) -> Option<PossibleValue> {
Some(match self {
Shell::Bash => PossibleValue::new("bash"),
})
}
}

0 comments on commit 830dd74

Please sign in to comment.