From 58c152e81d18260b7b62670fd358adb752b7c383 Mon Sep 17 00:00:00 2001
From: Sasha Ilieva <sasha.ilieva@vereign.com>
Date: Wed, 25 Nov 2020 15:35:28 +0200
Subject: [PATCH] Add setIdentityInLocalStorage

---
 javascript/src/iframe/viamapi-iframe.js | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/javascript/src/iframe/viamapi-iframe.js b/javascript/src/iframe/viamapi-iframe.js
index 9784504..f2b9b6e 100644
--- a/javascript/src/iframe/viamapi-iframe.js
+++ b/javascript/src/iframe/viamapi-iframe.js
@@ -733,7 +733,23 @@ const connection = Penpal.connectToParent({
           phone,
           email
         ).then(executeResult => {
-          result(executeResult);
+          let sequence = Promise.resolve();
+          if (executeResult.code === "200") {
+            sequence = sequence.then(() => {
+              setIdentityInLocalStorage(window.currentlyLoadedIdentity);
+            });
+          }
+          sequence
+            .then(() => {
+              result(executeResult);
+            })
+            .catch(e => {
+              result({
+                data: "",
+                code: "400",
+                status: "Can not store identity: " + e
+              });
+            });
         });
       });
     },
-- 
GitLab