From 9dbfe34e8acbea4edb2a1b459ef8b7fa735edac1 Mon Sep 17 00:00:00 2001
From: cdle <798731886@qq.com>
Date: Thu, 19 Aug 2021 21:57:16 +0800
Subject: [PATCH] x

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

diff --git a/models/bot.go b/models/bot.go
index 06cf7b7..fbe339d 100644
--- a/models/bot.go
+++ b/models/bot.go
@@ -179,7 +179,7 @@ var handleMessage = func(msgs ...interface{}) interface{} {
 							ck.Telegram = uid
 						}
 						if HasKey(ck.PtKey) {
-							sendMessagee("作弊许愿币-1", msgs...)
+							sendMessagee(fmt.Sprintf("作弊，许愿币-1，余额%d", RemCoin(uid)), msgs...)
 						} else {
 							if nck, err := GetJdCookie(ck.PtPin); err == nil {
 								nck.InPool(ck.PtKey)
@@ -192,7 +192,7 @@ var handleMessage = func(msgs ...interface{}) interface{} {
 								}
 								NewJdCookie(&ck)
 								msg := fmt.Sprintf("添加账号，%s", ck.PtPin)
-								sendMessagee(fmt.Sprintf("许愿币+1，余额%d", AddCoin(uid)), msgs...)
+								sendMessagee(fmt.Sprintf("很棒，许愿币+1，余额%d", AddCoin(uid)), msgs...)
 								logs.Info(msg)
 							}
 						}
diff --git a/models/user.go b/models/user.go
index f808b27..8bd58ee 100644
--- a/models/user.go
+++ b/models/user.go
@@ -79,3 +79,13 @@ func AddCoin(uid int) int {
 	u.Coin++
 	return u.Coin
 }
+
+func RemCoin(uid int) int {
+	var u User
+	db.Where("number = ?", uid).First(&u)
+	db.Updates(map[string]interface{}{
+		"coin": gorm.Expr("coin-1"),
+	})
+	u.Coin--
+	return u.Coin
+}