github.com/nicgrayson/terraform@v0.4.3-0.20150415203910-c4de50829380/config/module/detect_bitbucket.go (about)

     1  package module
     2  
     3  import (
     4  	"encoding/json"
     5  	"fmt"
     6  	"net/http"
     7  	"net/url"
     8  	"strings"
     9  )
    10  
    11  // BitBucketDetector implements Detector to detect BitBucket URLs and turn
    12  // them into URLs that the Git or Hg Getter can understand.
    13  type BitBucketDetector struct{}
    14  
    15  func (d *BitBucketDetector) Detect(src, _ string) (string, bool, error) {
    16  	if len(src) == 0 {
    17  		return "", false, nil
    18  	}
    19  
    20  	if strings.HasPrefix(src, "bitbucket.org/") {
    21  		return d.detectHTTP(src)
    22  	}
    23  
    24  	return "", false, nil
    25  }
    26  
    27  func (d *BitBucketDetector) detectHTTP(src string) (string, bool, error) {
    28  	u, err := url.Parse("https://" + src)
    29  	if err != nil {
    30  		return "", true, fmt.Errorf("error parsing BitBucket URL: %s", err)
    31  	}
    32  
    33  	// We need to get info on this BitBucket repository to determine whether
    34  	// it is Git or Hg.
    35  	var info struct {
    36  		SCM string `json:"scm"`
    37  	}
    38  	infoUrl := "https://api.bitbucket.org/1.0/repositories" + u.Path
    39  	resp, err := http.Get(infoUrl)
    40  	if err != nil {
    41  		return "", true, fmt.Errorf("error looking up BitBucket URL: %s", err)
    42  	}
    43  	if resp.StatusCode == 403 {
    44  		// A private repo
    45  		return "", true, fmt.Errorf(
    46  			"shorthand BitBucket URL can't be used for private repos, " +
    47  				"please use a full URL")
    48  	}
    49  	dec := json.NewDecoder(resp.Body)
    50  	if err := dec.Decode(&info); err != nil {
    51  		return "", true, fmt.Errorf("error looking up BitBucket URL: %s", err)
    52  	}
    53  
    54  	switch info.SCM {
    55  	case "git":
    56  		if !strings.HasSuffix(u.Path, ".git") {
    57  			u.Path += ".git"
    58  		}
    59  
    60  		return "git::" + u.String(), true, nil
    61  	case "hg":
    62  		return "hg::" + u.String(), true, nil
    63  	default:
    64  		return "", true, fmt.Errorf("unknown BitBucket SCM type: %s", info.SCM)
    65  	}
    66  }