From c8f2e8e82d25bf94661f7e624b9f3c056cedcedf Mon Sep 17 00:00:00 2001
From: Guillaume Roux <guillaumeroux123@gmail.com>
Date: Wed, 2 Aug 2023 15:16:43 +0200
Subject: [PATCH] Check for blocked snaps on extension startup (#20336)

* allow `SnapController` to call `SnapsRegistry:update`

* call `SnapController:updateBlockedSnaps` on extension startup

* only check if at least one snap is installed
---
 app/scripts/background.js          | 7 +++++++
 app/scripts/metamask-controller.js | 1 +
 2 files changed, 8 insertions(+)

diff --git a/app/scripts/background.js b/app/scripts/background.js
index 44f9d177d..c5ffd2d5d 100644
--- a/app/scripts/background.js
+++ b/app/scripts/background.js
@@ -821,6 +821,13 @@ export function setupController(
     });
   }
   ///: END:ONLY_INCLUDE_IN
+
+  ///: BEGIN:ONLY_INCLUDE_IN(snaps)
+  // Updates the snaps registry and check for newly blocked snaps to block if the user has at least one snap installed.
+  if (Object.keys(controller.snapController.state.snaps).length > 0) {
+    controller.snapController.updateBlockedSnaps();
+  }
+  ///: END:ONLY_INCLUDE_IN
 }
 
 //
diff --git a/app/scripts/metamask-controller.js b/app/scripts/metamask-controller.js
index 8c3dad7b0..341fae755 100644
--- a/app/scripts/metamask-controller.js
+++ b/app/scripts/metamask-controller.js
@@ -959,6 +959,7 @@ export default class MetamaskController extends EventEmitter {
         'ExecutionService:handleRpcRequest',
         'SnapsRegistry:get',
         'SnapsRegistry:getMetadata',
+        'SnapsRegistry:update',
       ],
     });