Expose addDetectedTokens action

MetaMask · Jan 24, 2024 · f230718 · f230718
1 parent 5aed7e0
commit f230718
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/packages/assets-controllers/src/TokensController.ts b/packages/assets-controllers/src/TokensController.ts
@@ -109,13 +109,20 @@ export type TokensState = BaseState &
  */
 const controllerName = 'TokensController';
 
-export type TokensControllerActions = TokensControllerGetStateAction;
+export type TokensControllerActions =
+  | TokensControllerGetStateAction
+  | TokensControllerAddDetectedTokensAction;
 
 export type TokensControllerGetStateAction = ControllerGetStateAction<
   typeof controllerName,
   TokensState
 >;
 
+export type TokensControllerAddDetectedTokensAction = {
+  type: `${typeof controllerName}:addDetectedTokens`;
+  handler: TokensController['addDetectedTokens'];
+};
+
 /**
  * The external actions available to the {@link TokensController}.
  */
@@ -246,6 +253,11 @@ export class TokensController extends BaseControllerV1<
 
     this.messagingSystem = messenger;
 
+    this.messagingSystem.registerActionHandler(
+      `${controllerName}:addDetectedTokens` as const,
+      this.addDetectedTokens.bind(this),
+    );
+
     this.messagingSystem.subscribe(
       'PreferencesController:stateChange',
       ({ selectedAddress }) => {

diff --git a/packages/assets-controllers/src/index.ts b/packages/assets-controllers/src/index.ts
@@ -21,7 +21,17 @@ export type {
 export { TokenDetectionController } from './TokenDetectionController';
 export * from './TokenListController';
 export * from './TokenRatesController';
-export * from './TokensController';
+export type {
+  TokensConfig,
+  TokensState,
+  TokensControllerActions,
+  TokensControllerGetStateAction,
+  TokensControllerAddDetectedTokensAction,
+  TokensControllerEvents,
+  TokensControllerStateChangeEvent,
+  TokensControllerMessenger,
+} from './TokensController';
+export { TokensController } from './TokensController';
 export {
   isTokenDetectionSupportedForNetwork,
   formatIconUrlWithProxy,