restart.go 1.1 KB
Newer Older
LinuxSuRen's avatar
LinuxSuRen 已提交
1 2 3 4 5
package cmd

import (
	"fmt"

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

LinuxSuRen's avatar
LinuxSuRen 已提交
11 12 13 14 15 16 17
// RestartOption holds the options for restart cmd
type RestartOption struct {
	BatchOption
}

var restartOption RestartOption

LinuxSuRen's avatar
LinuxSuRen 已提交
18 19
func init() {
	rootCmd.AddCommand(restartCmd)
LinuxSuRen's avatar
LinuxSuRen 已提交
20
	restartCmd.Flags().BoolVarP(&restartOption.Batch, "batch", "b", false, "Batch mode, no need confirm")
LinuxSuRen's avatar
LinuxSuRen 已提交
21 22 23 24 25 26
}

var restartCmd = &cobra.Command{
	Use:   "restart",
	Short: "Restart your Jenkins",
	Long:  `Restart your Jenkins`,
yJunS's avatar
yJunS 已提交
27
	Run: func(_ *cobra.Command, _ []string) {
LinuxSuRen's avatar
LinuxSuRen 已提交
28
		jenkins := getCurrentJenkinsFromOptionsOrDie()
LinuxSuRen's avatar
LinuxSuRen 已提交
29 30 31
		if !restartOption.Batch {
			confirm := false
			prompt := &survey.Confirm{
LinuxSuRen's avatar
LinuxSuRen 已提交
32
				Message: fmt.Sprintf("Are you sure to restart Jenkins %s?", jenkins.URL),
LinuxSuRen's avatar
LinuxSuRen 已提交
33 34 35 36 37
			}
			survey.AskOne(prompt, &confirm)
			if !confirm {
				return
			}
LinuxSuRen's avatar
LinuxSuRen 已提交
38 39
		}

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

LinuxSuRen's avatar
LinuxSuRen 已提交
47
		jclient.Restart()
LinuxSuRen's avatar
LinuxSuRen 已提交
48 49
	},
}