diff --git a/javascript/src/iframe/viamapi-iframe.js b/javascript/src/iframe/viamapi-iframe.js index e3a6fbf08d89784e538d17aa27c027d2082eac50..20da6bd9d0a02ffeaaee5b30a61d64ba55010e9b 100644 --- a/javascript/src/iframe/viamapi-iframe.js +++ b/javascript/src/iframe/viamapi-iframe.js @@ -561,6 +561,10 @@ const connection = Penpal.connectToParent({ newIdentity.setPinCode(pinCode); window.currentlyLoadedIdentity = newIdentity; + localStorage.setItem( + "currentlyLoadedIdentity", + JSON.stringify(newIdentity) + ); const { publicKey, x509Certificate } = newIdentity.authentication; window.loadedIdentities[publicKey] = newIdentity; @@ -984,10 +988,14 @@ const connection = Penpal.connectToParent({ return response; }, - contactsCheckAccountRecoveryStatus: async restoreAccessIdentity => { - window.currentlyLoadedIdentity = restoreAccessIdentity; - const { publicKey } = restoreAccessIdentity.authentication; - window.loadedIdentities[publicKey] = restoreAccessIdentity; + contactsCheckAccountRecoveryStatus: async () => { + const currentlyLoadedIdentity = localStorage.getItem( + "currentlyLoadedIdentity" + ); + const parsedIdentity = JSON.parse(currentlyLoadedIdentity); + window.currentlyLoadedIdentity = parsedIdentity; + const { publicKey } = parsedIdentity.authentication; + window.loadedIdentities[publicKey] = parsedIdentity; window.viamAnonymousApi.setIdentity(publicKey); window.viamApi.setSessionData("", ""); window.viamApi.setIdentity(publicKey); @@ -1021,9 +1029,10 @@ const connection = Penpal.connectToParent({ console.log({ identityLoginResponse }); const { code, data } = identityLoginResponse; if (code === "200") { - handleIdentityLogin(restoreAccessIdentity, data.Uuid, data.Session); - await getProfileData(restoreAccessIdentity); - await setIdentityInLocalStorage(restoreAccessIdentity); + handleIdentityLogin(parsedIdentity, data.Uuid, data.Session); + await getProfileData(parsedIdentity); + await setIdentityInLocalStorage(parsedIdentity); + localStorage.removeItem("currentlyLoadedIdentity"); } }