diff --git a/models/bot.go b/models/bot.go index b4e9677..de7ed72 100644 --- a/models/bot.go +++ b/models/bot.go @@ -96,78 +96,77 @@ var handleMessage = func(msgs ...interface{}) interface{} { logs.Info(rsp) if err != nil { return err - } - //else { - // ss1 := regexp.MustCompile(`pin=([^;=\s]+);wskey=([^;=\s]+)`).FindAllStringSubmatch(msg, -1) - // if len(ss1) > 0 { - // for _, s := range ss1 { - // ck1 := JdCookie{ - // PtPin: s[1], - // WsKey: s[2], - // } - // if sender.IsQQ() { - // ck1.QQ = sender.UserID - // } else if sender.IsTG() { - // ck1.Telegram = sender.UserID - // } - // if nck, err := GetJdCookie(ck1.PtPin); err == nil { - // if nck.WsKey == "" || len(nck.WsKey) == 0 { - // nck.InPoolWskey(ck1.WsKey) - // msg := fmt.Sprintf("写入WsKey,%s", ck1.PtPin) - // (&JdCookie{}).Push(msg) - // logs.Info(msg) - // } else { - // msg := fmt.Sprintf("重复写入") - // (&JdCookie{}).Push(msg) - // logs.Info(msg) - // } - // } - // - // } - // go func() { - // Save <- &JdCookie{} - // }() - // return nil - // } - //} - ss := regexp.MustCompile(`pt_key=([^;=\s]+);pt_pin=([^;=\s]+)`).FindAllStringSubmatch(rsp, -1) - if len(ss) > 0 { - xyb := 0 - for _, s := range ss { - ck := JdCookie{ - PtKey: s[1], - PtPin: s[2], - } - if CookieOK(&ck) { - xyb++ + } else { + ss1 := regexp.MustCompile(`pin=([^;=\s]+);wskey=([^;=\s]+)`).FindAllStringSubmatch(msg, -1) + if len(ss1) > 0 { + for _, s := range ss1 { + ck1 := JdCookie{ + PtPin: s[1], + WsKey: s[2], + } if sender.IsQQ() { - ck.QQ = sender.UserID + ck1.QQ = sender.UserID } else if sender.IsTG() { - ck.Telegram = sender.UserID + ck1.Telegram = sender.UserID } - if nck, err := GetJdCookie(ck.PtPin); err == nil { - nck.InPool(ck.PtKey) - msg := fmt.Sprintf("更新账号,%s", ck.PtPin) - (&JdCookie{}).Push(msg) - logs.Info(msg) - } else { - if Cdle { - ck.Hack = True + if nck, err := GetJdCookie(ck1.PtPin); err == nil { + if nck.WsKey == "" || len(nck.WsKey) == 0 { + nck.InPoolWskey(ck1.WsKey) + msg := fmt.Sprintf("写入WsKey,%s", ck1.PtPin) + (&JdCookie{}).Push(msg) + logs.Info(msg) + } else { + msg := fmt.Sprintf("重复写入") + (&JdCookie{}).Push(msg) + logs.Info(msg) } - NewJdCookie(&ck) - msg := fmt.Sprintf("添加账号,%s", ck.PtPin) - sender.Reply(fmt.Sprintf(msg, AddCoin(sender.UserID))) - logs.Info(msg) } - } else { - sender.Reply(fmt.Sprintf("无效,许愿币-1,余额%d", RemCoin(sender.UserID, 1))) + } + go func() { + Save <- &JdCookie{} + }() + return nil } - go func() { - Save <- &JdCookie{} - }() - return nil } + //ss := regexp.MustCompile(`pt_key=([^;=\s]+);pt_pin=([^;=\s]+)`).FindAllStringSubmatch(rsp, -1) + //if len(ss) > 0 { + // xyb := 0 + // for _, s := range ss { + // ck := JdCookie{ + // PtKey: s[1], + // PtPin: s[2], + // } + // if CookieOK(&ck) { + // xyb++ + // 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) + // msg := fmt.Sprintf("更新账号,%s", ck.PtPin) + // (&JdCookie{}).Push(msg) + // logs.Info(msg) + // } else { + // if Cdle { + // ck.Hack = True + // } + // NewJdCookie(&ck) + // msg := fmt.Sprintf("添加账号,%s", ck.PtPin) + // sender.Reply(fmt.Sprintf(msg, AddCoin(sender.UserID))) + // logs.Info(msg) + // } + // } else { + // sender.Reply(fmt.Sprintf("无效,许愿币-1,余额%d", RemCoin(sender.UserID, 1))) + // } + // } + // go func() { + // Save <- &JdCookie{} + // }() + // return nil + //} return rsp } }