login returnState conditions

src/app/accounts/accountsLoginController.js

@@ -7,25 +7,28 @@ angular
         $scope.twoFactorProviderConstants = constants.twoFactorProvider;
         $scope.rememberTwoFactor = { checked: false };
 
-        if ($state.current.name.indexOf('twoFactor') > -1 && (!$scope.twoFactorProviders || !$scope.twoFactorProviders.length)) {
+        $scope.returnState = $state.params.returnState;
-            $state.go('frontend.login.info', { returnState: returnState });
+        $scope.stateEmail = $state.params.email;
-        }
+        if (!$scope.returnState && $state.params.org) {
-
+            $scope.returnState = {
-        var returnState;
-        if (!$state.params.returnState && $state.params.org) {
-            returnState = {
                 name: 'backend.user.settingsCreateOrg',
                 params: { plan: $state.params.org }
             };
         }
-        else {
+        else if (!$scope.returnState && $state.params.premium) {
-            returnState = $state.params.returnState;
+            $scope.returnState = {
+                name: 'backend.user.settingsPremium'
+            };
+        }
+
+        if ($state.current.name.indexOf('twoFactor') > -1 && (!$scope.twoFactorProviders || !$scope.twoFactorProviders.length)) {
+            $state.go('frontend.login.info', { returnState: $scope.returnState });
         }
 
         var rememberedEmail = $cookies.get(constants.rememberedEmailCookieName);
-        if (rememberedEmail || $state.params.email) {
+        if (rememberedEmail || $scope.stateEmail) {
             $scope.model = {
-                email: $state.params.email ? $state.params.email : rememberedEmail,
+                email: $scope.stateEmail || rememberedEmail,
                 rememberEmail: rememberedEmail !== null
             };
 
@@ -67,7 +70,7 @@ angular
                     $scope.twoFactorProvider = getDefaultProvider(twoFactorProviders);
 
                     $analytics.eventTrack('Logged In To Two-step');
-                    $state.go('frontend.login.twoFactor', { returnState: returnState }).then(function () {
+                    $state.go('frontend.login.twoFactor', { returnState: $scope.returnState }).then(function () {
                         $timeout(function () {
                             $("#code").focus();
                             init();
@@ -164,8 +167,8 @@ angular
         };
 
         function loggedInGo() {
-            if (returnState) {
+            if ($scope.returnState) {
-                $state.go(returnState.name, returnState.params);
+                $state.go($scope.returnState.name, $scope.returnState.params);
             }
             else {
                 $state.go('backend.user.vault');

src/app/accounts/accountsRegisterController.js

@@ -13,6 +13,12 @@ angular
                 params: { plan: $state.params.org }
             };
         }
+        else if (!stateParams.returnState && stateParams.premium) {
+            $scope.returnState = {
+                name: 'backend.user.settingsPremium',
+                params: { plan: $state.params.org }
+            };
+        }
         else {
             $scope.returnState = stateParams.returnState;
         }

src/app/accounts/views/accountsLoginInfo.html

@@ -36,7 +36,7 @@
     <hr />
     <ul>
         <li>
-            <a ui-sref="frontend.register({returnState: state.params.returnState, email: state.params.email})">
+            <a ui-sref="frontend.register({returnState: returnState, email: stateEmail})">
                 Create a new account
             </a>
         </li>

src/app/config.js

@@ -7,7 +7,7 @@ angular
         $locationProvider.hashPrefix('');
         jwtOptionsProvider.config({
             urlParam: 'access_token3',
-            whiteListedDomains: ['api.bitwarden.com', 'preview-api.bitwarden.com', 'localhost', '192.168.1.4']
+            whiteListedDomains: ['api.bitwarden.com', 'preview-api.bitwarden.com', 'localhost', '192.168.1.3']
         });
         var refreshPromise;
         jwtInterceptorProvider.tokenGetter = /*@ngInject*/ function (options, tokenService, authService) {
@@ -214,14 +214,16 @@ angular
                 controller: 'accountsLoginController',
                 params: {
                     returnState: null,
-                    email: null
+                    email: null,
+                    premium: null,
+                    org: null
                 },
                 data: {
                     bodyClass: 'login-page'
                 }
             })
             .state('frontend.login.info', {
-                url: '^/?org',
+                url: '^/?org&premium&email',
                 templateUrl: 'app/accounts/views/accountsLoginInfo.html',
                 data: {
                     pageTitle: 'Log In'
@@ -260,12 +262,14 @@ angular
                 }
             })
             .state('frontend.register', {
-                url: '^/register?org',
+                url: '^/register?org&premium',
                 templateUrl: 'app/accounts/views/accountsRegister.html',
                 controller: 'accountsRegisterController',
                 params: {
                     returnState: null,
-                    email: null
+                    email: null,
+                    org: null,
+                    premium: null
                 },
                 data: {
                     pageTitle: 'Register',