aboutsummaryrefslogtreecommitdiff
path: root/apps/servers/octopus/supergit/src/commit.rs
blob: 58b57ce22f733ef12b65b1f38e8edf49291435de (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
use crate::{files::FileTree, Diff, HashId};
use git2::Repository;
use std::sync::Arc;

/// Represent a commit on a repository
///
/// When creating a commit object, it is guaranteed that it exists in
/// the repository.
#[derive(Clone)]
pub struct Commit {
    pub id: HashId,
    repo: Arc<Repository>,
}

impl Commit {
    /// Create a commit object and check if it exists in the repo
    pub(crate) fn new(r: &Arc<Repository>, id: HashId) -> Option<Self> {
        r.find_commit(id.to_oid()).ok().map(|_| Self {
            id,
            repo: Arc::clone(r),
        })
    }

    /// Get a utf-8 string representation of the commit ID
    pub fn id_str(&self) -> String {
        self.id.to_string()
    }

    /// Get the summary line as a utf-8 string
    pub fn summary(&self) -> String {
        self.find().summary().unwrap().into()
    }

    /// Get the number of parents
    pub fn parent_count(&self) -> usize {
        self.repo
            .find_commit(self.id.to_oid())
            .unwrap()
            .parent_count()
    }

    /// Return the first parent, if it exists
    pub fn first_parent(&self) -> Option<Self> {
        self.find()
            .parent(0)
            .ok()
            .and_then(|c| Self::new(&self.repo, c.id().into()))
    }

    /// Get a specific parent, if it exists
    pub fn parent(&self, num: usize) -> Option<Self> {
        self.find()
            .parent(num)
            .ok()
            .and_then(|c| Self::new(&self.repo, c.id().into()))
    }

    /// Get the set of parents as a vector
    ///
    /// Use this function if you suspect a commit has more than one
    /// parent.
    pub fn parents(&self) -> Vec<Commit> {
        self.find()
            .parents()
            .map(|c| Self::new(&self.repo, c.id().into()).unwrap())
            .collect()
    }

    /// Get the file tree for this commit
    pub fn tree(&self) -> FileTree {
        FileTree::new(Arc::clone(&self.repo), self.id.clone())
    }

    /// Get the list of paths in the repository touched by this commit
    ///
    /// Using this function directly is often not what you want.
    /// Instead, use the `get_history(...)` function on `FileTree`,
    /// which uses this function.
    pub fn get_paths(&self) -> Vec<String> {
        self.get_diff()
            .map(|d| Diff::from(d))
            .map_or(vec![], |d| d.get_paths())
    }

    /// Utility function to get a merged diff from all parents
    fn get_diff(&self) -> Option<git2::Diff> {
        // Get all diffs with parents
        let stree = self.find().tree().unwrap();
        let mut vec = self
            .parents()
            .into_iter()
            .filter_map(|p| {
                self.repo
                    .diff_tree_to_tree(Some(&stree), Some(p.find().tree().as_ref().unwrap()), None)
                    .ok()
            })
            .collect::<Vec<_>>();

        // If there are no parents
        if vec.len() == 0 {
            vec = vec![self.repo.diff_tree_to_tree(Some(&stree), None, None).ok()?];
        }

        // Take the first and merge onto
        let first = vec.remove(0);
        Some(vec.iter().fold(first, |mut acc, diff| {
            acc.merge(diff).unwrap();
            acc
        }))
    }

    fn find(&self) -> git2::Commit {
        self.repo.find_commit(self.id.to_oid()).unwrap()
    }
}