yk: new tool for testing yubikey

1 files changed, 48 insertions, 0 deletions

diff --git a/yk/main.go b/yk/main.go
new file mode 100644
index 0000000..56e8417
--- /dev/null
+++ b/yk/main.go
@@ -0,0 +1,48 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"log"
+	"os"
+	"strings"
+
+	"github.com/GeertJohan/yubigo"
+	_ "github.com/joho/godotenv/autoload"
+	"github.com/kr/pretty"
+)
+
+func main() {
+	yubiAuth, err := yubigo.NewYubiAuth(os.Getenv("YUBIKEY_CLIENT_ID"), os.Getenv("YUBIKEY_SECRET_KEY"))
+	if err != nil {
+		log.Fatal(err)
+	}
+	_ = yubiAuth
+
+	reader := bufio.NewReader(os.Stdin)
+	fmt.Print("yk> ")
+	text, err := reader.ReadString(byte('\n'))
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	text = strings.TrimSpace(text)
+
+	resp, _, err := yubiAuth.Verify(text)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	pretty.Println(resp)
+
+	if !resp.IsValidOTP() {
+		log.Fatal("invalid OTP")
+	}
+
+	prefix, _, err := yubigo.ParseOTP(text)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	log.Printf("uid: %s", prefix)
+}