mirrors/nginx-ui
1
0
Fork 0
mirror of https://github.com/0xJacky/nginx-ui.git synced 2025-05-11 18:35:51 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

fix: key user does not exist #483

Browse source
This commit is contained in:
Jacky 2024-08-02 20:14:45 +08:00
parent b0d64e41e0
commit 2cce7849f2
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D
2 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
api/user/otp.go
View file

@ -9,6 +9,7 @@ import (
"github.com/0xJacky/Nginx-UI/api"
"github.com/0xJacky/Nginx-UI/internal/crypto"
"github.com/0xJacky/Nginx-UI/internal/user"
"github.com/0xJacky/Nginx-UI/model"
"github.com/0xJacky/Nginx-UI/query"
"github.com/0xJacky/Nginx-UI/settings"
"github.com/gin-gonic/gin"
@ -165,14 +166,19 @@ func ResetOTP(c *gin.Context) {
}
func OTPStatus(c *gin.Context) {
status := false
u, ok := c.Get("user")
if ok {
status = u.(*model.Auth).EnabledOTP()
}
c.JSON(http.StatusOK, gin.H{
"status": len(api.CurrentUser(c).OTPSecret) > 0,
"status": status,
})
}
func SecureSessionStatus(c *gin.Context) {
cUser := api.CurrentUser(c)
if !cUser.EnabledOTP() {
u, ok := c.Get("user")
if !ok || !u.(*model.Auth).EnabledOTP() {
c.JSON(http.StatusOK, gin.H{
"status": false,
})
@ -189,12 +195,13 @@ func SecureSessionStatus(c *gin.Context) {
return
}
if user.VerifySecureSessionID(ssid, cUser.ID) {
if user.VerifySecureSessionID(ssid, u.(*model.Auth).ID) {
c.JSON(http.StatusOK, gin.H{
"status": true,
})
return
}
c.JSON(http.StatusOK, gin.H{
"status": false,
})