From 8e3c7a40c8780edd1f13dab77ce2e83e23337e11 Mon Sep 17 00:00:00 2001
From: jaketaylor <jake@privategrity.com>
Date: Sun, 24 Mar 2019 14:00:22 -0400
Subject: [PATCH] Cleanup

---
 cmd/root.go | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/cmd/root.go b/cmd/root.go
index 2685ce652..62002ce08 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -121,23 +121,27 @@ func sessionInitialization() {
 
 	// Register a new user if requested
 	if register {
+
 		regCode := "AAAA" // FIXME: Need to pass in registration code
 		fmt.Printf("Attempting to register with code %s...\n", regCode)
+
 		uid, err = bindings.Register(regCode, registrationAddr,
 			strings.Join(gwAddresses, ","), mint)
 		if err != nil {
 			fmt.Printf("Could Not Register User: %s\n", err.Error())
 			return
 		}
+
 		fmt.Printf("Successfully registered user %v!\n", uid)
-	}
 
-	// Log the user in, for now using the first gateway specified
-	if !register {
+	} else {
+
 		uid = id.NewUserFromUint(userId, nil).Bytes()
+
 	}
-	_, err = bindings.Login(uid[:], gwAddresses[0], "")
 
+	// Log the user in, for now using the first gateway specified
+	_, err = bindings.Login(uid[:], gwAddresses[0], "")
 	if err != nil {
 		fmt.Printf("Could Not Log In: %s\n", err)
 		return
-- 
GitLab