v2_impl.go

package bitbucket

import (
	"fmt"
	"github.com/dghubble/sling"
	"net/http"
)

type basicAuth struct {
	user, password string
}

type role string

const (
	AdminRole       role = "admin"
	ContributorRole role = "contributor"
	MemberRole      role = "member"
)

type V2Impl struct {
	client       *sling.Sling
	Teams        v2Teams
	Repositories v2Repositories
	Users        v2Users
	basicAuth    basicAuth
}

func NewV2Of(client *http.Client) *V2Impl {
	if client == nil {
		client = http.DefaultClient
	}

	base := sling.New().Base(v2BaseUrl).Client(client)

	impl := &V2Impl{
		client: base,
	}

	impl.Teams = &v2TeamsImpl{impl}
	impl.Repositories = &v2RepositoriesImpl{impl}
	impl.Users = &v2UsersImpl{impl}

	return impl
}

// anonymous v2 api
func NewV2() *V2Impl {
	return NewV2Of(nil)
}

func NewV2BasicAuth(user, pass string) *V2Impl {
	client := NewV2()
	client.basicAuth = basicAuth{
		user, pass,
	}

	return client
}

func (impl *V2Impl) DoList(req *http.Request) (*ListResult, error) {
	successV := &ListResult{}
	failureV := &BitbucketError{}
	resp, err := impl.DoCustom(req, successV, failureV)

	if err != nil {
		if resp.StatusCode == http.StatusUnauthorized {
			return nil, fmt.Errorf("unauthorized")
		}
		return nil, err
	}

	// this is a status code > 299 and there's formatted error returned from bitbucket
	if failureV.Err.Message != "" {
		return nil, failureV
	}

	return successV, nil
}

func (impl *V2Impl) Next(next string) (*ListResult, error) {
	req, err := impl.client.Path(next).Request()
	if err != nil {

	}
	return impl.DoList(req)
}

func (impl *V2Impl) DoCustom(req *http.Request, successV, failureV interface{}) (*http.Response, error) {
	req.Header.Set("Content-Type", "application/json")

	if impl.basicAuth.user != "" {
		req.SetBasicAuth(impl.basicAuth.user, impl.basicAuth.password)
	}

	resp, err := impl.client.Do(req, successV, failureV)
	return resp, err
}

func (impl *V2Impl) Do(req *http.Request) (map[string]interface{}, error) {
	successV := map[string]interface{}{}
	failureV := &BitbucketError{}

	_, err := impl.DoCustom(req, &successV, failureV)
	// this is a fatal error
	if err != nil {
		return nil, err
	}

	// this is a status code > 299 and there's formatted error returned from bitbucket
	if failureV.Err.Message != "" {
		return nil, failureV
	}

	return successV, nil
}