job_input.go

package cmd

import (
	"encoding/json"
	"fmt"
	"github.com/jenkins-zh/jenkins-cli/app/cmd/common"
	"log"
	"net/http"
	"strconv"

	"github.com/jenkins-zh/jenkins-cli/app/i18n"

	"github.com/AlecAivazis/survey/v2"
	"github.com/jenkins-zh/jenkins-cli/client"
	"github.com/spf13/cobra"
)

// JobInputOption is the job delete option
type JobInputOption struct {
	common.BatchOption

	Action string

	RoundTripper http.RoundTripper
}

var jobInputOption JobInputOption

func init() {
	jobCmd.AddCommand(jobInputCmd)
	jobInputCmd.Flags().StringVarP(&jobInputOption.Action, "action", "", "",
		i18n.T("The action whether you want to process or abort."))
	jobInputOption.Stdio = common.GetSystemStdio()
}

var jobInputCmd = &cobra.Command{
	Use:   "input <jobName> [buildID]",
	Short: i18n.T("Input a job in your Jenkins"),
	Long:  i18n.T("Input a job in your Jenkins"),
	Args:  cobra.MinimumNArgs(1),
	Run: func(cmd *cobra.Command, args []string) {
		jobName := args[0]
		buildID := -1

		if len(args) >= 2 {
			var err error
			if buildID, err = strconv.Atoi(args[1]); err != nil {
				cmd.PrintErrln(err)
			}
		}

		jclient := &client.JobClient{
			JenkinsCore: client.JenkinsCore{
				RoundTripper: jobInputOption.RoundTripper,
				Debug:        rootOptions.Debug,
			},
		}
		getCurrentJenkinsAndClientOrDie(&(jclient.JenkinsCore))

		if inputActions, err := jclient.GetJobInputActions(jobName, buildID); err != nil {
			log.Fatal(err)
		} else if len(inputActions) >= 1 {
			inputAction := inputActions[0]
			params := make(map[string]string)

			if len(inputAction.Inputs) > 0 {
				inputsJSON, _ := json.MarshalIndent(inputAction.Inputs, "", " ")
				content := string(inputsJSON)

				content, err = jobBuildOption.Editor(content, "Edit your pipeline input parameters")
				if err == nil {
					err = json.Unmarshal([]byte(content), &(inputAction.Inputs))
				}

				for _, input := range inputAction.Inputs {
					params[input.Name] = input.Value
				}
			}

			render := &survey.Renderer{}
			render.WithStdio(jobInputOption.Stdio)

			// allow users make their choice through cli arguments
			action := jobInputOption.Action
			if action == "" {
				prompt := &survey.Input{
					Renderer: *render,
					Message:  fmt.Sprintf("Are you going to process or abort this input: %s?", inputAction.Message),
				}
				survey.AskOne(prompt, &action)
			}

			if action == "process" {
				err = jclient.JobInputSubmit(jobName, inputAction.ID, buildID, false, params)
			} else if action == "abort" {
				err = jclient.JobInputSubmit(jobName, inputAction.ID, buildID, true, params)
			} else {
				cmd.PrintErrln("Only process or abort is accepted!")
			}

			if err != nil {
				cmd.PrintErrln(err)
			}
		}
	},
}