diff options
| -rw-r--r-- | src/init.rs | 28 |
1 files changed, 23 insertions, 5 deletions
diff --git a/src/init.rs b/src/init.rs index 5dc854d9a..4a237ac88 100644 --- a/src/init.rs +++ b/src/init.rs @@ -1,25 +1,43 @@ +use std::ffi::OsStr; +use std::path::Path; + pub fn init(shell_name: &str) { log::debug!("Shell name: {}", shell_name); - let setup_script = match shell_name { - "bash" => { + + let shell_basename = Path::new(shell_name).file_stem().and_then(OsStr::to_str); + + let setup_script = match shell_basename { + Some("bash") => { let script = "PS1=\"$(starship prompt --status=$?)\""; Some(script) } - "zsh" => { + Some("zsh") => { let script = "PROMPT=\"$(starship prompt --status=$?)\""; Some(script) } - "fish" => { + Some("fish") => { let script = "function fish_prompt; starship prompt --status=$status; end"; Some(script) } + None => { + println!( + "Invalid shell name provided: {}\\n\ + If this issue persists, please open an \ + issue in the starship repo: \\n\ + https://github.com/starship/starship/issues/new\\n\"", + shell_name + ); + None + } _ => { + /* Calling unwrap() here is fine because the None case will have + already matched on the previous arm */ println!( "printf \"\\n{0} is not yet supported by starship.\\n\ For the time being, we support bash, zsh, and fish.\\n\ Please open an issue in the starship repo if you would like to \ see support for {0}:\\nhttps://github.com/starship/starship/issues/new\"\\n\\n", - shell_name + shell_basename.unwrap() ); None } |