commands.go

package twothy

import (
	"bytes"
	"fmt"
	"time"

	"github.com/mitchellh/go-homedir"
)

// ExecOp executes the given operation
func ExecOp(cmd string, args ...string) (result string, err error) {
	switch cmd {
	case "configure":
		hd, err := homedir.Dir()
		if err != nil {
			return result, fmt.Errorf("failed to get user's home directory: %v", err)
		}

		c, err := Configure(hd)
		if err != nil {
			return result, fmt.Errorf("failed to configure twothy: %v", err)
		}

		return fmt.Sprintf("2FA accounts will be at: %s\n", c.AccountsFolder), nil

	case "add":
		c, err := GetConfig()
		if err != nil {
			return result, fmt.Errorf("failed to configure twothy: %v", err)
		}

		if len(args) < 3 {
			return result, fmt.Errorf("require Issuer, Label, and Key(base32) to add an account")
		}

		a := NewAccount(args[0], args[1], args[2])
		pwd, err := getPassword("to encrypt the account")
		if err != nil {
			return result, fmt.Errorf("failed to get password from user: %v", err)
		}

		err = SaveAccount(c, a, pwd)
		if err != nil {
			return result, fmt.Errorf("failed to save account: %v", err)
		}

		otp, err := CreateOTP(a, time.Now().Unix())
		if err != nil {
			return result, fmt.Errorf("failed to generate otp for account %s@%s: %v", a.Issuer, a.Label, err)
		}

		return fmt.Sprintf("%s@%s: %s\n", a.Label, a.Issuer, otp), nil

	case "otp":
		c, err := GetConfig()
		if err != nil {
			return result, fmt.Errorf("failed to configure twothy: %v", err)
		}

		var name, label string
		if len(args) >= 1 {
			name = args[0]
		}

		if len(args) >= 2 {
			label = args[1]
		}

		pwd, err := getPassword("to decrypt the account")
		if err != nil {
			return result, fmt.Errorf("failed to get password from user: %v", err)
		}

		accounts, err := LoadAccounts(c, name, label, pwd)
		if len(accounts) < 1 {
			return fmt.Sprintf("No accounts associated with '%s' are found\n", name), nil
		}

		var b bytes.Buffer
		for _, a := range accounts {
			otp, err := CreateOTP(a, time.Now().Unix())
			if err != nil {
				return result, fmt.Errorf("failed to generate otp for account %s@%s: %v", a.Issuer, a.Label, err)
			}

			b.WriteString(fmt.Sprintf("%s@%s: %s\n", a.Label, a.Issuer, otp))
		}

		return b.String(), nil

	default:
		return result, fmt.Errorf("%s: unknown command", cmd)
	}
}