From 18aea4bfe89b83c5e2d6d55daa68efa6180655cc Mon Sep 17 00:00:00 2001
From: Henning Perl <perl@cs.uni-bonn.de>
Date: Sun, 28 Dec 2014 23:07:33 +0100
Subject: Add git_diff_get_stats()

This commit adds git_diff_get_stats() as well as functions to query the
stats for insertions, deletions, and changed files.
---
 diff.go | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

(limited to 'diff.go')

diff --git a/diff.go b/diff.go
index 63f3453..63fa867 100644
--- a/diff.go
+++ b/diff.go
@@ -190,6 +190,46 @@ func (diff *Diff) FindSimilar(opts *DiffFindOptions) error {
 	return nil
 }
 
+type DiffStats struct {
+	ptr *C.git_diff_stats
+}
+
+func (stats *DiffStats) Free() error {
+	if stats.ptr == nil {
+		return ErrInvalid
+	}
+	runtime.SetFinalizer(stats, nil)
+	C.git_diff_stats_free(stats.ptr)
+	stats.ptr = nil
+	return nil
+}
+
+func (stats *DiffStats) Insertions() int {
+	return int(C.git_diff_stats_insertions(stats.ptr))
+}
+
+func (stats *DiffStats) Deletions() int {
+	return int(C.git_diff_stats_deletions(stats.ptr))
+}
+
+func (stats *DiffStats) FilesChanged() int {
+	return int(C.git_diff_stats_files_changed(stats.ptr))
+}
+
+func (diff *Diff) Stats() (*DiffStats, error) {
+	stats := new(DiffStats)
+
+	runtime.LockOSThread()
+	defer runtime.UnlockOSThread()
+
+	if ecode := C.git_diff_get_stats(&stats.ptr, diff.ptr); ecode < 0 {
+		return nil, MakeGitError(ecode)
+	}
+	runtime.SetFinalizer(stats, (*DiffStats).Free)
+
+	return stats, nil
+}
+
 type diffForEachData struct {
 	FileCallback DiffForEachFileCallback
 	HunkCallback DiffForEachHunkCallback
-- 
cgit v1.2.3