GitHubMirror/libmutton
1
0
Fork 0
mirror of https://github.com/rwinkhart/libmutton.git synced 2026-08-27 20:36:29 -04:00
Code Issues Packages Projects Releases Wiki Activity

Handle Steam TOTP directly in GenTOTP

Browse Source
This commit is contained in:
RandyTheSilly
2025-11-13 22:47:46 -05:00
parent 336dc5c158
commit 04e1b0dd37
1 changed files with 6 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
clip/totp.go
+6 -11
View File
@@ -11,13 +11,14 @@ import (
"github.com/pquerna/otp/totp"
)
// GenTOTP generates a TOTP token from a secret (supports standard and Steam TOTP).
func GenTOTP(secret string, time time.Time, forSteam bool) (string, error) {
// GenTOTP generates a TOTP token from a secret.
// Prefix `secret` with "steam@" for Steam TOTP format.
func GenTOTP(secret string, time time.Time) (string, error) {
var totpToken string
var err error
if forSteam {
totpToken, err = totp.GenerateCodeCustom(secret, time, totp.ValidateOpts{Period: 30, Digits: 5, Encoder: otp.EncoderSteam})
if strings.HasPrefix(secret, "steam@") {
totpToken, err = totp.GenerateCodeCustom(secret[6:], time, totp.ValidateOpts{Period: 30, Digits: 5, Encoder: otp.EncoderSteam})
} else {
totpToken, err = totp.GenerateCode(secret, time)
}
@@ -32,15 +33,9 @@ func GenTOTP(secret string, time time.Time, forSteam bool) (string, error) {
// TOTPCopier is meant to be run as a goroutine to keep
// the clipboard up-to-date with the latest TOTP token.
func TOTPCopier(secret string, errorChan chan<- error, done <-chan bool) {
var forSteam bool
if strings.HasPrefix(secret, "steam@") {
secret = secret[6:]
forSteam = true
}
for {
currentTime := time.Now()
token, err := GenTOTP(secret, currentTime, forSteam)
token, err := GenTOTP(secret, currentTime)
if err != nil {
errorChan <- err
}