get_pipeline.go

package commands

import (
	"encoding/json"
	"fmt"
	"os"

	"gopkg.in/yaml.v2"

	"github.com/concourse/atc"
	"github.com/concourse/fly/commands/internal/displayhelpers"
	"github.com/concourse/fly/rc"
	"github.com/concourse/go-concourse/concourse"
	"github.com/mattn/go-isatty"
)

type GetPipelineCommand struct {
	Pipeline string `short:"p" long:"pipeline" required:"true" description:"Get configuration of this pipeline"`
	JSON     bool   `short:"j" long:"json"                     description:"Print config as json instead of yaml"`
}

func (command *GetPipelineCommand) Execute(args []string) error {
	asJSON := command.JSON
	pipelineName := command.Pipeline

	client, err := rc.TargetClient(Fly.Target)
	if err != nil {
		return err
	}
	err = rc.ValidateClient(client, Fly.Target)
	if err != nil {
		return err
	}

	config, rawConfig, _, _, err := client.PipelineConfig(pipelineName)
	if err != nil {
		if _, ok := err.(concourse.PipelineConfigError); ok {
			dumpRawConfig(rawConfig, asJSON)
			command.showConfigWarning()
			return err
		} else {
			return err
		}
	}

	return dump(config, asJSON)
}

func dump(config atc.Config, asJSON bool) error {
	var payload []byte
	var err error
	if asJSON {
		payload, err = json.Marshal(config)
	} else {
		payload, err = yaml.Marshal(config)
	}
	if err != nil {
		return err
	}

	_, err = fmt.Printf("%s", payload)
	if err != nil {
		return err
	}

	return nil
}

func dumpRawConfig(rawConfig atc.RawConfig, asJSON bool) error {
	var payload []byte
	if asJSON {
		payload = []byte(rawConfig)
	} else {
		var config map[string]interface{}
		err := json.Unmarshal([]byte(rawConfig), &config)
		if err != nil {
			return err
		}

		payload, err = yaml.Marshal(config)
	}

	_, err := fmt.Printf("%s", payload)
	if err != nil {
		return err
	}

	return nil
}

func (command *GetPipelineCommand) showConfigWarning() {
	if isatty.IsTerminal(os.Stdout.Fd()) {
		fmt.Fprintln(os.Stderr, "")
	}
	displayhelpers.PrintWarningHeader()
	fmt.Fprintln(os.Stderr, "Existing config is invalid, it was returned as-is\n")
}