woodpecker/cmd/drone-build/run.go
2015-09-02 16:14:25 -07:00

167 lines
3.9 KiB
Go

package main
import (
"fmt"
"github.com/drone/drone/Godeps/_workspace/src/github.com/samalba/dockerclient"
common "github.com/drone/drone/pkg/types"
"github.com/drone/drone/pkg/yaml"
"github.com/drone/drone/pkg/yaml/inject"
)
type Context struct {
System *common.System `json:"system"`
Repo *common.Repo `json:"repo"`
Build *common.Build `json:"build"`
Job *common.Job `json:"job"`
Yaml []byte `json:"yaml"`
// todo re-factor these
Clone *common.Clone `json:"clone"`
Keys *common.Keypair `json:"keys"`
Netrc *common.Netrc `json:"netrc"`
Conf *common.Config `json:"-"`
infos []*dockerclient.ContainerInfo
client dockerclient.Client
}
func setup(c *Context) error {
var err error
var opts = parser.Opts{
Network: false,
Privileged: false,
Volumes: false,
Caching: false,
Whitelist: c.System.Plugins,
}
// if repository is trusted the build may specify
// custom volumes, networking and run in trusted mode.
if c.Repo.Trusted {
opts.Network = true
opts.Privileged = true
opts.Volumes = true
opts.Caching = true
}
// if repository is private enable caching
if c.Repo.Private {
opts.Caching = true
}
// inject the matrix parameters into the yaml
injected := inject.Inject(string(c.Yaml), c.Job.Environment)
c.Conf, err = parser.ParseSingle(injected, &opts, c.Repo)
if err != nil {
return err
}
// and append the matrix parameters as environment
// variables for the build
for k, v := range c.Job.Environment {
env := k + "=" + v
c.Conf.Build.Environment = append(c.Conf.Build.Environment, env)
}
// and append drone, jenkins, travis and other
// environment variables that may be of use.
for k, v := range toEnv(c) {
env := k + "=" + v
c.Conf.Build.Environment = append(c.Conf.Build.Environment, env)
}
pathv, ok := c.Conf.Clone.Config["path"]
if ok {
path, ok := pathv.(string)
if ok {
c.Clone.Dir = path
return nil
}
}
return fmt.Errorf("Workspace path not found")
}
type execFunc func(c *Context) (int, error)
func execClone(c *Context) (int, error) {
conf := toContainerConfig(c.Conf.Clone)
conf.Cmd = toCommand(c, c.Conf.Clone)
info, err := run(c.client, conf, c.Conf.Clone.Pull)
if err != nil {
return 255, err
}
return info.State.ExitCode, nil
}
func execBuild(c *Context) (int, error) {
conf := toContainerConfig(c.Conf.Build)
conf.Entrypoint = []string{"/bin/sh", "-e"}
conf.Cmd = []string{"/drone/bin/build.sh"}
info, err := run(c.client, conf, c.Conf.Build.Pull)
if err != nil {
return 255, err
}
return info.State.ExitCode, nil
}
func execSetup(c *Context) (int, error) {
conf := toContainerConfig(c.Conf.Setup)
conf.Cmd = toCommand(c, c.Conf.Setup)
info, err := run(c.client, conf, c.Conf.Setup.Pull)
if err != nil {
return 255, err
}
return info.State.ExitCode, nil
}
func execDeploy(c *Context) (int, error) {
return runSteps(c, c.Conf.Deploy)
}
func execPublish(c *Context) (int, error) {
return runSteps(c, c.Conf.Publish)
}
func execNotify(c *Context) (int, error) {
return runSteps(c, c.Conf.Notify)
}
func execCompose(c *Context) (int, error) {
for _, step := range c.Conf.Compose {
conf := toContainerConfig(step)
_, err := daemon(c.client, conf, step.Pull)
if err != nil {
return 0, err
}
}
return 0, nil
}
func runSteps(c *Context, steps map[string]*common.Step) (int, error) {
for _, step := range steps {
// verify the step matches the branch
// and other specifications
if step.Condition != nil {
if !step.Condition.MatchOwner(c.Repo.Owner) ||
!step.Condition.MatchBranch(c.Clone.Branch) ||
!step.Condition.MatchMatrix(c.Job.Environment) {
continue
}
}
conf := toContainerConfig(step)
conf.Cmd = toCommand(c, step)
// append global environment variables
conf.Env = append(conf.Env, c.System.Globals...)
info, err := run(c.client, conf, step.Pull)
if err != nil {
return 255, err
} else if info.State.ExitCode != 0 {
return info.State.ExitCode, nil
}
}
return 0, nil
}