mirror of https://github.com/authelia/authelia.git
67 lines
2.1 KiB
Go
67 lines
2.1 KiB
Go
package model
|
|
|
|
import (
|
|
"github.com/authelia/authelia/v4/internal/utils"
|
|
)
|
|
|
|
// UserInfo represents the user information required by the web UI.
|
|
type UserInfo struct {
|
|
// The users display name.
|
|
DisplayName string `db:"-" json:"display_name"`
|
|
|
|
// The preferred 2FA method.
|
|
Method string `db:"second_factor_method" json:"method" valid:"required"`
|
|
|
|
// True if a TOTP device has been registered.
|
|
HasTOTP bool `db:"has_totp" json:"has_totp" valid:"required"`
|
|
|
|
// True if a WebAuthn credential has been registered.
|
|
HasWebAuthn bool `db:"has_webauthn" json:"has_webauthn" valid:"required"`
|
|
|
|
// True if a duo device has been configured as the preferred.
|
|
HasDuo bool `db:"has_duo" json:"has_duo" valid:"required"`
|
|
}
|
|
|
|
// SetDefaultPreferred2FAMethod configures the default method based on what is configured as available and the users available methods.
|
|
func (i *UserInfo) SetDefaultPreferred2FAMethod(methods []string, fallback string) (changed bool) {
|
|
if len(methods) == 0 {
|
|
// No point attempting to change the method if no methods are available.
|
|
return false
|
|
}
|
|
|
|
before := i.Method
|
|
|
|
totp, webauthn, duo := utils.IsStringInSlice(SecondFactorMethodTOTP, methods), utils.IsStringInSlice(SecondFactorMethodWebAuthn, methods), utils.IsStringInSlice(SecondFactorMethodDuo, methods)
|
|
|
|
if i.Method == "" && utils.IsStringInSlice(fallback, methods) {
|
|
i.Method = fallback
|
|
} else if i.Method != "" && !utils.IsStringInSlice(i.Method, methods) {
|
|
i.Method = ""
|
|
}
|
|
|
|
if i.Method == "" {
|
|
i.setMethod(totp, webauthn, duo, methods, fallback)
|
|
}
|
|
|
|
return before != i.Method
|
|
}
|
|
|
|
func (i *UserInfo) setMethod(totp, webauthn, duo bool, methods []string, fallback string) {
|
|
switch {
|
|
case i.HasTOTP && totp:
|
|
i.Method = SecondFactorMethodTOTP
|
|
case i.HasWebAuthn && webauthn:
|
|
i.Method = SecondFactorMethodWebAuthn
|
|
case i.HasDuo && duo:
|
|
i.Method = SecondFactorMethodDuo
|
|
case fallback != "" && utils.IsStringInSlice(fallback, methods):
|
|
i.Method = fallback
|
|
case totp:
|
|
i.Method = SecondFactorMethodTOTP
|
|
case webauthn:
|
|
i.Method = SecondFactorMethodWebAuthn
|
|
case duo:
|
|
i.Method = SecondFactorMethodDuo
|
|
}
|
|
}
|