485 lines
11 KiB
Go
485 lines
11 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/drone/drone/agent"
|
|
"github.com/drone/drone/build/docker"
|
|
"github.com/drone/drone/model"
|
|
"github.com/drone/drone/queue"
|
|
"github.com/drone/drone/yaml"
|
|
|
|
"github.com/codegangsta/cli"
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
var execCmd = cli.Command{
|
|
Name: "exec",
|
|
Usage: "execute a local build",
|
|
Action: func(c *cli.Context) {
|
|
if err := exec(c); err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
},
|
|
Flags: []cli.Flag{
|
|
cli.BoolTFlag{
|
|
Name: "local",
|
|
Usage: "build from local directory",
|
|
EnvVar: "DRONE_LOCAL",
|
|
},
|
|
cli.StringSliceFlag{
|
|
Name: "plugin",
|
|
Usage: "plugin steps to enable",
|
|
EnvVar: "DRONE_PLUGIN_ENABLE",
|
|
},
|
|
cli.StringSliceFlag{
|
|
Name: "secret",
|
|
Usage: "build secrets in KEY=VALUE format",
|
|
EnvVar: "DRONE_SECRET",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "secrets-file",
|
|
Usage: "build secrets file in KEY=VALUE format",
|
|
EnvVar: "DRONE_SECRETS_FILE",
|
|
},
|
|
cli.StringSliceFlag{
|
|
Name: "matrix",
|
|
Usage: "build matrix in KEY=VALUE format",
|
|
EnvVar: "DRONE_MATRIX",
|
|
},
|
|
cli.DurationFlag{
|
|
Name: "timeout",
|
|
Usage: "build timeout",
|
|
Value: time.Hour,
|
|
EnvVar: "DRONE_TIMEOUT",
|
|
},
|
|
cli.DurationFlag{
|
|
Name: "timeout.inactivity",
|
|
Usage: "build timeout for inactivity",
|
|
Value: time.Minute * 15,
|
|
EnvVar: "DRONE_TIMEOUT_INACTIVITY",
|
|
},
|
|
cli.BoolFlag{
|
|
EnvVar: "DRONE_PLUGIN_PULL",
|
|
Name: "pull",
|
|
Usage: "always pull latest plugin images",
|
|
},
|
|
cli.StringFlag{
|
|
EnvVar: "DRONE_PLUGIN_NAMESPACE",
|
|
Name: "namespace",
|
|
Value: "plugins",
|
|
Usage: "default plugin image namespace",
|
|
},
|
|
cli.StringSliceFlag{
|
|
EnvVar: "DRONE_PLUGIN_PRIVILEGED",
|
|
Name: "privileged",
|
|
Usage: "plugins that require privileged mode",
|
|
Value: &cli.StringSlice{
|
|
"plugins/docker",
|
|
"plugins/docker:*",
|
|
"plugins/gcr",
|
|
"plugins/gcr:*",
|
|
"plugins/ecr",
|
|
"plugins/ecr:*",
|
|
},
|
|
},
|
|
|
|
// Docker daemon flags
|
|
|
|
cli.StringFlag{
|
|
EnvVar: "DOCKER_HOST",
|
|
Name: "docker-host",
|
|
Usage: "docker deamon address",
|
|
Value: "unix:///var/run/docker.sock",
|
|
},
|
|
cli.BoolFlag{
|
|
EnvVar: "DOCKER_TLS_VERIFY",
|
|
Name: "docker-tls-verify",
|
|
Usage: "docker daemon supports tlsverify",
|
|
},
|
|
cli.StringFlag{
|
|
EnvVar: "DOCKER_CERT_PATH",
|
|
Name: "docker-cert-path",
|
|
Usage: "docker certificate directory",
|
|
Value: "",
|
|
},
|
|
|
|
//
|
|
// Please note the below flags are mirrored in the plugin starter kit and
|
|
// should be kept synchronized.
|
|
// https://github.com/drone/drone-plugin-starter
|
|
//
|
|
|
|
cli.StringFlag{
|
|
Name: "repo.fullname",
|
|
Usage: "repository full name",
|
|
EnvVar: "DRONE_REPO",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "repo.owner",
|
|
Usage: "repository owner",
|
|
EnvVar: "DRONE_REPO_OWNER",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "repo.name",
|
|
Usage: "repository name",
|
|
EnvVar: "DRONE_REPO_NAME",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "repo.type",
|
|
Value: "git",
|
|
Usage: "repository type",
|
|
EnvVar: "DRONE_REPO_SCM",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "repo.link",
|
|
Usage: "repository link",
|
|
EnvVar: "DRONE_REPO_LINK",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "repo.avatar",
|
|
Usage: "repository avatar",
|
|
EnvVar: "DRONE_REPO_AVATAR",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "repo.branch",
|
|
Usage: "repository default branch",
|
|
EnvVar: "DRONE_REPO_BRANCH",
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "repo.private",
|
|
Usage: "repository is private",
|
|
EnvVar: "DRONE_REPO_PRIVATE",
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "repo.trusted",
|
|
Usage: "repository is trusted",
|
|
EnvVar: "DRONE_REPO_TRUSTED",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "remote.url",
|
|
Usage: "git remote url",
|
|
EnvVar: "DRONE_REMOTE_URL",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.sha",
|
|
Usage: "git commit sha",
|
|
EnvVar: "DRONE_COMMIT_SHA",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.ref",
|
|
Value: "refs/heads/master",
|
|
Usage: "git commit ref",
|
|
EnvVar: "DRONE_COMMIT_REF",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.branch",
|
|
Value: "master",
|
|
Usage: "git commit branch",
|
|
EnvVar: "DRONE_COMMIT_BRANCH",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.message",
|
|
Usage: "git commit message",
|
|
EnvVar: "DRONE_COMMIT_MESSAGE",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.link",
|
|
Usage: "git commit link",
|
|
EnvVar: "DRONE_COMMIT_LINK",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.author.name",
|
|
Usage: "git author name",
|
|
EnvVar: "DRONE_COMMIT_AUTHOR",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.author.email",
|
|
Usage: "git author email",
|
|
EnvVar: "DRONE_COMMIT_AUTHOR_EMAIL",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.author.avatar",
|
|
Usage: "git author avatar",
|
|
EnvVar: "DRONE_COMMIT_AUTHOR_AVATAR",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "build.event",
|
|
Value: "push",
|
|
Usage: "build event",
|
|
EnvVar: "DRONE_BUILD_EVENT",
|
|
},
|
|
cli.IntFlag{
|
|
Name: "build.number",
|
|
Usage: "build number",
|
|
EnvVar: "DRONE_BUILD_NUMBER",
|
|
},
|
|
cli.IntFlag{
|
|
Name: "build.created",
|
|
Usage: "build created",
|
|
EnvVar: "DRONE_BUILD_CREATED",
|
|
},
|
|
cli.IntFlag{
|
|
Name: "build.started",
|
|
Usage: "build started",
|
|
EnvVar: "DRONE_BUILD_STARTED",
|
|
},
|
|
cli.IntFlag{
|
|
Name: "build.finished",
|
|
Usage: "build finished",
|
|
EnvVar: "DRONE_BUILD_FINISHED",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "build.status",
|
|
Usage: "build status",
|
|
Value: "success",
|
|
EnvVar: "DRONE_BUILD_STATUS",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "build.link",
|
|
Usage: "build link",
|
|
EnvVar: "DRONE_BUILD_LINK",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "build.deploy",
|
|
Usage: "build deployment target",
|
|
EnvVar: "DRONE_DEPLOY_TO",
|
|
},
|
|
cli.BoolTFlag{
|
|
Name: "yaml.verified",
|
|
Usage: "build yaml is verified",
|
|
EnvVar: "DRONE_YAML_VERIFIED",
|
|
},
|
|
cli.BoolTFlag{
|
|
Name: "yaml.signed",
|
|
Usage: "build yaml is signed",
|
|
EnvVar: "DRONE_YAML_SIGNED",
|
|
},
|
|
cli.IntFlag{
|
|
Name: "prev.build.number",
|
|
Usage: "previous build number",
|
|
EnvVar: "DRONE_PREV_BUILD_NUMBER",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "prev.build.status",
|
|
Usage: "previous build status",
|
|
EnvVar: "DRONE_PREV_BUILD_STATUS",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "prev.commit.sha",
|
|
Usage: "previous build sha",
|
|
EnvVar: "DRONE_PREV_COMMIT_SHA",
|
|
},
|
|
|
|
cli.StringFlag{
|
|
Name: "netrc.username",
|
|
Usage: "previous build sha",
|
|
EnvVar: "DRONE_NETRC_USERNAME",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "netrc.password",
|
|
Usage: "previous build sha",
|
|
EnvVar: "DRONE_NETRC_PASSWORD",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "netrc.machine",
|
|
Usage: "previous build sha",
|
|
EnvVar: "DRONE_NETRC_MACHINE",
|
|
},
|
|
},
|
|
}
|
|
|
|
func exec(c *cli.Context) error {
|
|
sigterm := make(chan os.Signal, 1)
|
|
cancelc := make(chan bool, 1)
|
|
signal.Notify(sigterm, os.Interrupt)
|
|
go func() {
|
|
<-sigterm
|
|
cancelc <- true
|
|
}()
|
|
|
|
path := c.Args().First()
|
|
if path == "" {
|
|
path = ".drone.yml"
|
|
}
|
|
path, _ = filepath.Abs(path)
|
|
dir := filepath.Dir(path)
|
|
|
|
file, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
engine, err := docker.New(
|
|
c.String("docker-host"),
|
|
c.String("docker-cert-path"),
|
|
c.Bool("docker-tls-verify"),
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
a := agent.Agent{
|
|
Update: agent.NoopUpdateFunc,
|
|
Logger: agent.TermLoggerFunc,
|
|
Engine: engine,
|
|
Timeout: c.Duration("timeout.inactivity"),
|
|
Platform: "linux/amd64",
|
|
Namespace: c.String("namespace"),
|
|
Disable: c.StringSlice("plugin"),
|
|
Escalate: c.StringSlice("privileged"),
|
|
Netrc: []string{},
|
|
Local: dir,
|
|
Pull: c.Bool("pull"),
|
|
}
|
|
|
|
payload := &queue.Work{
|
|
Yaml: string(file),
|
|
Verified: c.BoolT("yaml.verified"),
|
|
Signed: c.BoolT("yaml.signed"),
|
|
Repo: &model.Repo{
|
|
FullName: c.String("repo.fullname"),
|
|
Owner: c.String("repo.owner"),
|
|
Name: c.String("repo.name"),
|
|
Kind: c.String("repo.type"),
|
|
Link: c.String("repo.link"),
|
|
Branch: c.String("repo.branch"),
|
|
Avatar: c.String("repo.avatar"),
|
|
Timeout: int64(c.Duration("timeout").Minutes()),
|
|
IsPrivate: c.Bool("repo.private"),
|
|
IsTrusted: c.Bool("repo.trusted"),
|
|
Clone: c.String("remote.url"),
|
|
},
|
|
System: &model.System{
|
|
Link: c.GlobalString("server"),
|
|
},
|
|
Secrets: getSecrets(c),
|
|
Netrc: &model.Netrc{
|
|
Login: c.String("netrc.username"),
|
|
Password: c.String("netrc.password"),
|
|
Machine: c.String("netrc.machine"),
|
|
},
|
|
Build: &model.Build{
|
|
Commit: c.String("commit.sha"),
|
|
Branch: c.String("commit.branch"),
|
|
Ref: c.String("commit.ref"),
|
|
Link: c.String("commit.link"),
|
|
Message: c.String("commit.message"),
|
|
Author: c.String("commit.author.name"),
|
|
Email: c.String("commit.author.email"),
|
|
Avatar: c.String("commit.author.avatar"),
|
|
Number: c.Int("build.number"),
|
|
Event: c.String("build.event"),
|
|
Deploy: c.String("build.deploy"),
|
|
},
|
|
BuildLast: &model.Build{
|
|
Number: c.Int("prev.build.number"),
|
|
Status: c.String("prev.build.status"),
|
|
Commit: c.String("prev.commit.sha"),
|
|
},
|
|
}
|
|
|
|
if len(c.StringSlice("matrix")) > 0 {
|
|
p := *payload
|
|
p.Job = &model.Job{
|
|
Environment: getMatrix(c),
|
|
}
|
|
return a.Run(&p, cancelc)
|
|
}
|
|
|
|
axes, err := yaml.ParseMatrix(file)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(axes) == 0 {
|
|
axes = append(axes, yaml.Axis{})
|
|
}
|
|
|
|
var jobs []*model.Job
|
|
count := 0
|
|
for _, axis := range axes {
|
|
jobs = append(jobs, &model.Job{
|
|
Number: count,
|
|
Environment: axis,
|
|
})
|
|
count++
|
|
}
|
|
|
|
for _, job := range jobs {
|
|
fmt.Printf("Running Matrix job #%d\n", job.Number)
|
|
p := *payload
|
|
p.Job = job
|
|
if err := a.Run(&p, cancelc); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// helper function to retrieve matrix variables.
|
|
func getMatrix(c *cli.Context) map[string]string {
|
|
envs := map[string]string{}
|
|
for _, s := range c.StringSlice("matrix") {
|
|
parts := strings.SplitN(s, "=", 2)
|
|
if len(parts) != 2 {
|
|
continue
|
|
}
|
|
k := parts[0]
|
|
v := parts[1]
|
|
envs[k] = v
|
|
}
|
|
return envs
|
|
}
|
|
|
|
// helper function to retrieve secret variables.
|
|
func getSecrets(c *cli.Context) []*model.Secret {
|
|
|
|
var secrets []*model.Secret
|
|
|
|
if c.String("secrets-file") != "" {
|
|
envs, _ := godotenv.Read(c.String("secrets-file"))
|
|
for k, v := range envs {
|
|
secret := &model.Secret{
|
|
Name: k,
|
|
Value: v,
|
|
Events: []string{
|
|
model.EventPull,
|
|
model.EventPush,
|
|
model.EventTag,
|
|
model.EventDeploy,
|
|
},
|
|
Images: []string{"*"},
|
|
}
|
|
secrets = append(secrets, secret)
|
|
}
|
|
}
|
|
|
|
for _, s := range c.StringSlice("secret") {
|
|
parts := strings.SplitN(s, "=", 2)
|
|
if len(parts) != 2 {
|
|
continue
|
|
}
|
|
secret := &model.Secret{
|
|
Name: parts[0],
|
|
Value: parts[1],
|
|
Events: []string{
|
|
model.EventPull,
|
|
model.EventPush,
|
|
model.EventTag,
|
|
model.EventDeploy,
|
|
},
|
|
Images: []string{"*"},
|
|
}
|
|
secrets = append(secrets, secret)
|
|
}
|
|
return secrets
|
|
}
|