x

models/bot.go

@@ -107,8 +107,8 @@ var handleMessage = func(msgs ...interface{}) interface{} {
 			return "你没有权限操作"
 		}
 		return Count()
-	// case "sign", "打卡":
-	// 	return "打卡成功，许愿币+1"
+	case "许愿币":
+		return fmt.Sprintf("许愿币余额%d", GetCoin(uid))
 	case "qrcode", "扫码", "二维码", "scan":
 		url := fmt.Sprintf("http://127.0.0.1:%d/api/login/qrcode.png?tp=%s&uid=%d&gid=%d", web.BConfig.Listen.HTTPPort, tp, uid, gid)
 		rsp, err := httplib.Get(url).Response()

models/user.go

@@ -89,3 +89,9 @@ func RemCoin(uid int) int {
 	u.Coin--
 	return u.Coin
 }
+
+func GetCoin(uid int) int {
+	var u User
+	db.Where("number = ?", uid).First(&u)
+	return u.Coin
+}