mirror of
https://github.com/gravitational/teleport
synced 2024-10-23 02:32:39 +00:00
56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
package utils
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base32"
|
|
"image/png"
|
|
"net/url"
|
|
|
|
"github.com/gravitational/trace"
|
|
|
|
"github.com/pquerna/otp"
|
|
)
|
|
|
|
// GenerateQRCode takes in a OTP Key URL and returns a PNG-encoded QR code.
|
|
func GenerateQRCode(u string) ([]byte, error) {
|
|
otpKey, err := otp.NewKeyFromURL(u)
|
|
if err != nil {
|
|
return nil, trace.Wrap(err)
|
|
}
|
|
|
|
otpImage, err := otpKey.Image(450, 450)
|
|
if err != nil {
|
|
return nil, trace.Wrap(err)
|
|
}
|
|
|
|
var otpQR bytes.Buffer
|
|
err = png.Encode(&otpQR, otpImage)
|
|
if err != nil {
|
|
return nil, trace.Wrap(err)
|
|
}
|
|
|
|
return otpQR.Bytes(), nil
|
|
}
|
|
|
|
// GenerateOTPURL returns a OTP Key URL that can be used to construct a HOTP or TOTP key. For more
|
|
// details see: https://github.com/google/google-authenticator/wiki/Key-Uri-Format
|
|
// Example: otpauth://totp/foo:bar@baz.com?secret=qux
|
|
func GenerateOTPURL(typ string, label string, parameters map[string][]byte) string {
|
|
var u url.URL
|
|
|
|
u.Scheme = "otpauth"
|
|
u.Host = typ
|
|
u.Path = label
|
|
|
|
var params url.Values = make(url.Values)
|
|
for k, v := range parameters {
|
|
if k == "secret" {
|
|
v = []byte(base32.StdEncoding.EncodeToString(v))
|
|
}
|
|
params.Add(k, string(v))
|
|
}
|
|
u.RawQuery = params.Encode()
|
|
|
|
return u.String()
|
|
}
|