2 files changed, 57 insertions, 0 deletions
diff --git a/models/branch.go b/models/branch.go
new file mode 100644
index 00000000..08b5b7c3
--- /dev/null
+++ b/models/branch.go
@@ -0,0 +1,43 @@
+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 GetBranchesByRepo(user,repo string) ([]*Branch, error) {
+	return GetBranchesByPath(RepoPath(user, repo))
+}
+
+func (br *Branch) GetCommit() (*git.Commit, error) {
+	gitRepo, err := git.OpenRepository(br.Path)
+	if err != nil {
+		return nil, err
+	}
+	return gitRepo.GetBranchCommit(br.Name)
+}
diff --git a/models/repo.go b/models/repo.go
index 8ce1f719..82fea00a 100644
--- a/models/repo.go
+++ b/models/repo.go
@@ -288,6 +288,20 @@ func (repo *Repository) GetMirror() (err error) {
 	return err
 }
 
+func (repo *Repository) GetBranch(br string) (_ *Branch, err error) {
+	if(!git.IsBranchExist(repo.RepoPath(), br)){
+		return nil, errors.New("Branch do not exist");
+	}
+	return &Branch{
+			Path: repo.RepoPath(),
+			Name: br,
+	},nil
+}
+
+func (repo *Repository) GetBranches() (_ []*Branch, err error) {
+	return GetBranchesByPath(repo.RepoPath())
+}
+
 func (repo *Repository) GetBaseRepo() (err error) {
 	if !repo.IsFork {
 		return nil