set_pipeline.go

package commands

import (
	"github.com/concourse/atc"
	"github.com/concourse/atc/web"
	"github.com/concourse/fly/commands/internal/flaghelpers"
	"github.com/concourse/fly/commands/internal/setpipelinehelpers"
	"github.com/concourse/fly/rc"
	"github.com/tedsuo/rata"
)

type SetPipelineCommand struct {
	SkipInteractive bool `short:"n"  long:"non-interactive"               description:"Skips interactions, uses default values"`

	Pipeline flaghelpers.PipelineFlag `short:"p"  long:"pipeline"  required:"true"  description:"Pipeline to configure"`
	Config   atc.PathFlag             `short:"c"  long:"config"    required:"true"  description:"Pipeline configuration file"`

	Var     []flaghelpers.VariablePairFlag     `short:"v"  long:"var"       value-name:"[NAME=STRING]"  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]"    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 *SetPipelineCommand) Execute(args []string) error {
	configPath := command.Config
	templateVariablesFiles := command.VarsFrom
	pipelineName := string(command.Pipeline)

	target, err := rc.LoadTarget(Fly.Target)
	if err != nil {
		return err
	}

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

	webRequestGenerator := rata.NewRequestGenerator(target.Client().URL(), web.Routes)

	atcConfig := setpipelinehelpers.ATCConfig{
		Team:                target.Team(),
		PipelineName:        pipelineName,
		WebRequestGenerator: webRequestGenerator,
		SkipInteraction:     command.SkipInteractive,
	}

	return atcConfig.Set(configPath, command.Var, command.YAMLVar, templateVariablesFiles)
}