Skip to content

Commit

Permalink
make lint happy
Browse files Browse the repository at this point in the history
  • Loading branch information
fumiama committed Aug 1, 2023
1 parent f466ca7 commit dae0378
Show file tree
Hide file tree
Showing 2 changed files with 55 additions and 57 deletions.
109 changes: 53 additions & 56 deletions cmd/gocq/login.go
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ import (
"strconv"
"strings"
"sync"
"sync/atomic"
"time"

"github.com/Mrs4s/MiraiGo/client"
Expand Down Expand Up @@ -398,52 +399,52 @@ func refreshToken(uin string) error {
}
msg := gjson.GetBytes(resp, "msg")
if gjson.GetBytes(resp, "code").Int() != 0 {
return errors.New(msg)
return errors.New(msg.String())
}
go callback(uin, gjson.GetBytes(resp, "data").Array(), "request token")
return true
return nil
}

var missTokenCount = uint64(0)

func sign(seq uint64, uin string, cmd string, qua string, buff []byte) (sign []byte, extra []byte, token []byte, err error) {
i := 0
for {
sign, extra, token, err = _sign(seq, uin, cmd, qua, buff)
if err != nil {
log.Warnf("获取sso sign时出现错误: %v server: %v", err, base.SignServer)
}
if i > 0 {
break
}
i++
if (!base.IsBelow110) && base.Account.AutoRegister && err == nil && len(sign) == 0 {
if registerLock.TryLock() { // 避免并发时多处同时销毁并重新注册
log.Warn("获取签名为空,实例可能丢失,正在尝试重新注册")
defer registerLock.Unlock()
err := destroySignServer(uin)
if err != nil {
log.Warnf(err)
return nil, nil, nil, err
}
register(base.Account.Uin, device.AndroidId, device.Guid, device.QImei36, base.Key)
}
continue
}
if (!base.IsBelow110) && base.Account.AutoRefreshToken && len(token) == 0 {
log.Warnf("token 已过期, 总丢失 token 次数为 %v", atomic.AddUint64(&missTokenCount, 1))
if registerLock.TryLock() {
defer registerLock.Unlock()
if err := refreshToken(uin); err != nil {
log.Warnf("刷新 token 出现错误: %v server: %v", err, base.SignServer)
} else {
log.Info("刷新 token 成功")
}
}
continue
}
break
}
i := 0
for {
sign, extra, token, err = _sign(seq, uin, cmd, qua, buff)
if err != nil {
log.Warnf("获取sso sign时出现错误: %v server: %v", err, base.SignServer)
}
if i > 0 {
break
}
i++
if (!base.IsBelow110) && base.Account.AutoRegister && err == nil && len(sign) == 0 {
if registerLock.TryLock() { // 避免并发时多处同时销毁并重新注册
log.Warn("获取签名为空,实例可能丢失,正在尝试重新注册")
defer registerLock.Unlock()
err := destroySignServer(uin)
if err != nil {
log.Warnln(err)
return nil, nil, nil, err
}
register(base.Account.Uin, device.AndroidId, device.Guid, device.QImei36, base.Key)
}
continue
}
if (!base.IsBelow110) && base.Account.AutoRefreshToken && len(token) == 0 {
log.Warnf("token 已过期, 总丢失 token 次数为 %v", atomic.AddUint64(&missTokenCount, 1))
if registerLock.TryLock() {
defer registerLock.Unlock()
if err := refreshToken(uin); err != nil {
log.Warnf("刷新 token 出现错误: %v server: %v", err, base.SignServer)
} else {
log.Info("刷新 token 成功")
}
}
continue
}
break
}
return sign, extra, token, err
}

Expand All @@ -453,24 +454,20 @@ func destroySignServer(uin string) error {
signServer += "/"
}
signVersion, err := getSignServerVersion()
if err != nil {
return errors.Wrapf(err, "获取签名服务版本出现错误, server: %v", signServer)
}
if err != nil {
return errors.Wrapf(err, "获取签名服务版本出现错误, server: %v", signServer)
}
if global.VersionNameCompare("v"+signVersion, "v1.1.6") {
return errors.Errorf("当前签名服务器版本 %v 低于 1.1.6,无法使用 destroy 接口", signVersion)
}
resp, err := download.Request{
Method: http.MethodGet,
URL: signServer + "destroy" + fmt.Sprintf("?uin=%v&key=%v", uin, base.Key),
}.Bytes()
if err != nil {
if err != nil || gjson.GetBytes(resp, "code").Int() != 0 {
return errors.Wrapf(err, "destroy 实例出现错误, server: %v", signServer)
}
msg := gjson.GetBytes(resp, "msg")
if gjson.GetBytes(resp, "code").Int() != 0 {
return errors.Wrapf(err, "destroy 实例出现错误, server: %v", signServer)
}
return nil
return nil
}

func getSignServerVersion() (version string, err error) {
Expand All @@ -485,7 +482,7 @@ func getSignServerVersion() (version string, err error) {
if gjson.GetBytes(resp, "code").Int() == 0 {
return gjson.GetBytes(resp, "data.version").String(), nil
}
return "", errors.New("empty version")
return "", errors.New("empty version")
}

// 定时刷新 token, interval 为间隔时间(分钟)
Expand All @@ -502,18 +499,18 @@ func startRefreshTokenTask(interval int64) {
log.Warn("间隔时间不能超过 60 分钟,已自动设置为 60 分钟")
interval = 60
}
t := time.Newticker(time.Duration(interval) * time.Minute)
defer t.Stop()
t := time.NewTicker(time.Duration(interval) * time.Minute)
defer t.Stop()
for range t.C {
err := refreshToken(strconv.FormatInt(base.Account.Uin, 10))
if err != nil {
log.Warnf("刷新 token 出现错误: %v server: %v", err, base.SignServer)
}
err := refreshToken(strconv.FormatInt(base.Account.Uin, 10))
if err != nil {
log.Warnf("刷新 token 出现错误: %v server: %v", err, base.SignServer)
}
}
}

func waitSignServer() bool {
t := time.NewTicker(time.Second*5)
t := time.NewTicker(time.Second * 5)
defer t.Stop()
i := 0
for range t.C {
Expand Down
3 changes: 2 additions & 1 deletion coolq/api.go
Original file line number Diff line number Diff line change
Expand Up @@ -1149,7 +1149,8 @@ func (bot *CQBot) CQDelGroupMemo(groupID int64, fid string) global.MSG {
// @rename(msg->message, block->reject_add_request)
func (bot *CQBot) CQSetGroupKick(groupID int64, userID int64, msg string, block bool) global.MSG {
if g := bot.Client.FindGroup(groupID); g != nil {
if m := g.FindMember(userID); m == nil {
m := g.FindMember(userID)
if m == nil {
return Failed(100, "MEMBER_IS_NOT_IN_GROUP", "人员不存在")
}
err := m.Kick(msg, block)
Expand Down

0 comments on commit dae0378

Please sign in to comment.