harness-drone/server/builds.go

package server

import (
	"fmt"
	"io"
	"net/http"
	"net/url"
	"strconv"
	"time"

	"github.com/drone/drone/common"
	"github.com/drone/drone/parser/inject"
	"github.com/drone/drone/queue"
	"github.com/gin-gonic/gin"
	"github.com/gin-gonic/gin/binding"
)

// GetBuild accepts a request to retrieve a build
// from the datastore for the given repository and
// build number.
//
//     GET /api/builds/:owner/:name/:number
//
func GetBuild(c *gin.Context) {
	store := ToDatastore(c)
	repo := ToRepo(c)
	num, err := strconv.Atoi(c.Params.ByName("number"))
	if err != nil {
		c.Fail(400, err)
		return
	}
	build, err := store.Build(repo.FullName, num)
	if err != nil {
		c.Fail(404, err)
	} else {
		c.JSON(200, build)
	}
}

// GetBuilds accepts a request to retrieve a list
// of builds from the datastore for the given repository.
//
//     GET /api/builds/:owner/:name
//
func GetBuilds(c *gin.Context) {
	store := ToDatastore(c)
	repo := ToRepo(c)
	builds, err := store.BuildList(repo.FullName)
	if err != nil {
		c.Fail(404, err)
	} else {
		c.JSON(200, builds)
	}
}

// GetBuildLogs accepts a request to retrieve logs from the
// datastore for the given repository, build and task
// number.
//
//     GET /api/repos/:owner/:name/logs/:number/:task
//
func GetBuildLogs(c *gin.Context) {
	store := ToDatastore(c)
	repo := ToRepo(c)
	full, _ := strconv.ParseBool(c.Params.ByName("full"))
	build, _ := strconv.Atoi(c.Params.ByName("number"))
	task, _ := strconv.Atoi(c.Params.ByName("task"))

	r, err := store.LogReader(repo.FullName, build, task)
	if err != nil {
		c.Fail(404, err)
	} else if full {
		io.Copy(c.Writer, r)
	} else {
		io.Copy(c.Writer, io.LimitReader(r, 2000000))
	}
}

// PostBuildStatus accepts a request to create a new build
// status. The created user status is returned in JSON
// format if successful.
//
//     POST /api/repos/:owner/:name/status/:number
//
func PostBuildStatus(c *gin.Context) {
	store := ToDatastore(c)
	repo := ToRepo(c)
	num, err := strconv.Atoi(c.Params.ByName("number"))
	if err != nil {
		c.Fail(400, err)
		return
	}
	in := &common.Status{}
	if !c.BindWith(in, binding.JSON) {
		c.AbortWithStatus(400)
		return
	}
	if err := store.SetBuildStatus(repo.Name, num, in); err != nil {
		c.Fail(400, err)
	} else {
		c.JSON(201, in)
	}
}

// RunBuild accepts a request to restart an existing build.
//
//     POST /api/builds/:owner/:name/builds/:number
//
func RunBuild(c *gin.Context) {
	remote := ToRemote(c)
	store := ToDatastore(c)
	queue_ := ToQueue(c)
	repo := ToRepo(c)
	num, err := strconv.Atoi(c.Params.ByName("number"))
	if err != nil {
		c.Fail(400, err)
		return
	}
	build, err := store.Build(repo.FullName, num)
	if err != nil {
		c.Fail(404, err)
		return
	}

	keys, err := store.RepoKeypair(repo.FullName)
	if err != nil {
		c.Fail(404, err)
		return
	}

	user, err := store.User(repo.User.Login)
	if err != nil {
		c.Fail(404, err)
		return
	}

	// must not restart a running build
	if build.State == common.StatePending || build.State == common.StateRunning {
		c.AbortWithStatus(409)
		return
	}

	build.State = common.StatePending
	build.Started = 0
	build.Finished = 0
	build.Duration = 0
	build.Statuses = []*common.Status{}
	for _, task := range build.Tasks {
		task.State = common.StatePending
		task.Started = 0
		task.Finished = 0
		task.ExitCode = 0
	}

	err = store.SetBuild(repo.FullName, build)
	if err != nil {
		c.Fail(500, err)
		return
	}

	netrc, err := remote.Netrc(user)
	if err != nil {
		c.Fail(500, err)
		return
	}

	// featch the .drone.yml file from the database
	raw, err := remote.Script(user, repo, build)
	if err != nil {
		c.Fail(404, err)
		return
	}

	// inject any private parameters into the .drone.yml
	params, _ := store.RepoParams(repo.FullName)
	if params != nil && len(params) != 0 {
		raw = []byte(inject.InjectSafe(string(raw), params))
	}

	queue_.Publish(&queue.Work{
		User:  user,
		Repo:  repo,
		Build: build,
		Keys:  keys,
		Netrc: netrc,
		Yaml:  raw,
	})

	c.JSON(202, build)
}

// KillBuild accepts a request to kill a running build.
//
//     DELETE /api/builds/:owner/:name/builds/:number
//
func KillBuild(c *gin.Context) {
	queue := ToQueue(c)
	store := ToDatastore(c)
	repo := ToRepo(c)
	num, err := strconv.Atoi(c.Params.ByName("number"))
	if err != nil {
		c.Fail(400, err)
		return
	}
	build, err := store.Build(repo.FullName, num)
	if err != nil {
		c.Fail(404, err)
		return
	}

	// must not restart a running build
	if build.State != common.StatePending && build.State != common.StateRunning {
		c.Fail(409, err)
		return
	}

	// remove from the queue if exists
	for _, item := range queue.Items() {
		if item.Repo.FullName == repo.FullName && item.Build.Number == build.Number {
			queue.Remove(item)
			break
		}
	}

	build.State = common.StateKilled
	build.Finished = time.Now().Unix()
	if build.Started == 0 {
		build.Started = build.Finished
	}
	build.Duration = build.Finished - build.Started
	for _, task := range build.Tasks {
		if task.State != common.StatePending && task.State != common.StateRunning {
			continue
		}
		task.State = common.StateKilled
		task.Started = build.Started
		task.Finished = build.Finished
	}
	err = store.SetBuild(repo.FullName, build)
	if err != nil {
		c.Fail(500, err)
		return
	}

	// get the agent from the repository so we can
	// notify the agent to kill the build.
	agent, err := store.BuildAgent(repo.FullName, build.Number)
	if err != nil {
		c.JSON(200, build)
		return
	}
	url_, _ := url.Parse("http://" + agent.Addr)
	url_.Path = fmt.Sprintf("/cancel/%s/%v", repo.FullName, build.Number)
	resp, err := http.Post(url_.String(), "application/json", nil)
	if err != nil {
		c.Fail(500, err)
		return
	}
	defer resp.Body.Close()

	c.JSON(200, build)
}