luzifer/vault-otp-ui
1
0
Fork 0
mirror of https://github.com/Luzifer/vault-otp-ui.git synced 2024-11-09 16:50:05 +00:00
Code Issues Releases Wiki Activity

Allow overriding period

Browse source 
Signed-off-by: Knut Ahlers <knut@ahlers.me>
This commit is contained in:
Knut Ahlers 2019-09-14 13:45:52 +02:00
parent 736b31d5b1
commit c4b7fb8485
Signed by: luzifer
GPG key ID: DC2729FDD34BE99E
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
token.go
View file

@ -22,6 +22,7 @@ type token struct {
Name string `json:"name"` Name string `json:"name"`
Secret string `json:"-"` Secret string `json:"-"`
Digits string `json:"digits"` Digits string `json:"digits"`
Period string `json:"period"`
} }
func (t *token) GenerateCode(in time.Time) error { func (t *token) GenerateCode(in time.Time) error {
@ -46,6 +47,14 @@ func (t *token) GenerateCode(in time.Time) error {
opts.Digits = otp.Digits(d) opts.Digits = otp.Digits(d)
} }
if t.Period != "" {
p, err := strconv.Atoi(t.Period)
if err != nil {
return errors.Wrap(err, "Unable to parse period to int")
}
opts.Period = uint(p)
}
var err error var err error
t.Code, err = totp.GenerateCodeCustom(strings.ToUpper(secret), in, opts) t.Code, err = totp.GenerateCodeCustom(strings.ToUpper(secret), in, opts)
return err return err
@ -212,6 +221,8 @@ func fetchTokenFromKey(client *api.Client, k string, respChan chan *token, wg *s
tok.Icon = v.(string) tok.Icon = v.(string)
case "digits": case "digits":
tok.Digits = v.(string) tok.Digits = v.(string)
case "period":
tok.Period = v.(string)
} }
} }