Llewellyn/starship
1
0
Fork 0
mirror of https://github.com/Llewellynvdm/starship.git synced 2025-02-09 15:38:27 +00:00
Code Issues Releases Activity

feat: add shell completion generation command (starship completions) (#881)

Browse Source
This commit is contained in:
Nathan West 2020-06-03 11:25:51 -04:00 committed by GitHub
parent a62712d33d
commit c04a0eb1d1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 28 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/main.rs
View File

@ -1,3 +1,4 @@
use std::io;
use std::time::SystemTime; use std::time::SystemTime;
#[macro_use] #[macro_use]
@ -19,7 +20,7 @@ mod segment;
mod utils; mod utils;
use crate::module::ALL_MODULES; use crate::module::ALL_MODULES;
use clap::{App, AppSettings, Arg, SubCommand}; use clap::{App, AppSettings, Arg, Shell, SubCommand};
fn main() { fn main() {
pretty_env_logger::init(); pretty_env_logger::init();
@ -71,7 +72,7 @@ fn main() {
.long("print-full-init") .long("print-full-init")
.help("Print the main initialization script (as opposed to the init stub)"); .help("Print the main initialization script (as opposed to the init stub)");
let matches = let mut app =
App::new("starship") App::new("starship")
.about("The cross-shell prompt for astronauts. ☄🌌️") .about("The cross-shell prompt for astronauts. ☄🌌️")
// pull the version number from Cargo.toml // pull the version number from Cargo.toml
@ -139,7 +140,21 @@ fn main() {
.subcommand( .subcommand(
SubCommand::with_name("explain").about("Explains the currently showing modules"), SubCommand::with_name("explain").about("Explains the currently showing modules"),
) )
.get_matches(); .subcommand(
SubCommand::with_name("completions")
.about("Generate starship shell completions for your shell to stdout")
.arg(
Arg::with_name("shell")
.takes_value(true)
.possible_values(&Shell::variants())
.help("the shell to generate completions for")
.value_name("SHELL")
.required(true)
.env("STARSHIP_SHELL"),
),
);
let matches = app.clone().get_matches();
match matches.subcommand() { match matches.subcommand() {
("init", Some(sub_m)) => { ("init", Some(sub_m)) => {
@ -183,6 +198,15 @@ fn main() {
} }
} }
("explain", Some(sub_m)) => print::explain(sub_m.clone()), ("explain", Some(sub_m)) => print::explain(sub_m.clone()),
_ => {} ("completions", Some(sub_m)) => {
let shell: Shell = sub_m
.value_of("shell")
.expect("Shell name missing.")
.parse()
.expect("Invalid shell");
app.gen_completions_to("starship", shell, &mut io::stdout().lock());
}
(command, _) => unreachable!("Invalid subcommand: {}", command),
} }
} }