oiupoyt/discordo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Implement MFA login (#225)

Browse source
This commit is contained in:
ayntgl 2022-09-26 19:24:04 +05:30 • committed by GitHub
parent 0bbe6bf93a
commit 3c507d4b34
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
ui/login_view.go
View file

@ -61,6 +61,7 @@ func newEmailView(c *Core) *EmailView {
v.AddInputField("Email", "", 0, nil, nil)
v.AddPasswordField("Password", "", 0, 0, nil)
v.AddPasswordField("Code (optional)", "", 0, 0, nil)
v.AddButton("Login", v.onLoginButtonSelected)
return v
@ -82,17 +83,27 @@ func (v *EmailView) onLoginButtonSelected() {
log.Fatal(err)
}
if l.Token != "" && !l.MFA {
err = v.core.Run(l.Token)
// If the token is not dispatched in the response and the "mfa" field is set as true, login using MFA instead.
if l.Token == "" && l.MFA {
code := v.GetFormItem(2).(*tview.InputField).GetText()
if code == "" {
return
}
// Retry logging in with a 2FA token
l, err = client.TOTP(code, l.Ticket)
if err != nil {
log.Fatal(err)
}
v.core.Draw()
go keyring.Set(config.Name, "token", l.Token)
}
// TODO: MFA login
err = v.core.Run(l.Token)
if err != nil {
log.Fatal(err)
}
v.core.Draw()
go keyring.Set(config.Name, "token", l.Token)
}
type TokenView struct {