diff --git a/models/bot.go b/models/bot.go index 088a043..ec0fdf3 100644 --- a/models/bot.go +++ b/models/bot.go @@ -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{})