fix: incorrect status of enabled 2fa of user #577

This commit is contained in:
Jacky 2024-09-26 12:57:34 +08:00
parent f2a586ca78
commit 2f7070377f
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D
4 changed files with 31 additions and 11 deletions

View file

@ -3,15 +3,17 @@ package model
import (
"github.com/go-webauthn/webauthn/webauthn"
"github.com/spf13/cast"
"gorm.io/gorm"
)
type User struct {
Model
Name string `json:"name"`
Password string `json:"-"`
Status bool `json:"status" gorm:"default:1"`
OTPSecret []byte `json:"-" gorm:"type:blob"`
Name string `json:"name"`
Password string `json:"-"`
Status bool `json:"status" gorm:"default:1"`
OTPSecret []byte `json:"-" gorm:"type:blob"`
EnabledTwoFA bool `json:"enabled_2fa" gorm:"-"`
}
type AuthToken struct {
@ -24,6 +26,11 @@ func (u *User) TableName() string {
return "auths"
}
func (u *User) AfterFind(_ *gorm.DB) error {
u.EnabledTwoFA = u.Enabled2FA()
return nil
}
func (u *User) EnabledOTP() bool {
return len(u.OTPSecret) != 0
}