Source Update

src/functions/login.sh

@@ -4,7 +4,7 @@ wx-login() {
     echo -n "$wxNormal";
     echo "------------------------------"
 
-    ORG=$2
+    ORG=$1
 
     if [[ -z "$HOSTNAME" || ${#HOSTNAME} -lt 5 ]]; then
         wx-stop
@@ -92,7 +92,7 @@ wx-login() {
     if [[ $VAULT_STATUS -eq 200 ]]
     then
         IDM_STATUS=$(curl -s -o /dev/null -w "%{http_code}" https://$IDM_DOMAIN)
-        if [[ $IDM_STATUS -eq 302 ]]
+        if [[ $IDM_STATUS -eq 301 ]]
         then
             echo -n $wxBold
             echo -n $ORG
@@ -138,8 +138,8 @@ wx-login() {
                     wx-stop
                 fi
             fi
-            echo ""
 
+            config[${ORG}]=$VAULT_LOGIN
             wx-start
         else
             echo -n $wxBold
@@ -171,6 +171,7 @@ wx-login() {
                 wx-stop
             fi
 
+            config[${ORG}]=$VAULT_LOGIN
             wx-start
         fi
     else