job_build.go 1.9 KB
Newer Older
LinuxSuRen's avatar
LinuxSuRen 已提交
1 2 3
package cmd

import (
4
	"encoding/json"
LinuxSuRen's avatar
LinuxSuRen 已提交
5
	"fmt"
6
	"log"
LinuxSuRen's avatar
LinuxSuRen 已提交
7

LinuxSuRen's avatar
LinuxSuRen 已提交
8
	"github.com/AlecAivazis/survey/v2"
9
	"github.com/jenkins-zh/jenkins-cli/client"
LinuxSuRen's avatar
LinuxSuRen 已提交
10 11 12
	"github.com/spf13/cobra"
)

LinuxSuRen's avatar
LinuxSuRen 已提交
13 14 15 16 17 18
type JobBuildOption struct {
	BatchOption
}

var jobBuildOption JobBuildOption

LinuxSuRen's avatar
LinuxSuRen 已提交
19 20
func init() {
	jobCmd.AddCommand(jobBuildCmd)
LinuxSuRen's avatar
LinuxSuRen 已提交
21
	jobBuildCmd.Flags().BoolVarP(&jobBuildOption.Batch, "batch", "b", false, "Batch mode, no need confirm")
LinuxSuRen's avatar
LinuxSuRen 已提交
22 23 24
}

var jobBuildCmd = &cobra.Command{
LinuxSuRen's avatar
LinuxSuRen 已提交
25
	Use:   "build <jobName>",
LinuxSuRen's avatar
LinuxSuRen 已提交
26 27 28
	Short: "Build the job of your Jenkins",
	Long:  `Build the job of your Jenkins`,
	Run: func(cmd *cobra.Command, args []string) {
LinuxSuRen's avatar
LinuxSuRen 已提交
29
		if len(args) == 0 {
30 31
			cmd.Help()
			return
LinuxSuRen's avatar
LinuxSuRen 已提交
32 33
		}

LinuxSuRen's avatar
LinuxSuRen 已提交
34 35 36
		name := args[0]

		if !jobBuildOption.Confirm(fmt.Sprintf("Are you sure to build job %s", name)) {
LinuxSuRen's avatar
LinuxSuRen 已提交
37 38 39
			return
		}

LinuxSuRen's avatar
LinuxSuRen 已提交
40 41 42 43 44 45 46 47
		jenkins := getCurrentJenkins()
		jclient := &client.JobClient{}
		jclient.URL = jenkins.URL
		jclient.UserName = jenkins.UserName
		jclient.Token = jenkins.Token
		jclient.Proxy = jenkins.Proxy
		jclient.ProxyAuth = jenkins.ProxyAuth

48 49
		paramDefs := []client.ParameterDefinition{}
		hasParam := false
LinuxSuRen's avatar
LinuxSuRen 已提交
50
		if job, err := jclient.GetJob(name); err == nil {
51 52 53 54 55 56 57 58 59 60 61 62
			if len(job.Property) != 0 {
				for _, pro := range job.Property {
					if len(pro.ParameterDefinitions) == 0 {
						continue
					}

					if data, err := json.MarshalIndent(pro.ParameterDefinitions, "", " "); err == nil {
						content := string(data)
						prompt := &survey.Editor{
							Message:       "Edit your pipeline script",
							FileName:      "*.sh",
							Default:       content,
LinuxSuRen's avatar
LinuxSuRen 已提交
63
							HideDefault:   true,
64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
							AppendDefault: true,
						}

						if err = survey.AskOne(prompt, &content); err != nil {
							log.Fatal(err)
						}

						if err = json.Unmarshal([]byte(content), &paramDefs); err != nil {
							log.Fatal(err)
						}
					}
					hasParam = true
					break
				}
			}
		}

		if hasParam {
LinuxSuRen's avatar
LinuxSuRen 已提交
82
			jclient.BuildWithParams(name, paramDefs)
83
		} else {
LinuxSuRen's avatar
LinuxSuRen 已提交
84
			jclient.Build(name)
85
		}
LinuxSuRen's avatar
LinuxSuRen 已提交
86 87
	},
}