mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-06-02 21:49:25 +00:00
124 lines
2.8 KiB
Go
124 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/laszlocph/woodpecker/cncd/pipeline/pipeline"
|
|
"github.com/laszlocph/woodpecker/cncd/pipeline/pipeline/backend"
|
|
"github.com/laszlocph/woodpecker/cncd/pipeline/pipeline/backend/docker"
|
|
"github.com/laszlocph/woodpecker/cncd/pipeline/pipeline/backend/kubernetes"
|
|
"github.com/laszlocph/woodpecker/cncd/pipeline/pipeline/interrupt"
|
|
"github.com/laszlocph/woodpecker/cncd/pipeline/pipeline/multipart"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
var executeCommand = cli.Command{
|
|
Name: "exec",
|
|
Usage: "execute the compiled file",
|
|
Action: executeAction,
|
|
Flags: []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "in",
|
|
Value: "pipeline.json",
|
|
},
|
|
cli.DurationFlag{
|
|
Name: "timeout",
|
|
EnvVar: "CI_TIMEOUT",
|
|
Value: time.Hour,
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "kubernetes",
|
|
EnvVar: "CI_KUBERNETES",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "kubernetes-namepsace",
|
|
EnvVar: "CI_KUBERNETES_NAMESPACE",
|
|
Value: "default",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "kubernetes-endpoint",
|
|
EnvVar: "CI_KUBERNETES_ENDPOINT",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "kubernetes-token",
|
|
EnvVar: "CI_KUBERNETES_TOKEN",
|
|
},
|
|
},
|
|
}
|
|
|
|
func executeAction(c *cli.Context) (err error) {
|
|
path := c.Args().First()
|
|
if path == "" {
|
|
path = c.String("in")
|
|
}
|
|
|
|
var reader io.ReadCloser
|
|
if path == "-" {
|
|
reader = os.Stdin
|
|
} else {
|
|
reader, err = os.Open(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
defer reader.Close()
|
|
|
|
config, err := pipeline.Parse(reader)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var engine backend.Engine
|
|
if c.Bool("kubernetes") {
|
|
engine = kubernetes.New(
|
|
c.String("kubernetes-namepsace"),
|
|
c.String("kubernetes-endpoint"),
|
|
c.String("kubernetes-token"),
|
|
)
|
|
} else {
|
|
engine, err = docker.NewEnv()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), c.Duration("timeout"))
|
|
defer cancel()
|
|
ctx = interrupt.WithContext(ctx)
|
|
|
|
return pipeline.New(config,
|
|
pipeline.WithContext(ctx),
|
|
pipeline.WithLogger(defaultLogger),
|
|
pipeline.WithTracer(defaultTracer),
|
|
pipeline.WithEngine(engine),
|
|
).Run()
|
|
}
|
|
|
|
var defaultLogger = pipeline.LogFunc(func(proc *backend.Step, rc multipart.Reader) error {
|
|
part, err := rc.NextPart()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
io.Copy(os.Stderr, part)
|
|
return nil
|
|
})
|
|
|
|
var defaultTracer = pipeline.TraceFunc(func(state *pipeline.State) error {
|
|
if state.Process.Exited {
|
|
fmt.Printf("proc %q exited with status %d\n", state.Pipeline.Step.Name, state.Process.ExitCode)
|
|
} else {
|
|
fmt.Printf("proc %q started\n", state.Pipeline.Step.Name)
|
|
state.Pipeline.Step.Environment["CI_BUILD_STATUS"] = "success"
|
|
state.Pipeline.Step.Environment["CI_BUILD_FINISHED"] = strconv.FormatInt(time.Now().Unix(), 10)
|
|
if state.Pipeline.Error != nil {
|
|
state.Pipeline.Step.Environment["CI_BUILD_STATUS"] = "failure"
|
|
}
|
|
}
|
|
return nil
|
|
})
|