From 69ee6327ed607d63a80a27f127ba822d87e5796b Mon Sep 17 00:00:00 2001
From: Zhao Xiaojie <LinuxSuRen@gmail.com>
Date: Wed, 26 Jun 2019 14:28:47 +0800
Subject: [PATCH] Add option to check update center site and open the browse

---
 app/cmd/plugin.go | 43 +++++++++++++++++++++++++++++++++++++++----
 1 file changed, 39 insertions(+), 4 deletions(-)

diff --git a/app/cmd/plugin.go b/app/cmd/plugin.go
index 63743c7..cc7e173 100644
--- a/app/cmd/plugin.go
+++ b/app/cmd/plugin.go
@@ -14,29 +14,34 @@ import (
 	"strings"
 
 	"github.com/gosuri/uiprogress"
+	"github.com/linuxsuren/jenkins-cli/client"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 )
 
 // PluginOptions contains the command line options
 type PluginOptions struct {
-	Upload bool
+	Upload      bool
+	CheckUpdate bool
+	Open        bool
 }
 
 func init() {
 	rootCmd.AddCommand(pluginCmd)
-	pluginCmd.PersistentFlags().BoolVarP(&author.Upload, "upload", "u", false, "Upload plugin to your Jenkins server")
+	pluginCmd.PersistentFlags().BoolVarP(&pluginOpt.Upload, "upload", "u", false, "Upload plugin to your Jenkins server")
+	pluginCmd.PersistentFlags().BoolVarP(&pluginOpt.CheckUpdate, "check", "c", false, "Checkout update center server")
+	pluginCmd.PersistentFlags().BoolVarP(&pluginOpt.Open, "open", "o", false, "Open the browse with the address of plugin manager")
 	viper.BindPFlag("upload", pluginCmd.PersistentFlags().Lookup("upload"))
 }
 
-var author PluginOptions
+var pluginOpt PluginOptions
 
 var pluginCmd = &cobra.Command{
 	Use:   "plugin",
 	Short: "Manage the plugins of Jenkins",
 	Long:  `Manage the plugins of Jenkins`,
 	Run: func(cmd *cobra.Command, args []string) {
-		if author.Upload {
+		if pluginOpt.Upload {
 			crumb, config := getCrumb()
 
 			api := fmt.Sprintf("%s/pluginManager/uploadPlugin", config.URL)
@@ -74,6 +79,36 @@ var pluginCmd = &cobra.Command{
 				log.Fatal(err)
 			}
 		}
+
+		if pluginOpt.CheckUpdate {
+			jclient := &client.PluginManager{}
+
+			crumb, config := getCrumb()
+			jclient.CrumbRequestField = crumb.CrumbRequestField
+			jclient.Crumb = crumb.Crumb
+			jclient.URL = config.URL
+			jclient.UserName = config.UserName
+			jclient.Token = config.Token
+			jclient.CheckUpdate(func(response *http.Response) {
+				code := response.StatusCode
+				if code == 200 {
+					fmt.Println("update site updated.")
+				} else {
+					contentData, _ := ioutil.ReadAll(response.Body)
+					log.Fatal(fmt.Sprintf("response code is %d, content: ",
+						code, string(contentData)))
+				}
+			})
+		}
+
+		if pluginOpt.Open {
+			jenkins := getCurrentJenkins()
+			if jenkins.URL != "" {
+				open(fmt.Sprintf("%s/pluginManager", jenkins.URL))
+			} else {
+				log.Fatal(fmt.Sprintf("No URL fond from %s", jenkins.Name))
+			}
+		}
 	},
 }
 
-- 
GitLab