From b82ff321fa663806800c3b7e856863744af51f0a Mon Sep 17 00:00:00 2001 From: marblenix Date: Sat, 28 Dec 2019 21:20:36 -0600 Subject: [PATCH] feat(git_status): Add a stash count segment (#598) --- docs/config/README.md | 1 + src/configs/git_status.rs | 2 ++ src/modules/git_status.rs | 47 ++++++++++++++++++++----------- tests/testsuite/git_status.rs | 53 +++++++++++++++++++++++++++++------ 4 files changed, 77 insertions(+), 26 deletions(-) diff --git a/docs/config/README.md b/docs/config/README.md index 8e83fe90..265077c2 100644 --- a/docs/config/README.md +++ b/docs/config/README.md @@ -533,6 +533,7 @@ current directory. | `untracked` | `"?"` | There are untracked files in the working directory. | | `untracked_count` | [link](#git-status-counts) | Show and style the number of untracked files. | | `stashed` | `"$"` | A stash exists for the local repository. | +| `stashed_count` | [link](#git-status-counts) | Show and style the number of stashes. | | `modified` | `"!"` | There are file modifications in the working directory. | | `modified_count` | [link](#git-status-counts) | Show and style the number of modified files. | | `staged` | `"+"` | A new file has been added to the staging area. | diff --git a/src/configs/git_status.rs b/src/configs/git_status.rs index 028274df..9afdcba7 100644 --- a/src/configs/git_status.rs +++ b/src/configs/git_status.rs @@ -6,6 +6,7 @@ use starship_module_config_derive::ModuleConfig; #[derive(Clone, ModuleConfig)] pub struct GitStatusConfig<'a> { pub stashed: SegmentConfig<'a>, + pub stashed_count: CountConfig, pub ahead: SegmentConfig<'a>, pub behind: SegmentConfig<'a>, pub diverged: SegmentConfig<'a>, @@ -32,6 +33,7 @@ impl<'a> RootModuleConfig<'a> for GitStatusConfig<'a> { fn new() -> Self { GitStatusConfig { stashed: SegmentConfig::new("$"), + stashed_count: CountConfig::default(), ahead: SegmentConfig::new("⇡"), behind: SegmentConfig::new("⇣"), diverged: SegmentConfig::new("⇕"), diff --git a/src/modules/git_status.rs b/src/modules/git_status.rs index 0d126799..4a558faf 100644 --- a/src/modules/git_status.rs +++ b/src/modules/git_status.rs @@ -4,6 +4,7 @@ use super::{Context, Module, RootModuleConfig}; use crate::config::SegmentConfig; use crate::configs::git_status::{CountConfig, GitStatusConfig}; +use std::borrow::BorrowMut; /// Creates a module with the Git branch in the current directory /// @@ -23,7 +24,7 @@ pub fn module<'a>(context: &'a Context) -> Option> { let repo = context.get_repo().ok()?; let branch_name = repo.branch.as_ref()?; let repo_root = repo.root.as_ref()?; - let repository = Repository::open(repo_root).ok()?; + let mut repository = Repository::open(repo_root).ok()?; let mut module = context.new_module("git_status"); let config: GitStatusConfig = GitStatusConfig::try_load(module.config); @@ -38,6 +39,9 @@ pub fn module<'a>(context: &'a Context) -> Option> { .set_style(config.style); module.set_style(config.style); + let repo_status = get_repo_status(repository.borrow_mut()); + log::debug!("Repo status: {:?}", repo_status); + let ahead_behind = get_ahead_behind(&repository, branch_name); if ahead_behind == Ok((0, 0)) { log::trace!("No ahead/behind found"); @@ -45,16 +49,6 @@ pub fn module<'a>(context: &'a Context) -> Option> { log::debug!("Repo ahead/behind: {:?}", ahead_behind); } - let stash_object = repository.revparse_single("refs/stash"); - if stash_object.is_ok() { - log::debug!("Stash object: {:?}", stash_object); - } else { - log::trace!("No stash object found"); - } - - let repo_status = get_repo_status(&repository); - log::debug!("Repo status: {:?}", repo_status); - // Add the conflicted segment if let Ok(repo_status) = repo_status { create_segment_with_count( @@ -113,8 +107,14 @@ pub fn module<'a>(context: &'a Context) -> Option> { } // Add the stashed segment - if stash_object.is_ok() { - module.create_segment("stashed", &config.stashed); + if let Ok(repo_status) = repo_status { + create_segment_with_count( + &mut module, + "stashed", + repo_status.stashed, + &config.stashed, + config.stashed_count, + ); } // Add all remaining status segments @@ -187,16 +187,18 @@ fn create_segment_with_count<'a>( } /// Gets the number of files in various git states (staged, modified, deleted, etc...) -fn get_repo_status(repository: &Repository) -> Result { +fn get_repo_status(repository: &mut Repository) -> Result { let mut status_options = git2::StatusOptions::new(); match repository.config()?.get_entry("status.showUntrackedFiles") { Ok(entry) => status_options.include_untracked(entry.value() != Some("no")), _ => status_options.include_untracked(true), }; - status_options.renames_from_rewrites(true); - status_options.renames_head_to_index(true); - status_options.renames_index_to_workdir(true); + status_options + .renames_from_rewrites(true) + .renames_head_to_index(true) + .renames_index_to_workdir(true) + .include_unmodified(true); let statuses: Vec = repository .statuses(Some(&mut status_options))? @@ -215,6 +217,7 @@ fn get_repo_status(repository: &Repository) -> Result { modified: statuses.iter().filter(|s| is_modified(**s)).count(), staged: statuses.iter().filter(|s| is_staged(**s)).count(), untracked: statuses.iter().filter(|s| is_untracked(**s)).count(), + stashed: stashed_count(repository)?, }; Ok(repo_status) @@ -244,6 +247,15 @@ fn is_untracked(status: Status) -> bool { status.is_wt_new() } +fn stashed_count(repository: &mut Repository) -> Result { + let mut count = 0; + repository.stash_foreach(|_, _, _| { + count += 1; + true + })?; + Result::Ok(count) +} + /// Compares the current branch with the branch it is tracking to determine how /// far ahead or behind it is in relation fn get_ahead_behind( @@ -268,4 +280,5 @@ struct RepoStatus { modified: usize, staged: usize, untracked: usize, + stashed: usize, } diff --git a/tests/testsuite/git_status.rs b/tests/testsuite/git_status.rs index dc32c988..d1ecf085 100644 --- a/tests/testsuite/git_status.rs +++ b/tests/testsuite/git_status.rs @@ -264,15 +264,7 @@ fn shows_stashed() -> io::Result<()> { let repo_dir = common::create_fixture_repo()?; barrier(); - File::create(repo_dir.join("readme.md"))?.sync_all()?; - - barrier(); - - Command::new("git") - .args(&["stash", "--all"]) - .current_dir(repo_dir.as_path()) - .output()?; - barrier(); + create_stash(&repo_dir)?; Command::new("git") .args(&["reset", "--hard", "HEAD"]) @@ -292,6 +284,36 @@ fn shows_stashed() -> io::Result<()> { Ok(()) } +#[test] +fn shows_stashed_with_count() -> io::Result<()> { + let repo_dir = common::create_fixture_repo()?; + barrier(); + + create_stash(&repo_dir)?; + barrier(); + + Command::new("git") + .args(&["reset", "--hard", "HEAD"]) + .current_dir(repo_dir.as_path()) + .output()?; + barrier(); + + let output = common::render_module("git_status") + .use_config(toml::toml! { + [git_status] + stashed_count.enabled = true + }) + .arg("--path") + .arg(repo_dir) + .output()?; + let actual = String::from_utf8(output.stdout).unwrap(); + let expected = Color::Red.bold().paint(format!("[{}] ", "$1")).to_string(); + + assert_eq!(expected, actual); + + Ok(()) +} + #[test] #[ignore] fn shows_modified() -> io::Result<()> { @@ -580,6 +602,19 @@ fn create_conflict(repo_dir: &PathBuf) -> io::Result<()> { Ok(()) } +fn create_stash(repo_dir: &PathBuf) -> io::Result<()> { + File::create(repo_dir.join("readme.md"))?.sync_all()?; + barrier(); + + Command::new("git") + .args(&["stash", "--all"]) + .current_dir(repo_dir.as_path()) + .output()?; + barrier(); + + Ok(()) +} + fn create_untracked(repo_dir: &PathBuf) -> io::Result<()> { File::create(repo_dir.join("license"))?.sync_all()?;