vault-totp/token.go

package main

import (
	"fmt"
	"strings"
	"time"

	"github.com/pquerna/otp/totp"
)

type token struct {
	Name   string
	Secret string
}

func (t token) GetCode(in time.Time) (string, error) {
	secret := t.Secret

	if n := len(secret) % 8; n != 0 {
		secret = secret + strings.Repeat("=", 8-n)
	}

	return totp.GenerateCode(strings.ToUpper(secret), in)
}

func (t token) BuildOutput(showName bool, namePad int) (string, error) {
	// Output: "123456 (Valid 12s)", "123456 (Valid 1s)"

	n := time.Now()
	code, err := t.GetCode(n)
	if err != nil {
		return "", err
	}

	var output string
	if showName {
		output = fmt.Sprintf(fmt.Sprintf("%%-%ds", namePad+2), fmt.Sprintf("%s:", t.Name))
	}

	output = fmt.Sprintf("%s%s", output, code)

	if !cfg.NoTime {
		remain := 30 - (n.Second() % 30)
		output = fmt.Sprintf("%s (Valid %ds) ", output, remain)
	}

	return output, nil
}

// Sorter interface

type tokenList []token

func (t tokenList) Len() int           { return len(t) }
func (t tokenList) Less(i, j int) bool { return t[i].Name < t[j].Name }
func (t tokenList) Swap(i, j int)      { t[i], t[j] = t[j], t[i] }

func (t tokenList) LongestName() (l int) {
	for _, s := range t {
		if ll := len(s.Name); ll > l {
			l = ll
		}
	}

	return
}