From bee6812543a98756ef9fc0527e641f78cb94a3f2 Mon Sep 17 00:00:00 2001
From: ZJY <764763903@qq.com>
Date: Thu, 2 Sep 2021 09:30:17 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B5=8B=E8=AF=95=E3=80=82=E3=80=82=E3=80=82?=
 =?UTF-8?q?=E3=80=82=E8=AF=B7=E5=8B=BF=E6=8B=89=E5=8F=96=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controllers/account.go | 6 ------
 controllers/login.go   | 6 ++++++
 main.go                | 3 ++-
 theme/admin.html       | 3 ++-
 4 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/controllers/account.go b/controllers/account.go
index a90631b..c37c4fe 100644
--- a/controllers/account.go
+++ b/controllers/account.go
@@ -75,9 +75,3 @@ func (c *AccountController) CreateOrUpdate() {
 func (c *AccountController) Admin() {
 	c.Ctx.WriteString(models.Admin)
 }
-
-func (c *AccountController) IsAdmin() {
-	var pin = c.GetString("pin")
-	c.SetSession("pin", pin)
-	c.Ctx.WriteString("登录")
-}
diff --git a/controllers/login.go b/controllers/login.go
index 3c5a138..b3c9691 100644
--- a/controllers/login.go
+++ b/controllers/login.go
@@ -324,6 +324,12 @@ func FetchJdCookieValue(key string, cookies string) string {
 	}
 }
 
+func (c *LoginController) IsAdmin() {
+	pin := c.GetString("pin")
+	c.SetSession("pin", pin)
+	c.Ctx.WriteString("登录")
+}
+
 func (c *LoginController) Cookie() {
 	cookies := c.Ctx.Input.Header("Set-Cookie")
 	pt_key := FetchJdCookieValue("pt_key", cookies)
diff --git a/main.go b/main.go
index 9ec9065..7da21c3 100644
--- a/main.go
+++ b/main.go
@@ -55,10 +55,11 @@ func main() {
 	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/login/admin", &controllers.LoginController{}, "post:IsAdmin")
 	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{}, "post:IsAdmin")
+	web.Router("/admin", &controllers.AccountController{}, "get:Admin")
 	if models.Config.Static == "" {
 		models.Config.Static = "./static"
 	}
diff --git a/theme/admin.html b/theme/admin.html
index 4046d7e..0394937 100644
--- a/theme/admin.html
+++ b/theme/admin.html
@@ -78,6 +78,7 @@
     function login()
     {
         const userName = document.querySelector('#pin').value
+        sessionStorage.setItem('pin',document.getElementById("pin").value);
         var xmlhttp = window.XMLHttpRequest ? new XMLHttpRequest() : new ActiveXObject("Microsoft.XMLHTTP");
         xmlhttp.onreadystatechange = function() {
             if (xmlhttp.readyState == 4 && xmlhttp.status == 200) {
@@ -104,7 +105,7 @@
                 }
             }
         }
-        xmlhttp.open("POST","/admin", true);
+        xmlhttp.open("POST","/api/login/admin", true);
         xmlhttp.setRequestHeader("Content-Type","application/x-www-form-urlencoded")
         xmlhttp.send("pin="+userName);
     }