From 27285f2ccef115c1dca9af489e52266afe6a7fcf Mon Sep 17 00:00:00 2001
From: cdle <798731886@qq.com>
Date: Thu, 19 Aug 2021 22:02:16 +0800
Subject: [PATCH] x

---
 models/bot.go  | 4 ++--
 models/user.go | 6 ++++++
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/models/bot.go b/models/bot.go
index fbe339d..be40d3e 100644
--- a/models/bot.go
+++ b/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()
diff --git a/models/user.go b/models/user.go
index 8bd58ee..dca8c8b 100644
--- a/models/user.go
+++ b/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
+}