82 lines
2.6 KiB
Go
82 lines
2.6 KiB
Go
|
package postbank
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Cmd go run main.go postbank
|
||
|
|
var Cmd = &cobra.Command{
|
||
|
|
Use: "postbank",
|
||
|
|
Short: "邮储国密SM辅助工具",
|
||
|
|
}
|
||
|
|
|
||
|
|
// merchantId, privateKey, sopPublicKey, inputJson, true
|
||
|
|
const (
|
||
|
|
merchantIdKey = "merchantId"
|
||
|
|
privateKeyKey = "priKey"
|
||
|
|
sopPublicKeyKey = "sopPublicKey"
|
||
|
|
inputKey = "input"
|
||
|
|
isRequestKey = "isRequest"
|
||
|
|
accessTokenKey = "accessToken"
|
||
|
|
|
||
|
|
shortMerchantIdKey = "m"
|
||
|
|
shortPrivateKeyKey = "p"
|
||
|
|
shortSopPublicKeyKey = "s"
|
||
|
|
shortInputKey = "i"
|
||
|
|
shortIsRequestKey = "r"
|
||
|
|
shortAccessTokenKey = "a"
|
||
|
|
|
||
|
|
merchantIdContextKey = "merchantId"
|
||
|
|
privateKeyContextKey = "priKey"
|
||
|
|
sopPublicKeyContextKey = "sopPublicKey"
|
||
|
|
inputContextKey = "input"
|
||
|
|
isRequestContextKey = "isRequest"
|
||
|
|
accessTokenContextKey = "accessToken"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
encryptCmd.PersistentFlags().StringP(merchantIdKey, shortMerchantIdKey, "", "商户号")
|
||
|
|
encryptCmd.PersistentFlags().StringP(privateKeyKey, shortPrivateKeyKey, "", "私钥")
|
||
|
|
encryptCmd.PersistentFlags().StringP(sopPublicKeyKey, shortSopPublicKeyKey, "", "对方公钥")
|
||
|
|
encryptCmd.PersistentFlags().StringP(inputKey, shortInputKey, "", "待加密的JSON字符串")
|
||
|
|
encryptCmd.PersistentFlags().BoolP(isRequestKey, shortIsRequestKey, false, "请求加密")
|
||
|
|
encryptCmd.PersistentFlags().StringP(accessTokenKey, shortAccessTokenKey, "", "accessToken")
|
||
|
|
|
||
|
|
decryptCmd.PersistentFlags().StringP(merchantIdKey, shortMerchantIdKey, "", "商户号")
|
||
|
|
decryptCmd.PersistentFlags().StringP(privateKeyKey, shortPrivateKeyKey, "", "私钥")
|
||
|
|
decryptCmd.PersistentFlags().StringP(sopPublicKeyKey, shortSopPublicKeyKey, "", "对方公钥")
|
||
|
|
decryptCmd.PersistentFlags().StringP(inputKey, shortInputKey, "", "待解密的JSON字符串")
|
||
|
|
decryptCmd.PersistentFlags().BoolP(isRequestKey, shortIsRequestKey, false, "请求加密")
|
||
|
|
|
||
|
|
Cmd.AddCommand(encryptCmd, decryptCmd, generateCmd)
|
||
|
|
}
|
||
|
|
|
||
|
|
func forceArgsToStringContext(cmd *cobra.Command, key, ctxKey string) error {
|
||
|
|
data, err := cmd.Flags().GetString(key)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("缺少参数:<%s>%w", key, err)
|
||
|
|
}
|
||
|
|
cmd.SetContext(context.WithValue(cmd.Context(), ctxKey, data))
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func argsToStringContext(cmd *cobra.Command, key, ctxKey string) error {
|
||
|
|
data, err := cmd.Flags().GetString(key)
|
||
|
|
if err != nil {
|
||
|
|
data = ""
|
||
|
|
}
|
||
|
|
cmd.SetContext(context.WithValue(cmd.Context(), ctxKey, data))
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func forceArgsToBoolContext(cmd *cobra.Command, key, ctxKey string) error {
|
||
|
|
data, err := cmd.Flags().GetBool(key)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("缺少参数:<%s>%w", key, err)
|
||
|
|
}
|
||
|
|
cmd.SetContext(context.WithValue(cmd.Context(), ctxKey, data))
|
||
|
|
return nil
|
||
|
|
}
|