94 lines
2.1 KiB
Go
94 lines
2.1 KiB
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)
|
|
}
|