harness-drone/remote/bitbucketserver/client.go
2016-04-20 11:59:47 -07:00

52 lines
1.2 KiB
Go

package bitbucketserver
import (
"crypto/tls"
"crypto/x509"
"encoding/pem"
log "github.com/Sirupsen/logrus"
"github.com/mrjones/oauth"
"io/ioutil"
"net/http"
)
func NewClient(ConsumerRSA string, ConsumerKey string, URL string) *oauth.Consumer {
//TODO: make this configurable
privateKeyFileContents, err := ioutil.ReadFile(ConsumerRSA)
log.Info("Tried to read the key")
if err != nil {
log.Error(err)
}
block, _ := pem.Decode([]byte(privateKeyFileContents))
privateKey, err := x509.ParsePKCS1PrivateKey(block.Bytes)
if err != nil {
log.Error(err)
}
c := oauth.NewRSAConsumer(
ConsumerKey,
privateKey,
oauth.ServiceProvider{
RequestTokenUrl: URL + "/plugins/servlet/oauth/request-token",
AuthorizeTokenUrl: URL + "/plugins/servlet/oauth/authorize",
AccessTokenUrl: URL + "/plugins/servlet/oauth/access-token",
HttpMethod: "POST",
})
c.HttpClient = &http.Client{
Transport: &http.Transport{
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
},
}
return c
}
func NewClientWithToken(Consumer *oauth.Consumer, AccessToken string) *http.Client {
var token oauth.AccessToken
token.Token = AccessToken
client, err := Consumer.MakeHttpClient(&token)
if err != nil {
log.Error(err)
}
return client
}