make it a pointer

O3Labs · May 30, 2018 · 4e64572 · 4e64572
1 parent 33239fc
commit 4e64572
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/neoutils/nep2.go b/neoutils/nep2.go
@@ -2,8 +2,17 @@ package neoutils
 
 import "github.com/o3labs/neo-utils/neoutils/nep2"
 
-func NEP2Encrypt(wif string, passphrase string) (s string, address string, err error) {
-	return nep2.NEP2Encrypt(wif, passphrase)
+type NEP2 struct {
+	EncryptedKey string
+	Address      string
+}
+
+func NEP2Encrypt(wif string, passphrase string) (*NEP2, error) {
+	encryptedKey, address, err := nep2.NEP2Encrypt(wif, passphrase)
+	return &NEP2{
+		EncryptedKey: encryptedKey,
+		Address:      address,
+	}, err
 }
 
 func NEP2Decrypt(key, passphrase string) (s string, err error) {