Add option to toggle verbose rust version (spaceship-prompt#521)

georgegg · Nov 21, 2018 · 6792870 · 6792870
1 parent 8b0e0d7
commit 6792870
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/docs/Options.md b/docs/Options.md
@@ -351,6 +351,7 @@ Rust section is shown only in directories that contain `Cargo.toml` or any other
 | `SPACESHIP_RUST_SUFFIX` | `$SPACESHIP_PROMPT_DEFAULT_SUFFIX` | Suffix after the Rust section |
 | `SPACESHIP_RUST_SYMBOL` | `𝗥·` | Character to be shown before Rust version |
 | `SPACESHIP_RUST_COLOR` | `red` | Color of Rust section |
+| `SPACESHIP_RUST_VERBOSE_VERSION` | `false` | Show what branch is being used, if any. (Beta, Nightly) |
 
 ### Haskell (`haskell`)
 

diff --git a/sections/rust.zsh b/sections/rust.zsh
@@ -13,6 +13,7 @@ SPACESHIP_RUST_PREFIX="${SPACESHIP_RUST_PREFIX="$SPACESHIP_PROMPT_DEFAULT_PREFIX
 SPACESHIP_RUST_SUFFIX="${SPACESHIP_RUST_SUFFIX="$SPACESHIP_PROMPT_DEFAULT_SUFFIX"}"
 SPACESHIP_RUST_SYMBOL="${SPACESHIP_RUST_SYMBOL="𝗥 "}"
 SPACESHIP_RUST_COLOR="${SPACESHIP_RUST_COLOR="red"}"
+SPACESHIP_RUST_VERBOSE_VERSION="${SPACESHIP_RUST_VERBOSE_VERSION=false}"
 
 # ------------------------------------------------------------------------------
 # Section
@@ -29,6 +30,10 @@ spaceship_rust() {
 
   local rust_version=$(rustc --version | cut -d' ' -f2)
 
+  if [[ $SPACESHIP_RUST_VERBOSE_VERSION == false ]]; then
+  	local rust_version=$(echo $rust_version | cut -d'-' -f1) # Cut off -suffixes from version. "v1.30.0-beta.11" or "v1.30.0-nightly"
+  fi
+
   spaceship::section \
     "$SPACESHIP_RUST_COLOR" \
     "$SPACESHIP_RUST_PREFIX" \