mirror of
https://github.com/Llewellynvdm/starship.git
synced 2024-11-28 07:46:28 +00:00
d06ba072a8
* feat: allow printing default and computed config * fix custom modules * actually fix custom modules
267 lines
9.6 KiB
Rust
267 lines
9.6 KiB
Rust
use clap::crate_authors;
|
|
use std::io;
|
|
use std::time::SystemTime;
|
|
|
|
use clap::{App, AppSettings, Arg, Shell, SubCommand};
|
|
use rand::distributions::Alphanumeric;
|
|
use rand::Rng;
|
|
use starship::module::ALL_MODULES;
|
|
use starship::*;
|
|
|
|
fn main() {
|
|
// Configure the current terminal on windows to support ANSI escape sequences.
|
|
#[cfg(windows)]
|
|
let _ = ansi_term::enable_ansi_support();
|
|
logger::init();
|
|
|
|
let status_code_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);
|
|
|
|
let path_arg = Arg::with_name("path")
|
|
.short("p")
|
|
.long("path")
|
|
.value_name("PATH")
|
|
.help("The path that the prompt should render for.")
|
|
.takes_value(true);
|
|
|
|
let logical_path_arg = Arg::with_name("logical_path")
|
|
.short("P")
|
|
.long("logical-path")
|
|
.value_name("LOGICAL_PATH")
|
|
.help(concat!(
|
|
"The logical path that the prompt should render for. ",
|
|
"This path should be a virtual/logical representation of the PATH argument."
|
|
))
|
|
.takes_value(true);
|
|
|
|
let shell_arg = Arg::with_name("shell")
|
|
.value_name("SHELL")
|
|
.help(
|
|
"The name of the currently running shell\nCurrently supported options: bash, zsh, fish, powershell, ion, elvish, tcsh",
|
|
)
|
|
.required(true);
|
|
|
|
let cmd_duration_arg = Arg::with_name("cmd_duration")
|
|
.short("d")
|
|
.long("cmd-duration")
|
|
.value_name("CMD_DURATION")
|
|
.help("The execution duration of the last command, in milliseconds")
|
|
.takes_value(true);
|
|
|
|
let keymap_arg = Arg::with_name("keymap")
|
|
.short("k")
|
|
.long("keymap")
|
|
.value_name("KEYMAP")
|
|
// fish/zsh only
|
|
.help("The keymap of fish/zsh")
|
|
.takes_value(true);
|
|
|
|
let jobs_arg = Arg::with_name("jobs")
|
|
.short("j")
|
|
.long("jobs")
|
|
.value_name("JOBS")
|
|
.help("The number of currently running jobs")
|
|
.takes_value(true);
|
|
|
|
let init_scripts_arg = Arg::with_name("print_full_init")
|
|
.long("print-full-init")
|
|
.help("Print the main initialization script (as opposed to the init stub)");
|
|
|
|
let long_version = crate::shadow::clap_version();
|
|
let mut app = App::new("starship")
|
|
.about("The cross-shell prompt for astronauts. ☄🌌️")
|
|
// pull the version number from Cargo.toml
|
|
.version(shadow::PKG_VERSION)
|
|
.long_version(long_version.as_str())
|
|
// pull the authors from Cargo.toml
|
|
.author(crate_authors!())
|
|
.after_help("https://github.com/starship/starship")
|
|
.setting(AppSettings::SubcommandRequiredElseHelp)
|
|
.subcommand(
|
|
SubCommand::with_name("init")
|
|
.about("Prints the shell function used to execute starship")
|
|
.arg(&shell_arg)
|
|
.arg(&init_scripts_arg),
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("prompt")
|
|
.about("Prints the full starship prompt")
|
|
.arg(&status_code_arg)
|
|
.arg(&path_arg)
|
|
.arg(&logical_path_arg)
|
|
.arg(&cmd_duration_arg)
|
|
.arg(&keymap_arg)
|
|
.arg(&jobs_arg),
|
|
)
|
|
.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)
|
|
.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)
|
|
.arg(&logical_path_arg)
|
|
.arg(&cmd_duration_arg)
|
|
.arg(&keymap_arg)
|
|
.arg(&jobs_arg),
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("config")
|
|
.alias("configure")
|
|
.about("Edit the starship configuration")
|
|
.arg(
|
|
Arg::with_name("name")
|
|
.help("Configuration key to edit")
|
|
.required(false)
|
|
.requires("value"),
|
|
)
|
|
.arg(Arg::with_name("value").help("Value to place into that key")),
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("print-config")
|
|
.about("Prints the computed starship configuration")
|
|
.arg(
|
|
Arg::with_name("default")
|
|
.short("d")
|
|
.long("default")
|
|
.help("Print the default instead of the computed config")
|
|
.takes_value(false),
|
|
),
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("toggle")
|
|
.about("Toggle a given starship module")
|
|
.arg(
|
|
Arg::with_name("name")
|
|
.help("The name of the module to be toggled")
|
|
.required(true),
|
|
)
|
|
.arg(
|
|
Arg::with_name("key")
|
|
.help("The key of the config to be toggled")
|
|
.required(false)
|
|
.required_unless("name"),
|
|
),
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("bug-report").about(
|
|
"Create a pre-populated GitHub issue with information about your configuration",
|
|
),
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("time")
|
|
.about("Prints time in milliseconds")
|
|
.settings(&[AppSettings::Hidden]),
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("explain").about("Explains the currently showing modules"),
|
|
)
|
|
.subcommand(SubCommand::with_name("timings").about("Prints timings of all active modules"))
|
|
.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"),
|
|
),
|
|
)
|
|
.subcommand(SubCommand::with_name("session").about("Generate random session key"));
|
|
|
|
let matches = app.clone().get_matches();
|
|
|
|
match matches.subcommand() {
|
|
("init", Some(sub_m)) => {
|
|
let shell_name = sub_m.value_of("shell").expect("Shell name missing.");
|
|
if sub_m.is_present("print_full_init") {
|
|
init::init_main(shell_name).expect("can't init_main");
|
|
} else {
|
|
init::init_stub(shell_name).expect("can't init_stub");
|
|
}
|
|
}
|
|
("prompt", Some(sub_m)) => print::prompt(sub_m.clone()),
|
|
("module", Some(sub_m)) => {
|
|
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());
|
|
}
|
|
}
|
|
("config", Some(sub_m)) => {
|
|
if let Some(name) = sub_m.value_of("name") {
|
|
if let Some(value) = sub_m.value_of("value") {
|
|
configure::update_configuration(name, value)
|
|
}
|
|
} else {
|
|
configure::edit_configuration()
|
|
}
|
|
}
|
|
("print-config", Some(sub_m)) => {
|
|
let print_default = sub_m.is_present("default");
|
|
configure::print_configuration(print_default)
|
|
}
|
|
("toggle", Some(sub_m)) => {
|
|
if let Some(name) = sub_m.value_of("name") {
|
|
if let Some(value) = sub_m.value_of("key") {
|
|
configure::toggle_configuration(name, value)
|
|
} else {
|
|
configure::toggle_configuration(name, "disabled")
|
|
}
|
|
}
|
|
}
|
|
("bug-report", Some(_)) => bug_report::create(),
|
|
("time", _) => {
|
|
match SystemTime::now()
|
|
.duration_since(SystemTime::UNIX_EPOCH)
|
|
.ok()
|
|
{
|
|
Some(time) => println!("{}", time.as_millis()),
|
|
None => println!("{}", -1),
|
|
}
|
|
}
|
|
("explain", Some(sub_m)) => print::explain(sub_m.clone()),
|
|
("timings", Some(sub_m)) => print::timings(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());
|
|
}
|
|
("session", _) => println!(
|
|
"{}",
|
|
rand::thread_rng()
|
|
.sample_iter(&Alphanumeric)
|
|
.take(16)
|
|
.map(char::from)
|
|
.collect::<String>()
|
|
),
|
|
(command, _) => unreachable!("Invalid subcommand: {}", command),
|
|
}
|
|
}
|