Skip to content

Commit

Permalink
chore: some improvements
Browse files Browse the repository at this point in the history
  • Loading branch information
@cnlancehu
cnlancehu committed Aug 29, 2023
1 parent e1c6a5b commit 80317b0
Show file tree
Hide file tree
Showing 2 changed files with 49 additions and 41 deletions.
81 changes: 44 additions & 37 deletions main.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ import (
"net/http"
"os"
"reflect"
"runtime"
"sort"
"strconv"
"strings"
Expand All @@ -22,26 +23,26 @@ import (
)

var (
version string = "0.0.0" // 程序版本号
goos string = "windows" // 程序运行的操作系统
goarch string = "arch" // 程序运行的操作系统架构
buildTime string = "time" // 程序编译时间
action string // 程序运行的行为
notifa bool // 是否启用 windows 通知
notifyHelpMsg string = "" // 帮助信息中的通知信息
ua string = "qcip/" + version // 请求的 User-Agent
confPath string = "config.json" // 默认配置文件路径
errmsglist map[int]string
errhandletimes int = 0
httpClient *http.Client = &http.Client{
version string = "0.0.0" // 程序版本号
goos string = runtime.GOOS // 程序运行的操作系统
goarch string = runtime.GOARCH // 程序运行的操作系统架构
buildTime string = "buildTime" // 程序编译时间
action string // 程序运行的行为
notifa bool // 是否启用 windows 通知
notifyHelpMsg string = "" // 帮助信息中的通知信息
ua string = "qcip/" + version // 请求的 User-Agent
confPath string = "config.json" // 默认配置文件路径
errmsglist map[int]string // 错误信息列表
errhandletimes int = 0 // 错误输出的次数
httpClient *http.Client = &http.Client{
Timeout: time.Second * 10,
Transport: &http.Transport{
DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
return net.Dial("tcp4", addr)
},
},
}
notify = func(title, msg string, succeed bool) {} // 禁用的通知函数
notify = func(title, msg string, succeed bool) {} // 默认禁用的通知函数
)

type Config struct {
Expand Down Expand Up @@ -277,11 +278,11 @@ func getconfig(confPath string) Config {
errhandle("Config error:")
for _, key := range requiredKeys {
if _, ok := reflect.TypeOf(configData).FieldByName(key); !ok {
errhandle(" " + key + " not found")
errhandle("\t" + key + " not found")
} else if reflect.ValueOf(configData).FieldByName(key).String() == "" {
errhandle(" " + key + " is empty")
errhandle("\t" + key + " is empty")
} else if reflect.ValueOf(configData).FieldByName(key).String() == key {
errhandle(" " + key + " is incorrect")
errhandle("\t" + key + " is incorrect")
}
}
errexit()
Expand Down Expand Up @@ -310,7 +311,7 @@ func getip(api string, maxretries int) string {
failed = true
if i == 0 {
errhandle("IP API calling error:")
errhandle(" Error detail: " + err.Error())
errhandle("\tError detail: " + err.Error())
continue
}
if maxretries != 0 {
Expand All @@ -333,7 +334,7 @@ func getip(api string, maxretries int) string {
err := Body.Close()
if err != nil {
errhandle("IP API calling error")
errhandle(" Error detail: " + err.Error())
errhandle("\tError detail: " + err.Error())
errexit()
return
}
Expand All @@ -352,7 +353,7 @@ func getip(api string, maxretries int) string {
err := json.Unmarshal(fetchapi("https://myip.ipip.net/ip"), &r)
if err != nil {
errhandle("IP API calling error: " + err.Error())
errhandle(" Error detail: " + err.Error())
errhandle("\tError detail: " + err.Error())
errexit()
}
return r.IP
Expand All @@ -370,6 +371,7 @@ func getip(api string, maxretries int) string {
// 轻量应用服务器部分
func lhgetrules(credential *common.Credential, InstanceRegion string, InstanceId string) []*lighthouse.FirewallRuleInfo {
cpf := profile.NewClientProfile()
cpf.NetworkFailureMaxRetries = 3
cpf.HttpProfile.Endpoint = "lighthouse.tencentcloudapi.com"
client, _ := lighthouse.NewClient(credential, InstanceRegion, cpf)
request := lighthouse.NewDescribeFirewallRulesRequest()
Expand All @@ -379,7 +381,7 @@ func lhgetrules(credential *common.Credential, InstanceRegion string, InstanceId
response, err := client.DescribeFirewallRules(request)
if _, ok := err.(*errors.TencentCloudSDKError); ok {
errhandle("Error while fetching rules for lighthouse:")
errhandle(" " + err.Error())
errhandle("\t" + err.Error())
errexit()
}
return response.Response.FirewallRuleSet
Expand Down Expand Up @@ -414,6 +416,7 @@ func lhmodifyrules(credential *common.Credential, InstanceRegion string, Instanc
}
}
cpf := profile.NewClientProfile()
cpf.NetworkFailureMaxRetries = 3
cpf.HttpProfile.Endpoint = "lighthouse.tencentcloudapi.com"
client, _ := lighthouse.NewClient(credential, InstanceRegion, cpf)
request := lighthouse.NewModifyFirewallRulesRequest()
Expand All @@ -422,7 +425,7 @@ func lhmodifyrules(credential *common.Credential, InstanceRegion string, Instanc
_, err := client.ModifyFirewallRules(request)
if _, ok := err.(*errors.TencentCloudSDKError); ok {
errhandle("Error while modifying rules for lighthouse:")
errhandle(" " + err.Error())
errhandle("\t" + err.Error())
errexit()
return
}
Expand All @@ -431,14 +434,15 @@ func lhmodifyrules(credential *common.Credential, InstanceRegion string, Instanc
// 云服务器安全组部分
func sggetrules(credential *common.Credential, SecurityGroupId string, SecurityGroupRegion string) *vpc.SecurityGroupPolicySet {
cpf := profile.NewClientProfile()
cpf.NetworkFailureMaxRetries = 3
cpf.HttpProfile.Endpoint = "vpc.tencentcloudapi.com"
client, _ := vpc.NewClient(credential, SecurityGroupRegion, cpf)
request := vpc.NewDescribeSecurityGroupPoliciesRequest()
request.SecurityGroupId = common.StringPtr(SecurityGroupId)
response, err := client.DescribeSecurityGroupPolicies(request)
if _, ok := err.(*errors.TencentCloudSDKError); ok {
errhandle("Error while fetching rules for security group:")
errhandle(" " + err.Error())
errhandle("\t" + err.Error())
errexit()
}
return response.Response.SecurityGroupPolicySet
Expand All @@ -463,6 +467,7 @@ func sgmatch(rules *vpc.SecurityGroupPolicySet, ip string, config Config) (*vpc.

func sgmodifyrules(credential *common.Credential, SecurityGroupId string, SecurityGroupRegion string, rules *vpc.SecurityGroupPolicySet) {
cpf := profile.NewClientProfile()
cpf.NetworkFailureMaxRetries = 3
cpf.HttpProfile.Endpoint = "vpc.tencentcloudapi.com"
client, _ := vpc.NewClient(credential, SecurityGroupRegion, cpf)
request := vpc.NewModifySecurityGroupPoliciesRequest()
Expand All @@ -483,7 +488,7 @@ func sgmodifyrules(credential *common.Credential, SecurityGroupId string, Securi
_, err := client.ModifySecurityGroupPolicies(request)
if _, ok := err.(*errors.TencentCloudSDKError); ok {
errhandle("Error while modifying rules for security group:")
errhandle(" " + err.Error())
errhandle("\t" + err.Error())
errexit()
}
}
Expand Down Expand Up @@ -513,25 +518,27 @@ func processRules(ptrRules interface{}) interface{} {
}

func errhandle(errmsg string) {
errhandletimes++
errmsglist[errhandletimes] = errmsg
if notifa {
errhandletimes++
errmsglist[errhandletimes] = errmsg
}
fmt.Printf("\033[31m%s\033[0m\n", errmsg)
}

func errexit() {
var (
keys []int
allerrmsg string
)
for k := range errmsglist {
keys = append(keys, k)
}
sort.Ints(keys)
for _, k := range keys {
allerrmsg += errmsglist[k] + "\n"
}
allerrmsg = strings.ReplaceAll(allerrmsg, "\t", " ")
if notifa {
var (
keys []int
allerrmsg string
)
for k := range errmsglist {
keys = append(keys, k)
}
sort.Ints(keys)
for _, k := range keys {
allerrmsg += errmsglist[k] + "\n"
}
allerrmsg = strings.ReplaceAll(allerrmsg, "\t", " ")
notify("QCIP | Error", allerrmsg, false)
}
os.Exit(1)
Expand Down
9 changes: 5 additions & 4 deletions winnotify.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,14 +14,14 @@ var (
successicon []byte
//go:embed static/failed.ico
failedicon []byte
err error
err error = nil
)

func init() {
// 系统为 windows 时，启用 winnotify 并显示对应的帮助信息
var erroroccurred bool = false
notifyHelpMsg = "\n -n, --winnotify Send notifacation cards, only available on Windows"
notifyerrhandle := func(err error) {
notifyerrhandle := func() {
if err != nil {
err = nil
erroroccurred = true
Expand All @@ -30,14 +30,14 @@ func init() {
notify = func(title string, message string, succeed bool) {
var tmpFile *os.File
tmpFile, err = os.CreateTemp("", "qcip-*.ico")
notifyerrhandle(err)
notifyerrhandle()
if succeed {
tmpFile.Write(successicon)
} else {
tmpFile.Write(failedicon)
}
err = tmpFile.Close()
notifyerrhandle(err)
notifyerrhandle()
iconpath := tmpFile.Name()
var notification toast.Notification
if !erroroccurred {
Expand All @@ -55,6 +55,7 @@ func init() {
}
}
err = notification.Push()
notifyerrhandle()
if erroroccurred {
errhandle("Error occurred when sending notification cards")
}
Expand Down

0 comments on commit 80317b0

Please sign in to comment.