From 1e0c43ae4e2e134a9f55beadd58848a74dc82e67 Mon Sep 17 00:00:00 2001
From: ksripathi <kammari.sripathi@gmail.com>
Date: Thu, 4 Jan 2018 16:17:03 +0530
Subject: [PATCH] fixed redirect page to enrolled course

---
 src/runtime/authentication.org | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/runtime/authentication.org b/src/runtime/authentication.org
index 5cdc0a7..64984c4 100644
--- a/src/runtime/authentication.org
+++ b/src/runtime/authentication.org
@@ -31,7 +31,7 @@ var setLoginUrlInContentIframe = function(event) {
     document.cookie = "session-status='logged-out'";
     iframe.src = hosted_base + "/logout";
     toggleBtns();
-    loadAllDisciplines();
+    setTimeout(loadAllDisciplines, 600);
   }
 };
 
@@ -168,16 +168,20 @@ function processPostMessage(evt) {
       toggleBackButtonFlag(loadAllDisciplines);
     }
   };
-
-  var post_msg = evt.data.toLocaleLowerCase();
+  var post_msg_dict = JSON.parse(evt.data);
+  var post_msg = post_msg_dict.message.toLocaleLowerCase();
+  var referrer = post_msg_dict.referrer;
+  var iframe = document.getElementById("contentIframe");
 
   if (post_msg.indexOf("my courses") != -1) {
     document.cookie = "session-status='logged-in'";
-    if (window.lab_url == "") {
+    if (window.loggedInState == true) {
+      iframe.src = referrer;
+    }
+    else if (window.lab_url == "") {
       loadAllDisciplines();
     }
     else {
-      var iframe = document.getElementById("contentIframe");
       iframe.src = window.lab_url;
     }
   } else {