ignore bots in commits

Author: o2sh

Diff for: src/onefetch/cli.rs

@@ -27,6 +27,7 @@ pub struct Cli {
     pub image_backend: Option<Box<dyn image_backends::ImageBackend>>,
     pub image_color_resolution: usize,
     pub no_merges: bool,
+    pub no_bots: bool,
     pub no_color_palette: bool,
     pub number_of_authors: usize,
     pub excluded: Vec<String>,
@@ -208,6 +209,11 @@ impl Cli {
             .long("no-merges")
             .help("Ignores merge commits."),
         )
+        .arg(
+            Arg::with_name("no-bots")
+            .long("no-bots")
+            .help("Ignore bots."),
+        )
         .arg(
             Arg::with_name("isotime")
             .short("z")
@@ -263,6 +269,7 @@ impl Cli {
         };
         let no_bold = matches.is_present("no-bold");
         let no_merges = matches.is_present("no-merges");
+        let no_bots = matches.is_present("no-bots");
         let no_color_palette = matches.is_present("no-palette");
         let print_languages = matches.is_present("languages");
         let print_package_managers = matches.is_present("package-managers");
@@ -360,6 +367,7 @@ impl Cli {
             image_backend,
             image_color_resolution,
             no_merges,
+            no_bots,
             no_color_palette,
             number_of_authors,
             excluded,

Diff for: src/onefetch/info.rs

@@ -211,7 +211,7 @@ impl Info {
     pub fn new(config: Cli) -> Result<Info> {
         let git_version = cli_utils::get_git_version();
         let repo = Repository::discover(&config.repo_path)?;
-        let internal_repo = Repo::new(&repo, config.no_merges)?;
+        let internal_repo = Repo::new(&repo, config.no_merges, config.no_bots)?;
         let (repo_name, repo_url) = internal_repo.get_name_and_url()?;
         let head_refs = internal_repo.get_head_refs()?;
         let pending_changes = internal_repo.get_pending_changes()?;

Diff for: src/onefetch/repo.rs

@@ -1,6 +1,7 @@
 use crate::onefetch::{commit_info::CommitInfo, error::*, utils};
 use git2::{
-    BranchType, Commit, Repository, RepositoryOpenFlags, Status, StatusOptions, StatusShow,
+    BranchType, Commit, Repository, RepositoryOpenFlags, Signature, Status, StatusOptions,
+    StatusShow,
 };
 use regex::Regex;
 use std::path::Path;
@@ -11,12 +12,12 @@ pub struct Repo<'a> {
 }
 
 impl<'a> Repo<'a> {
-    pub fn new(repo: &'a Repository, no_merges: bool) -> Result<Self> {
-        let logs = Repo::get_logs(repo, no_merges)?;
+    pub fn new(repo: &'a Repository, no_merges: bool, no_bots: bool) -> Result<Self> {
+        let logs = Repo::get_logs(repo, no_merges, no_bots)?;
         Ok(Self { repo, logs })
     }
 
-    fn get_logs(repo: &'a Repository, no_merges: bool) -> Result<Vec<Commit<'a>>> {
+    fn get_logs(repo: &'a Repository, no_merges: bool, no_bots: bool) -> Result<Vec<Commit<'a>>> {
         let mut revwalk = repo.revwalk()?;
         revwalk.push_head()?;
         let logs: Vec<Commit<'a>> = revwalk
@@ -25,7 +26,8 @@ impl<'a> Repo<'a> {
                 Ok(r) => repo
                     .find_commit(r)
                     .ok()
-                    .filter(|commit| !(no_merges && commit.parents().len() > 1)),
+                    .filter(|commit| !(no_merges && commit.parents().len() > 1))
+                    .filter(|commit| !(no_bots && is_bot(commit.author()))),
             })
             .collect();
 
@@ -269,3 +271,8 @@ pub fn is_valid(repo_path: &str) -> Result<bool> {
     let repo = Repository::open_ext(repo_path, RepositoryOpenFlags::empty(), Vec::<&Path>::new());
     Ok(repo.is_ok() && !repo?.is_bare())
 }
+
+pub fn is_bot(author: Signature) -> bool {
+    let author_name = String::from_utf8_lossy(author.name_bytes()).into_owned();
+    author_name.contains("[bot]")
+}