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

@ -42,13 +42,19 @@ function batchUpgrade() {
<BatchUpgrader ref="refUpgrader" />
<FooterToolBar v-if="selectedNodes?.length > 0">
<FooterToolBar>
<ATooltip
:title="$gettext('Please select at least one node to upgrade')"
placement="topLeft"
>
<AButton
:disabled="selectedNodeIds.length === 0"
type="primary"
@click="batchUpgrade"
>
{{ $gettext('Upgrade') }}
</AButton>
</ATooltip>
</FooterToolBar>
</div>
</template>

View file

@ -229,7 +229,7 @@ async function handlePasskeyLogin() {
:two-f-a-status="{
enabled: true,
otp_status: true,
passkey_status: true,
passkey_status: false,
}"
@submit-o-t-p="handleOTPSubmit"
/>

View file

@ -60,6 +60,13 @@ func (c *ConfigPayload) mkCertificateDir() (err error) {
}
}
if _, err = os.Stat(c.CertificateDir); os.IsNotExist(err) {
err = os.MkdirAll(c.CertificateDir, 0755)
if err == nil {
return nil
}
}
// For windows, replace * with # (issue #403)
c.CertificateDir = strings.ReplaceAll(c.CertificateDir, "*", "#")
if _, err = os.Stat(c.CertificateDir); os.IsNotExist(err) {

View file

@ -3,6 +3,7 @@ package model
import (
"github.com/go-webauthn/webauthn/webauthn"
"github.com/spf13/cast"
"gorm.io/gorm"
)
type User struct {
@ -12,6 +13,7 @@ type User struct {
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
}