harness-drone/plugins/sender/builtin.go

package sender

import (
	"github.com/drone/drone/model"
)

type builtin struct {
	store model.SenderStore
	conf  model.ConfigStore
}

// New returns a new local gating service.
func New(store model.SenderStore, conf model.ConfigStore) model.SenderService {
	return &builtin{store, conf}
}

func (b *builtin) SenderAllowed(user *model.User, repo *model.Repo, build *model.Build, conf *model.Config) (bool, error) {
	if build.Event == model.EventPull && build.Sender != user.Login {
		// check to see if the configuration has already been used in an
		// existing build. If yes it is considered approved.
		if ok, _ := b.conf.ConfigFindApproved(conf); ok {
			return true, nil
		}
		// else check to see if the configuration is sent from a user
		// account that is a repositroy approver themselves.
		sender, err := b.store.SenderFind(repo, build.Sender)
		if err != nil || sender.Block {
			return false, nil
		}
	}
	return true, nil
}

func (b *builtin) SenderCreate(repo *model.Repo, sender *model.Sender) error {
	return b.store.SenderCreate(sender)
}

func (b *builtin) SenderUpdate(repo *model.Repo, sender *model.Sender) error {
	return b.store.SenderUpdate(sender)
}

func (b *builtin) SenderDelete(repo *model.Repo, login string) error {
	sender, err := b.store.SenderFind(repo, login)
	if err != nil {
		return err
	}
	return b.store.SenderDelete(sender)
}

func (b *builtin) SenderList(repo *model.Repo) ([]*model.Sender, error) {
	return b.store.SenderList(repo)
}