add --vpc support

Author: JacksonTian

config/configure.go

@@ -46,9 +46,7 @@ func NewConfigureCommand() *cli.Command {
 			if len(args) > 0 {
 				return cli.NewInvalidCommandError(args[0], ctx)
 			}
-			profileName, _ := ProfileFlag(ctx.Flags()).GetValue()
-			mode, _ := ModeFlag(ctx.Flags()).GetValue()
-			return doConfigure(ctx, profileName, mode)
+			return doConfigure(ctx)
 		},
 	}
 
@@ -59,14 +57,17 @@ func NewConfigureCommand() *cli.Command {
 	return c
 }
 
-func doConfigure(ctx *cli.Context, profileName string, mode string) error {
+func doConfigure(ctx *cli.Context) error {
 	w := ctx.Writer()
 
 	conf, err := hookLoadConfiguration(LoadConfiguration)(GetConfigPath() + "/" + configFile)
 	if err != nil {
 		return err
 	}
 
+	profileName, _ := ProfileFlag(ctx.Flags()).GetValue()
+	mode, _ := ModeFlag(ctx.Flags()).GetValue()
+
 	if profileName == "" {
 		if conf.CurrentProfile == "" {
 			profileName = "default"

config/flags.go

@@ -21,6 +21,7 @@ import (
 const (
 	ProfileFlagName         = "profile"
 	ModeFlagName            = "mode"
+	VpcFlagName             = "vpc"
 	AccessKeyIdFlagName     = "access-key-id"
 	AccessKeySecretFlagName = "access-key-secret"
 	StsTokenFlagName        = "sts-token"
@@ -62,6 +63,7 @@ func AddFlags(fs *cli.FlagSet) {
 	fs.Add(NewSkipSecureVerify())
 	fs.Add(NewExpiredSecondsFlag())
 	fs.Add(NewProcessCommandFlag())
+	fs.Add(NewVpcFlag())
 }
 
 func ConnectTimeoutFlag(fs *cli.FlagSet) *cli.Flag {
@@ -76,6 +78,10 @@ func ModeFlag(fs *cli.FlagSet) *cli.Flag {
 	return fs.Get(ModeFlagName)
 }
 
+func VpcFlag(fs *cli.FlagSet) *cli.Flag {
+	return fs.Get(VpcFlagName)
+}
+
 func AccessKeyIdFlag(fs *cli.FlagSet) *cli.Flag {
 	return fs.Get(AccessKeyIdFlagName)
 }
@@ -131,12 +137,15 @@ func RetryCountFlag(fs *cli.FlagSet) *cli.Flag {
 func SkipSecureVerify(fs *cli.FlagSet) *cli.Flag {
 	return fs.Get(SkipSecureVerifyName)
 }
+
 func ConfigurePathFlag(fs *cli.FlagSet) *cli.Flag {
 	return fs.Get(ConfigurePathFlagName)
 }
+
 func ExpiredSecondsFlag(fs *cli.FlagSet) *cli.Flag {
 	return fs.Get(ExpiredSecondsFlagName)
 }
+
 func ProcessCommandFlag(fs *cli.FlagSet) *cli.Flag {
 	return fs.Get(ProcessCommandFlagName)
 }
@@ -170,6 +179,16 @@ func NewModeFlag() *cli.Flag {
 			"使用 `--mode {AK|StsToken|RamRoleArn|EcsRamRole|RsaKeyPair|RamRoleArnWithRoleName}` 指定认证方式")}
 }
 
+func NewVpcFlag() *cli.Flag {
+	return &cli.Flag{
+		Category:     "config",
+		Name:         VpcFlagName,
+		AssignedMode: cli.AssignedNone,
+		Short: i18n.T(
+			"use `--vpc` to enable vpc endpoint",
+			"使用 `--vpc` 启用 VPC 接入点地址")}
+}
+
 func NewAccessKeyIdFlag() *cli.Flag {
 	return &cli.Flag{
 		Category:     "config",

config/hello.go

@@ -24,7 +24,11 @@ import (
 func DoHello(ctx *cli.Context, profile *Profile) {
 	w := ctx.Writer()
 
+	vpc := VpcFlag(ctx.Flags()).IsAssigned()
 	client, err := profile.GetClient(ctx)
+	if vpc {
+		client.Network = "vpc"
+	}
 
 	if err != nil {
 		cli.Println(w, "-----------------------------------------------")
@@ -36,6 +40,7 @@ func DoHello(ctx *cli.Context, profile *Profile) {
 		cli.Println(w, "-----------------------------------------------")
 		return
 	}
+
 	request := ecs.CreateDescribeRegionsRequest()
 	response := ecs.CreateDescribeRegionsResponse()
 	if vendorEnv, ok := os.LookupEnv("ALIBABA_CLOUD_VENDOR"); ok {

openapi/flags.go

@@ -21,6 +21,7 @@ import (
 func AddFlags(fs *cli.FlagSet) {
 	fs.Add(NewSecureFlag())
 	fs.Add(NewForceFlag())
+	// fs.Add(NewVpcFlag())
 	fs.Add(NewEndpointFlag())
 	fs.Add(NewVersionFlag())
 	fs.Add(NewHeaderFlag())
@@ -38,6 +39,7 @@ func AddFlags(fs *cli.FlagSet) {
 const (
 	SecureFlagName   = "secure"
 	ForceFlagName    = "force"
+	VpcFlagName      = "vpc"
 	EndpointFlagName = "endpoint"
 	VersionFlagName  = "version"
 	HeaderFlagName   = "header"
@@ -62,6 +64,10 @@ func ForceFlag(fs *cli.FlagSet) *cli.Flag {
 	return fs.Get(ForceFlagName)
 }
 
+func VpcFlag(fs *cli.FlagSet) *cli.Flag {
+	return fs.Get(VpcFlagName)
+}
+
 func EndpointFlag(fs *cli.FlagSet) *cli.Flag {
 	return fs.Get(EndpointFlagName)
 }
@@ -125,6 +131,14 @@ func NewForceFlag() *cli.Flag {
 			"添加 `--force` 开关可跳过API与参数的合法性检查")}
 }
 
+func NewVpcFlag() *cli.Flag {
+	return &cli.Flag{Category: "caller",
+		Name: VpcFlagName, AssignedMode: cli.AssignedOnce,
+		Short: i18n.T(
+			"use `--vpc` to enable vpc endpoint",
+			"使用 `--vpc` 来启用 VPC 接入点地址")}
+}
+
 func NewEndpointFlag() *cli.Flag {
 	return &cli.Flag{Category: "caller",
 		Name: EndpointFlagName, AssignedMode: cli.AssignedOnce,