mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2025-10-22 06:13:24 +00:00
fixed compile error in queue
This commit is contained in:
Brad
309
client/main.go
309
client/main.go
@@ -1,9 +1,308 @@
|
||||
package main
|
||||
|
||||
var (
|
||||
// commit sha for the current build.
|
||||
version string = "0.2-dev"
|
||||
revision string
|
||||
import (
|
||||
"bytes"
|
||||
"flag"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/drone/drone/shared/build"
|
||||
"github.com/drone/drone/shared/build/docker"
|
||||
"github.com/drone/drone/shared/build/log"
|
||||
"github.com/drone/drone/shared/build/repo"
|
||||
"github.com/drone/drone/shared/build/script"
|
||||
|
||||
"gopkg.in/yaml.v1"
|
||||
)
|
||||
|
||||
func main() {}
|
||||
var (
|
||||
// identity file (id_rsa) that will be injected
|
||||
// into the container if specified
|
||||
identity = flag.String("identity", "", "")
|
||||
|
||||
// runs Drone in parallel mode if True
|
||||
parallel = flag.Bool("parallel", false, "")
|
||||
|
||||
// build will timeout after N milliseconds.
|
||||
// this will default to 500 minutes (6 hours)
|
||||
timeout = flag.Duration("timeout", 300*time.Minute, "")
|
||||
|
||||
// build will run in a privileged container
|
||||
privileged = flag.Bool("privileged", false, "")
|
||||
|
||||
// runs Drone with verbose output if True
|
||||
verbose = flag.Bool("v", false, "")
|
||||
|
||||
// displays the help / usage if True
|
||||
help = flag.Bool("h", false, "")
|
||||
|
||||
// version number, currently deterined by the
|
||||
// git revision number (sha)
|
||||
version string
|
||||
)
|
||||
|
||||
func init() {
|
||||
// default logging
|
||||
log.SetPrefix("\033[2m[DRONE] ")
|
||||
log.SetSuffix("\033[0m\n")
|
||||
log.SetOutput(os.Stdout)
|
||||
log.SetPriority(log.LOG_NOTICE)
|
||||
}
|
||||
|
||||
func main() {
|
||||
// Parse the input parameters
|
||||
flag.Usage = usage
|
||||
flag.Parse()
|
||||
|
||||
if *help {
|
||||
flag.Usage()
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
if *verbose {
|
||||
log.SetPriority(log.LOG_DEBUG)
|
||||
}
|
||||
|
||||
// Must speicify a command
|
||||
args := flag.Args()
|
||||
if len(args) == 0 {
|
||||
flag.Usage()
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
switch {
|
||||
// run drone build assuming the current
|
||||
// working directory contains the drone.yml
|
||||
case args[0] == "build" && len(args) == 1:
|
||||
path, _ := os.Getwd()
|
||||
path = filepath.Join(path, ".drone.yml")
|
||||
run(path)
|
||||
|
||||
// run drone build where the path to the
|
||||
// source directory is provided
|
||||
case args[0] == "build" && len(args) == 2:
|
||||
path := args[1]
|
||||
path = filepath.Clean(path)
|
||||
path, _ = filepath.Abs(path)
|
||||
path = filepath.Join(path, ".drone.yml")
|
||||
run(path)
|
||||
|
||||
// run drone vet where the path to the
|
||||
// source directory is provided
|
||||
case args[0] == "vet" && len(args) == 2:
|
||||
path := args[1]
|
||||
path = filepath.Clean(path)
|
||||
path, _ = filepath.Abs(path)
|
||||
path = filepath.Join(path, ".drone.yml")
|
||||
vet(path)
|
||||
|
||||
// run drone vet assuming the current
|
||||
// working directory contains the drone.yml
|
||||
case args[0] == "vet" && len(args) == 1:
|
||||
path, _ := os.Getwd()
|
||||
path = filepath.Join(path, ".drone.yml")
|
||||
vet(path)
|
||||
|
||||
// print the version / revision number
|
||||
case args[0] == "version" && len(args) == 1:
|
||||
println(version)
|
||||
|
||||
// print the help message
|
||||
case args[0] == "help" && len(args) == 1:
|
||||
flag.Usage()
|
||||
}
|
||||
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
func vet(path string) {
|
||||
// parse the Drone yml file
|
||||
script, err := script.ParseBuildFile(path)
|
||||
if err != nil {
|
||||
log.Err(err.Error())
|
||||
os.Exit(1)
|
||||
return
|
||||
}
|
||||
|
||||
// print the Drone yml as parsed
|
||||
out, _ := yaml.Marshal(script)
|
||||
log.Noticef("parsed yaml:\n%s", string(out))
|
||||
}
|
||||
|
||||
func run(path string) {
|
||||
dockerClient := docker.New()
|
||||
|
||||
// parse the Drone yml file
|
||||
s, err := script.ParseBuildFile(path)
|
||||
if err != nil {
|
||||
log.Err(err.Error())
|
||||
os.Exit(1)
|
||||
return
|
||||
}
|
||||
|
||||
// get the repository root directory
|
||||
dir := filepath.Dir(path)
|
||||
code := repo.Repo{
|
||||
Name: filepath.Base(dir),
|
||||
Branch: "HEAD", // should we do this?
|
||||
Path: dir,
|
||||
}
|
||||
|
||||
// does the local repository match the
|
||||
// $GOPATH/src/{package} pattern? This is
|
||||
// important so we know the target location
|
||||
// where the code should be copied inside
|
||||
// the container.
|
||||
if gopath, ok := getRepoPath(dir); ok {
|
||||
code.Dir = gopath
|
||||
|
||||
} else if gopath, ok := getGoPath(dir); ok {
|
||||
// in this case we found a GOPATH and
|
||||
// reverse engineered the package path
|
||||
code.Dir = gopath
|
||||
|
||||
} else {
|
||||
// otherwise just use directory name
|
||||
code.Dir = filepath.Base(dir)
|
||||
}
|
||||
|
||||
// this is where the code gets uploaded to the container
|
||||
// TODO move this code to the build package
|
||||
code.Dir = filepath.Join("/var/cache/drone/src", filepath.Clean(code.Dir))
|
||||
|
||||
// track all build results
|
||||
var builders []*build.Builder
|
||||
|
||||
// ssh key to import into container
|
||||
var key []byte
|
||||
if len(*identity) != 0 {
|
||||
key, err = ioutil.ReadFile(*identity)
|
||||
if err != nil {
|
||||
fmt.Printf("[Error] Could not find or read identity file %s\n", *identity)
|
||||
os.Exit(1)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
builds := []*script.Build{s}
|
||||
|
||||
// loop through and create builders
|
||||
for _, b := range builds { //script.Builds {
|
||||
builder := build.New(dockerClient)
|
||||
builder.Build = b
|
||||
builder.Repo = &code
|
||||
builder.Key = key
|
||||
builder.Stdout = os.Stdout
|
||||
builder.Timeout = *timeout
|
||||
builder.Privileged = *privileged
|
||||
|
||||
if *parallel == true {
|
||||
var buf bytes.Buffer
|
||||
builder.Stdout = &buf
|
||||
}
|
||||
|
||||
builders = append(builders, builder)
|
||||
}
|
||||
|
||||
switch *parallel {
|
||||
case false:
|
||||
runSequential(builders)
|
||||
case true:
|
||||
runParallel(builders)
|
||||
}
|
||||
|
||||
// if in parallel mode, print out the buffer
|
||||
// if we had a failure
|
||||
for _, builder := range builders {
|
||||
if builder.BuildState.ExitCode == 0 {
|
||||
continue
|
||||
}
|
||||
|
||||
if buf, ok := builder.Stdout.(*bytes.Buffer); ok {
|
||||
log.Noticef("printing stdout for failed build %s", builder.Build.Name)
|
||||
println(buf.String())
|
||||
}
|
||||
}
|
||||
|
||||
// this exit code is initially 0 and will
|
||||
// be set to an error code if any of the
|
||||
// builds fail.
|
||||
var exit int
|
||||
|
||||
fmt.Printf("\nDrone Build Results \033[90m(%v)\033[0m\n", len(builders))
|
||||
|
||||
// loop through and print results
|
||||
for _, builder := range builders {
|
||||
build := builder.Build
|
||||
res := builder.BuildState
|
||||
duration := time.Duration(res.Finished - res.Started)
|
||||
switch {
|
||||
case builder.BuildState.ExitCode == 0:
|
||||
fmt.Printf(" \033[32m\u2713\033[0m %v \033[90m(%v)\033[0m\n", build.Name, humanizeDuration(duration*time.Second))
|
||||
case builder.BuildState.ExitCode != 0:
|
||||
fmt.Printf(" \033[31m\u2717\033[0m %v \033[90m(%v)\033[0m\n", build.Name, humanizeDuration(duration*time.Second))
|
||||
exit = builder.BuildState.ExitCode
|
||||
}
|
||||
}
|
||||
|
||||
os.Exit(exit)
|
||||
}
|
||||
|
||||
func runSequential(builders []*build.Builder) {
|
||||
// loop through and execute each build
|
||||
for _, builder := range builders {
|
||||
if err := builder.Run(); err != nil {
|
||||
log.Errf("Error executing build: %s", err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func runParallel(builders []*build.Builder) {
|
||||
// spawn four worker goroutines
|
||||
var wg sync.WaitGroup
|
||||
for _, builder := range builders {
|
||||
// Increment the WaitGroup counter
|
||||
wg.Add(1)
|
||||
// Launch a goroutine to run the build
|
||||
go func(builder *build.Builder) {
|
||||
defer wg.Done()
|
||||
builder.Run()
|
||||
}(builder)
|
||||
time.Sleep(500 * time.Millisecond) // get weird iptables failures unless we sleep.
|
||||
}
|
||||
|
||||
// wait for the workers to finish
|
||||
wg.Wait()
|
||||
}
|
||||
|
||||
var usage = func() {
|
||||
fmt.Println(`Drone is a tool for building and testing code in Docker containers.
|
||||
|
||||
Usage:
|
||||
|
||||
drone command [arguments]
|
||||
|
||||
The commands are:
|
||||
|
||||
build build and test the repository
|
||||
version print the version number
|
||||
vet validate the yaml configuration file
|
||||
|
||||
-v runs drone with verbose output
|
||||
-h display this help and exit
|
||||
--parallel runs drone build tasks in parallel
|
||||
--timeout=300ms timeout build after 300 milliseconds
|
||||
--privileged runs drone build in a privileged container
|
||||
|
||||
Examples:
|
||||
drone build builds the source in the pwd
|
||||
drone build /path/to/repo builds the source repository
|
||||
|
||||
Use "drone help [command]" for more information about a command.
|
||||
`)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user