andrei/starship
1
0
Fork 0
mirror of https://github.com/starship/starship.git synced 2026-06-23 02:05:51 +07:00
Code Issues Packages Projects Releases 10 Wiki Activity

fix: Change panicking on unknown module to error print and support module listing flag (#197)

Browse Source
This commit is contained in:
Saurav Sharma
2019-08-20 10:27:25 +05:45
committed by Matan Kushner
parent 0e82c19f37
commit 075a76897c
6 changed files with 76 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main.rs
+19 -3
View File
@@ -10,6 +10,7 @@ mod print;
mod segment;
mod utils;
use crate::module::ALL_MODULES;
use clap::{App, AppSettings, Arg, SubCommand};
fn main() {
@@ -91,7 +92,14 @@ fn main() {
.arg(
Arg::with_name("name")
.help("The name of the module to be printed")
.required(true),
.required(true)
.required_unless("list"),
)
.arg(
Arg::with_name("list")
.short("l")
.long("list")
.help("List out all supported modules"),
)
.arg(&status_code_arg)
.arg(&path_arg)
@@ -112,8 +120,16 @@ fn main() {
}
("prompt", Some(sub_m)) => print::prompt(sub_m.clone()),
("module", Some(sub_m)) => {
let module_name = sub_m.value_of("name").expect("Module name missing.");
print::module(module_name, sub_m.clone());
if sub_m.is_present("list") {
println!("Supported modules list");
println!("----------------------");
for modules in ALL_MODULES {
println!("{}", modules);
}
}
if let Some(module_name) = sub_m.value_of("name") {
print::module(module_name, sub_m.clone());
}
}
_ => {}
}