harness-drone/pkg/runner/builtin/updater.go

package builtin

import (
	"encoding/json"
	"fmt"
	"io"

	"github.com/drone/drone/pkg/bus"
	"github.com/drone/drone/pkg/remote"
	"github.com/drone/drone/pkg/store"
	common "github.com/drone/drone/pkg/types"
)

type Updater interface {
	SetCommit(*common.User, *common.Repo, *common.Commit) error
	SetBuild(*common.Repo, *common.Commit, *common.Build) error
	SetLogs(*common.Repo, *common.Commit, *common.Build, io.ReadCloser) error
}

// NewUpdater returns an implementation of the Updater interface
// that directly modifies the database and sends messages to the bus.
func NewUpdater(bus bus.Bus, store store.Store, rem remote.Remote) Updater {
	return &updater{bus, store, rem}
}

type updater struct {
	bus    bus.Bus
	store  store.Store
	remote remote.Remote
}

func (u *updater) SetCommit(user *common.User, r *common.Repo, c *common.Commit) error {
	err := u.store.SetCommit(c)
	if err != nil {
		return err
	}

	err = u.remote.Status(user, r, c)
	if err != nil {
		// log err
	}

	// we need this because builds coming from
	// a remote agent won't have the embedded
	// build list. we should probably just rethink
	// the messaging instead of this hack.
	if c.Builds == nil || len(c.Builds) == 0 {
		c.Builds, _ = u.store.BuildList(c)
	}

	msg, err := json.Marshal(c)
	if err != nil {
		return err
	}

	u.bus.Send(&bus.Event{
		Name: r.FullName,
		Kind: bus.EventRepo,
		Msg:  msg,
	})
	return nil
}

func (u *updater) SetBuild(r *common.Repo, c *common.Commit, b *common.Build) error {
	err := u.store.SetBuild(b)
	if err != nil {
		return err
	}

	// we need this because builds coming from
	// a remote agent won't have the embedded
	// build list. we should probably just rethink
	// the messaging instead of this hack.
	if c.Builds == nil || len(c.Builds) == 0 {
		c.Builds, _ = u.store.BuildList(c)
	}

	msg, err := json.Marshal(c)
	if err != nil {
		return err
	}

	u.bus.Send(&bus.Event{
		Name: r.FullName,
		Kind: bus.EventRepo,
		Msg:  msg,
	})
	return nil
}

func (u *updater) SetLogs(r *common.Repo, c *common.Commit, b *common.Build, rc io.ReadCloser) error {
	path := fmt.Sprintf("/logs/%s/%v/%v", r.FullName, c.Sequence, b.Sequence)
	return u.store.SetBlobReader(path, rc)
}