Skip to content

Commit

Permalink
Update proto
Browse files Browse the repository at this point in the history
  • Loading branch information
dim13 committed Mar 2, 2023
1 parent 7b529ae commit 8a5f990
Show file tree
Hide file tree
Showing 4 changed files with 154 additions and 152 deletions.
30 changes: 15 additions & 15 deletions migration/convert.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,19 +7,19 @@ import (
)

var (
typeString = map[Payload_OtpType]string{
Payload_OTP_TYPE_HOTP: "hotp",
Payload_OTP_TYPE_TOTP: "totp",
typeString = map[Payload_OtpParameters_OtpType]string{
Payload_OtpParameters_OTP_TYPE_HOTP: "hotp",
Payload_OtpParameters_OTP_TYPE_TOTP: "totp",
}
algString = map[Payload_Algorithm]string{
Payload_ALGORITHM_SHA1: "SHA1",
Payload_ALGORITHM_SHA256: "SHA256",
Payload_ALGORITHM_SHA512: "SHA512",
Payload_ALGORITHM_MD5: "MD5",
algString = map[Payload_OtpParameters_Algorithm]string{
Payload_OtpParameters_ALGORITHM_SHA1: "SHA1",
Payload_OtpParameters_ALGORITHM_SHA256: "SHA256",
Payload_OtpParameters_ALGORITHM_SHA512: "SHA512",
Payload_OtpParameters_ALGORITHM_MD5: "MD5",
}
digitsString = map[Payload_DigitCount]string{
Payload_DIGIT_COUNT_SIX: "6",
Payload_DIGIT_COUNT_EIGHT: "8",
digitsString = map[Payload_OtpParameters_DigitCount]string{
Payload_OtpParameters_DIGIT_COUNT_SIX: "6",
Payload_OtpParameters_DIGIT_COUNT_EIGHT: "8",
}
)

Expand Down Expand Up @@ -49,19 +49,19 @@ func (op *Payload_OtpParameters) URL() *url.URL {
v.Add("issuer", op.Issuer)
}
// optional
if op.Algorithm != Payload_ALGORITHM_UNSPECIFIED {
if op.Algorithm != Payload_OtpParameters_ALGORITHM_UNSPECIFIED {
v.Add("algorithm", algString[op.Algorithm])
}
// optional
if op.Digits != Payload_DIGIT_COUNT_UNSPECIFIED {
if op.Digits != Payload_OtpParameters_DIGIT_COUNT_UNSPECIFIED {
v.Add("digits", digitsString[op.Digits])
}
// required if type is hotp
if op.Type == Payload_OTP_TYPE_HOTP {
if op.Type == Payload_OtpParameters_OTP_TYPE_HOTP {
v.Add("counter", fmt.Sprint(op.Counter))
}
// optional if type is totp
if op.Type == Payload_OTP_TYPE_TOTP {
if op.Type == Payload_OtpParameters_OTP_TYPE_TOTP {
v.Add("period", "30") // default value
}
return &url.URL{
Expand Down
28 changes: 14 additions & 14 deletions migration/evaluate.go
Original file line number Diff line number Diff line change
Expand Up @@ -14,22 +14,22 @@ import (
)

var (
hashFunc = map[Payload_Algorithm]func() hash.Hash{
Payload_ALGORITHM_UNSPECIFIED: sha1.New, // default
Payload_ALGORITHM_SHA1: sha1.New,
Payload_ALGORITHM_SHA256: sha256.New,
Payload_ALGORITHM_SHA512: sha512.New,
Payload_ALGORITHM_MD5: md5.New,
hashFunc = map[Payload_OtpParameters_Algorithm]func() hash.Hash{
Payload_OtpParameters_ALGORITHM_UNSPECIFIED: sha1.New, // default
Payload_OtpParameters_ALGORITHM_SHA1: sha1.New,
Payload_OtpParameters_ALGORITHM_SHA256: sha256.New,
Payload_OtpParameters_ALGORITHM_SHA512: sha512.New,
Payload_OtpParameters_ALGORITHM_MD5: md5.New,
}
digitCount = map[Payload_DigitCount]int{
Payload_DIGIT_COUNT_UNSPECIFIED: 6, // default
Payload_DIGIT_COUNT_SIX: 6,
Payload_DIGIT_COUNT_EIGHT: 8,
digitCount = map[Payload_OtpParameters_DigitCount]int{
Payload_OtpParameters_DIGIT_COUNT_UNSPECIFIED: 6, // default
Payload_OtpParameters_DIGIT_COUNT_SIX: 6,
Payload_OtpParameters_DIGIT_COUNT_EIGHT: 8,
}
countFunc = map[Payload_OtpType]func(*Payload_OtpParameters) uint64{
Payload_OTP_TYPE_UNSPECIFIED: totp, // default
Payload_OTP_TYPE_HOTP: hotp,
Payload_OTP_TYPE_TOTP: totp,
countFunc = map[Payload_OtpParameters_OtpType]func(*Payload_OtpParameters) uint64{
Payload_OtpParameters_OTP_TYPE_UNSPECIFIED: totp, // default
Payload_OtpParameters_OTP_TYPE_HOTP: hotp,
Payload_OtpParameters_OTP_TYPE_TOTP: totp,
}
)

Expand Down
Loading

0 comments on commit 8a5f990

Please sign in to comment.