user_token.go 1.2 KB
Newer Older
LinuxSuRen's avatar
LinuxSuRen 已提交
1 2 3 4 5 6
package cmd

import (
	"fmt"
	"log"

7
	"github.com/jenkins-zh/jenkins-cli/client"
LinuxSuRen's avatar
LinuxSuRen 已提交
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
	"github.com/spf13/cobra"
)

type UserTokenOption struct {
	Generate bool
	Name     string
}

var userTokenOption UserTokenOption

func init() {
	userCmd.AddCommand(userTokenCmd)
	userTokenCmd.Flags().BoolVarP(&userTokenOption.Generate, "generate", "g", false, "Generate the token")
	userTokenCmd.Flags().StringVarP(&userTokenOption.Name, "name", "n", "", "Name of the token")
}

var userTokenCmd = &cobra.Command{
	Use:   "token",
	Short: "Token the user of your Jenkins",
	Long:  `Token the user of your Jenkins`,
yJunS's avatar
yJunS 已提交
28
	Run: func(cmd *cobra.Command, _ []string) {
LinuxSuRen's avatar
LinuxSuRen 已提交
29 30 31 32 33
		if !userTokenOption.Generate {
			cmd.Help()
			return
		}

LinuxSuRen's avatar
LinuxSuRen 已提交
34
		jenkins := getCurrentJenkinsFromOptionsOrDie()
LinuxSuRen's avatar
LinuxSuRen 已提交
35 36 37 38 39 40 41 42
		jclient := &client.UserClient{}
		jclient.URL = jenkins.URL
		jclient.UserName = jenkins.UserName
		jclient.Token = jenkins.Token
		jclient.Proxy = jenkins.Proxy
		jclient.ProxyAuth = jenkins.ProxyAuth

		tokenName := userTokenOption.Name
43
		if status, err := jclient.CreateToken(tokenName); err == nil {
LinuxSuRen's avatar
LinuxSuRen 已提交
44 45 46 47 48 49 50 51 52 53 54
			var data []byte
			if data, err = userOption.Output(status); err == nil {
				fmt.Printf("%s\n", string(data))
			} else {
				log.Fatal(err)
			}
		} else {
			log.Fatal(err)
		}
	},
}