From bc7a3e953f5ead0ac77911c577a86a1c1807c9c1 Mon Sep 17 00:00:00 2001
From: infl00p <fanis@infloop.gr>
Date: Fri, 23 Jul 2021 12:29:55 +0300
Subject: [PATCH] tests and various adds

---
 main.go | 16 ++++++++++++----
 otp.go  | 11 +++++++++++
 2 files changed, 23 insertions(+), 4 deletions(-)
 create mode 100644 otp.go

diff --git a/main.go b/main.go
index 27cafde..d8855d0 100644
--- a/main.go
+++ b/main.go
@@ -5,9 +5,9 @@ import (
 	"os/exec"
 	"strings"
 
+	"github.com/99designs/keyring"
 	"github.com/godbus/dbus"
 	"github.com/godbus/dbus/introspect"
-	"github.com/99designs/keyring"
 )
 
 var messages = map[string]string{
@@ -70,12 +70,17 @@ func main() {
 		AllowedBackends: []keyring.BackendType{keyring.KWalletBackend},
 		ServiceName: "kdewallet",
 		KWalletFolder: "otprunner",
+		KWalletAppID: "otprunner",
 	})
 	if err != nil {
 		log.Fatal("Unable to Open KDE Wallet ", err.Error())
 	}
 	r.wallet = wallet
 
+	// test data
+	r.wallet.Set(keyring.Item{"koko1",[]byte("lala1"),"Label koko1", "Description1", true, true})
+	r.wallet.Set(keyring.Item{"koko2",[]byte("lala2"),"Label koko2", "Description2", true, true})
+
 	conn.Export(r, ifacepath, ifacename)
 	conn.Export(introspect.Introspectable(intro), ifacepath, "org.freedesktop.DBus.Introspectable")
 
@@ -122,9 +127,12 @@ func (r runner) Match(query string) ([]match, *dbus.Error) {
 }
 
 func (r runner) Run(selection string, id string) *dbus.Error {
-	code, _ := r.wallet.Get(selection)
-	log.Println(code.Data)
-	err := AddToClipboard(string(code.Data))
+	code, err := r.wallet.Get(selection)
+	if err != nil {
+		log.Println(code)
+		log.Println(err)
+	}
+	err = AddToClipboard(string(code.Data))
 	if err != nil {
 		log.Fatal(err)
 	}
diff --git a/otp.go b/otp.go
new file mode 100644
index 0000000..9fe092a
--- /dev/null
+++ b/otp.go
@@ -0,0 +1,11 @@
+package main
+
+import (
+"crypto/hmac"
+
+)
+
+func GenerateCode() {
+}
+
+extract