Add support for new Snaps state methods

MetaMask · Dec 9, 2024 · 3646fe2 · 3646fe2
1 parent cbb57a1
commit 3646fe2
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/app/scripts/controllers/permissions/specifications.js b/app/scripts/controllers/permissions/specifications.js
@@ -408,13 +408,16 @@ export const unrestrictedMethods = Object.freeze([
   'wallet_invokeSnap',
   'wallet_invokeKeyring',
   'snap_getClientStatus',
+  'snap_getCurrencyRate',
+  'snap_clearState',
   'snap_getFile',
+  'snap_getState',
   'snap_createInterface',
   'snap_updateInterface',
   'snap_getInterfaceState',
   'snap_getInterfaceContext',
   'snap_resolveInterface',
-  'snap_getCurrencyRate',
+  'snap_setState',
   ///: BEGIN:ONLY_INCLUDE_IF(build-mmi)
   'metamaskinstitutional_authenticate',
   'metamaskinstitutional_reauthenticate',

diff --git a/app/scripts/metamask-controller.js b/app/scripts/metamask-controller.js
@@ -6113,6 +6113,10 @@ export default class MetamaskController extends EventEmitter {
 
     engine.push(
       createSnapsMethodMiddleware(subjectType === SubjectType.Snap, {
+        clearSnapState: this.controllerMessenger.call.bind(
+          this.controllerMessenger,
+          'SnapController:clearSnapState',
+        ),
         getUnlockPromise: this.appStateController.getUnlockPromise.bind(
           this.appStateController,
         ),
@@ -6135,6 +6139,14 @@ export default class MetamaskController extends EventEmitter {
           'SnapController:getFile',
           origin,
         ),
+        getSnapState: this.controllerMessenger.call.bind(
+          this.controllerMessenger,
+          'SnapController:getSnapState',
+        ),
+        updateSnapState: this.controllerMessenger.call.bind(
+          this.controllerMessenger,
+          'SnapController:updateSnapState',
+        ),
         installSnaps: this.controllerMessenger.call.bind(
           this.controllerMessenger,
           'SnapController:install',