25 lines
497 B
Go
25 lines
497 B
Go
|
package auth
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/pquerna/otp/totp"
|
||
|
|
)
|
||
|
|
|
||
|
|
// GenerateTOTP 生成 TOTP 密钥和二维码
|
||
|
|
func GenerateTOTP(user string, issuer string) (string, string, error) {
|
||
|
|
// 生成 TOTP 密钥
|
||
|
|
key, err := totp.Generate(totp.GenerateOpts{
|
||
|
|
Issuer: issuer,
|
||
|
|
AccountName: user,
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return "", "", err
|
||
|
|
}
|
||
|
|
|
||
|
|
return key.Secret(), key.URL(), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// VerifyTOTP 验证 TOTP 密码
|
||
|
|
func VerifyTOTP(secret string, code string) bool {
|
||
|
|
return totp.Validate(code, secret)
|
||
|
|
}
|