Git-Trees API (#5403)

* Git-Trees API

* update vendor'd libs

* added comments to exported function and formatted.

* make fmt

* update per @lafirks feedback

2 files changed, 93 insertions, 0 deletions

diff --git a/routers/api/v1/api.go b/routers/api/v1/api.go
index 47e556c881..0284e845ff 100644
--- a/routers/api/v1/api.go
+++ b/routers/api/v1/api.go
@@ -610,6 +610,7 @@ func RegisterRoutes(m *macaron.Macaron) {
 				m.Group("/git", func() {
 					m.Get("/refs", repo.GetGitAllRefs)
 					m.Get("/refs/*", repo.GetGitRefs)
+					m.Combo("/trees/:sha", context.RepoRef()).Get(repo.GetTree)
 				}, reqRepoReader(models.UnitTypeCode))
 			}, repoAssignment())
 		})
diff --git a/routers/api/v1/repo/tree.go b/routers/api/v1/repo/tree.go
new file mode 100644
index 0000000000..fd0bf84132
--- /dev/null
+++ b/routers/api/v1/repo/tree.go
@@ -0,0 +1,92 @@
+// Copyright 2018 The Gitea 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 repo
+
+import (
+	"fmt"
+	"strings"
+
+	"code.gitea.io/git"
+	"code.gitea.io/gitea/modules/context"
+	"code.gitea.io/gitea/modules/setting"
+	"code.gitea.io/sdk/gitea"
+)
+
+// GetTree get the tree of a repository.
+func GetTree(ctx *context.APIContext) {
+	sha := ctx.Params("sha")
+	if len(sha) == 0 {
+		ctx.Error(400, "sha not provided", nil)
+		return
+	}
+	tree := GetTreeBySHA(ctx, sha)
+	if tree != nil {
+		ctx.JSON(200, tree)
+	} else {
+		ctx.Error(400, "sha invalid", nil)
+	}
+}
+
+// GetTreeBySHA get the GitTreeResponse of a repository using a sha hash.
+func GetTreeBySHA(ctx *context.APIContext, sha string) *gitea.GitTreeResponse {
+	gitTree, err := ctx.Repo.GitRepo.GetTree(sha)
+	if err != nil || gitTree == nil {
+		return nil
+	}
+	tree := new(gitea.GitTreeResponse)
+	repoID := strings.TrimRight(setting.AppURL, "/") + "/api/v1/repos/" + ctx.Repo.Repository.Owner.Name + "/" + ctx.Repo.Repository.Name
+	tree.SHA = gitTree.ID.String()
+	tree.URL = repoID + "/git/trees/" + tree.SHA
+	var entries git.Entries
+	if ctx.QueryBool("recursive") {
+		entries, err = gitTree.ListEntriesRecursive()
+	} else {
+		entries, err = gitTree.ListEntries()
+	}
+	if err != nil {
+		return tree
+	}
+	repoIDLen := len(repoID)
+
+	// 51 is len(sha1) + len("/git/blobs/"). 40 + 11.
+	blobURL := make([]byte, repoIDLen+51)
+	copy(blobURL[:], repoID)
+	copy(blobURL[repoIDLen:], "/git/blobs/")
+
+	// 51 is len(sha1) + len("/git/trees/"). 40 + 11.
+	treeURL := make([]byte, repoIDLen+51)
+	copy(treeURL[:], repoID)
+	copy(treeURL[repoIDLen:], "/git/trees/")
+
+	// 40 is the size of the sha1 hash in hexadecimal format.
+	copyPos := len(treeURL) - 40
+
+	if len(entries) > 1000 {
+		tree.Entries = make([]gitea.GitEntry, 1000)
+	} else {
+		tree.Entries = make([]gitea.GitEntry, len(entries))
+	}
+	for e := range entries {
+		if e > 1000 {
+			tree.Truncated = true
+			break
+		}
+
+		tree.Entries[e].Path = entries[e].Name()
+		tree.Entries[e].Mode = fmt.Sprintf("%06x", entries[e].Mode())
+		tree.Entries[e].Type = string(entries[e].Type)
+		tree.Entries[e].Size = entries[e].Size()
+		tree.Entries[e].SHA = entries[e].ID.String()
+
+		if entries[e].IsDir() {
+			copy(treeURL[copyPos:], entries[e].ID.String())
+			tree.Entries[e].URL = string(treeURL[:])
+		} else {
+			copy(blobURL[copyPos:], entries[e].ID.String())
+			tree.Entries[e].URL = string(blobURL[:])
+		}
+	}
+	return tree
+}