Wire up to SnapInterfaceController

Author: GuillaumeRx

packages/snaps-controllers/coverage.json

@@ -1,6 +1,6 @@
 {
-  "branches": 93.31,
-  "functions": 97.05,
-  "lines": 98.25,
-  "statements": 97.98
+  "branches": 93.36,
+  "functions": 97.06,
+  "lines": 98.26,
+  "statements": 97.99
 }

packages/snaps-controllers/src/interface/SnapInterfaceController.test.tsx

@@ -9,6 +9,7 @@ import {
   ContentType,
 } from '@metamask/snaps-sdk';
 import {
+  AssetSelector,
   Box,
   Field,
   FileInput,
@@ -159,6 +160,36 @@ describe('SnapInterfaceController', () => {
       expect(state).toStrictEqual({ foo: { bar: null } });
     });
 
+    it('calls the multichain asset controller to construct an asset selector state', async () => {
+      const rootMessenger = getRootSnapInterfaceControllerMessenger();
+      const controllerMessenger =
+        getRestrictedSnapInterfaceControllerMessenger(rootMessenger);
+
+      // eslint-disable-next-line no-new
+      new SnapInterfaceController({
+        messenger: controllerMessenger,
+      });
+
+      const components = (
+        <AssetSelector
+          name="foo"
+          addresses={['eip155:0:0x1234567890123456789012345678901234567890']}
+          value="eip155:1/erc20:0x6b175474e89094c44da98b954eedeac495271d0f"
+        />
+      );
+
+      await rootMessenger.call(
+        'SnapInterfaceController:createInterface',
+        MOCK_SNAP_ID,
+        components,
+      );
+
+      expect(rootMessenger.call).toHaveBeenNthCalledWith(
+        3,
+        'MultichainAssetsController:getState',
+      );
+    });
+
     it('can create a new interface from JSX', async () => {
       const rootMessenger = getRootSnapInterfaceControllerMessenger();
       const controllerMessenger =

packages/snaps-controllers/src/interface/SnapInterfaceController.ts

@@ -17,11 +17,12 @@ import type {
   SnapId,
   ComponentOrElement,
   InterfaceContext,
+  FungibleAssetMetadata,
 } from '@metamask/snaps-sdk';
 import { ContentType } from '@metamask/snaps-sdk';
 import type { JSXElement } from '@metamask/snaps-sdk/jsx';
 import { getJsonSizeUnsafe, validateJsxLinks } from '@metamask/snaps-utils';
-import type { Json } from '@metamask/utils';
+import type { CaipAssetType, Json } from '@metamask/utils';
 import { assert, hasProperty } from '@metamask/utils';
 import { castDraft } from 'immer';
 import { nanoid } from 'nanoid';
@@ -72,12 +73,23 @@ export type SnapInterfaceControllerGetStateAction = ControllerGetStateAction<
   SnapInterfaceControllerState
 >;
 
+type MultichainAssetsControllerGetStateAction = ControllerGetStateAction<
+  'MultichainAssetsController',
+  {
+    assetsMetadata: {
+      [asset: CaipAssetType]: FungibleAssetMetadata;
+    };
+    accountsAssets: { [account: string]: CaipAssetType[] };
+  }
+>;
+
 export type SnapInterfaceControllerAllowedActions =
   | TestOrigin
   | MaybeUpdateState
   | HasApprovalRequest
   | AcceptRequest
-  | GetSnap;
+  | GetSnap
+  | MultichainAssetsControllerGetStateAction;
 
 export type SnapInterfaceControllerActions =
   | CreateInterface
@@ -249,7 +261,9 @@ export class SnapInterfaceController extends BaseController<
     validateInterfaceContext(context);
 
     const id = nanoid();
-    const componentState = constructState({}, element);
+    const componentState = constructState({}, element, {
+      getAssetMetadata: this.#getAssetMetadata.bind(this),
+    });
 
     this.update((draftState) => {
       // @ts-expect-error - TS2589: Type instantiation is excessively deep and
@@ -299,7 +313,9 @@ export class SnapInterfaceController extends BaseController<
     validateInterfaceContext(context);
 
     const oldState = this.state.interfaces[id].state;
-    const newState = constructState(oldState, element);
+    const newState = constructState(oldState, element, {
+      getAssetMetadata: this.#getAssetMetadata.bind(this),
+    });
 
     this.update((draftState) => {
       draftState.interfaces[id].state = newState;
@@ -426,6 +442,15 @@ export class SnapInterfaceController extends BaseController<
     );
   }
 
+  #getAssetMetadata(assetId: CaipAssetType) {
+    // TODO: Introduce an action in MultichainAssetsController to get a specific asset metadata.
+    const assets = this.messagingSystem.call(
+      'MultichainAssetsController:getState',
+    );
+
+    return assets.assetsMetadata[assetId];
+  }
+
   /**
    * Utility function to validate the components of an interface.
    * Throws if something is invalid.