perf(config):Add list_policies switch

404tk · 404tk · commit fe9d8436dc39 · 2023-12-04T22:31:36.000+08:00
diff --git a/pkg/providers/alibaba/ram/users.go b/pkg/providers/alibaba/ram/users.go
@@ -5,6 +5,7 @@ import (
 	"time"
 
 	"github.com/404tk/cloudtoolkit/pkg/schema"
+	"github.com/404tk/cloudtoolkit/utils"
 	"github.com/404tk/cloudtoolkit/utils/logger"
 	"github.com/aliyun/alibaba-cloud-sdk-go/sdk"
 	"github.com/aliyun/alibaba-cloud-sdk-go/sdk/auth/credentials"
@@ -77,7 +78,9 @@ func (d *Driver) GetRamUser(ctx context.Context) ([]schema.User, error) {
 				}
 			}
 
-			_user.Policies = listPoliciesForUser(client, _user.UserName)
+			if utils.ListPolicies {
+				_user.Policies = listPoliciesForUser(client, _user.UserName)
+			}
 
 			list = append(list, _user)
 			select {
diff --git a/pkg/providers/tencent/cam/users.go b/pkg/providers/tencent/cam/users.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 
 	"github.com/404tk/cloudtoolkit/pkg/schema"
+	"github.com/404tk/cloudtoolkit/utils"
 	"github.com/404tk/cloudtoolkit/utils/logger"
 	cam "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/cam/v20190116"
 	"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
@@ -46,7 +47,9 @@ func (d *Driver) GetCamUser(ctx context.Context) ([]schema.User, error) {
 		if *user.ConsoleLogin == 1 {
 			_user.EnableLogin = true
 		}
-		_user.Policies = listAttachedUserAllPolicies(client, user.Uin)
+		if utils.ListPolicies {
+			_user.Policies = listAttachedUserAllPolicies(client, user.Uin)
+		}
 
 		list = append(list, _user)
 	}
diff --git a/runner/banner.go b/runner/banner.go
@@ -12,7 +12,7 @@ const banner = `
 `
 
 // Version is the current version of cloudtoolkit
-const Version = `0.1.2`
+const Version = `0.1.3`
 
 // showBanner is used to show the banner to the user
 func ShowBanner() {
diff --git a/runner/init.go b/runner/init.go
@@ -24,6 +24,7 @@ func InitConfig() {
 	}
 
 	utils.DoSave = viper.GetBool("common.log_enable")
+	utils.ListPolicies = viper.GetBool("common.list_policies")
 	utils.LogDir = viper.GetString("common.log_dir")
 	utils.Cloudlist = viper.GetStringSlice("cloudlist")
 
@@ -40,7 +41,8 @@ func InitConfig() {
 }
 
 const defaultConfigFile = `common:
-  log_enable: true
+  log_enable: false
+  list_policies: false
   log_dir: logs
 
 cloudlist:
diff --git a/utils/const.go b/utils/const.go
@@ -29,6 +29,7 @@ const (
 
 var (
 	DoSave       bool
+	ListPolicies bool
 	LogDir       string
 	Cloudlist    []string
 	BackdoorUser string

Original file line number	Diff line number	Diff line change
@@ -12,7 +12,7 @@ const banner = `
`12`	`12`	`
`13`	`13`
`14`	`14`	`// Version is the current version of cloudtoolkit`
`15`		-const Version = `0.1.2`
	`15`	+const Version = `0.1.3`
`16`	`16`
`17`	`17`	`// showBanner is used to show the banner to the user`
`18`	`18`	`func ShowBanner() {`
Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,7 @@ func InitConfig() {`
`24`	`24`	`}`
`25`	`25`
`26`	`26`	`utils.DoSave = viper.GetBool("common.log_enable")`
	`27`	`+ utils.ListPolicies = viper.GetBool("common.list_policies")`
`27`	`28`	`utils.LogDir = viper.GetString("common.log_dir")`
`28`	`29`	`utils.Cloudlist = viper.GetStringSlice("cloudlist")`
`29`	`30`
`@@ -40,7 +41,8 @@ func InitConfig() {`
`40`	`41`	`}`
`41`	`42`
`42`	`43`	const defaultConfigFile = `common:
`43`		`- log_enable: true`
	`44`	`+ log_enable: false`
	`45`	`+ list_policies: false`
`44`	`46`	`log_dir: logs`
`45`	`47`
`46`	`48`	`cloudlist:`