Handle redirects on login page

web/src/cluster/components/Logs/Logs.jsx

@@ -20,7 +20,7 @@ import cfg from 'app/config';
 import LogViewer from 'app/components/LogViewer';
 import LogProvider from './LogProvider';
 import QueryEditorBasic from './QueryEditorBasic';
-import { getUrlParameter } from 'app/services/browser';
+import { getUrlParameter } from 'app/services/history';
 import * as Alerts from 'shared/components/Alert';
 import { Box, Flex, Input, ButtonSecondary } from 'shared/components';
 import { Cog } from 'shared/components/Icon';

web/src/components/Login/LoginFailed.jsx

@@ -15,7 +15,7 @@ limitations under the License.
 */
 
 import React from 'react';
-import { getUrlParameter } from 'app/services/browser';
+import { getUrlParameter } from 'app/services/history';
 import cfg from 'app/config';
 import LogoHero from './../LogoHero';
 import { LoginFailed as CardFailed}  from 'shared/components/CardError';

web/src/flux/user/actions.js

@@ -27,12 +27,12 @@ const logger = Logger.create('user/actions');
 
 export function login(userId, password, token) {
   const promise = auth.login(userId, password, token);
-  return _handleLoginPromise(promise);
+  return handleLoginPromise(promise);
 }
 
 export function loginWithU2f(userId, password) {
   const promise = auth.loginWithU2f(userId, password);
-  return _handleLoginPromise(promise);
+  return handleLoginPromise(promise);
 }
 
 export function loginWithSso(providerName, redirectUrl) {
@@ -59,12 +59,7 @@ export function changePassword(oldPsw, newPsw, token){
   return api.put(cfg.getSiteChangePasswordUrl(), data);
 }
 
-function getEntryRoute() {
-  const entryUrl = history.getRedirectParam();
-  return !entryUrl ? cfg.routes.app : entryUrl;
-}
-
-function _handleLoginPromise(promise) {
+function handleLoginPromise(promise) {
   return promise.done(() => {
     const redirect = getEntryRoute();
     const withPageRefresh = true;
@@ -73,4 +68,16 @@ function _handleLoginPromise(promise) {
   .fail(err => {
     logger.error('login', err);
   });
+}
+
+function getEntryRoute() {
+  let entryUrl = history.getRedirectParam();
+  if (entryUrl) {
+    entryUrl = history.ensureKnownRoute(entryUrl);
+  } else {
+    entryUrl = cfg.routes.app;
+  }
+
+  return history.ensureBaseUrl(entryUrl);
+
 }

web/src/services/browser.js

@@ -23,19 +23,4 @@ function detectPlatform() {
   }
 }
 
-export function getUrlParameter(name, path) {
-  path = path || window.location.search;
-  const query = path.substring(1);
-  const vars = query.split('&');
-  for (var i = 0; i < vars.length; i++) {
-      var pair = vars[i].split('=');
-      if (decodeURIComponent(pair[0]) == name) {
-          return decodeURIComponent(pair[1]);
-      }
-  }
-
-  return '';
-}
-
-export const platform = detectPlatform();
-
+export const platform = detectPlatform();

web/src/services/history.js

@@ -62,12 +62,7 @@ const history = {
   },
 
   getRedirectParam() {
-    let loc = this.original().location;
-    if (loc.query && loc.query.redirect_uri) {
-      return loc.query.redirect_uri;
-    }
-
-    return '';
+    return getUrlParameter('redirect_uri', this.original().location.search);
   },
 
   ensureKnownRoute(url) {
@@ -117,4 +112,18 @@ const match = url => route => {
   })
 }
 
-export default history;
+export default history;
+
+export function getUrlParameter(name, path) {
+  path = path || window.location.search;
+  const query = path.substring(1);
+  const vars = query.split('&');
+  for (var i = 0; i < vars.length; i++) {
+      var pair = vars[i].split('=');
+      if (decodeURIComponent(pair[0]) == name) {
+          return decodeURIComponent(pair[1]);
+      }
+  }
+
+  return '';
+}