Merge pull request #565 from Psychedelic/fix/app-canister-list

Fix/app canister list
Psychedelic · Aug 22, 2022 · 74ebb99 · 74ebb99
2 parents 53f2027 + 86f4f65
commit 74ebb99
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/source/Modules/Controller/connection.js b/source/Modules/Controller/connection.js
@@ -131,6 +131,11 @@ export class ConnectionModule extends ControllerModuleBase {
 
         const date = new Date().toISOString();
 
+        const populatedWhitelist = canistersInfo.reduce(
+          (accum, canisterInfo) => ({ ...accum, [canisterInfo.id]: canisterInfo }),
+          {},
+        );
+
         getApps(this.keyring?.currentWalletId.toString(), (apps = {}) => {
           const newApps = {
             ...apps,
@@ -144,7 +149,7 @@ export class ConnectionModule extends ControllerModuleBase {
               events: [
                 ...apps[domainUrl]?.events || [],
               ],
-              whitelist,
+              whitelist: populatedWhitelist,
               host,
             },
           };
@@ -209,7 +214,7 @@ export class ConnectionModule extends ControllerModuleBase {
             ? CONNECTION_STATUS.accepted
             : response.status;
           const whitelist = response.status === CONNECTION_STATUS.accepted
-            ? response.whitelist
+            ? apps[url]?.whitelist
             : [];
 
           const date = new Date().toISOString();

diff --git a/source/views/Popup/components/AllowAgent/index.jsx b/source/views/Popup/components/AllowAgent/index.jsx
@@ -58,14 +58,10 @@ const AllowAgent = ({
   const [expand, setExpand] = useState(false);
 
   const handleAllowAgent = async (status) => {
-    const whitelist = canisters.reduce(
-      (accum, canisterInfo) => ({ ...accum, [canisterInfo.id]: canisterInfo }),
-      {},
-    );
     reviewPendingTransaction(transactionId, () => {});
     const success = await portRPC.call('handleAllowAgent', [
       url,
-      { status, whitelist },
+      { status },
       callId,
       portId,
       transactionId,