github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/fly/commands/execute.go

package commands

import (
	"fmt"
	"net/url"
	"os"
	"os/signal"
	"strconv"
	"syscall"
	"time"

	"github.com/pf-qiu/concourse/v6/atc"
	"github.com/pf-qiu/concourse/v6/fly/commands/internal/displayhelpers"
	"github.com/pf-qiu/concourse/v6/fly/commands/internal/executehelpers"
	"github.com/pf-qiu/concourse/v6/fly/commands/internal/flaghelpers"
	"github.com/pf-qiu/concourse/v6/fly/commands/internal/templatehelpers"
	"github.com/pf-qiu/concourse/v6/fly/config"
	"github.com/pf-qiu/concourse/v6/fly/eventstream"
	"github.com/pf-qiu/concourse/v6/fly/rc"
	"github.com/pf-qiu/concourse/v6/fly/ui"
	"github.com/pf-qiu/concourse/v6/fly/ui/progress"
	"github.com/pf-qiu/concourse/v6/go-concourse/concourse"
	"github.com/vbauerster/mpb/v4"
)

type ExecuteCommand struct {
	TaskConfig     atc.PathFlag                       `short:"c" long:"config" required:"true"                description:"The task config to execute"`
	Privileged     bool                               `short:"p" long:"privileged"                            description:"Run the task with full privileges"`
	IncludeIgnored bool                               `          long:"include-ignored"                       description:"Including .gitignored paths. Disregards .gitignore entries and uploads everything"`
	Inputs         []flaghelpers.InputPairFlag        `short:"i" long:"input"       value-name:"NAME=PATH"    description:"An input to provide to the task (can be specified multiple times)"`
	InputMappings  []flaghelpers.InputMappingPairFlag `short:"m" long:"input-mapping"       value-name:"[NAME=STRING]"    description:"Map a resource to a different name as task input"`
	InputsFrom     flaghelpers.JobFlag                `short:"j" long:"inputs-from" value-name:"PIPELINE/JOB" description:"A job to base the inputs on"`
	Outputs        []flaghelpers.OutputPairFlag       `short:"o" long:"output"      value-name:"NAME=PATH"    description:"An output to fetch from the task (can be specified multiple times)"`
	Image          string                             `long:"image" description:"Image resource for the one-off build"`
	Tags           []string                           `          long:"tag"         value-name:"TAG"          description:"A tag for a specific environment (can be specified multiple times)"`
	Var            []flaghelpers.VariablePairFlag     `short:"v"  long:"var"       value-name:"[NAME=STRING]"  unquote:"false"  description:"Specify a string value to set for a variable in the pipeline"`
	YAMLVar        []flaghelpers.YAMLVariablePairFlag `short:"y"  long:"yaml-var"  value-name:"[NAME=YAML]"    unquote:"false"  description:"Specify a YAML value to set for a variable in the pipeline"`
	VarsFrom       []atc.PathFlag                     `short:"l"  long:"load-vars-from"  description:"Variable flag that can be used for filling in template values in configuration from a YAML file"`
}

func (command *ExecuteCommand) Execute(args []string) error {
	target, err := rc.LoadTarget(Fly.Target, Fly.Verbose)
	if err != nil {
		return err
	}

	err = target.Validate()
	if err != nil {
		return err
	}

	taskConfig, err := command.CreateTaskConfig(args)
	if err != nil {
		return err
	}

	planFactory := atc.NewPlanFactory(time.Now().Unix())

	inputs, inputMappings, imageResource, resourceTypes, err := executehelpers.DetermineInputs(
		planFactory,
		target.Team(),
		taskConfig.Inputs,
		command.Inputs,
		command.InputMappings,
		command.Image,
		command.InputsFrom,
		command.IncludeIgnored,
		taskConfig.Platform,
		command.Tags,
	)
	if err != nil {
		return err
	}

	if imageResource != nil {
		taskConfig.ImageResource = imageResource
	}

	outputs, err := executehelpers.DetermineOutputs(
		planFactory,
		taskConfig.Outputs,
		command.Outputs,
	)
	if err != nil {
		return err
	}

	plan, err := executehelpers.CreateBuildPlan(
		planFactory,
		target,
		command.Privileged,
		inputs,
		inputMappings,
		resourceTypes,
		outputs,
		taskConfig,
		command.Tags,
	)

	if err != nil {
		return err
	}

	client := target.Client()
	clientURL, err := url.Parse(client.URL())
	if err != nil {
		return err
	}

	var build atc.Build
	var buildURL *url.URL

	if command.InputsFrom.PipelineRef.Name != "" {
		build, err = target.Team().CreatePipelineBuild(command.InputsFrom.PipelineRef, plan)
		if err != nil {
			return err
		}
	} else {
		build, err = target.Team().CreateBuild(plan)
		if err != nil {
			return err
		}
	}

	buildURL, err = url.Parse(fmt.Sprintf("/builds/%d", build.ID))
	if err != nil {
		return err
	}

	fmt.Printf("executing build %d at %s\n", build.ID, clientURL.ResolveReference(buildURL))

	terminate := make(chan os.Signal, 1)

	go abortOnSignal(client, terminate, build)

	signal.Notify(terminate, syscall.SIGINT, syscall.SIGTERM)

	eventSource, err := client.BuildEvents(strconv.Itoa(build.ID))
	if err != nil {
		return err
	}

	renderOptions := eventstream.RenderOptions{}

	exitCode := eventstream.Render(os.Stdout, eventSource, renderOptions)
	eventSource.Close()

	artifactList, err := client.ListBuildArtifacts(strconv.Itoa(build.ID))
	if err != nil {
		return err
	}

	artifacts := map[string]atc.WorkerArtifact{}

	for _, artifact := range artifactList {
		artifacts[artifact.Name] = artifact
	}

	prog := progress.New()

	for _, output := range outputs {
		name := output.Name
		path := output.Path

		artifact, ok := artifacts[name]
		if !ok {
			continue
		}

		prog.Go("downloading "+output.Name, func(bar *mpb.Bar) error {
			return executehelpers.Download(bar, target.Team(), artifact.ID, path)
		})
	}

	err = prog.Wait()
	if err != nil {
		displayhelpers.FailWithErrorf("downloading failed: %s", err)
		return err
	}

	os.Exit(exitCode)

	return nil
}

func (command *ExecuteCommand) CreateTaskConfig(args []string) (atc.TaskConfig, error) {

	taskTemplate := templatehelpers.NewYamlTemplateWithParams(
		command.TaskConfig,
		command.VarsFrom,
		command.Var,
		command.YAMLVar,
		nil,
	)

	taskTemplateEvaluated, err := taskTemplate.Evaluate(false, false)
	if err != nil {
		return atc.TaskConfig{}, err
	}

	return config.OverrideTaskParams(taskTemplateEvaluated, args)
}

func abortOnSignal(
	client concourse.Client,
	terminate <-chan os.Signal,
	build atc.Build,
) {
	<-terminate

	fmt.Fprintf(ui.Stderr, "\naborting...\n")

	err := client.AbortBuild(strconv.Itoa(build.ID))
	if err != nil {
		fmt.Fprintln(ui.Stderr, "failed to abort:", err)
		os.Exit(2)
	}

	// if told to terminate again, exit immediately
	<-terminate
	fmt.Fprintln(ui.Stderr, "exiting immediately")
	os.Exit(2)
}