mirror of
https://github.com/Llewellynvdm/starship.git
synced 2024-09-27 20:59:02 +00:00
9f70ffb7a7
A couple of optimizations are done in this PR. One, we now will check config ahead of time to see if a module is disabled before running any module code. Also, we won't try to discover a git repository unless the module requests access to it.
163 lines
5.3 KiB
Rust
163 lines
5.3 KiB
Rust
use ansi_term::Color;
|
|
use git2::RepositoryState;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
use super::{Context, Module};
|
|
|
|
/// Creates a module with the state of the git repository at the current directory
|
|
///
|
|
/// During a git operation it will show: REBASING, BISECTING, MERGING, etc.
|
|
/// If the progress information is available (e.g. rebasing 3/10), it will show that too.
|
|
pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
|
|
let mut module = context.new_module("git_state");
|
|
|
|
let repo = context.get_repo().ok()?;
|
|
let repo_root = repo.root.as_ref()?;
|
|
let repo_state = repo.state?;
|
|
let state_description = get_state_description(repo_state, repo_root);
|
|
|
|
if let StateDescription::Clean = state_description {
|
|
return None;
|
|
}
|
|
|
|
let module_style = module
|
|
.config_value_style("style")
|
|
.unwrap_or_else(|| Color::Yellow.bold());
|
|
module.set_style(module_style);
|
|
module.get_prefix().set_value("(");
|
|
module.get_suffix().set_value(") ");
|
|
|
|
let label = match state_description {
|
|
StateDescription::Label(label) => label,
|
|
StateDescription::LabelAndProgress(label, _) => label,
|
|
// Should only be possible if you've added a new variant to StateDescription
|
|
_ => panic!("Expected to have a label at this point in the control flow."),
|
|
};
|
|
|
|
module.new_segment(label.segment_name, label.message_default);
|
|
|
|
if let StateDescription::LabelAndProgress(_, progress) = state_description {
|
|
module.new_segment("progress_current", &format!(" {}", progress.current));
|
|
module.new_segment("progress_divider", "/");
|
|
module.new_segment("progress_total", &format!("{}", progress.total));
|
|
}
|
|
|
|
Some(module)
|
|
}
|
|
|
|
static MERGE_LABEL: StateLabel = StateLabel {
|
|
segment_name: "merge",
|
|
message_default: "MERGING",
|
|
};
|
|
|
|
static REVERT_LABEL: StateLabel = StateLabel {
|
|
segment_name: "revert",
|
|
message_default: "REVERTING",
|
|
};
|
|
|
|
static CHERRY_LABEL: StateLabel = StateLabel {
|
|
segment_name: "cherry_pick",
|
|
message_default: "CHERRY-PICKING",
|
|
};
|
|
|
|
static BISECT_LABEL: StateLabel = StateLabel {
|
|
segment_name: "bisect",
|
|
message_default: "BISECTING",
|
|
};
|
|
|
|
static AM_LABEL: StateLabel = StateLabel {
|
|
segment_name: "am",
|
|
message_default: "AM",
|
|
};
|
|
|
|
static REBASE_LABEL: StateLabel = StateLabel {
|
|
segment_name: "rebase",
|
|
message_default: "REBASING",
|
|
};
|
|
|
|
static AM_OR_REBASE_LABEL: StateLabel = StateLabel {
|
|
segment_name: "am_or_rebase",
|
|
message_default: "AM/REBASE",
|
|
};
|
|
|
|
/// Returns the state of the current repository
|
|
///
|
|
/// During a git operation it will show: REBASING, BISECTING, MERGING, etc.
|
|
fn get_state_description(state: RepositoryState, root: &PathBuf) -> StateDescription {
|
|
match state {
|
|
RepositoryState::Clean => StateDescription::Clean,
|
|
RepositoryState::Merge => StateDescription::Label(&MERGE_LABEL),
|
|
RepositoryState::Revert => StateDescription::Label(&REVERT_LABEL),
|
|
RepositoryState::RevertSequence => StateDescription::Label(&REVERT_LABEL),
|
|
RepositoryState::CherryPick => StateDescription::Label(&CHERRY_LABEL),
|
|
RepositoryState::CherryPickSequence => StateDescription::Label(&CHERRY_LABEL),
|
|
RepositoryState::Bisect => StateDescription::Label(&BISECT_LABEL),
|
|
RepositoryState::ApplyMailbox => StateDescription::Label(&AM_LABEL),
|
|
RepositoryState::ApplyMailboxOrRebase => StateDescription::Label(&AM_OR_REBASE_LABEL),
|
|
RepositoryState::Rebase => describe_rebase(root),
|
|
RepositoryState::RebaseInteractive => describe_rebase(root),
|
|
RepositoryState::RebaseMerge => describe_rebase(root),
|
|
}
|
|
}
|
|
|
|
fn describe_rebase(root: &PathBuf) -> StateDescription {
|
|
/*
|
|
* Sadly, libgit2 seems to have some issues with reading the state of
|
|
* interactive rebases. So, instead, we'll poke a few of the .git files
|
|
* ourselves. This might be worth re-visiting this in the future...
|
|
*
|
|
* The following is based heavily on: https://github.com/magicmonty/bash-git-prompt
|
|
*/
|
|
|
|
let just_label = StateDescription::Label(&REBASE_LABEL);
|
|
|
|
let dot_git = root.join(".git");
|
|
|
|
let has_path = |relative_path: &str| {
|
|
let path = dot_git.join(Path::new(relative_path));
|
|
path.exists()
|
|
};
|
|
|
|
let file_to_usize = |relative_path: &str| {
|
|
let path = dot_git.join(Path::new(relative_path));
|
|
let contents = crate::utils::read_file(path).ok()?;
|
|
let quantity = contents.trim().parse::<usize>().ok()?;
|
|
Some(quantity)
|
|
};
|
|
|
|
let paths_to_progress = |current_path: &str, total_path: &str| {
|
|
let current = file_to_usize(current_path)?;
|
|
let total = file_to_usize(total_path)?;
|
|
Some(StateProgress { current, total })
|
|
};
|
|
|
|
let progress = if has_path("rebase-merge") {
|
|
paths_to_progress("rebase-merge/msgnum", "rebase-merge/end")
|
|
} else if has_path("rebase-apply") {
|
|
paths_to_progress("rebase-apply/next", "rebase-apply/last")
|
|
} else {
|
|
None
|
|
};
|
|
|
|
match progress {
|
|
None => just_label,
|
|
Some(progress) => StateDescription::LabelAndProgress(&REBASE_LABEL, progress),
|
|
}
|
|
}
|
|
|
|
enum StateDescription {
|
|
Clean,
|
|
Label(&'static StateLabel),
|
|
LabelAndProgress(&'static StateLabel, StateProgress),
|
|
}
|
|
|
|
struct StateLabel {
|
|
segment_name: &'static str,
|
|
message_default: &'static str,
|
|
}
|
|
|
|
struct StateProgress {
|
|
current: usize,
|
|
total: usize,
|
|
}
|