cmd: add signature command to calculate ucloud signature (#63)

Author: fioncat

CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.1.40 (2022-08-31)
+
+* add `signature` command to calculate signature quickly.
+
 ## 0.1.39 (2022-06-13)
 
 * fix init failure when user donot have a default project in ucloud console.

Makefile

@@ -1,4 +1,4 @@
-export VERSION=0.1.39
+export VERSION=0.1.40
 GOFMT_FILES?=$$(find . -name '*.go' |grep -v vendor)
 
 .PHONY : install

cmd/pathx.go

@@ -174,7 +174,7 @@ func NewCmdUGA3Create(out io.Writer) *cobra.Command {
 	bindRegion(createPathxReq, flags)
 	bindZone(createPathxReq, flags)
 
-	createPathxReq.Bandwidth = flags.String("bandwidth", "",
+	createPathxReq.Bandwidth = flags.Int("bandwidth", 0,
 		"Required. Shared bandwidth of the resource")
 	flags.String("area-code", "",
 		"Optional. When it is empty,the nearest zone will be selected based on the origin-domain and origin-ip. "+

cmd/root.go

@@ -129,6 +129,7 @@ func addChildren(root *cobra.Command) {
 	root.AddCommand(NewCmdMemcache())
 	root.AddCommand(NewCmdExt())
 	root.AddCommand(NewCmdAPI(out))
+	root.AddCommand(NewCmdSignature())
 	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")

cmd/signature.go

@@ -0,0 +1,93 @@
+package cmd
+
+import (
+	"bytes"
+	"fmt"
+	"net/url"
+	"strings"
+
+	"github.com/fatih/color"
+	"github.com/spf13/cobra"
+	"github.com/ucloud/ucloud-sdk-go/ucloud/auth"
+)
+
+func NewCmdSignature() *cobra.Command {
+	var (
+		rawParams  []string
+		privateKey string
+		rawURL     string
+	)
+	cmd := &cobra.Command{
+		Use:   "signature",
+		Short: "Calculate ucloud signature",
+		Long:  "Calculate ucloud signature",
+
+		Aliases: []string{"sign"},
+
+		Run: func(cmd *cobra.Command, args []string) {
+			var params map[string]interface{}
+			if rawURL != "" {
+				// Parse params from exists url
+				parsedURL, err := url.Parse(rawURL)
+				if err != nil {
+					fmt.Printf("error: failed to parse url %q: %v\n", rawURL, err)
+					return
+				}
+				query := parsedURL.Query()
+				params = make(map[string]interface{}, len(query))
+				for key, values := range query {
+					if key == "Signature" {
+						fmt.Println("error: the `Signature` cannot be placed in url")
+						return
+					}
+					if len(values) == 0 {
+						continue
+					}
+					val := values[0]
+					params[key] = val
+				}
+			}
+			if len(rawParams) > 0 {
+				if params == nil {
+					params = make(map[string]interface{}, len(rawParams))
+				}
+				for _, rawParam := range rawParams {
+					kv := strings.Split(rawParam, "=")
+					if len(kv) != 2 {
+						fmt.Printf("error: param %q is invalid\n", rawParam)
+						return
+					}
+					params[kv[0]] = kv[1]
+				}
+			}
+			if len(params) == 0 {
+				fmt.Println("error: missing param")
+				return
+			}
+
+			r := auth.CalculateSignature(params, privateKey)
+
+			var colorParamBuf bytes.Buffer
+			for _, key := range r.SortedKeys {
+				val := params[key]
+				colorParamBuf.WriteString(color.GreenString(key))
+				colorParamBuf.WriteString(color.CyanString("%v", val))
+			}
+			colorParamBuf.WriteString(color.MagentaString(privateKey))
+			fmt.Println("")
+			fmt.Printf("ParamStr: %s\n", colorParamBuf.String())
+			fmt.Println("")
+
+			fmt.Printf("Signature: %s\n", color.BlueString(r.Sign))
+		},
+	}
+
+	flags := cmd.Flags()
+	flags.SortFlags = false
+	flags.StringArrayVarP(&rawParams, "param", "m", nil, "Request params")
+	flags.StringVarP(&privateKey, "private-key", "k", "", "Private key")
+	flags.StringVarP(&rawURL, "url", "u", "", "Request url without signature")
+	cmd.MarkFlagRequired("private-key")
+
+	return cmd
+}

go.mod

@@ -3,12 +3,13 @@ module github.com/ucloud/ucloud-cli
 go 1.12
 
 require (
+	github.com/fatih/color v1.13.0
 	github.com/kr/pretty v0.1.0 // indirect
 	github.com/satori/go.uuid v1.2.0
 	github.com/sirupsen/logrus v1.3.0
 	github.com/spf13/cobra v0.0.3
 	github.com/spf13/pflag v1.0.3
-	github.com/ucloud/ucloud-sdk-go v0.21.21
+	github.com/ucloud/ucloud-sdk-go v0.21.42
 	golang.org/x/sys v0.0.0-20220422013727-9388b58f7150
 	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect
 )

go.sum

@@ -6,6 +6,8 @@ github.com/cpuguy83/go-md2man v1.0.10 h1:BSKMNlYxDvnunlTymqtgONjNnaRV1sTpcovwwjF
 github.com/cpuguy83/go-md2man v1.0.10/go.mod h1:SmD6nW6nTyfqj6ABTjUi3V3JVMnlJmwcJI5acqYI6dE=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/fatih/color v1.13.0 h1:8LOYc1KYPPmyKMuN8QV2DNRWNbLo6LZ0iLs8+mlH53w=
+github.com/fatih/color v1.13.0/go.mod h1:kLAiJbzzSOZDVNGyDpeOxJ47H46qBXwg5ILebYFFOfk=
 github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
 github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=
 github.com/inconshreveable/mousetrap v1.0.0 h1:Z8tu5sraLXCXIcARxBp/8cbvlwVa7Z1NHg9XEKhtSvM=
@@ -22,6 +24,11 @@ github.com/lixiaojun629/cobra v0.0.10/go.mod h1:6VKYqzoixuRlMBmzm3rHPS0sRYVhT3zX
 github.com/lixiaojun629/pflag v1.0.5 h1:plFJ2SBJd2S2Fc7ZwwFZ3682IvxBiUkhRuJS40OvEMs=
 github.com/lixiaojun629/pflag v1.0.5/go.mod h1:uchrjsiFxJj1XOBpO4YJCZwpqXHsCHovxY91tyFoUrg=
 github.com/magiconair/properties v1.8.0/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ=
+github.com/mattn/go-colorable v0.1.9 h1:sqDoxXbdeALODt0DAeJCVp38ps9ZogZEAXjus69YV3U=
+github.com/mattn/go-colorable v0.1.9/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc=
+github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
+github.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9Y=
+github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94=
 github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=
 github.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
 github.com/pelletier/go-toml v1.2.0/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic=
@@ -42,8 +49,8 @@ github.com/spf13/viper v1.3.2/go.mod h1:ZiWeW+zYFKm7srdB9IoDzzZXaJaI5eL9QjNiN/DM
 github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w=
 github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
-github.com/ucloud/ucloud-sdk-go v0.21.21 h1:c4UtP8dR3vfYY9bzxaOtRc3L1jdv9R7GteomTqI3vV8=
-github.com/ucloud/ucloud-sdk-go v0.21.21/go.mod h1:dyLmFHmUfgb4RZKYQP9IArlvQ2pxzFthfhwxRzOEPIw=
+github.com/ucloud/ucloud-sdk-go v0.21.42 h1:WGFpTRrnAL6yEZVJsZxNH5kSMNQNCYBsB8pq159QRG4=
+github.com/ucloud/ucloud-sdk-go v0.21.42/go.mod h1:dyLmFHmUfgb4RZKYQP9IArlvQ2pxzFthfhwxRzOEPIw=
 github.com/ugorji/go/codec v0.0.0-20181204163529-d75b2dcb6bc8/go.mod h1:VFNgLljTbGfSG7qAOspJ7OScBnGdDN/yBr0sguwnwf0=
 github.com/xordataexchange/crypt v0.0.3-0.20170626215501-b2862e3d0a77/go.mod h1:aYKd//L2LvnjZzWKhF00oedf4jCCReLcmhLdhm1A27Q=
 golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
@@ -56,6 +63,9 @@ golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5h
 golang.org/x/sys v0.0.0-20181205085412-a5c9d58dba9a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220422013727-9388b58f7150 h1:xHms4gcpe1YE7A3yIllJXP16CMAGuqwO2lX1mTyyRRc=
 golang.org/x/sys v0.0.0-20220422013727-9388b58f7150/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=