From 754b724ec46dd2d122050bc50b1b08382f6003b2 Mon Sep 17 00:00:00 2001
From: Sasha Ilieva <sasha.ilieva@vereign.com>
Date: Fri, 14 Aug 2020 14:11:38 +0300
Subject: [PATCH] Use loadedIdentities or identity

---
 javascript/src/iframe/viamapi-iframe.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/javascript/src/iframe/viamapi-iframe.js b/javascript/src/iframe/viamapi-iframe.js
index b0e228d..082601a 100644
--- a/javascript/src/iframe/viamapi-iframe.js
+++ b/javascript/src/iframe/viamapi-iframe.js
@@ -277,13 +277,14 @@ window.lastTimeGetProfile = 0;
 let iframeParent = null;
 
 const handleIdentityLogin = (identity, uuid, token) => {
-  const { viamApi } = window;
+  const { viamApi, loadedIdentities } = window;
   const { publicKey } = identity.authentication;
   viamApi.setSessionData(uuid, token);
   localStorage.setItem("uuid", uuid);
   localStorage.setItem("token", token);
   localStorage.setItem("authenticatedIdentity", publicKey);
-  window.currentlyAuthenticatedIdentity = identity;
+  window.currentlyAuthenticatedIdentity =
+    loadedIdentities[publicKey] || identity;
   window.lastTimeGetProfile = 0;
   setKeyForUUID(uuid, publicKey);
 };
-- 
GitLab