package builtin import ( "errors" "io" "io/ioutil" "time" "github.com/drone/drone/common" "github.com/samalba/dockerclient" ) var ( ErrTimeout = errors.New("Timeout") ErrLogging = errors.New("Logs not available") ) var ( // options to fetch the stdout and stderr logs logOpts = &dockerclient.LogOptions{ Stdout: true, Stderr: true, } // options to fetch the stdout and stderr logs // by tailing the output. logOptsTail = &dockerclient.LogOptions{ Follow: true, Stdout: true, Stderr: true, } ) var ( // name of the build agent container. DefaultAgent = "drone/drone-build:latest" // default name of the build agent executable DefaultEntrypoint = []string{"/go/bin/drone-build"} // default argument to invoke build steps DefaultBuildArgs = []string{"--build", "--clone", "--publish", "--deploy"} // default arguments to invoke notify steps DefaultNotifyArgs = []string{"--notify"} // default arguments to invoke notify steps DefaultNotifyTimeout = time.Minute * 5 ) type work struct { Repo *common.Repo `json:"repo"` Commit *common.Commit `json:"commit"` Build *common.Build `json:"build"` Keys *common.Keypair `json:"keys"` Netrc *common.Netrc `json:"netrc"` Yaml []byte `json:"yaml"` } type worker struct { timeout time.Duration client dockerclient.Client build *dockerclient.ContainerInfo notify *dockerclient.ContainerInfo } func newWorker(client dockerclient.Client) *worker { return newWorkerTimeout(client, 60) // default 60 minute timeout } func newWorkerTimeout(client dockerclient.Client, timeout int64) *worker { return &worker{ timeout: time.Duration(timeout) * time.Minute, client: client, } } // Build executes the clone, build and deploy steps. func (w *worker) Build(name string, stdin []byte) (_ int, err error) { // the command line arguments passed into the // build agent container. args := DefaultBuildArgs args = append(args, "--") args = append(args, string(stdin)) conf := &dockerclient.ContainerConfig{ Image: DefaultAgent, Entrypoint: DefaultEntrypoint, Cmd: args, HostConfig: dockerclient.HostConfig{ Binds: []string{"/var/run/docker.sock:/var/run/docker.sock"}, }, Volumes: map[string]struct{}{ "/drone": struct{}{}, "/var/run/docker.sock": struct{}{}, }, } w.build, err = run(w.client, conf, name, w.timeout) if err != nil { return 1, err } return w.build.State.ExitCode, err } // Notify executes the notification steps. func (w *worker) Notify(stdin []byte) error { // use the affinity parameter in case we are // using Docker swarm as a backend. environment := []string{"affinity:container==" + w.build.Id} // the build container is acting as an ambassador container // with a shared filesystem . volume := []string{w.build.Id} // the command line arguments passed into the // build agent container. args := DefaultNotifyArgs args = append(args, "--") args = append(args, string(stdin)) conf := &dockerclient.ContainerConfig{ Image: DefaultAgent, Entrypoint: DefaultEntrypoint, Cmd: args, Env: environment, HostConfig: dockerclient.HostConfig{ VolumesFrom: volume, }, } var err error w.notify, err = run(w.client, conf, "", DefaultNotifyTimeout) return err } // Logs returns a multi-reader that fetches the logs // from the build and deploy agents. func (w *worker) Logs() (io.ReadCloser, error) { if w.build == nil { return nil, ErrLogging } return w.client.ContainerLogs(w.build.Id, logOpts) } // Remove stops and removes the build, deploy and // notification agents created for the build task. func (w *worker) Remove() { if w.notify != nil { w.client.KillContainer(w.notify.Id, "9") w.client.RemoveContainer(w.notify.Id, true, true) } if w.build != nil { w.client.KillContainer(w.build.Id, "9") w.client.RemoveContainer(w.build.Id, true, true) } } // run is a helper function that creates and starts a container, // blocking until either complete or the timeout is reached. If // the timeout is reached an ErrTimeout is returned, else the // container info is returned. func run(client dockerclient.Client, conf *dockerclient.ContainerConfig, name string, timeout time.Duration) (*dockerclient.ContainerInfo, error) { // attempts to create the contianer id, err := client.CreateContainer(conf, name) if err != nil { // and pull the image and re-create if that fails client.PullImage(conf.Image, nil) id, err = client.CreateContainer(conf, name) // make sure the container is removed in // the event of a creation error. if err != nil && len(id) != 0 { client.RemoveContainer(id, true, true) } if err != nil { return nil, err } } // ensures the container is always stopped // and ready to be removed. defer func() { client.StopContainer(id, 5) client.KillContainer(id, "9") }() // fetches the container information. info, err := client.InspectContainer(id) if err != nil { return nil, err } // channel listening for errors while the // container is running async. errc := make(chan error, 1) infoc := make(chan *dockerclient.ContainerInfo, 1) go func() { // starts the container err := client.StartContainer(id, &conf.HostConfig) if err != nil { errc <- err return } // blocks and waits for the container to finish // by streaming the logs (to /dev/null). Ideally // we could use the `wait` function instead rc, err := client.ContainerLogs(id, logOptsTail) if err != nil { errc <- err return } io.Copy(ioutil.Discard, rc) rc.Close() // fetches the container information info, err := client.InspectContainer(id) if err != nil { errc <- err return } infoc <- info }() select { case info := <-infoc: return info, nil case err := <-errc: return info, err case <-time.After(timeout): return info, ErrTimeout } }