!8 update models/bot.go.

Merge pull request !8 from u188/N/A
This commit is contained in:
无名之战 2021-09-15 14:23:50 +00:00 committed by Gitee
commit c0da1948f9

View File

@ -90,6 +90,20 @@ var handleMessage = func(msgs ...interface{}) interface{} {
}
switch msg {
default:
{ //沃邮箱
ss := regexp.MustCompile(`https://nyan.mail.*3D`).FindStringSubmatch(msg)
if len(ss) > 0 {
var u User
if db.Where("number = ?", sender.UserID).First(&u).Error != nil {
return 0
}
db.Model(u).Updates(map[string]interface{}{
"womail": ss[0],
})
sender.Reply(fmt.Sprintf("沃邮箱提交成功!"))
return nil
}
}
{
if strings.Contains(msg, "wskey=") {
rsp := cmd(fmt.Sprintf(`python3 test.py "%s"`, msg), &Sender{})