559 lines
15 KiB
Go
559 lines
15 KiB
Go
package client
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/tls"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
"github.com/drone/drone/model"
|
|
"github.com/drone/drone/queue"
|
|
"github.com/gorilla/websocket"
|
|
"golang.org/x/net/context"
|
|
"golang.org/x/net/context/ctxhttp"
|
|
"golang.org/x/oauth2"
|
|
)
|
|
|
|
const (
|
|
pathPull = "%s/api/queue/pull/%s/%s"
|
|
pathWait = "%s/api/queue/wait/%d"
|
|
pathStream = "%s/api/queue/stream/%d"
|
|
pathPush = "%s/api/queue/status/%d"
|
|
pathPing = "%s/api/queue/ping"
|
|
pathLogs = "%s/api/queue/logs/%d"
|
|
pathLogsAuth = "%s/api/queue/logs/%d?access_token=%s"
|
|
|
|
pathSelf = "%s/api/user"
|
|
pathFeed = "%s/api/user/feed"
|
|
pathRepos = "%s/api/user/repos"
|
|
pathRepo = "%s/api/repos/%s/%s"
|
|
pathChown = "%s/api/repos/%s/%s/chown"
|
|
pathEncrypt = "%s/api/repos/%s/%s/encrypt"
|
|
pathBuilds = "%s/api/repos/%s/%s/builds"
|
|
pathBuild = "%s/api/repos/%s/%s/builds/%v"
|
|
pathJob = "%s/api/repos/%s/%s/builds/%d/%d"
|
|
pathLog = "%s/api/repos/%s/%s/logs/%d/%d"
|
|
pathKey = "%s/api/repos/%s/%s/key"
|
|
pathSign = "%s/api/repos/%s/%s/sign"
|
|
pathRepoSecrets = "%s/api/repos/%s/%s/secrets"
|
|
pathRepoSecret = "%s/api/repos/%s/%s/secrets/%s"
|
|
pathTeamSecrets = "%s/api/teams/%s/secrets"
|
|
pathTeamSecret = "%s/api/teams/%s/secrets/%s"
|
|
pathNodes = "%s/api/nodes"
|
|
pathNode = "%s/api/nodes/%d"
|
|
pathUsers = "%s/api/users"
|
|
pathUser = "%s/api/users/%s"
|
|
pathBuildQueue = "%s/api/builds"
|
|
pathAgent = "%s/api/agents"
|
|
)
|
|
|
|
type client struct {
|
|
client *http.Client
|
|
token string // auth token
|
|
base string // base url
|
|
}
|
|
|
|
// NewClient returns a client at the specified url.
|
|
func NewClient(uri string) Client {
|
|
return &client{client: http.DefaultClient, base: uri}
|
|
}
|
|
|
|
// NewClientToken returns a client at the specified url that authenticates all
|
|
// outbound requests with the given token.
|
|
func NewClientToken(uri, token string) Client {
|
|
config := new(oauth2.Config)
|
|
auther := config.Client(oauth2.NoContext, &oauth2.Token{AccessToken: token})
|
|
return &client{client: auther, base: uri, token: token}
|
|
}
|
|
|
|
// NewClientTokenTLS returns a client at the specified url that authenticates
|
|
// all outbound requests with the given token and tls.Config if provided.
|
|
func NewClientTokenTLS(uri, token string, c *tls.Config) Client {
|
|
config := new(oauth2.Config)
|
|
auther := config.Client(oauth2.NoContext, &oauth2.Token{AccessToken: token})
|
|
if c != nil {
|
|
if trans, ok := auther.Transport.(*oauth2.Transport); ok {
|
|
trans.Base = &http.Transport{
|
|
TLSClientConfig: c,
|
|
Proxy: http.ProxyFromEnvironment,
|
|
}
|
|
}
|
|
}
|
|
return &client{client: auther, base: uri, token: token}
|
|
}
|
|
|
|
// Self returns the currently authenticated user.
|
|
func (c *client) Self() (*model.User, error) {
|
|
out := new(model.User)
|
|
uri := fmt.Sprintf(pathSelf, c.base)
|
|
err := c.get(uri, out)
|
|
return out, err
|
|
}
|
|
|
|
// User returns a user by login.
|
|
func (c *client) User(login string) (*model.User, error) {
|
|
out := new(model.User)
|
|
uri := fmt.Sprintf(pathUser, c.base, login)
|
|
err := c.get(uri, out)
|
|
return out, err
|
|
}
|
|
|
|
// UserList returns a list of all registered users.
|
|
func (c *client) UserList() ([]*model.User, error) {
|
|
var out []*model.User
|
|
uri := fmt.Sprintf(pathUsers, c.base)
|
|
err := c.get(uri, &out)
|
|
return out, err
|
|
}
|
|
|
|
// UserPost creates a new user account.
|
|
func (c *client) UserPost(in *model.User) (*model.User, error) {
|
|
out := new(model.User)
|
|
uri := fmt.Sprintf(pathUsers, c.base)
|
|
err := c.post(uri, in, out)
|
|
return out, err
|
|
}
|
|
|
|
// UserPatch updates a user account.
|
|
func (c *client) UserPatch(in *model.User) (*model.User, error) {
|
|
out := new(model.User)
|
|
uri := fmt.Sprintf(pathUser, c.base, in.Login)
|
|
err := c.patch(uri, in, out)
|
|
return out, err
|
|
}
|
|
|
|
// UserDel deletes a user account.
|
|
func (c *client) UserDel(login string) error {
|
|
uri := fmt.Sprintf(pathUser, c.base, login)
|
|
err := c.delete(uri)
|
|
return err
|
|
}
|
|
|
|
// Repo returns a repository by name.
|
|
func (c *client) Repo(owner string, name string) (*model.Repo, error) {
|
|
out := new(model.Repo)
|
|
uri := fmt.Sprintf(pathRepo, c.base, owner, name)
|
|
err := c.get(uri, out)
|
|
return out, err
|
|
}
|
|
|
|
// RepoList returns a list of all repositories to which
|
|
// the user has explicit access in the host system.
|
|
func (c *client) RepoList() ([]*model.Repo, error) {
|
|
var out []*model.Repo
|
|
uri := fmt.Sprintf(pathRepos, c.base)
|
|
err := c.get(uri, &out)
|
|
return out, err
|
|
}
|
|
|
|
// RepoPost activates a repository.
|
|
func (c *client) RepoPost(owner string, name string) (*model.Repo, error) {
|
|
out := new(model.Repo)
|
|
uri := fmt.Sprintf(pathRepo, c.base, owner, name)
|
|
err := c.post(uri, nil, out)
|
|
return out, err
|
|
}
|
|
|
|
// RepoChow updates a repository owner.
|
|
func (c *client) RepoChown(owner string, name string) (*model.Repo, error) {
|
|
out := new(model.Repo)
|
|
uri := fmt.Sprintf(pathChown, c.base, owner, name)
|
|
err := c.post(uri, nil, out)
|
|
return out, err
|
|
}
|
|
|
|
// RepoPatch updates a repository.
|
|
func (c *client) RepoPatch(in *model.Repo) (*model.Repo, error) {
|
|
out := new(model.Repo)
|
|
uri := fmt.Sprintf(pathRepo, c.base, in.Owner, in.Name)
|
|
err := c.patch(uri, in, out)
|
|
return out, err
|
|
}
|
|
|
|
// RepoDel deletes a repository.
|
|
func (c *client) RepoDel(owner, name string) error {
|
|
uri := fmt.Sprintf(pathRepo, c.base, owner, name)
|
|
err := c.delete(uri)
|
|
return err
|
|
}
|
|
|
|
// Build returns a repository build by number.
|
|
func (c *client) Build(owner, name string, num int) (*model.Build, error) {
|
|
out := new(model.Build)
|
|
uri := fmt.Sprintf(pathBuild, c.base, owner, name, num)
|
|
err := c.get(uri, out)
|
|
return out, err
|
|
}
|
|
|
|
// Build returns the latest repository build by branch.
|
|
func (c *client) BuildLast(owner, name, branch string) (*model.Build, error) {
|
|
out := new(model.Build)
|
|
uri := fmt.Sprintf(pathBuild, c.base, owner, name, "latest")
|
|
if len(branch) != 0 {
|
|
uri += "?branch=" + branch
|
|
}
|
|
err := c.get(uri, out)
|
|
return out, err
|
|
}
|
|
|
|
// BuildList returns a list of recent builds for the
|
|
// the specified repository.
|
|
func (c *client) BuildList(owner, name string) ([]*model.Build, error) {
|
|
var out []*model.Build
|
|
uri := fmt.Sprintf(pathBuilds, c.base, owner, name)
|
|
err := c.get(uri, &out)
|
|
return out, err
|
|
}
|
|
|
|
// BuildQueue returns a list of enqueued builds.
|
|
func (c *client) BuildQueue() ([]*model.Feed, error) {
|
|
var out []*model.Feed
|
|
uri := fmt.Sprintf(pathBuildQueue, c.base)
|
|
err := c.get(uri, &out)
|
|
return out, err
|
|
}
|
|
|
|
// BuildStart re-starts a stopped build.
|
|
func (c *client) BuildStart(owner, name string, num int, params map[string]string) (*model.Build, error) {
|
|
out := new(model.Build)
|
|
val := parseToQueryParams(params)
|
|
uri := fmt.Sprintf(pathBuild+"?"+val.Encode(), c.base, owner, name, num)
|
|
err := c.post(uri, nil, out)
|
|
return out, err
|
|
}
|
|
|
|
// BuildStop cancels the running job.
|
|
func (c *client) BuildStop(owner, name string, num, job int) error {
|
|
uri := fmt.Sprintf(pathJob, c.base, owner, name, num, job)
|
|
err := c.delete(uri)
|
|
return err
|
|
}
|
|
|
|
// BuildFork re-starts a stopped build with a new build number,
|
|
// preserving the prior history.
|
|
func (c *client) BuildFork(owner, name string, num int, params map[string]string) (*model.Build, error) {
|
|
out := new(model.Build)
|
|
val := parseToQueryParams(params)
|
|
val.Set("fork", "true")
|
|
uri := fmt.Sprintf(pathBuild+"?"+val.Encode(), c.base, owner, name, num)
|
|
err := c.post(uri, nil, out)
|
|
return out, err
|
|
}
|
|
|
|
// BuildLogs returns the build logs for the specified job.
|
|
func (c *client) BuildLogs(owner, name string, num, job int) (io.ReadCloser, error) {
|
|
uri := fmt.Sprintf(pathLog, c.base, owner, name, num, job)
|
|
return stream(c.client, uri, "GET", nil, nil)
|
|
}
|
|
|
|
// Deploy triggers a deployment for an existing build using the
|
|
// specified target environment.
|
|
func (c *client) Deploy(owner, name string, num int, env string, params map[string]string) (*model.Build, error) {
|
|
out := new(model.Build)
|
|
val := parseToQueryParams(params)
|
|
val.Set("fork", "true")
|
|
val.Set("event", "deployment")
|
|
val.Set("deploy_to", env)
|
|
uri := fmt.Sprintf(pathBuild+"?"+val.Encode(), c.base, owner, name, num)
|
|
err := c.post(uri, nil, out)
|
|
return out, err
|
|
}
|
|
|
|
// SecretList returns a list of a repository secrets.
|
|
func (c *client) SecretList(owner, name string) ([]*model.Secret, error) {
|
|
var out []*model.Secret
|
|
uri := fmt.Sprintf(pathRepoSecrets, c.base, owner, name)
|
|
err := c.get(uri, &out)
|
|
return out, err
|
|
}
|
|
|
|
// SecretPost create or updates a repository secret.
|
|
func (c *client) SecretPost(owner, name string, secret *model.Secret) error {
|
|
uri := fmt.Sprintf(pathRepoSecrets, c.base, owner, name)
|
|
return c.post(uri, secret, nil)
|
|
}
|
|
|
|
// SecretDel deletes a named repository secret.
|
|
func (c *client) SecretDel(owner, name, secret string) error {
|
|
uri := fmt.Sprintf(pathRepoSecret, c.base, owner, name, secret)
|
|
return c.delete(uri)
|
|
}
|
|
|
|
// TeamSecretList returns a list of a repository secrets.
|
|
func (c *client) TeamSecretList(team string) ([]*model.Secret, error) {
|
|
var out []*model.Secret
|
|
uri := fmt.Sprintf(pathTeamSecrets, c.base, team)
|
|
err := c.get(uri, &out)
|
|
return out, err
|
|
}
|
|
|
|
// TeamSecretPost create or updates a repository secret.
|
|
func (c *client) TeamSecretPost(team string, secret *model.Secret) error {
|
|
uri := fmt.Sprintf(pathTeamSecrets, c.base, team)
|
|
return c.post(uri, secret, nil)
|
|
}
|
|
|
|
// TeamSecretDel deletes a named repository secret.
|
|
func (c *client) TeamSecretDel(team, secret string) error {
|
|
uri := fmt.Sprintf(pathTeamSecret, c.base, team, secret)
|
|
return c.delete(uri)
|
|
}
|
|
|
|
// Sign returns a cryptographic signature for the input string.
|
|
func (c *client) Sign(owner, name string, in []byte) ([]byte, error) {
|
|
uri := fmt.Sprintf(pathSign, c.base, owner, name)
|
|
rc, err := stream(c.client, uri, "POST", in, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rc.Close()
|
|
return ioutil.ReadAll(rc)
|
|
}
|
|
|
|
// AgentList returns a list of build agents.
|
|
func (c *client) AgentList() ([]*model.Agent, error) {
|
|
var out []*model.Agent
|
|
uri := fmt.Sprintf(pathAgent, c.base)
|
|
err := c.get(uri, &out)
|
|
return out, err
|
|
}
|
|
|
|
//
|
|
// below items for Queue (internal use only)
|
|
//
|
|
|
|
// Pull pulls work from the server queue.
|
|
func (c *client) Pull(os, arch string) (*queue.Work, error) {
|
|
out := new(queue.Work)
|
|
uri := fmt.Sprintf(pathPull, c.base, os, arch)
|
|
err := c.post(uri, nil, out)
|
|
return out, err
|
|
}
|
|
|
|
// Push pushes an update to the server.
|
|
func (c *client) Push(p *queue.Work) error {
|
|
uri := fmt.Sprintf(pathPush, c.base, p.Job.ID)
|
|
err := c.post(uri, p, nil)
|
|
return err
|
|
}
|
|
|
|
// Ping pings the server.
|
|
func (c *client) Ping() error {
|
|
uri := fmt.Sprintf(pathPing, c.base)
|
|
err := c.post(uri, nil, nil)
|
|
return err
|
|
}
|
|
|
|
// Stream streams the build logs to the server.
|
|
func (c *client) Stream(id int64, rc io.ReadCloser) error {
|
|
uri := fmt.Sprintf(pathStream, c.base, id)
|
|
err := c.post(uri, rc, nil)
|
|
|
|
return err
|
|
}
|
|
|
|
// LogPost sends the full build logs to the server.
|
|
func (c *client) LogPost(id int64, rc io.ReadCloser) error {
|
|
uri := fmt.Sprintf(pathLogs, c.base, id)
|
|
return c.post(uri, rc, nil)
|
|
}
|
|
|
|
// StreamWriter implements a special writer for streaming log entries to the
|
|
// central Drone server. The standard implementation is the gorilla.Connection.
|
|
type StreamWriter interface {
|
|
Close() error
|
|
WriteJSON(interface{}) error
|
|
}
|
|
|
|
// LogStream streams the build logs to the server.
|
|
func (c *client) LogStream(id int64) (StreamWriter, error) {
|
|
rawurl := fmt.Sprintf(pathLogsAuth, c.base, id, c.token)
|
|
uri, err := url.Parse(rawurl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if uri.Scheme == "https" {
|
|
uri.Scheme = "wss"
|
|
} else {
|
|
uri.Scheme = "ws"
|
|
}
|
|
|
|
// TODO need TLS client configuration
|
|
|
|
conn, _, err := websocket.DefaultDialer.Dial(uri.String(), nil)
|
|
return conn, err
|
|
}
|
|
|
|
// Wait watches and waits for the build to cancel or finish.
|
|
func (c *client) Wait(id int64) *Wait {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
return &Wait{id, c, ctx, cancel}
|
|
}
|
|
|
|
type Wait struct {
|
|
id int64
|
|
client *client
|
|
|
|
ctx context.Context
|
|
cancel context.CancelFunc
|
|
}
|
|
|
|
func (w *Wait) Done() (*model.Job, error) {
|
|
uri := fmt.Sprintf(pathWait, w.client.base, w.id)
|
|
req, err := w.client.createRequest(uri, "POST", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
res, err := ctxhttp.Do(w.ctx, w.client.client, req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer res.Body.Close()
|
|
job := &model.Job{}
|
|
err = json.NewDecoder(res.Body).Decode(&job)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return job, nil
|
|
}
|
|
|
|
func (w *Wait) Cancel() {
|
|
w.cancel()
|
|
}
|
|
|
|
//
|
|
// http request helper functions
|
|
//
|
|
|
|
// helper function for making an http GET request.
|
|
func (c *client) get(rawurl string, out interface{}) error {
|
|
return c.do(rawurl, "GET", nil, out)
|
|
}
|
|
|
|
// helper function for making an http POST request.
|
|
func (c *client) post(rawurl string, in, out interface{}) error {
|
|
return c.do(rawurl, "POST", in, out)
|
|
}
|
|
|
|
// helper function for making an http PUT request.
|
|
func (c *client) put(rawurl string, in, out interface{}) error {
|
|
return c.do(rawurl, "PUT", in, out)
|
|
}
|
|
|
|
// helper function for making an http PATCH request.
|
|
func (c *client) patch(rawurl string, in, out interface{}) error {
|
|
return c.do(rawurl, "PATCH", in, out)
|
|
}
|
|
|
|
// helper function for making an http DELETE request.
|
|
func (c *client) delete(rawurl string) error {
|
|
return c.do(rawurl, "DELETE", nil, nil)
|
|
}
|
|
|
|
// helper function to make an http request
|
|
func (c *client) do(rawurl, method string, in, out interface{}) error {
|
|
// executes the http request and returns the body as
|
|
// and io.ReadCloser
|
|
body, err := c.open(rawurl, method, in, out)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer body.Close()
|
|
|
|
// if a json response is expected, parse and return
|
|
// the json response.
|
|
if out != nil {
|
|
return json.NewDecoder(body).Decode(out)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// helper function to open an http request
|
|
func (c *client) open(rawurl, method string, in, out interface{}) (io.ReadCloser, error) {
|
|
uri, err := url.Parse(rawurl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// creates a new http request to bitbucket.
|
|
req, err := http.NewRequest(method, uri.String(), nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// if we are posting or putting data, we need to
|
|
// write it to the body of the request.
|
|
if in != nil {
|
|
rc, ok := in.(io.ReadCloser)
|
|
if ok {
|
|
req.Body = rc
|
|
req.Header.Set("Content-Type", "plain/text")
|
|
} else {
|
|
inJson, err := json.Marshal(in)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
buf := bytes.NewBuffer(inJson)
|
|
req.Body = ioutil.NopCloser(buf)
|
|
|
|
req.ContentLength = int64(len(inJson))
|
|
req.Header.Set("Content-Length", strconv.Itoa(len(inJson)))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
}
|
|
}
|
|
resp, err := c.client.Do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.StatusCode > http.StatusPartialContent {
|
|
defer resp.Body.Close()
|
|
out, _ := ioutil.ReadAll(resp.Body)
|
|
return nil, fmt.Errorf("client error %d: %s", resp.StatusCode, string(out))
|
|
}
|
|
return resp.Body, nil
|
|
}
|
|
|
|
// createRequest is a helper function that builds an http.Request.
|
|
func (c *client) createRequest(rawurl, method string, in interface{}) (*http.Request, error) {
|
|
uri, err := url.Parse(rawurl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// if we are posting or putting data, we need to
|
|
// write it to the body of the request.
|
|
var buf io.ReadWriter
|
|
if in != nil {
|
|
buf = new(bytes.Buffer)
|
|
err := json.NewEncoder(buf).Encode(in)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
// creates a new http request to bitbucket.
|
|
req, err := http.NewRequest(method, uri.String(), buf)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if in != nil {
|
|
req.Header.Set("Content-Type", "application/json")
|
|
}
|
|
return req, nil
|
|
}
|
|
|
|
// parseToQueryParams parses a map of strings and returns url.Values
|
|
func parseToQueryParams(p map[string]string) url.Values {
|
|
values := url.Values{}
|
|
for k, v := range p {
|
|
values.Add(k, v)
|
|
}
|
|
return values
|
|
}
|