From ca43e883006a58d92eec519925a007c0c82c55c6 Mon Sep 17 00:00:00 2001
From: Derrick Hammer <derrick@derrickhammer.com>
Date: Fri, 23 Jun 2023 18:48:43 -0400
Subject: [PATCH] fix: improve portal api

---
 src/portal.ts | 67 ++++++++++++++++++++++++++++++++-------------------
 1 file changed, 42 insertions(+), 25 deletions(-)

diff --git a/src/portal.ts b/src/portal.ts
index 3c0bd7b..48cf78d 100644
--- a/src/portal.ts
+++ b/src/portal.ts
@@ -10,7 +10,7 @@ export const DEFAULT_PORTAL_LIST: Portal[] = [
   { id: "lumeweb", url: "https://web3portal.com", name: "web3portal.com" },
 ];
 
-const ACTIVE_PORTALS = new Set<Client>();
+let ACTIVE_PORTALS = new Set<Client>();
 
 type PortalSessionsStore = { [id: string]: string };
 
@@ -19,31 +19,8 @@ export function maybeInitDefaultPortals(): ErrTuple {
     return [null, "activePortalMasterKey not set"];
   }
 
-  let portalSessionsData = window.localStorage.getItem("portals");
-  let portalSessions: PortalSessionsStore = {};
-  if (portalSessions) {
-    portalSessions = JSON.parse(
-      portalSessionsData as string,
-    ) as PortalSessionsStore;
-  }
-
   for (const portal of DEFAULT_PORTAL_LIST) {
-    let jwt: string | null = null;
-
-    if (portalSessions) {
-      if (portal.id in portalSessions) {
-        jwt = portalSessions[portal.id];
-      }
-    }
-
-    const client = new Client({
-      email: generatePortalEmail(portal),
-      portalUrl: portal.url,
-      privateKey: generatePortalKeyPair(portal).privateKey,
-      jwt: jwt as string,
-    });
-
-    ACTIVE_PORTALS.add(client);
+    initPortal(portal);
   }
 
   return [null, null];
@@ -76,3 +53,43 @@ export function generatePortalKeyPair(portal: Portal): KeyPair {
 export function getActivePortals(): Set<Client> {
   return ACTIVE_PORTALS;
 }
+
+export function addActivePortal(portal: Client) {
+  ACTIVE_PORTALS.add(portal);
+}
+
+export function initPortal(portal: Portal) {
+  const sessions = getPortalSessions();
+  let jwt: string | null = null;
+  if (sessions) {
+    if (portal.id in sessions) {
+      jwt = sessions[portal.id];
+    }
+  }
+
+  const client = new Client({
+    email: generatePortalEmail(portal),
+    portalUrl: portal.url,
+    privateKey: generatePortalKeyPair(portal).privateKey,
+    jwt: jwt as string,
+  });
+
+  addActivePortal(client);
+}
+export function getPortalSessions() {
+  let portalSessionsData = window.localStorage.getItem("portals");
+  let portalSessions: PortalSessionsStore = {};
+  if (portalSessions) {
+    portalSessions = JSON.parse(
+      portalSessionsData as string,
+    ) as PortalSessionsStore;
+
+    return portalSessions;
+  }
+
+  return undefined;
+}
+
+export function setActivePortals(portals: Client[]) {
+  ACTIVE_PORTALS = new Set<Client>(portals);
+}