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); }