starship/src/modules/git_branch.rs

use ansi_term::Color;

use super::{Context, Module};

/// Creates a module with the Git branch in the current directory
///
/// Will display the branch name if the current directory is a git repo
pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
    const GIT_BRANCH_CHAR: &str = " ";

    let branch_name = context.branch_name.as_ref()?;
    let segment_color = Color::Purple.bold();

    let mut module = context.new_module("git_branch")?;
    module.set_style(segment_color);
    module.get_prefix().set_value("on ");

    module.new_segment("symbol", GIT_BRANCH_CHAR);
    module.new_segment("name", branch_name);

    Some(module)
}