From 789b21cca9c8d1fd102e65b3f25710dbfee98639 Mon Sep 17 00:00:00 2001
From: ZJY <764763903@qq.com>
Date: Sat, 25 Sep 2021 10:34:03 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E8=B0=83=E8=AF=95=E7=99=BB=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controllers/account.go | 24 +++++++++++++++++++++++-
 controllers/login.go   | 20 ++++++++++++++++++++
 main.go                |  1 +
 models/command.go      | 32 ++++++++++++++++++++++++++++++++
 4 files changed, 76 insertions(+), 1 deletion(-)

diff --git a/controllers/account.go b/controllers/account.go
index 74a75c3..740fe8b 100644
--- a/controllers/account.go
+++ b/controllers/account.go
@@ -1,6 +1,8 @@
 package controllers
 
 import (
+	"encoding/json"
+	"fmt"
 	"github.com/beego/beego/v2/core/logs"
 	"github.com/cdle/xdd/models"
 )
@@ -61,8 +63,28 @@ func (c *AccountController) GetUserInfo() {
 	cookie, err := models.GetJdCookie(pin)
 	if err != nil {
 		logs.Error(err)
+		result := Result{
+			Data:    "null",
+			Code:    1,
+			Message: "查无匹配的pin",
+		}
+		jsons, errs := json.Marshal(result) //转换成JSON返回的是byte[]
+		if errs != nil {
+			fmt.Println(errs.Error())
+		}
+		c.Ctx.WriteString(string(jsons))
+	} else {
+		result := Result{
+			Data:    cookie.Query(),
+			Code:    0,
+			Message: "查询成功",
+		}
+		jsons, errs := json.Marshal(result) //转换成JSON返回的是byte[]
+		if errs != nil {
+			fmt.Println(errs.Error())
+		}
+		c.Ctx.WriteString(string(jsons))
 	}
-	cookie.Query()
 
 }
 
diff --git a/controllers/login.go b/controllers/login.go
index 3d01fd7..83885b6 100644
--- a/controllers/login.go
+++ b/controllers/login.go
@@ -44,6 +44,26 @@ type Result struct {
 var JdCookieRunners sync.Map
 var jdua = models.GetUserAgent
 
+//func (c *LoginController) GetQrcode1() {
+//	rsp, err := httplib.Get("https://api.kukuqaq.com/jd/qrcode").Response()
+//	if err != nil {
+//		logs.Info(err)
+//	}
+//	body, err1 := ioutil.ReadAll(rsp.Body)
+//	if err1 == nil {
+//		fmt.Println(string(body))
+//	}
+//	s := &models.QQuery{}
+//	if len(body) > 0 {
+//		json.Unmarshal(body, &s)
+//	}
+//	jsonByte, _ := json.Marshal(s)
+//	jsonStr := string(jsonByte)
+//	fmt.Printf("%v", jsonStr)
+//	c.Ctx.WriteString(`{"url":"` + "url" + `","img":"` + base64.StdEncoding.EncodeToString(data) + `"}`) //"data:image/png;base64," +
+//
+//}
+
 func (c *LoginController) GetQrcode() {
 	if v := c.GetSession("jd_token"); v != nil {
 		token := v.(string)
diff --git a/main.go b/main.go
index ad147ba..574fda1 100644
--- a/main.go
+++ b/main.go
@@ -53,6 +53,7 @@ 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/qrcode1", &controllers.LoginController{}, "get:GetQrcode1")
 	web.Router("/api/login/query", &controllers.LoginController{}, "get:Query")
 	web.Router("/api/login/cookie", &controllers.LoginController{}, "get:Cookie")
 	web.Router("/api/login/admin", &controllers.LoginController{}, "post:IsAdmin")
diff --git a/models/command.go b/models/command.go
index c4b6e49..8120b3a 100644
--- a/models/command.go
+++ b/models/command.go
@@ -117,6 +117,38 @@ var codeSignals = []CodeSignal{
 			return Count()
 		},
 	},
+	{
+		Command: []string{"删除WCK"},
+		Admin:   true,
+		Handle: func(sender *Sender) interface{} {
+			sender.handleJdCookies(func(ck *JdCookie) {
+				ck.Update(WsKey, "")
+				sender.Reply(fmt.Sprintf("已删除WCK,%s", ck.Nickname))
+			})
+			return nil
+		},
+	},
+	//{
+	//	Command: []string{"qrcode", "扫码", "二维码", "scan"},
+	//	Handle: func(sender *Sender) interface{} {
+	//		rsp, err := httplib.Get("https://api.kukuqaq.com/jd/qrcode").Response()
+	//		if err != nil {
+	//			return nil
+	//		}
+	//		body, err1 := ioutil.ReadAll(rsp.Body)
+	//		if err1 == nil {
+	//			fmt.Println(string(body))
+	//		}
+	//		s := &QQuery{}
+	//		if len(body) > 0 {
+	//			json.Unmarshal(body, &s)
+	//		}
+	//		jsonByte, _ := json.Marshal(s)
+	//		jsonStr := string(jsonByte)
+	//		fmt.Printf("%v", jsonStr)
+	//		return `{"url":"` + "http://www.baidu.com" + `","img":"` + s.Data.QqLoginQrcode.Bytes + `"}`
+	//	},
+	//},
 	{
 		Command: []string{`raw ^(\d{11})$`},
 		Handle: func(s *Sender) interface{} {