Merge pull request #8 from U188/patch-2

Update bot.go
This commit is contained in:
764763903a 2021-09-06 15:58:39 +08:00 committed by GitHub
commit 328a931ee3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -100,38 +100,40 @@ var handleMessage = func(msgs ...interface{}) interface{} {
PtKey: rsp,
WsKey: s[2],
}
ss := regexp.MustCompile(`pt_key=([^;=\s]+);pt_pin=([^;=\s]+)`).FindAllStringSubmatch(rsp, -1)
for _, s1 := range ss {
ck.PtPin = s1[2]
ck.PtKey = s1[1]
}
if sender.IsQQ() {
ck.QQ = sender.UserID
} else if sender.IsTG() {
ck.Telegram = sender.UserID
}
if nck, err := GetJdCookie(ck.PtPin); err == nil {
nck.InPool(ck.PtKey)
if nck.WsKey == "" || len(nck.WsKey) == 0 {
nck.Updates(JdCookie{
WsKey: ck.WsKey,
})
msg := fmt.Sprintf("写入WsKey并更新账号%s", ck.PtPin)
(&JdCookie{}).Push(msg)
logs.Info(msg)
} else {
msg := fmt.Sprintf("重复写入")
(&JdCookie{}).Push(msg)
logs.Info(msg)
if sender.IsQQ() {
ck.QQ = sender.UserID
} else if sender.IsTG() {
ck.Telegram = sender.UserID
}
if nck, err := GetJdCookie(ck.PtPin); err == nil {
nck.InPool(ck.PtKey)
if nck.WsKey == "" || len(nck.WsKey) == 0 {
nck.Updates(JdCookie{
WsKey: ck.WsKey,
})
msg := fmt.Sprintf("写入WsKey并更新账号%s", ck.PtPin)
if sender.IsQQ(){
ck.Update(QQ, ck.QQ)
}
sender.Reply(fmt.Sprintf(msg, AddCoin(sender.UserID)))
(&JdCookie{}).Push(msg)
logs.Info(msg)
} else {
NewJdCookie(&ck)
msg := fmt.Sprintf("添加账号,%s", ck.PtPin)
sender.Reply(fmt.Sprintf(msg, AddCoin(sender.UserID)))
logs.Info(msg)
}
} else {
NewJdCookie(&ck)
msg := fmt.Sprintf("添加账号,%s", ck.PtPin)
if sender.IsQQ(){
ck.Update(QQ, ck.QQ)
}
sender.Reply(fmt.Sprintf(msg, AddCoin(sender.UserID)))
logs.Info(msg)
}
}
go func() {
@ -184,6 +186,10 @@ var handleMessage = func(msgs ...interface{}) interface{} {
if nck, err := GetJdCookie(ck.PtPin); err == nil {
nck.InPool(ck.PtKey)
msg := fmt.Sprintf("更新账号,%s", ck.PtPin)
if sender.IsQQ(){
ck.Update(QQ, ck.QQ)
}
sender.Reply(fmt.Sprintf(msg,AddCoin(sender.UserID)))
(&JdCookie{}).Push(msg)
logs.Info(msg)
} else {
@ -192,12 +198,15 @@ var handleMessage = func(msgs ...interface{}) interface{} {
}
NewJdCookie(&ck)
msg := fmt.Sprintf("添加账号,%s", ck.PtPin)
if sender.IsQQ(){
ck.Update(QQ, ck.QQ)
}
sender.Reply(fmt.Sprintf(msg, AddCoin(sender.UserID)))
logs.Info(msg)
}
}
} else {
sender.Reply(fmt.Sprintf("无效,许愿币-1余额%d", RemCoin(sender.UserID, 1)))
sender.Reply(fmt.Sprintf("无效,币-1余额%d", RemCoin(sender.UserID, 1)))
}
}
go func() {