support call any api with command 'ucloud api'

Author: lixiaojun629

README-CN.md

@@ -216,3 +216,39 @@ gssh[uga-0psxxx] created
 $ ssh root@152.32.140.92.ipssh.net
 root@152.32.140.92.ipssh.net's password: password of the uhost instance
 ```
+
+使用"ucloud api"命令调用任意API，根据API文档把某个API的参数依次填入。此命令比较特殊，不支持--public-key,--private-key,--debug,--profile,--timeout-sec等公共参数，如果要开启debug模式，可以设置环境变量$UCLOUD_CLI_DEBUG=on
+
+```
+$ ucloud api --Action <APIName>  --Param1 <value> --Param2 <value> ...
+```
+或者把API参数写到JSON文件中，举例如下
+```
+$ ucloud api --local-file ./create_uhost.json
+
+//create_uhost.json文件内容
+{
+    "Action":"CreateUHostInstance",
+    "Region":"cn-bj2",
+    "Zone":"cn-bj2-02",
+    "ImageId":"uimage-gk2x3x",
+    "NetworkInterface": [{
+        "EIP":{
+            "Bandwidth":1,
+            "OperatorName":"Bgp",
+            "PayMode": "Bandwidth"
+        }
+    }],
+    "LoginMode":"Password",
+    "Password":"dGVzdGx4ajEy",
+    "CPU":1,
+    "Memory":2048,
+    "Disks":[
+        {
+            "Size":20,
+            "Type":"LOCAL_NORMAL",
+            "IsBoot":"true"
+        }
+    ]
+}
+```

README.md

@@ -225,3 +225,39 @@ gssh[uga-0psxxx] created
 $ ssh root@152.32.140.92.ipssh.net
 root@152.32.140.92.ipssh.net's password: password of the uhost instance
 ```
+
+Using command "ucloud api" to call any API.Fill in the parameters of an API in sequence according to the API documentation. This command is quite special, and public parameters such as --public-key,--private-key,--debug,--profile,--timeout-sec are not supported. If you want to tune on debug mode, set environment variable $UCLOUD_CLI_DEBUG=on 
+
+```
+$ ucloud api --Action <APIName>  --Param1 <value> --Param2 <value> ...
+```
+You can also put those API parameters into a json file, like this.
+```
+$ ucloud api --local-file ./create_uhost.json
+
+//create_uhost.json文件内容
+{
+    "Action":"CreateUHostInstance",
+    "Region":"cn-bj2",
+    "Zone":"cn-bj2-02",
+    "ImageId":"uimage-gk2x3x",
+    "NetworkInterface": [{
+        "EIP":{
+            "Bandwidth":1,
+            "OperatorName":"Bgp",
+            "PayMode": "Bandwidth"
+        }
+    }],
+    "LoginMode":"Password",
+    "Password":"dGVzdGx4ajEy",
+    "CPU":1,
+    "Memory":2048,
+    "Disks":[
+        {
+            "Size":20,
+            "Type":"LOCAL_NORMAL",
+            "IsBoot":"true"
+        }
+    ]
+}
+```

cmd/api.go

@@ -0,0 +1,89 @@
+package cmd
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"strings"
+
+	"github.com/spf13/cobra"
+
+	"github.com/ucloud/ucloud-cli/base"
+)
+
+//NewCmdAPI ucloud api --xkey xvalue
+func NewCmdAPI(out io.Writer) *cobra.Command {
+	return &cobra.Command{
+		Use:   "api",
+		Short: "Call API",
+		Long:  "Call API",
+		Run: func(c *cobra.Command, args []string) {
+			params, err := parseParamsFromCmdLine(args)
+			if err != nil {
+				fmt.Fprintln(out, err)
+				return
+			}
+
+			if params["local-file"] != nil {
+				file, ok := params["local-file"].(string)
+				if !ok {
+					fmt.Fprintf(out, "local-file should be a string\n")
+				}
+				params, err = parseParamsFromJSONFile(file)
+				if err != nil {
+					fmt.Fprintln(out, err)
+					return
+				}
+			}
+
+			req := base.BizClient.UAccountClient.NewGenericRequest()
+			err = req.SetPayload(params)
+			if err != nil {
+				fmt.Fprintf(out, "error: %v\n", err)
+				return
+			}
+
+			resp, err := base.BizClient.UAccountClient.GenericInvoke(req)
+			if err != nil {
+				fmt.Fprintf(out, "error: %v\n", err)
+				return
+			}
+
+			data, err := json.MarshalIndent(resp.GetPayload(), "", "  ")
+			if err != nil {
+				fmt.Fprintf(out, "error: %v\n", err)
+				return
+			}
+			fmt.Fprintln(out, string(data))
+		},
+	}
+}
+
+func parseParamsFromJSONFile(path string) (map[string]interface{}, error) {
+	content, err := ioutil.ReadFile(path)
+	if err != nil {
+		return nil, fmt.Errorf("read file error: %w", err)
+	}
+	params := make(map[string]interface{})
+	err = json.Unmarshal(content, &params)
+	if err != nil {
+		return nil, fmt.Errorf("parse json error: %w", err)
+	}
+	return params, err
+}
+
+func parseParamsFromCmdLine(args []string) (map[string]interface{}, error) {
+	if len(args)%2 != 0 {
+		return nil, errors.New("the key value pairs of api parameters do not match")
+	}
+	params := make(map[string]interface{})
+	for i := 0; i < len(args)-1; i += 2 {
+		if strings.HasPrefix(args[i], "--") {
+			args[i] = args[i][2:]
+		}
+		params[args[i]] = args[i+1]
+	}
+	return params, nil
+}

cmd/root.go

@@ -21,9 +21,8 @@ import (
 
 	"github.com/spf13/cobra"
 
-	"github.com/ucloud/ucloud-sdk-go/ucloud/log"
-
 	"github.com/ucloud/ucloud-cli/base"
+	"github.com/ucloud/ucloud-sdk-go/ucloud/log"
 )
 
 var global = &base.Global
@@ -129,6 +128,7 @@ func addChildren(root *cobra.Command) {
 	root.AddCommand(NewCmdRedis())
 	root.AddCommand(NewCmdMemcache())
 	root.AddCommand(NewCmdExt())
+	root.AddCommand(NewCmdAPI(out))
 	for _, c := range root.Commands() {
 		if c.Name() != "init" && c.Name() != "gendoc" && c.Name() != "config" {
 			c.PersistentFlags().StringVar(&global.PublicKey, "public-key", global.PublicKey, "Set public-key to override the public-key in local config file")
@@ -152,7 +152,22 @@ func Execute() {
 		}
 	}
 	base.InitConfig()
+	mode := os.Getenv("UCLOUD_CLI_DEBUG")
+	if mode == "on" || global.Debug {
+		base.ClientConfig.LogLevel = log.DebugLevel
+		base.BizClient = base.NewClient(base.ClientConfig, base.AuthCredential)
+	}
+
 	addChildren(cmd)
+
+	targetCmd, flags, err := cmd.Find(os.Args[1:])
+	if err == nil {
+		if targetCmd.Use == "api" {
+			targetCmd.Run(targetCmd, flags)
+			return
+		}
+	}
+
 	if err := cmd.Execute(); err != nil {
 		os.Exit(1)
 	}
@@ -220,12 +235,6 @@ func initialize(cmd *cobra.Command) {
 		base.ClientConfig.Zone = zone
 	}
 
-	mode := os.Getenv("UCLOUD_CLI_DEBUG")
-	if mode == "on" || global.Debug {
-		base.ClientConfig.LogLevel = log.DebugLevel
-		base.BizClient = base.NewClient(base.ClientConfig, base.AuthCredential)
-	}
-
 	if (cmd.Name() != "config" && cmd.Name() != "init" && cmd.Name() != "version") && (cmd.Parent() != nil && cmd.Parent().Name() != "config") {
 		if base.InCloudShell {
 			return