summaryrefslogtreecommitdiffstats
path: root/src/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs67
1 files changed, 54 insertions, 13 deletions
diff --git a/src/main.rs b/src/main.rs
index 1f69d4ed9..4707cd816 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,36 +1,77 @@
#[macro_use]
extern crate clap;
-extern crate ansi_term;
-extern crate battery;
-extern crate dirs;
-extern crate git2;
-extern crate pretty_env_logger;
-
mod context;
mod module;
mod modules;
mod print;
mod segment;
-use clap::{App, Arg};
+use clap::{App, Arg, SubCommand};
fn main() {
pretty_env_logger::init();
- let args = App::new("Starship")
+ let matches = App::new("Starship")
.about("The cross-shell prompt for astronauts. ✨🚀")
// pull the version number from Cargo.toml
.version(crate_version!())
// pull the authors from Cargo.toml
.author(crate_authors!())
.after_help("https://github.com/matchai/starship")
- .arg(
- Arg::with_name("status_code")
- .help("The status code of the previously run command")
- .required(true),
+ .subcommand(
+ SubCommand::with_name("prompt")
+ .about("Prints the full starship prompt")
+ .arg(
+ Arg::with_name("status_code")
+ .short("s")
+ .long("status")
+ .value_name("STATUS_CODE")
+ .help("The status code of the previously run command")
+ .takes_value(true),
+ )
+ .arg(
+ Arg::with_name("path")
+ .short("p")
+ .long("path")
+ .value_name("PATH")
+ .help("The path that the prompt should render for ($PWD by default)")
+ .takes_value(true),
+ ),
+ )
+ .subcommand(
+ SubCommand::with_name("module")
+ .about("Prints a specific prompt module")
+ .arg(
+ Arg::with_name("name")
+ .help("The name of the module to be printed")
+ .required(true),
+ )
+ .arg(
+ Arg::with_name("status_code")
+ .short("s")
+ .long("status")
+ .value_name("STATUS_CODE")
+ .help("The status code of the previously run command")
+ .takes_value(true),
+ )
+ .arg(
+ Arg::with_name("path")
+ .short("p")
+ .long("path")
+ .value_name("PATH")
+ .help("The path the prompt should render for ($PWD by default)")
+ .takes_value(true),
+ ),
)
.get_matches();
- print::prompt(args);
+ match matches.subcommand() {
+ ("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());
+ }
+ _ => {}
+ }
}