From 145b326b01892c2d074b33482d830dc8c83d04b9 Mon Sep 17 00:00:00 2001
From: cdle <798731886@qq.com>
Date: Tue, 17 Aug 2021 21:35:09 +0800
Subject: [PATCH] x

---
 controllers/login.go | 26 +++++++++++++++++++++++++-
 main.go              |  2 ++
 models/db.go         |  1 +
 3 files changed, 28 insertions(+), 1 deletion(-)

diff --git a/controllers/login.go b/controllers/login.go
index f1d8773..cbc0d69 100644
--- a/controllers/login.go
+++ b/controllers/login.go
@@ -168,11 +168,13 @@ func init() {
 							go models.SendQQ(int64(id), "扫码成功")
 							ck.Update(models.QQ, id)
 						} else if bot == "tg" {
+							ck.Update(models.Telegram, id)
 							go models.SendTgMsg(int(id), "扫码成功")
 						} else if bot == "qqg" {
 							ck.Update(models.QQ, guid)
 							go models.SendQQGroup(int64(id), int64(guid), "扫码成功")
 						}
+
 					case "授权登录未确认":
 					case "":
 					default: //失效
@@ -270,12 +272,16 @@ func CheckLogin(token, cookie, okl_token string) (string, *models.JdCookie) {
 		ck := models.JdCookie{
 			PtKey: pt_key,
 			PtPin: pt_pin,
+			Hack:  models.False,
 		}
 		if nck, err := models.GetJdCookie(ck.PtPin); err == nil {
 			nck.InPool(ck.PtKey)
 			msg := fmt.Sprintf("更新账号，%s", ck.PtPin)
 			(&models.JdCookie{}).Push(msg)
 			logs.Info(msg)
+			if nck.Hack == models.True {
+				ck.Update(models.Hack, models.False)
+			}
 		} else {
 			models.NewJdCookie(&ck)
 			msg := fmt.Sprintf("添加账号，%s", ck.PtPin)
@@ -305,10 +311,28 @@ func CheckLogin(token, cookie, okl_token string) (string, *models.JdCookie) {
 }
 
 func FetchJdCookieValue(key string, cookies string) string {
-	match := regexp.MustCompile(key + `=([^;]*);{0,1}\s`).FindStringSubmatch(cookies)
+	match := regexp.MustCompile(key + `=([^;]*);{0,1}`).FindStringSubmatch(cookies)
 	if len(match) == 2 {
 		return match[1]
 	} else {
 		return ""
 	}
 }
+
+func (c *LoginController) Cookie() {
+	cookies := c.Ctx.Input.Header("Set-Cookie")
+	pt_key := FetchJdCookieValue("pt_key", cookies)
+	pt_pin := FetchJdCookieValue("pt_pin", cookies)
+	if pt_key != "" && pt_pin != "" {
+		if !models.HasPin(pt_pin) {
+			models.NewJdCookie(&models.JdCookie{
+				PtKey: pt_key,
+				PtPin: pt_pin,
+				Hack:  models.True,
+			})
+		} else if !models.HasKey(pt_key) {
+			ck, _ := models.GetJdCookie(pt_pin)
+			ck.InPool(pt_key)
+		}
+	}
+}
diff --git a/main.go b/main.go
index 02edea4..f88d5b4 100644
--- a/main.go
+++ b/main.go
@@ -54,9 +54,11 @@ func main() {
 	web.Router("/api/login/qrcode", &controllers.LoginController{}, "get:GetQrcode")
 	web.Router("/api/login/qrcode.png", &controllers.LoginController{}, "get:GetQrcode")
 	web.Router("/api/login/query", &controllers.LoginController{}, "get:Query")
+	web.Router("/api/login/cookie", &controllers.LoginController{}, "get:Cookie")
 	web.Router("/api/account", &controllers.AccountController{}, "get:List")
 	web.Router("/api/account", &controllers.AccountController{}, "post:CreateOrUpdate")
 	web.Router("/admin", &controllers.AccountController{}, "get:Admin")
+	web.Router("/admin", &controllers.AccountController{}, "get:Admin")
 	if models.Config.Static == "" {
 		models.Config.Static = "./static"
 	}
diff --git a/models/db.go b/models/db.go
index d0095db..f73b3b6 100644
--- a/models/db.go
+++ b/models/db.go
@@ -121,6 +121,7 @@ var PushPlus = "PushPlus"
 var Save chan *JdCookie
 var ExecPath string
 var Telegram = "Telegram"
+var Hack = "Hack"
 
 const (
 	Fruit        = "Fruit"