add GetApiHostnameURL to client

Author: cgroschupp

client.go

@@ -25,25 +25,25 @@ const DEFAULT_PAGE_LENGTH = 10
 const DEFAULT_MAX_DEPTH = 1
 const DEFAULT_BITBUCKET_API_BASE_URL = "https://api.bitbucket.org/2.0"
 
-func apiBaseUrl() string {
+func apiBaseUrl() (*url.URL, error) {
 	ev := os.Getenv("BITBUCKET_API_BASE_URL")
-	if ev != "" {
-		return ev
+	if ev == "" {
+		ev = DEFAULT_BITBUCKET_API_BASE_URL
 	}
 
-	return DEFAULT_BITBUCKET_API_BASE_URL
+	return url.Parse(ev)
 }
 
 type Client struct {
-	Auth         *auth
-	Users        users
-	User         user
-	Teams        teams
-	Repositories *Repositories
-	Workspaces   *Workspace
-	Pagelen      uint64
-	MaxDepth     uint64
-	apiBaseURL   string
+	Auth            *auth
+	Users           users
+	User            user
+	Teams           teams
+	Repositories    *Repositories
+	Workspaces      *Workspace
+	Pagelen         uint64
+	MaxDepth        uint64
+	apiBaseURL      *url.URL
 
 	HttpClient *http.Client
 }
@@ -135,7 +135,11 @@ func NewBasicAuth(u, p string) *Client {
 }
 
 func injectClient(a *auth) *Client {
-	c := &Client{Auth: a, Pagelen: DEFAULT_PAGE_LENGTH, MaxDepth: DEFAULT_MAX_DEPTH, apiBaseURL: apiBaseUrl()}
+	bitbucketUrl, err := apiBaseUrl()
+	if err != nil {
+		log.Fatalf("invalid bitbucket url")
+	}
+	c := &Client{Auth: a, Pagelen: DEFAULT_PAGE_LENGTH, MaxDepth: DEFAULT_MAX_DEPTH, apiBaseURL: bitbucketUrl}
 	c.Repositories = &Repositories{
 		c:                  c,
 		PullRequests:       &PullRequests{c: c},
@@ -156,11 +160,15 @@ func injectClient(a *auth) *Client {
 }
 
 func (c *Client) GetApiBaseURL() string {
-	return c.apiBaseURL
+	return fmt.Sprintf("%s://%s%s", c.apiBaseURL.Scheme, c.apiBaseURL.Host, c.apiBaseURL.Path)
+}
+
+func (c *Client) GetApiHostnameURL() string {
+	return fmt.Sprintf("%s://%s", c.apiBaseURL.Scheme, c.apiBaseURL.Host)
 }
 
-func (c *Client) SetApiBaseURL(urlStr string) {
-	c.apiBaseURL = urlStr
+func (c *Client) SetApiBaseURL(urlStr url.URL) {
+	c.apiBaseURL = &urlStr
 }
 
 func (c *Client) executeRaw(method string, urlStr string, text string) (io.ReadCloser, error) {
@@ -386,7 +394,7 @@ func unexpectedHttpStatusCode(statusCode int) bool {
 func (c *Client) requestUrl(template string, args ...interface{}) string {
 
 	if len(args) == 1 && args[0] == "" {
-		return c.apiBaseURL + template
+		return c.GetApiBaseURL() + template
 	}
-	return c.apiBaseURL + fmt.Sprintf(template, args...)
+	return c.GetApiBaseURL() + fmt.Sprintf(template, args...)
 }