use crate::commit::models::Commit; use chrono::{DateTime, Local}; use git2::{BranchType, Error, Repository, Time}; use std::fs::remove_dir_all; /// Return the temporary folder of the git repository fn get_tmp_dir(repo_name: &String) -> String { format!("/tmp/{}", &repo_name) } /// Clone a repository `repo` in a temporary folder. It uses only GitHub at /// this moment. pub fn clone_repo(repo_name: &String) -> Result { let url: &str = &format!("https://github.com/{}", &repo_name).to_owned()[..]; let tmp_dir: String = get_tmp_dir(&repo_name); let cloned = Repository::clone(url, tmp_dir)?; Ok(cloned) } /// Check if a `branch` exists inside the repository and then set the head of /// `repo` to that branch pub fn get_branch(repo: &Repository, branch: &str) -> Result<(), Error> { if let Err(e) = repo.find_branch(&branch, BranchType::Local) { return Err(e); } repo.set_head(&format!("refs/heads/{}", branch).to_owned()[..]) } /// Get a `git2::Commit` and returns a date in RFC2822 format pub fn get_date(commit: &git2::Commit) -> String { let t: Time = commit.time(); let (offset, sign) = match t.offset_minutes() { n if n < 0 => (-n, '-'), n => (n, '+'), }; let (hours, minutes) = (offset / 60, offset % 60); let ts = time::Timespec::new(t.seconds() + (t.offset_minutes() as i64) * 60, 0); let time = time::at(ts); let result = format!( "{} {}{:02}{:02}", time.strftime("%a, %e %b %Y %T").unwrap(), sign, hours, minutes ); result } /// Get a `git2::Commit` and returns a valid `Commit` to upload to the database fn get_commit(gcommit: &git2::Commit, repo_name: &String) -> Commit { let hash = gcommit.id().to_string(); let tree = match gcommit.parent_id(0) { Ok(parent) => Some(parent.to_string()), Err(_) => None, }; let mut text = "".to_string(); for line in String::from_utf8_lossy(gcommit.message_bytes()).lines() { text += line; text += "\n"; } // Remove the last "\n" let _ = text.pop(); let date: DateTime = match DateTime::parse_from_rfc2822(&get_date(&gcommit)) { Ok(date) => date.into(), Err(e) => { // This is a real problem! // TODO: manage this error panic!(e) } }; let author_email = gcommit.author().email().unwrap().to_string(); let author_name = gcommit.author().name().unwrap().to_string(); let committer_email = gcommit.committer().email().unwrap().to_string(); let committer_name = gcommit.committer().name().unwrap().to_string(); Commit { hash, tree, text, date, author_email, author_name, committer_email, committer_name, repository_url: repo_name.clone(), } } /// Get all commits from a Git repository. Returns an array of `Commit`. /// First, clone the repo into a temporary folder. /// Then, open the repository. /// Then, get commits /// Finally, remove the temporary folder pub fn repo_commits( repo_name: &String, branch: &String, ) -> Result, Error> { // Remove a possible already cloned repository let _ = remove_dir_all(get_tmp_dir(&repo_name)); // Try to clone the repo. If it returns an error, it's useless to go ahead: // raises an error. let repo = match clone_repo(&repo_name) { Ok(r) => r, Err(e) => { return Err(e); } }; if let Err(e) = get_branch(&repo, branch) { return Err(e); } let mut revwalk = repo.revwalk()?; let head = repo.head().unwrap().target().unwrap(); if let Err(e) = revwalk.push(head) { return Err(e); } let mut commits: Vec = vec![]; for commit in revwalk { let hash = repo.find_commit(commit?)?; commits.push(get_commit(&hash, &repo_name)); } // Remove the cloned repository folder let _ = remove_dir_all(get_tmp_dir(&repo_name)); Ok(commits) }