Added support for max_depth (#73)

Author: sffortytwo

bitbucket.go

@@ -138,6 +138,7 @@ type RepositoryBranchOptions struct {
 	Sort     string `json:"sort"`
 	PageNum  int    `json:"page"`
 	Pagelen  int    `json:"pagelen"`
+	MaxDepth int    `json:"max_depth"`
 }
 
 type PullRequestsOptions struct {
@@ -238,5 +239,6 @@ type DownloadsOptions struct {
 type PageRes struct {
 	Page    int32 `json:"page"`
 	PageLen int32 `json:"pagelen"`
+	MaxDepth int32 `json:"max_depth"`
 	Size    int32 `json:"size"`
 }

client.go

@@ -23,6 +23,7 @@ import (
 )
 
 const DEFAULT_PAGE_LENGTH = 10
+const DEFAULT_MAX_DEPTH = 1
 
 type Client struct {
 	Auth         *auth
@@ -31,6 +32,7 @@ type Client struct {
 	Teams        teams
 	Repositories *Repositories
 	Pagelen      uint64
+	MaxDepth     uint64
 
 	HttpClient *http.Client
 }
@@ -122,7 +124,7 @@ func NewBasicAuth(u, p string) *Client {
 }
 
 func injectClient(a *auth) *Client {
-	c := &Client{Auth: a, Pagelen: DEFAULT_PAGE_LENGTH}
+	c := &Client{Auth: a, Pagelen: DEFAULT_PAGE_LENGTH, MaxDepth: DEFAULT_MAX_DEPTH}
 	c.Repositories = &Repositories{
 		c:                  c,
 		PullRequests:       &PullRequests{c: c},
@@ -169,6 +171,17 @@ func (c *Client) execute(method string, urlStr string, text string) (interface{}
 			urlObj.RawQuery = q.Encode()
 			urlStr = urlObj.String()
 		}
+
+		if c.MaxDepth != DEFAULT_MAX_DEPTH {
+			urlObj, err := url.Parse(urlStr)
+			if err != nil {
+				return nil, err
+			}
+			q := urlObj.Query()
+			q.Set("max_depth", strconv.FormatUint(c.MaxDepth, DEC_RADIX))
+			urlObj.RawQuery = q.Encode()
+			urlStr = urlObj.String()
+		}
 	}
 
 	body := strings.NewReader(text)
@@ -217,6 +230,7 @@ func (c *Client) execute(method string, urlStr string, text string) (interface{}
 					resultMap["values"] = valuesSlice
 					delete(resultMap, "page")
 					delete(resultMap, "pagelen")
+					delete(resultMap, "max_depth")
 					delete(resultMap, "size")
 					result = resultMap
 				}

repositories.go

@@ -23,6 +23,7 @@ type Repositories struct {
 type RepositoriesRes struct {
 	Page    int32
 	Pagelen int32
+	MaxDepth int32
 	Size    int32
 	Items   []Repository
 }
@@ -85,6 +86,10 @@ func decodeRepositorys(reposResponse interface{}) (*RepositoriesRes, error) {
 	if !ok {
 		pagelen = 0
 	}
+	max_depth, ok := reposResponseMap["max_width"].(float64)
+	if !ok {
+		max_depth = 0
+	}
 	size, ok := reposResponseMap["size"].(float64)
 	if !ok {
 		size = 0
@@ -93,6 +98,7 @@ func decodeRepositorys(reposResponse interface{}) (*RepositoriesRes, error) {
 	repositories := RepositoriesRes{
 		Page:    int32(page),
 		Pagelen: int32(pagelen),
+		MaxDepth: int32(max_depth),
 		Size:    int32(size),
 		Items:   repos,
 	}

repository.go

@@ -46,6 +46,7 @@ type RepositoryBlob struct {
 type RepositoryBranches struct {
 	Page     int
 	Pagelen  int
+	MaxDepth int
 	Size     int
 	Next     string
 	Branches []RepositoryBranch
@@ -106,6 +107,7 @@ func (r *Repository) Get(ro *RepositoryOptions) (*Repository, error) {
 func (r *Repository) ListFiles(ro *RepositoryFilesOptions) ([]RepositoryFile, error) {
 	filePath := path.Join("/repositories", ro.Owner, ro.RepoSlug, "src", ro.Ref, ro.Path) + "/"
 	urlStr := r.c.requestUrl(filePath)
+	print(urlStr)
 	response, err := r.c.execute("GET", urlStr, "")
 	if err != nil {
 		return nil, err
@@ -146,6 +148,10 @@ func (r *Repository) ListBranches(rbo *RepositoryBranchOptions) (*RepositoryBran
 		params.Add("pagelen", strconv.Itoa(rbo.Pagelen))
 	}
 
+	if rbo.MaxDepth > 0 {
+		params.Add("max_depth", strconv.Itoa(rbo.MaxDepth))
+	}
+
 	urlStr := r.c.requestUrl("/repositories/%s/%s/refs/branches?%s", rbo.Owner, rbo.RepoSlug, params.Encode())
 	response, err := r.c.executeRaw("GET", urlStr, "")
 	if err != nil {
@@ -349,6 +355,10 @@ func decodeRepositoryBranches(branchResponse interface{}) (*RepositoryBranches,
 	if !ok {
 		pagelen = 0
 	}
+	max_depth, ok := branchResponseMap["max_depth"].(float64)
+	if !ok {
+		max_depth = 0
+	}
 	size, ok := branchResponseMap["size"].(float64)
 	if !ok {
 		size = 0
@@ -362,6 +372,7 @@ func decodeRepositoryBranches(branchResponse interface{}) (*RepositoryBranches,
 	repositoryBranches := RepositoryBranches{
 		Page:     int(page),
 		Pagelen:  int(pagelen),
+		MaxDepth: int(max_depth),
 		Size:     int(size),
 		Next:     next,
 		Branches: branches,