2ce566483b
Fixed returning the http(s) version of the clone link which had the username in it however should not to work with netrc Quick typo fix
292 lines
7.8 KiB
Go
292 lines
7.8 KiB
Go
package agent
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/url"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/drone/drone/build"
|
|
"github.com/drone/drone/model"
|
|
"github.com/drone/drone/queue"
|
|
"github.com/drone/drone/version"
|
|
"github.com/drone/drone/yaml"
|
|
"github.com/drone/drone/yaml/expander"
|
|
"github.com/drone/drone/yaml/transform"
|
|
)
|
|
|
|
type Logger interface {
|
|
Write(*build.Line)
|
|
}
|
|
|
|
type Agent struct {
|
|
Update UpdateFunc
|
|
Logger LoggerFunc
|
|
Engine build.Engine
|
|
Timeout time.Duration
|
|
Platform string
|
|
Namespace string
|
|
Disable []string
|
|
Escalate []string
|
|
Netrc []string
|
|
Local string
|
|
Pull bool
|
|
}
|
|
|
|
func (a *Agent) Poll() error {
|
|
|
|
// logrus.Infof("Starting build %s/%s#%d.%d",
|
|
// payload.Repo.Owner, payload.Repo.Name, payload.Build.Number, payload.Job.Number)
|
|
//
|
|
//
|
|
// logrus.Infof("Finished build %s/%s#%d.%d",
|
|
// payload.Repo.Owner, payload.Repo.Name, payload.Build.Number, payload.Job.Number)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *Agent) Run(payload *queue.Work, cancel <-chan bool) error {
|
|
|
|
payload.Job.Status = model.StatusRunning
|
|
payload.Job.Started = time.Now().Unix()
|
|
|
|
spec, err := a.prep(payload)
|
|
if err != nil {
|
|
payload.Job.Error = err.Error()
|
|
payload.Job.ExitCode = 255
|
|
payload.Job.Finished = payload.Job.Started
|
|
payload.Job.Status = model.StatusError
|
|
a.Update(payload)
|
|
return err
|
|
}
|
|
a.Update(payload)
|
|
err = a.exec(spec, payload, cancel)
|
|
|
|
if err != nil {
|
|
payload.Job.ExitCode = 255
|
|
payload.Job.Error = err.Error()
|
|
}
|
|
if exitErr, ok := err.(*build.ExitError); ok {
|
|
payload.Job.ExitCode = exitErr.Code
|
|
payload.Job.Error = "" // exit errors are already written to the log
|
|
}
|
|
|
|
payload.Job.Finished = time.Now().Unix()
|
|
|
|
switch payload.Job.ExitCode {
|
|
case 128, 130, 137:
|
|
payload.Job.Status = model.StatusKilled
|
|
case 0:
|
|
payload.Job.Status = model.StatusSuccess
|
|
default:
|
|
payload.Job.Status = model.StatusFailure
|
|
}
|
|
|
|
a.Update(payload)
|
|
|
|
return err
|
|
}
|
|
|
|
func (a *Agent) prep(w *queue.Work) (*yaml.Config, error) {
|
|
|
|
envs := toEnv(w)
|
|
w.Yaml = expander.ExpandString(w.Yaml, envs)
|
|
|
|
// inject the netrc file into the clone plugin if the repository is
|
|
// private and requires authentication.
|
|
var secrets []*model.Secret
|
|
if w.Verified {
|
|
secrets = append(secrets, w.Secrets...)
|
|
}
|
|
|
|
if w.Repo.IsPrivate {
|
|
secrets = append(secrets, &model.Secret{
|
|
Name: "DRONE_NETRC_USERNAME",
|
|
Value: w.Netrc.Login,
|
|
Images: []string{"*"},
|
|
Events: []string{"*"},
|
|
})
|
|
secrets = append(secrets, &model.Secret{
|
|
Name: "DRONE_NETRC_PASSWORD",
|
|
Value: w.Netrc.Password,
|
|
Images: []string{"*"},
|
|
Events: []string{"*"},
|
|
})
|
|
secrets = append(secrets, &model.Secret{
|
|
Name: "DRONE_NETRC_MACHINE",
|
|
Value: w.Netrc.Machine,
|
|
Images: []string{"*"},
|
|
Events: []string{"*"},
|
|
})
|
|
}
|
|
|
|
conf, err := yaml.ParseString(w.Yaml)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
src := "src"
|
|
if url, _ := url.Parse(w.Repo.Link); url != nil {
|
|
host, _, err := net.SplitHostPort(url.Host)
|
|
if err == nil {
|
|
url.Host = host
|
|
}
|
|
src = filepath.Join(src, url.Host, url.Path)
|
|
}
|
|
|
|
transform.Clone(conf, w.Repo.Kind)
|
|
transform.Environ(conf, envs)
|
|
transform.DefaultFilter(conf)
|
|
if w.BuildLast != nil {
|
|
transform.ChangeFilter(conf, w.BuildLast.Status)
|
|
}
|
|
|
|
transform.ImageSecrets(conf, secrets, w.Build.Event)
|
|
transform.Identifier(conf)
|
|
transform.WorkspaceTransform(conf, "/drone", src)
|
|
|
|
if err := transform.Check(conf, w.Repo.IsTrusted); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
transform.CommandTransform(conf)
|
|
transform.ImagePull(conf, a.Pull)
|
|
transform.ImageTag(conf)
|
|
transform.ImageName(conf)
|
|
transform.ImageNamespace(conf, a.Namespace)
|
|
transform.ImageEscalate(conf, a.Escalate)
|
|
transform.PluginParams(conf)
|
|
|
|
if a.Local != "" {
|
|
transform.PluginDisable(conf, a.Disable)
|
|
transform.ImageVolume(conf, []string{a.Local + ":" + conf.Workspace.Path})
|
|
}
|
|
|
|
transform.Pod(conf)
|
|
|
|
return conf, nil
|
|
}
|
|
|
|
func (a *Agent) exec(spec *yaml.Config, payload *queue.Work, cancel <-chan bool) error {
|
|
|
|
conf := build.Config{
|
|
Engine: a.Engine,
|
|
Buffer: 500,
|
|
}
|
|
|
|
pipeline := conf.Pipeline(spec)
|
|
defer pipeline.Teardown()
|
|
|
|
// setup the build environment
|
|
if err := pipeline.Setup(); err != nil {
|
|
return err
|
|
}
|
|
|
|
timeout := time.After(time.Duration(payload.Repo.Timeout) * time.Minute)
|
|
|
|
for {
|
|
select {
|
|
case <-pipeline.Done():
|
|
return pipeline.Err()
|
|
case <-cancel:
|
|
pipeline.Stop()
|
|
return fmt.Errorf("termination request received, build cancelled")
|
|
case <-timeout:
|
|
pipeline.Stop()
|
|
return fmt.Errorf("maximum time limit exceeded, build cancelled")
|
|
case <-time.After(a.Timeout):
|
|
pipeline.Stop()
|
|
return fmt.Errorf("terminal inactive for %v, build cancelled", a.Timeout)
|
|
case <-pipeline.Next():
|
|
|
|
// TODO(bradrydzewski) this entire block of code should probably get
|
|
// encapsulated in the pipeline.
|
|
status := model.StatusSuccess
|
|
if pipeline.Err() != nil {
|
|
status = model.StatusFailure
|
|
}
|
|
// updates the build status passed into each container. I realize this is
|
|
// a bit out of place and will work to resolve.
|
|
pipeline.Head().Environment["DRONE_BUILD_STATUS"] = status
|
|
|
|
if !pipeline.Head().Constraints.Match(
|
|
a.Platform,
|
|
payload.Build.Deploy,
|
|
payload.Build.Event,
|
|
payload.Build.Branch,
|
|
status, payload.Job.Environment) { // TODO: fix this whole section
|
|
|
|
pipeline.Skip()
|
|
} else {
|
|
pipeline.Exec()
|
|
}
|
|
case line := <-pipeline.Pipe():
|
|
a.Logger(line)
|
|
}
|
|
}
|
|
}
|
|
|
|
func toEnv(w *queue.Work) map[string]string {
|
|
envs := map[string]string{
|
|
"CI": "drone",
|
|
"DRONE": "true",
|
|
"DRONE_ARCH": "linux/amd64",
|
|
"DRONE_REPO": w.Repo.FullName,
|
|
"DRONE_REPO_SCM": w.Repo.Kind,
|
|
"DRONE_REPO_OWNER": w.Repo.Owner,
|
|
"DRONE_REPO_NAME": w.Repo.Name,
|
|
"DRONE_REPO_LINK": w.Repo.Link,
|
|
"DRONE_REPO_AVATAR": w.Repo.Avatar,
|
|
"DRONE_REPO_BRANCH": w.Repo.Branch,
|
|
"DRONE_REPO_PRIVATE": fmt.Sprintf("%v", w.Repo.IsPrivate),
|
|
"DRONE_REPO_TRUSTED": fmt.Sprintf("%v", w.Repo.IsTrusted),
|
|
"DRONE_REMOTE_URL": w.Repo.Clone,
|
|
"DRONE_COMMIT_SHA": w.Build.Commit,
|
|
"DRONE_COMMIT_REF": w.Build.Ref,
|
|
"DRONE_COMMIT_BRANCH": w.Build.Branch,
|
|
"DRONE_COMMIT_LINK": w.Build.Link,
|
|
"DRONE_COMMIT_MESSAGE": w.Build.Message,
|
|
"DRONE_COMMIT_AUTHOR": w.Build.Author,
|
|
"DRONE_COMMIT_AUTHOR_EMAIL": w.Build.Email,
|
|
"DRONE_COMMIT_AUTHOR_AVATAR": w.Build.Avatar,
|
|
"DRONE_BUILD_NUMBER": fmt.Sprintf("%d", w.Build.Number),
|
|
"DRONE_BUILD_EVENT": w.Build.Event,
|
|
"DRONE_BUILD_STATUS": w.Build.Status,
|
|
"DRONE_BUILD_LINK": fmt.Sprintf("%s/%s/%d", w.System.Link, w.Repo.FullName, w.Build.Number),
|
|
"DRONE_BUILD_CREATED": fmt.Sprintf("%d", w.Build.Created),
|
|
"DRONE_BUILD_STARTED": fmt.Sprintf("%d", w.Build.Started),
|
|
"DRONE_BUILD_FINISHED": fmt.Sprintf("%d", w.Build.Finished),
|
|
"DRONE_YAML_VERIFIED": fmt.Sprintf("%v", w.Verified),
|
|
"DRONE_YAML_SIGNED": fmt.Sprintf("%v", w.Signed),
|
|
"DRONE_BRANCH": w.Build.Branch,
|
|
"DRONE_COMMIT": w.Build.Commit,
|
|
"DRONE_VERSION": version.Version,
|
|
}
|
|
|
|
if w.Build.Event == model.EventTag {
|
|
envs["DRONE_TAG"] = strings.TrimPrefix(w.Build.Ref, "refs/tags/")
|
|
}
|
|
if w.Build.Event == model.EventPull {
|
|
envs["DRONE_PULL_REQUEST"] = pullRegexp.FindString(w.Build.Ref)
|
|
}
|
|
if w.Build.Event == model.EventDeploy {
|
|
envs["DRONE_DEPLOY_TO"] = w.Build.Deploy
|
|
}
|
|
|
|
if w.BuildLast != nil {
|
|
envs["DRONE_PREV_BUILD_STATUS"] = w.BuildLast.Status
|
|
envs["DRONE_PREV_BUILD_NUMBER"] = fmt.Sprintf("%v", w.BuildLast.Number)
|
|
envs["DRONE_PREV_COMMIT_SHA"] = w.BuildLast.Commit
|
|
}
|
|
|
|
// inject matrix values as environment variables
|
|
for key, val := range w.Job.Environment {
|
|
envs[key] = val
|
|
}
|
|
return envs
|
|
}
|
|
|
|
var pullRegexp = regexp.MustCompile("\\d+")
|