Browse Source

add flags for host update

Anish Mukherjee 2 years ago
parent
commit
905f5ef5e2
1 changed files with 28 additions and 8 deletions
  1. 28 8
      cli/cmd/host/update.go

+ 28 - 8
cli/cmd/host/update.go

@@ -23,23 +23,43 @@ var (
 )
 )
 
 
 var hostUpdateCmd = &cobra.Command{
 var hostUpdateCmd = &cobra.Command{
-	Use:   "update HostID /path/to/host_definition.json",
-	Args:  cobra.ExactArgs(2),
+	Use:   "update HostID",
+	Args:  cobra.ExactArgs(1),
 	Short: "Update a host",
 	Short: "Update a host",
 	Long:  `Update a host`,
 	Long:  `Update a host`,
 	Run: func(cmd *cobra.Command, args []string) {
 	Run: func(cmd *cobra.Command, args []string) {
 		apiHost := &models.ApiHost{}
 		apiHost := &models.ApiHost{}
-		content, err := os.ReadFile(args[1])
-		if err != nil {
-			log.Fatal("Error when opening file: ", err)
-		}
-		if err := json.Unmarshal(content, apiHost); err != nil {
-			log.Fatal(err)
+		if apiHostFilePath != "" {
+			content, err := os.ReadFile(apiHostFilePath)
+			if err != nil {
+				log.Fatal("Error when opening file: ", err)
+			}
+			if err := json.Unmarshal(content, apiHost); err != nil {
+				log.Fatal(err)
+			}
+		} else {
+			apiHost.EndpointIP = endpoint
+			apiHost.Name = name
+			apiHost.ListenPort = listenPort
+			apiHost.ProxyListenPort = proxyListenPort
+			apiHost.MTU = mtu
+			apiHost.ProxyEnabled = proxyEnabled
+			apiHost.IsStatic = isStatic
+			apiHost.IsDefault = isDefault
 		}
 		}
 		functions.PrettyPrint(functions.UpdateHost(args[0], apiHost))
 		functions.PrettyPrint(functions.UpdateHost(args[0], apiHost))
 	},
 	},
 }
 }
 
 
 func init() {
 func init() {
+	hostUpdateCmd.Flags().StringVar(&apiHostFilePath, "file", "", "Path to host_definition.json")
+	hostUpdateCmd.Flags().StringVar(&endpoint, "endpoint", "", "Endpoint of the Host")
+	hostUpdateCmd.Flags().StringVar(&name, "name", "", "Host name")
+	hostUpdateCmd.Flags().IntVar(&listenPort, "listen_port", 0, "Listen port of the host")
+	hostUpdateCmd.Flags().IntVar(&proxyListenPort, "proxy_listen_port", 0, "Proxy listen port of the host")
+	hostUpdateCmd.Flags().IntVar(&mtu, "mtu", 0, "Host MTU size")
+	hostUpdateCmd.Flags().BoolVar(&proxyEnabled, "proxy", false, "Enable proxy ?")
+	hostUpdateCmd.Flags().BoolVar(&isStatic, "static", false, "Make Host Static ?")
+	hostUpdateCmd.Flags().BoolVar(&isDefault, "default", false, "Make Host Default ?")
 	rootCmd.AddCommand(hostUpdateCmd)
 	rootCmd.AddCommand(hostUpdateCmd)
 }
 }