@subwallet/extension-base 1.0.1 → 1.0.2-1b

package/koni/background/handlers/State.d.ts

@@ -1,11 +1,13 @@
 /// <reference types="chrome" />
 import { _AssetRef, _AssetType, _ChainAsset, _ChainInfo, _MultiChainAsset } from '@subwallet/chain-list/types';
-import { AddTokenRequestExternal, ApiMap, AuthRequestV2, BalanceItem, BalanceJson, BrowserConfirmationType, ChainStakingMetadata, ConfirmationsQueue, CrowdloanItem, CrowdloanJson, CurrentAccountInfo, EvmSendTransactionParams, ExternalRequestPromise, KeyringState, NftCollection, NftItem, NftJson, NominatorMetadata, RequestAccountExportPrivateKey, RequestCheckPublicAndSecretKey, RequestConfirmationComplete, RequestSettingsType, ResponseAccountExportPrivateKey, ResponseCheckPublicAndSecretKey, ServiceInfo, SingleModeJson, StakingItem, StakingJson, StakingRewardItem, StakingRewardJson, StakingType, ThemeNames, UiSettings } from '@subwallet/extension-base/background/KoniTypes';
+import { AddTokenRequestExternal, ApiMap, AuthRequestV2, BalanceItem, BalanceJson, BrowserConfirmationType, ChainStakingMetadata, ConfirmationsQueue, CrowdloanItem, CrowdloanJson, CurrentAccountInfo, EvmSendTransactionParams, ExternalRequestPromise, NftCollection, NftItem, NftJson, NominatorMetadata, RequestAccountExportPrivateKey, RequestCheckPublicAndSecretKey, RequestConfirmationComplete, RequestSettingsType, ResponseAccountExportPrivateKey, ResponseCheckPublicAndSecretKey, ServiceInfo, SingleModeJson, StakingItem, StakingJson, StakingRewardItem, StakingRewardJson, StakingType, ThemeNames, UiSettings } from '@subwallet/extension-base/background/KoniTypes';
 import { AccountJson, RequestAuthorizeTab, RequestRpcSend, RequestRpcSubscribe, RequestRpcUnsubscribe, RequestSign, ResponseRpcListProviders, ResponseSigning } from '@subwallet/extension-base/background/types';
 import { BalanceService } from '@subwallet/extension-base/services/balance-service';
 import { ChainService } from '@subwallet/extension-base/services/chain-service';
 import { _ChainConnectionStatus, _ChainState, _NetworkUpsertParams, _ValidateCustomAssetRequest } from '@subwallet/extension-base/services/chain-service/types';
+import { EventService } from '@subwallet/extension-base/services/event-service';
 import { HistoryService } from '@subwallet/extension-base/services/history-service';
+import { KeyringService } from '@subwallet/extension-base/services/keyring-service';
 import MigrationService from '@subwallet/extension-base/services/migration-service';
 import NotificationService from '@subwallet/extension-base/services/notification-service/NotificationService';
 import { PriceService } from '@subwallet/extension-base/services/price-service';
@@ -13,6 +15,7 @@ import RequestService from '@subwallet/extension-base/services/request-service';
 import { AuthUrls, MetaRequest, SignRequest } from '@subwallet/extension-base/services/request-service/types';
 import SettingService from '@subwallet/extension-base/services/setting-service/SettingService';
 import DatabaseService from '@subwallet/extension-base/services/storage-service/DatabaseService';
+import { SubscanService } from '@subwallet/extension-base/services/subscan-service';
 import TransactionService from '@subwallet/extension-base/services/transaction-service';
 import { MetadataDef, ProviderMeta } from '@subwallet/extension-inject/types';
 import SimpleKeyring from 'eth-simple-keyring';
@@ -26,12 +29,8 @@ export default class KoniState {
     private injectedProviders;
     private readonly providers;
     private readonly unsubscriptionMap;
-    private readonly currentAccountStore;
     private readonly accountRefStore;
-    private readonly keyringStateSubject;
     private externalRequest;
-    private keyringState;
-    private serviceInfoSubject;
     private balanceMap;
     private balanceSubject;
     private crowdloanMap;
@@ -44,6 +43,8 @@ export default class KoniState {
     private stakingRewardState;
     private lazyMap;
     readonly notificationService: NotificationService;
+    readonly eventService: EventService;
+    readonly keyringService: KeyringService;
     readonly chainService: ChainService;
     readonly dbService: DatabaseService;
     private cron;
@@ -57,6 +58,7 @@ export default class KoniState {
     readonly priceService: PriceService;
     readonly balanceService: BalanceService;
     readonly migrationService: MigrationService;
+    readonly subscanService: SubscanService;
     constructor(providers?: Providers);
     get knownMetadata(): MetadataDef[];
     injectMetadata(url: string, request: MetadataDef): Promise<boolean>;
@@ -76,9 +78,7 @@ export default class KoniState {
     private startSubscription;
     onReady(): void;
     isReady(): boolean;
-    getKeyringState(): KeyringState;
-    subscribeKeyringState(): Subject<KeyringState>;
-    setKeyringState(data: KeyringState, callback?: () => void): void;
+    updateKeyringState(isReady?: boolean, callback?: () => void): void;
     private lazyNext;
     getAuthRequestV2(id: string): AuthRequestV2;
     setAuthorize(data: AuthUrls, callback?: () => void): void;
@@ -106,10 +106,10 @@ export default class KoniState {
     setNftCollection(network: string, data: NftCollection, callback?: (data: NftCollection) => void): void;
     getNftCollection(): import("dexie").PromiseExtended<NftCollection[]>;
     subscribeNftCollection(): import("dexie").Observable<NftCollection[]>;
-    resetNft(newAddress: string): Promise<void>;
+    resetNft(newAddress: string): void;
     updateNftData(network: string, nftData: NftItem, address: string, callback?: (nftData: NftItem) => void): void;
-    removeNfts(chain: string, address: string, collectionId: string, nftIds: string[]): import("dexie").PromiseExtended<number>;
     deleteNftCollection(chain: string, collectionId: string): Promise<void>;
+    cleanUpNfts(chain: string, owner: string, collectionId: string, nftIds: string[]): void;
     getNft(): Promise<NftJson | undefined>;
     subscribeNft(): Subject<NftJson>;
     resetStakingReward(): void;
@@ -120,7 +120,6 @@ export default class KoniState {
     removeAccountRef(address: string, callback: () => void): void;
     getStakingReward(update: (value: StakingRewardJson) => void): void;
     subscribeStakingReward(): Subject<StakingRewardJson>;
-    getCurrentAccount(update: (value: CurrentAccountInfo) => void): void;
     setCurrentAccount(data: CurrentAccountInfo, callback?: () => void): void;
     setAccountTie(address: string, genesisHash: string | null): boolean;
     switchEvmNetworkByUrl(shortenUrl: string, networkKey: string): Promise<void>;
@@ -135,17 +134,16 @@ export default class KoniState {
     setBrowserConfirmationType(browserConfirmationType: BrowserConfirmationType, callback?: (settingData: UiSettings) => void): void;
     setCamera(value: boolean): void;
     subscribeSettingsSubject(): Subject<RequestSettingsType>;
-    subscribeCurrentAccount(): Subject<CurrentAccountInfo>;
-    getAccountAddress(): Promise<string | null | undefined>;
-    getDecodedAddresses(address?: string): Promise<string[]>;
+    getAccountAddress(): string | null;
+    getDecodedAddresses(address?: string): string[];
     getAllAddresses(): string[];
     private removeInactiveChainBalances;
     getBalance(reset?: boolean): BalanceJson;
     getStoredBalance(address: string): Promise<Record<string, BalanceItem>>;
-    switchAccount(newAddress: string): Promise<void>;
+    handleSwitchAccount(newAddress: string): Promise<void>;
     resetBalanceMap(newAddress: string): Promise<void>;
     resetCrowdloanMap(newAddress: string): Promise<void>;
-    resetStaking(newAddress: string): Promise<void>;
+    resetStaking(newAddress: string): void;
     setBalanceItem(tokenSlug: string, item: BalanceItem): void;
     private updateBalanceStore;
     subscribeBalance(): Subject<BalanceJson>;
@@ -154,7 +152,6 @@ export default class KoniState {
     setCrowdloanItem(networkKey: string, item: CrowdloanItem): void;
     private updateCrowdloanStore;
     subscribeCrowdloan(): Subject<CrowdloanJson>;
-    getAllPriceIds(): string[];
     getSmartContractNfts(): _ChainAsset[];
     getChainInfoMap(): Record<string, _ChainInfo>;
     getChainStateMap(): Record<string, _ChainState>;
@@ -183,7 +180,6 @@ export default class KoniState {
     disableChain(chainSlug: string): Promise<boolean>;
     enableChain(chainSlug: string, enableTokens?: boolean): Promise<boolean>;
     resetDefaultChains(): boolean;
-    updateNetworkStatus(networkKey: string, status: _ChainConnectionStatus): void;
     getSubstrateApiMap(): Record<string, import("@subwallet/extension-base/services/chain-service/types")._SubstrateApi>;
     getSubstrateApi(networkKey: string): import("@subwallet/extension-base/services/chain-service/types")._SubstrateApi;
     getEvmApiMap(): Record<string, import("@subwallet/extension-base/services/chain-service/types")._EvmApi>;
@@ -191,8 +187,7 @@ export default class KoniState {
     getApiMap(): ApiMap;
     refreshSubstrateApi(key: string): boolean;
     refreshWeb3Api(key: string): void;
-    subscribeServiceInfo(): Subject<ServiceInfo>;
-    updateServiceInfo(): void;
+    getServiceInfo(): ServiceInfo;
     getExternalRequestMap(): Record<string, ExternalRequestPromise>;
     setExternalRequestMap(id: string, value: ExternalRequestPromise): void;
     getExternalRequest(id: string): ExternalRequestPromise;
@@ -220,5 +215,11 @@ export default class KoniState {
     wakeup(): Promise<void>;
     cancelSubscription(id: string): boolean;
     createUnsubscriptionHandle(id: string, unsubscribe: () => void): void;
+    updateChainConnectionStatus(chain: string, status: _ChainConnectionStatus): void;
+    autoEnableChains(addresses: string[]): Promise<void>;
+    onAccountAdd(): void;
+    onAccountRemove(): void;
+    reloadNft(): Promise<boolean>;
+    reloadStaking(): Promise<boolean>;
 }
 export {};

package/koni/background/handlers/State.js

@@ -11,15 +11,18 @@ import { BalanceService } from '@subwallet/extension-base/services/balance-servi
 import { ChainService } from '@subwallet/extension-base/services/chain-service';
 import { _PREDEFINED_SINGLE_MODES } from '@subwallet/extension-base/services/chain-service/constants';
 import { _getEvmChainId, _getSubstrateGenesisHash, _isAssetFungibleToken, _isChainEnabled, _isChainTestNet, _isSubstrateParachain, _parseMetadataForSmartContractAsset } from '@subwallet/extension-base/services/chain-service/utils';
+import { EventService } from '@subwallet/extension-base/services/event-service';
 import { HistoryService } from '@subwallet/extension-base/services/history-service';
+import { KeyringService } from '@subwallet/extension-base/services/keyring-service';
 import MigrationService from '@subwallet/extension-base/services/migration-service';
 import NotificationService from '@subwallet/extension-base/services/notification-service/NotificationService';
 import { PriceService } from '@subwallet/extension-base/services/price-service';
 import RequestService from '@subwallet/extension-base/services/request-service';
 import SettingService from '@subwallet/extension-base/services/setting-service/SettingService';
 import DatabaseService from '@subwallet/extension-base/services/storage-service/DatabaseService';
+import { SubscanService } from '@subwallet/extension-base/services/subscan-service';
+import { SUBSCAN_CHAIN_MAP_REVERSE } from '@subwallet/extension-base/services/subscan-service/subscan-chain-map';
 import TransactionService from '@subwallet/extension-base/services/transaction-service';
-import { CurrentAccountStore } from '@subwallet/extension-base/stores';
 import AccountRefStore from '@subwallet/extension-base/stores/AccountRef';
 import { isContractAddress, parseContractInput } from '@subwallet/extension-base/utils/eth/parseTransaction';
 import { decodePair } from '@subwallet/keyring/pair/decode';
@@ -54,16 +57,8 @@ const generateDefaultCrowdloanMap = () => {
 export default class KoniState {
   injectedProviders = new Map();
   unsubscriptionMap = {};
-  currentAccountStore = new CurrentAccountStore();
   accountRefStore = new AccountRefStore();
-  keyringStateSubject = new Subject();
   externalRequest = {};
-  keyringState = {
-    isReady: false,
-    isLocked: true,
-    hasMasterPassword: false
-  };
-  serviceInfoSubject = new Subject();
   balanceMap = {};
   balanceSubject = new Subject();
   crowdloanMap = generateDefaultCrowdloanMap();
@@ -80,18 +75,20 @@ export default class KoniState {
   };
   lazyMap = {};
   ready = false;
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
   constructor(providers = {}) {
     this.providers = providers;
     this.dbService = new DatabaseService();
+    this.eventService = new EventService();
+    this.subscanService = new SubscanService();
+    this.keyringService = new KeyringService(this.eventService);
     this.notificationService = new NotificationService();
-    this.chainService = new ChainService(this.dbService);
+    this.chainService = new ChainService(this.dbService, this.eventService);
     this.settingService = new SettingService();
     this.requestService = new RequestService(this.chainService, this.settingService);
-    this.priceService = new PriceService(this.serviceInfoSubject, this.dbService, this.chainService);
+    this.priceService = new PriceService(this.dbService, this.eventService, this.chainService);
     this.balanceService = new BalanceService(this.chainService);
-    this.historyService = new HistoryService(this.dbService, this.chainService);
-    this.transactionService = new TransactionService(this.chainService, this.requestService, this.balanceService, this.historyService, this.notificationService, this.dbService);
+    this.historyService = new HistoryService(this.dbService, this.chainService, this.eventService);
+    this.transactionService = new TransactionService(this.chainService, this.eventService, this.requestService, this.balanceService, this.historyService, this.notificationService, this.dbService);
     this.migrationService = new MigrationService(this);
     this.subscription = new KoniSubscription(this, this.dbService);
     this.cron = new KoniCron(this, this.subscription, this.dbService);
@@ -198,8 +195,10 @@ export default class KoniState {
     await this.chainService.init();
     await this.migrationService.run();
     this.startSubscription();
-    this.updateServiceInfo();
+    this.eventService.emit('chain.ready', true);
     this.onReady();
+    this.onAccountAdd();
+    this.onAccountRemove();
     this.logger.log('Done init state');
   }
   startSubscription() {
@@ -219,15 +218,8 @@ export default class KoniState {
   isReady() {
     return this.ready;
   }
-  getKeyringState() {
-    return this.keyringState;
-  }
-  subscribeKeyringState() {
-    return this.keyringStateSubject;
-  }
-  setKeyringState(data, callback) {
-    this.keyringStateSubject.next(data);
-    this.keyringState = data;
+  updateKeyringState(isReady = true, callback) {
+    this.keyringService.updateKeyringState(isReady);
     callback && callback();
   }
   lazyNext = (key, callback) => {
@@ -285,7 +277,7 @@ export default class KoniState {
     return this.dbService.getNominatorMetadata();
   }
   async getStaking() {
-    const addresses = await this.getDecodedAddresses();
+    const addresses = this.getDecodedAddresses();
     const stakings = await this.dbService.getStakings(addresses, this.activeChainSlugs);
     return {
       ready: true,
@@ -303,7 +295,7 @@ export default class KoniState {
     return addresses;
   }
   async getPooledStakingRecordsByAddress(addresses) {
-    return await this.dbService.getPooledStakings(addresses, this.activeChainSlugs);
+    return this.dbService.getPooledStakings(addresses, this.activeChainSlugs);
   }
 
   // TODO: delete later
@@ -344,12 +336,12 @@ export default class KoniState {
   subscribeNftCollection() {
     return this.dbService.stores.nftCollection.subscribeNftCollection(this.activeChainSlugs);
   }
-  async resetNft(newAddress) {
+  resetNft(newAddress) {
     this.getNft().then(data => this.nftSubject.next(data || {
       nftList: [],
       total: 0
     })).catch(e => this.logger.warn(e));
-    const addresses = await this.getDecodedAddresses(newAddress);
+    const addresses = this.getDecodedAddresses(newAddress);
     this.dbService.subscribeNft(addresses, this.activeChainSlugs, nfts => {
       this.nftSubject.next({
         nftList: nfts,
@@ -361,14 +353,14 @@ export default class KoniState {
     this.dbService.addNft(address, nftData).catch(e => this.logger.warn(e));
     callback && callback(nftData);
   }
-  removeNfts(chain, address, collectionId, nftIds) {
-    return this.dbService.removeNfts(chain, address, collectionId, nftIds);
-  }
   deleteNftCollection(chain, collectionId) {
     return this.dbService.deleteNftCollection(chain, collectionId);
   }
+  cleanUpNfts(chain, owner, collectionId, nftIds) {
+    this.dbService.cleanUpNft(chain, owner, collectionId, nftIds).catch(e => this.logger.warn(e));
+  }
   async getNft() {
-    const addresses = await this.getDecodedAddresses();
+    const addresses = this.getDecodedAddresses();
     if (!addresses.length) {
       return;
     }
@@ -442,9 +434,6 @@ export default class KoniState {
   subscribeStakingReward() {
     return this.stakingRewardSubject;
   }
-  getCurrentAccount(update) {
-    this.currentAccountStore.get('CurrentAccountInfo', update);
-  }
   setCurrentAccount(data, callback) {
     const {
       address,
@@ -454,9 +443,9 @@ export default class KoniState {
       ...data
     };
     if (address === ALL_ACCOUNT_KEY) {
-      const pairs = keyring.getPairs();
+      const pairs = keyring.getAccounts();
       const pair = pairs[0];
-      const pairGenesisHash = pair.meta.genesisHash;
+      const pairGenesisHash = (pair === null || pair === void 0 ? void 0 : pair.meta.genesisHash) || '';
       if (pairs.length > 1 || !pair) {
         result.allGenesisHash = currentGenesisHash || undefined;
       } else {
@@ -465,10 +454,8 @@ export default class KoniState {
         result.allGenesisHash = pairGenesisHash || undefined;
       }
     }
-    this.currentAccountStore.set('CurrentAccountInfo', result, () => {
-      this.updateServiceInfo();
-      callback && callback();
-    });
+    this.keyringService.setCurrentAccount(result);
+    callback && callback();
   }
   setAccountTie(address, genesisHash) {
     if (address !== ALL_ACCOUNT_KEY) {
@@ -479,12 +466,11 @@ export default class KoniState {
         genesisHash
       });
     }
-    this.getCurrentAccount(accountInfo => {
-      if (address === accountInfo.address) {
-        accountInfo.currentGenesisHash = genesisHash || ALL_GENESIS_HASH;
-        this.setCurrentAccount(accountInfo);
-      }
-    });
+    const accountInfo = this.keyringService.currentAccount;
+    if (address === accountInfo.address) {
+      accountInfo.currentGenesisHash = genesisHash || ALL_GENESIS_HASH;
+      this.setCurrentAccount(accountInfo);
+    }
     return true;
   }
   async switchEvmNetworkByUrl(shortenUrl, networkKey) {
@@ -507,9 +493,7 @@ export default class KoniState {
     const {
       address,
       currentGenesisHash
-    } = await new Promise(resolve => {
-      this.getCurrentAccount(resolve);
-    });
+    } = this.keyringService.currentAccount;
     return this.requestService.addConfirmation(id, url, 'switchNetworkRequest', {
       networkKey,
       address: changeAddress
@@ -624,35 +608,28 @@ export default class KoniState {
   subscribeSettingsSubject() {
     return this.settingService.getSubject();
   }
-  subscribeCurrentAccount() {
-    return this.currentAccountStore.getSubject();
-  }
   getAccountAddress() {
-    return new Promise((resolve, reject) => {
-      this.getCurrentAccount(account => {
-        if (account) {
-          resolve(account.address);
-        } else {
-          resolve(null);
-        }
-      });
-    });
+    const address = this.keyringService.currentAccount.address;
+    if (address === '') {
+      return null;
+    }
+    return address;
   }
-  async getDecodedAddresses(address) {
+  getDecodedAddresses(address) {
     let checkingAddress = address;
     if (!address) {
-      checkingAddress = await this.getAccountAddress();
+      checkingAddress = this.getAccountAddress();
     }
     if (!checkingAddress) {
       return [];
     }
     if (checkingAddress === ALL_ACCOUNT_KEY) {
-      return Object.keys(accounts.subject.value);
+      return this.getAllAddresses();
     }
     return [checkingAddress];
   }
   getAllAddresses() {
-    return Object.keys(accounts.subject.value);
+    return keyring.getAccounts().map(account => account.address);
   }
   removeInactiveChainBalances(balanceMap) {
     const activeBalanceMap = {};
@@ -678,7 +655,7 @@ export default class KoniState {
     const items = await this.dbService.stores.balance.getBalanceMapByAddress(address);
     return items || {};
   }
-  async switchAccount(newAddress) {
+  async handleSwitchAccount(newAddress) {
     await Promise.all([this.resetBalanceMap(newAddress), this.resetCrowdloanMap(newAddress)]);
   }
   async resetBalanceMap(newAddress) {
@@ -700,11 +677,11 @@ export default class KoniState {
     };
     this.publishCrowdloan(true);
   }
-  async resetStaking(newAddress) {
+  resetStaking(newAddress) {
     this.getStaking().then(data => {
       this.stakingSubject.next(data);
     }).catch(e => this.logger.warn(e));
-    const addresses = await this.getDecodedAddresses(newAddress);
+    const addresses = this.getDecodedAddresses(newAddress);
     this.dbService.subscribeStaking(addresses, this.activeChainSlugs, stakings => {
       this.stakingSubject.next({
         ready: true,
@@ -723,9 +700,8 @@ export default class KoniState {
     });
   }
   updateBalanceStore(item) {
-    this.getCurrentAccount(currentAccountInfo => {
-      this.dbService.updateBalanceStore(currentAccountInfo.address, item).catch(e => this.logger.warn(e));
-    });
+    const currentAccountInfo = this.keyringService.currentAccount;
+    this.dbService.updateBalanceStore(currentAccountInfo.address, item).catch(e => this.logger.warn(e));
   }
   subscribeBalance() {
     return this.balanceSubject;
@@ -754,16 +730,12 @@ export default class KoniState {
     });
   }
   updateCrowdloanStore(networkKey, item) {
-    this.getCurrentAccount(currentAccountInfo => {
-      this.dbService.updateCrowdloanStore(networkKey, currentAccountInfo.address, item).catch(e => this.logger.warn(e));
-    });
+    const currentAccountInfo = this.keyringService.currentAccount;
+    this.dbService.updateCrowdloanStore(networkKey, currentAccountInfo.address, item).catch(e => this.logger.warn(e));
   }
   subscribeCrowdloan() {
     return this.crowdloanSubject;
   }
-  getAllPriceIds() {
-    return this.chainService.getAllPriceIds();
-  }
   getSmartContractNfts() {
     return this.chainService.getSmartContractNfts();
   }
@@ -821,12 +793,13 @@ export default class KoniState {
       await this.chainService.updateAssetSetting(tokenSlug, {
         visible: true
       });
+      this.eventService.emit('asset.updateState', tokenSlug);
+    } else {
+      this.eventService.emit('asset.updateState', tokenSlug);
     }
-    this.updateServiceInfo();
   }
   deleteCustomAssets(targetTokens) {
     this.chainService.deleteCustomAssets(targetTokens);
-    this.updateServiceInfo();
   }
   async validateCustomChain(provider, existedChainSlug) {
     return await this.chainService.validateCustomChain(provider, existedChainSlug);
@@ -849,14 +822,12 @@ export default class KoniState {
       await this.chainService.updateAssetSetting(newNativeTokenSlug, {
         visible: true
       });
+      this.eventService.emit('asset.updateState', newNativeTokenSlug);
     }
-    this.updateServiceInfo();
     return true;
   }
   removeCustomChain(networkKey) {
-    const result = this.chainService.removeCustomChain(networkKey);
-    this.updateServiceInfo();
-    return result;
+    return this.chainService.removeCustomChain(networkKey);
   }
 
   // TODO: avoids turning off chains related to ledger account
@@ -882,29 +853,18 @@ export default class KoniState {
   async disableChain(chainSlug) {
     // const defaultChains = this.getDefaultNetworkKeys();
     await this.chainService.updateAssetSettingByChain(chainSlug, false);
-    const result = this.chainService.disableChain(chainSlug);
-    this.updateServiceInfo();
-    return result;
+    return this.chainService.disableChain(chainSlug);
   }
   async enableChain(chainSlug, enableTokens = true) {
     if (enableTokens) {
       await this.chainService.updateAssetSettingByChain(chainSlug, true);
     }
-    const result = this.chainService.enableChain(chainSlug);
-    this.updateServiceInfo();
-    return result;
+    return this.chainService.enableChain(chainSlug);
   }
   resetDefaultChains() {
     const defaultChains = this.getDefaultNetworkKeys();
     return this.chainService.resetChainInfoMap(defaultChains);
   }
-  updateNetworkStatus(networkKey, status) {
-    const chainState = this.chainService.getChainStateByKey(networkKey);
-    if (chainState.connectionStatus === status) {
-      return;
-    }
-    this.chainService.setChainConnectionStatus(networkKey, status);
-  }
   getSubstrateApiMap() {
     return this.chainService.getSubstrateApiMap();
   }
@@ -930,20 +890,14 @@ export default class KoniState {
   refreshWeb3Api(key) {
     this.chainService.refreshEvmApi(key);
   }
-  subscribeServiceInfo() {
-    return this.serviceInfoSubject;
-  }
-  updateServiceInfo() {
-    this.logger.log('<---Update serviceInfo--->');
-    this.getCurrentAccount(value => {
-      this.serviceInfoSubject.next({
-        chainInfoMap: this.chainService.getChainInfoMap(),
-        chainApiMap: this.getApiMap(),
-        currentAccountInfo: value,
-        assetRegistry: this.chainService.getAssetRegistry(),
-        chainStateMap: this.chainService.getChainStateMap()
-      });
-    });
+  getServiceInfo() {
+    return {
+      chainInfoMap: this.chainService.getChainInfoMap(),
+      chainApiMap: this.getApiMap(),
+      currentAccountInfo: this.keyringService.currentAccount,
+      assetRegistry: this.chainService.getAssetRegistry(),
+      chainStateMap: this.chainService.getChainStateMap()
+    };
   }
   getExternalRequestMap() {
     return this.externalRequest;
@@ -1374,4 +1328,83 @@ export default class KoniState {
   createUnsubscriptionHandle(id, unsubscribe) {
     this.unsubscriptionMap[id] = unsubscribe;
   }
+  updateChainConnectionStatus(chain, status) {
+    this.chainService.setChainConnectionStatus(chain, status);
+  }
+  async autoEnableChains(addresses) {
+    const assetMap = this.chainService.getAssetRegistry();
+    const promiseList = addresses.map(address => {
+      return this.subscanService.getMultiChainBalance(address).catch(e => {
+        console.error(e);
+        return null;
+      });
+    });
+    const needEnableChains = [];
+    const needActiveTokens = [];
+    const currentAssetSettings = await this.chainService.getAssetSettings();
+    const chainMap = this.chainService.getChainInfoMap();
+    const balanceDataList = await Promise.all(promiseList);
+    balanceDataList.forEach(balanceData => {
+      balanceData && balanceData.forEach(({
+        balance,
+        bonded,
+        category,
+        locked,
+        network,
+        symbol
+      }) => {
+        var _currentAssetSettings;
+        const chain = SUBSCAN_CHAIN_MAP_REVERSE[network];
+        const chainInfo = chain ? chainMap[chain] : null;
+        const balanceIsEmpty = (!balance || balance === '0') && !locked && locked === '0' && (!bonded || bonded === '0');
+
+        // Cancel if chain is not supported or is testnet or balance is 0
+        if (!chainInfo || chainInfo.isTestnet || balanceIsEmpty) {
+          return;
+        }
+        const tokenKey = `${chain}-${category === 'native' ? 'NATIVE' : 'LOCAL'}-${symbol.toUpperCase()}`;
+        if (assetMap[tokenKey] && !((_currentAssetSettings = currentAssetSettings[tokenKey]) !== null && _currentAssetSettings !== void 0 && _currentAssetSettings.visible)) {
+          needEnableChains.push(chain);
+          needActiveTokens.push(tokenKey);
+          currentAssetSettings[tokenKey] = {
+            visible: true
+          };
+        }
+      });
+    });
+    if (needActiveTokens.length) {
+      this.chainService.enableChains(needEnableChains);
+      this.chainService.setAssetSettings({
+        ...currentAssetSettings
+      });
+    }
+  }
+  onAccountAdd() {
+    this.eventService.on('account.add', address => {
+      this.autoEnableChains([address]).catch(this.logger.error);
+    });
+  }
+  onAccountRemove() {
+    this.eventService.on('account.remove', address => {
+      // Some separate service like historyService will listen to this event and remove inside that service
+
+      const stores = this.dbService.stores;
+
+      // Remove Balance
+      stores.balance.removeAllByAddress(address).catch(console.error);
+      stores.balance.removeAllByAddress(ALL_ACCOUNT_KEY).catch(console.error);
+
+      // Remove NFT
+      stores.nft.deleteNftByAddress([address]).catch(console.error);
+
+      // Remove Staking Data
+      stores.staking.removeAllByAddress(address).catch(console.error);
+    });
+  }
+  async reloadNft() {
+    return await this.cron.reloadNft();
+  }
+  async reloadStaking() {
+    return await this.cron.reloadStaking();
+  }
 }