Add getSupportedAccounts

Author: FrederikBolding

packages/snaps-controllers/src/multichain/MultichainRouter.test.ts

@@ -367,4 +367,40 @@ describe('MultichainRouter', () => {
       ).toStrictEqual(['getVersion']);
     });
   });
+
+  describe('getSupportedAccounts', () => {
+    it('returns a set of both protocol and account Snap methods', async () => {
+      const rootMessenger = getRootMultichainRouterMessenger();
+      const messenger = getRestrictedMultichainRouterMessenger(rootMessenger);
+      const withSnapKeyring = getMockWithSnapKeyring();
+
+      /* eslint-disable-next-line no-new */
+      new MultichainRouter({
+        messenger,
+        withSnapKeyring,
+      });
+
+      rootMessenger.registerActionHandler('SnapController:getAll', () => {
+        return [getTruncatedSnap()];
+      });
+
+      rootMessenger.registerActionHandler(
+        'AccountsController:listMultichainAccounts',
+        () => MOCK_SOLANA_ACCOUNTS,
+      );
+
+      rootMessenger.registerActionHandler(
+        'PermissionController:getPermissions',
+        () => MOCK_SOLANA_SNAP_PERMISSIONS,
+      );
+
+      expect(
+        messenger.call('MultichainRouter:getSupportedAccounts', {
+          scope: SOLANA_CAIP2,
+        }),
+      ).toStrictEqual([
+        'solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp:7S3P4HxJpyyigGzodYwHtCxZyUQe9JiBMHyRWXArAaKv',
+      ]);
+    });
+  });
 });

packages/snaps-controllers/src/multichain/MultichainRouter.ts

@@ -28,6 +28,11 @@ export type MultichainRouterGetSupportedMethodsAction = {
   handler: MultichainRouter['getSupportedMethods'];
 };
 
+export type MultichainRouterGetSupportedAccountsAction = {
+  type: `${typeof name}:getSupportedAccounts`;
+  handler: MultichainRouter['getSupportedAccounts'];
+};
+
 // Since the AccountsController depends on snaps-controllers we manually type this
 type InternalAccount = {
   id: string;
@@ -62,7 +67,8 @@ export type AccountsControllerListMultichainAccountsAction = {
 
 export type MultichainRouterActions =
   | MultichainRouterHandleRequestAction
-  | MultichainRouterGetSupportedMethodsAction;
+  | MultichainRouterGetSupportedMethodsAction
+  | MultichainRouterGetSupportedAccountsAction;
 
 export type MultichainRouterAllowedActions =
   | GetAllSnaps
@@ -110,6 +116,11 @@ export class MultichainRouter {
       `${name}:getSupportedMethods`,
       (...args) => this.getSupportedMethods(...args),
     );
+
+    this.#messenger.registerActionHandler(
+      `${name}:getSupportedAccounts`,
+      (...args) => this.getSupportedAccounts(...args),
+    );
   }
 
   async #resolveRequestAddress(
@@ -300,4 +311,18 @@ export class MultichainRouter {
 
     return Array.from(new Set([...accountMethods, ...protocolMethods]));
   }
+
+  /**
+   * Get a list of supported accounts for a given scope.
+   *
+   * @param options - An options bag.
+   * @param options.scope - The CAIP-2 scope.
+   * @returns A list of CAIP-10 addresses.
+   */
+  getSupportedAccounts({ scope }: { scope: CaipChainId }): string[] {
+    return this.#messenger
+      .call('AccountsController:listMultichainAccounts', scope)
+      .filter((account: InternalAccount) => account.metadata.snap?.enabled)
+      .map((account) => `${scope}:${account.address}`);
+  }
 }