From b7b30cd85e5cabd9d643013ffb10dafd133b18ea Mon Sep 17 00:00:00 2001
From: Antoine GIRARD <sapk@sapk.fr>
Date: Thu, 28 Jan 2016 20:49:05 +0100
Subject: Corrections following recommendations

---
 models/repo_branch.go | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 models/repo_branch.go

(limited to 'models/repo_branch.go')

diff --git a/models/repo_branch.go b/models/repo_branch.go
new file mode 100644
index 00000000..b784f7d5
--- /dev/null
+++ b/models/repo_branch.go
@@ -0,0 +1,43 @@
+// Copyright 2016 The Gogs Authors. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package models
+
+import (
+	"github.com/gogits/git-module"
+)
+
+type Branch struct {
+	Path		string
+	Name      	string
+}
+
+func GetBranchesByPath(path string) ([]*Branch, error) {
+	gitRepo, err := git.OpenRepository(path)
+	if err != nil {
+		return nil, err
+	}
+
+	brs, err := gitRepo.GetBranches()
+	if err != nil {
+		return nil, err
+	}
+
+	Branches := make([]*Branch, len(brs))
+	for i := range brs {
+		Branches[i] = &Branch{
+			Path: path,
+			Name: brs[i],
+		}
+	}
+	return Branches, nil
+}
+
+func (br *Branch) GetCommit() (*git.Commit, error) {
+	gitRepo, err := git.OpenRepository(br.Path)
+	if err != nil {
+		return nil, err
+	}
+	return gitRepo.GetBranchCommit(br.Name)
+}
-- 
cgit v1.2.3