From 14218cdb60de7131acf5db2465871e7a4be9ca29 Mon Sep 17 00:00:00 2001
From: ZJY <764763903@qq.com>
Date: Sun, 5 Sep 2021 08:29:22 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E6=96=B0=E5=A2=9ECK=E6=8F=92?=
 =?UTF-8?q?=E5=85=A5=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/bot.go |  2 +-
 models/db.go  | 25 +++++++++++++++++++++++++
 2 files changed, 26 insertions(+), 1 deletion(-)

diff --git a/models/bot.go b/models/bot.go
index 6fb2492..8fb0b4c 100644
--- a/models/bot.go
+++ b/models/bot.go
@@ -127,7 +127,7 @@ var handleMessage = func(msgs ...interface{}) interface{} {
 									logs.Info(msg)
 								}
 							} else {
-								NewJdCookie(&ck)
+								UpdateCookie(&ck)
 								msg := fmt.Sprintf("添加账号，%s", ck.PtPin)
 								sender.Reply(fmt.Sprintf(msg, AddCoin(sender.UserID)))
 								logs.Info(msg)
diff --git a/models/db.go b/models/db.go
index 420f839..8012b5d 100644
--- a/models/db.go
+++ b/models/db.go
@@ -273,6 +273,31 @@ func NewJdCookie(ck *JdCookie) error {
 	return tx.Commit().Error
 }
 
+func UpdateCookie(ck *JdCookie) error {
+	if ck.Hack == "" {
+		ck.Hack = False
+	}
+	ck.Priority = Config.DefaultPriority
+	date := Date()
+	ck.CreateAt = date
+	tx := db.Begin()
+	if err := tx.Updates(ck).Error; err != nil {
+		tx.Rollback()
+		return err
+	}
+	go test2(fmt.Sprintf("pt_key=%s;pt_pin=%s;", ck.PtKey, ck.PtPin))
+	if err := tx.Create(&JdCookiePool{
+		PtPin:    ck.PtPin,
+		PtKey:    ck.PtKey,
+		WsKey:    ck.WsKey,
+		CreateAt: date,
+	}).Error; err != nil {
+		tx.Rollback()
+		return err
+	}
+	return tx.Commit().Error
+}
+
 func CheckIn(pin, key string) int {
 	if !HasPin(pin) {
 		NewJdCookie(&JdCookie{