feat: Add git_commit module (#673)

1 files changed, 54 insertions, 0 deletions

diff --git a/src/modules/git_commit.rs b/src/modules/git_commit.rs
new file mode 100644
index 000000000..cacfa3982
--- /dev/null
+++ b/src/modules/git_commit.rs
@@ -0,0 +1,54 @@
+use super::{Context, Module, RootModuleConfig};
+use git2::Repository;
+
+use crate::configs::git_commit::GitCommitConfig;
+
+/// Creates a module with the Git commit in the current directory
+///
+/// Will display the commit hash if the current directory is a git repo
+pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
+    let mut module = context.new_module("git_commit");
+    let config = GitCommitConfig::try_load(module.config);
+    if config.disabled {
+        return None;
+    };
+
+    module
+        .get_prefix()
+        .set_value(config.prefix)
+        .set_style(config.style);
+    module
+        .get_suffix()
+        .set_value(config.suffix)
+        .set_style(config.style);
+    module.set_style(config.style);
+
+    let repo = context.get_repo().ok()?;
+    let repo_root = repo.root.as_ref()?;
+    let git_repo = Repository::open(repo_root).ok()?;
+
+    let git_head = git_repo.head().ok()?;
+    let head_commit = git_head.peel_to_commit().ok()?;
+    let commit_oid = head_commit.id();
+    module.create_segment(
+        "hash",
+        &config.hash.with_value(&id_to_hex_abbrev(
+            commit_oid.as_bytes(),
+            config.commit_hash_length,
+        )),
+    );
+
+    Some(module)
+}
+
+/// len specifies length of hex encoded string
+pub fn id_to_hex_abbrev(bytes: &[u8], len: usize) -> String {
+    bytes
+        .iter()
+        .map(|b| format!("{:02x}", b))
+        .collect::<Vec<String>>()
+        .join("")
+        .chars()
+        .take(len)
+        .collect()
+}