@subwallet/extension-base 1.1.23-0 → 1.1.24-1

package/cjs/services/subscan-service/index.js

@@ -129,11 +129,19 @@ class SubscanService {
   }
   getExtrinsicsList(chain, address) {
     let page = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : 0;
+    let blockRange = arguments.length > 3 ? arguments[3] : undefined;
+    const _blockRange = (() => {
+      if (!blockRange || !blockRange.to) {
+        return null;
+      }
+      return `${blockRange.from || 0}-${blockRange.to}`;
+    })();
     return this.addRequest(async () => {
       const rs = await this.postRequest(this.getApiUrl(chain, 'api/scan/extrinsics'), {
         page,
         row: QUERY_ROW,
-        address
+        address,
+        block_range: _blockRange
       });
       if (rs.status !== 200) {
         throw new _SWError.SWError('SubscanService.getExtrinsicsList', await rs.text());
@@ -143,22 +151,38 @@ class SubscanService {
     });
   }
   async fetchAllPossibleExtrinsicItems(chain, address, cbAfterEachRequest) {
-    let count = 0;
+    let limit = arguments.length > 3 && arguments[3] !== undefined ? arguments[3] : {
+      page: 10,
+      record: 1000
+    };
+    let maxCount = 0;
+    let currentCount = 0;
+    const blockRange = {
+      from: null,
+      to: null
+    };
     const resultMap = {};
     const _getExtrinsicItems = async page => {
-      const res = await this.getExtrinsicsList(chain, address, page);
+      const res = await this.getExtrinsicsList(chain, address, page, blockRange);
       if (!res || !res.count || !res.extrinsics || !res.extrinsics.length) {
         return;
       }
-      if (res.count > count) {
-        count = res.count;
+      if (res.count > maxCount) {
+        maxCount = res.count;
       }
       cbAfterEachRequest === null || cbAfterEachRequest === void 0 ? void 0 : cbAfterEachRequest(res.extrinsics);
       res.extrinsics.forEach(item => {
         resultMap[item.extrinsic_hash] = item;
       });
-      if (Object.values(resultMap).length < count) {
+      currentCount += res.extrinsics.length;
+      if (page > limit.page || currentCount > limit.record) {
+        return;
+      }
+      if (currentCount < maxCount) {
         await (0, _utils.wait)(100);
+        if (page === 0) {
+          blockRange.to = res.extrinsics[0].block_num;
+        }
         await _getExtrinsicItems(++page);
       }
     };
@@ -168,16 +192,16 @@ class SubscanService {
   getTransfersList(chain, address) {
     let page = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : 0;
     let direction = arguments.length > 3 ? arguments[3] : undefined;
-    const requestBody = {
-      page,
-      row: QUERY_ROW,
-      address
-    };
-    if (direction) {
-      requestBody.direction = direction;
-    }
+    let blockRange = arguments.length > 4 ? arguments[4] : undefined;
     return this.addRequest(async () => {
-      const rs = await this.postRequest(this.getApiUrl(chain, 'api/v2/scan/transfers'), requestBody);
+      const rs = await this.postRequest(this.getApiUrl(chain, 'api/v2/scan/transfers'), {
+        page,
+        row: QUERY_ROW,
+        address,
+        direction: direction || null,
+        from_block: (blockRange === null || blockRange === void 0 ? void 0 : blockRange.from) || null,
+        to_block: (blockRange === null || blockRange === void 0 ? void 0 : blockRange.to) || null
+      });
       if (rs.status !== 200) {
         throw new _SWError.SWError('SubscanService.getTransfersList', await rs.text());
       }
@@ -186,27 +210,47 @@ class SubscanService {
     });
   }
   async fetchAllPossibleTransferItems(chain, address, direction, cbAfterEachRequest) {
-    let count = 0;
+    let limit = arguments.length > 4 && arguments[4] !== undefined ? arguments[4] : {
+      page: 10,
+      record: 1000
+    };
+    let maxCount = 0;
+    let currentCount = 0;
+    const blockRange = {
+      from: null,
+      to: null
+    };
     const resultMap = {};
     const _getTransferItems = async page => {
-      const res = await this.getTransfersList(chain, address, page, direction);
+      const res = await this.getTransfersList(chain, address, page, direction, blockRange);
       if (!res || !res.count || !res.transfers || !res.transfers.length) {
         return;
       }
-      if (res.count > count) {
-        count = res.count;
+      if (res.count > maxCount) {
+        maxCount = res.count;
       }
       cbAfterEachRequest === null || cbAfterEachRequest === void 0 ? void 0 : cbAfterEachRequest(res.transfers);
       res.transfers.forEach(item => {
-        resultMap[item.hash] = item;
+        if (!resultMap[item.hash]) {
+          resultMap[item.hash] = [item];
+        } else {
+          resultMap[item.hash].push(item);
+        }
       });
-      if (Object.values(resultMap).length < count) {
+      currentCount += res.transfers.length;
+      if (page > limit.page || currentCount > limit.record) {
+        return;
+      }
+      if (currentCount < maxCount) {
         await (0, _utils.wait)(100);
+        if (page === 0) {
+          blockRange.to = res.transfers[0].block_num;
+        }
         await _getTransferItems(++page);
       }
     };
     await _getTransferItems(0);
-    return Object.values(resultMap);
+    return resultMap;
   }
 }
 exports.SubscanService = SubscanService;

package/cjs/services/transaction-service/index.js

@@ -121,7 +121,8 @@ class TransactionService {
             if (!web3) {
               validationResponse.errors.push(new _TransactionError.TransactionError(_KoniTypes.BasicTxErrorType.CHAIN_DISCONNECTED, undefined));
             } else {
-              const gasPrice = await web3.api.eth.getGasPrice();
+              const _price = await web3.api.eth.getGasPrice();
+              const gasPrice = (0, _eth.recalculateGasPrice)(_price, chainInfo.slug);
               const gasLimit = await web3.api.eth.estimateGas(transaction);
               estimateFee.value = (gasLimit * parseInt(gasPrice)).toString();
             }
@@ -932,8 +933,8 @@ class TransactionService {
           }).forEach(_ref11 => {
             let {
               event: {
-                method,
-                data: [error]
+                data: [error],
+                method
               }
             } = _ref11;
             if (method === 'ExtrinsicFailed') {

package/cjs/types/balance.js

@@ -0,0 +1 @@
+"use strict";

package/cjs/types/index.js

@@ -3,6 +3,17 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
+var _balance = require("./balance");
+Object.keys(_balance).forEach(function (key) {
+  if (key === "default" || key === "__esModule") return;
+  if (key in exports && exports[key] === _balance[key]) return;
+  Object.defineProperty(exports, key, {
+    enumerable: true,
+    get: function () {
+      return _balance[key];
+    }
+  });
+});
 var _buy = require("./buy");
 Object.keys(_buy).forEach(function (key) {
   if (key === "default" || key === "__esModule") return;

package/cjs/utils/{address.js → account.js}

@@ -1,12 +1,14 @@
 "use strict";
 
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.convertSubjectInfoToAddresses = void 0;
+exports.getAccountJsonByAddress = exports.convertSubjectInfoToAddresses = void 0;
 exports.quickFormatAddressToCompare = quickFormatAddressToCompare;
 exports.simpleAddress = void 0;
 var _index = require("@subwallet/extension-base/utils/index");
+var _uiKeyring = _interopRequireDefault(require("@subwallet/ui-keyring"));
 var _utilCrypto = require("@polkadot/util-crypto");
 // Copyright 2019-2022 @subwallet/extension-base authors & contributors
 // SPDX-License-Identifier: Apache-2.0
@@ -31,4 +33,32 @@ const convertSubjectInfoToAddresses = subjectInfo => {
     ...info.json.meta
   }));
 };
-exports.convertSubjectInfoToAddresses = convertSubjectInfoToAddresses;
+
+/**
+ * @function getAccountJsonByAddress
+ * @desc Get account info by address
+ * <p>
+ *   Note: Use on the background only
+ * </p>
+ * @param {string} address - Address
+ * @returns {AccountJson|null}  - Account info or null if not found
+ */
+exports.convertSubjectInfoToAddresses = convertSubjectInfoToAddresses;
+const getAccountJsonByAddress = address => {
+  try {
+    const pair = _uiKeyring.default.getPair(address);
+    if (pair) {
+      return {
+        address: pair.address,
+        type: pair.type,
+        ...pair.meta
+      };
+    } else {
+      return null;
+    }
+  } catch (e) {
+    console.warn(e);
+    return null;
+  }
+};
+exports.getAccountJsonByAddress = getAccountJsonByAddress;

package/cjs/utils/eth.js

@@ -4,7 +4,8 @@ var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefau
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.signatureToHex = exports.createTransactionFromRLP = exports.anyNumberToBN = exports.Transaction = void 0;
+exports.signatureToHex = exports.recalculateGasPrice = exports.createTransactionFromRLP = exports.anyNumberToBN = exports.Transaction = void 0;
+var _constants = require("@subwallet/extension-base/constants");
 var _bignumber = _interopRequireDefault(require("bignumber.js"));
 var _ethers = require("ethers");
 var _util = require("@polkadot/util");
@@ -69,4 +70,8 @@ const signatureToHex = sig => {
   const hexV = (0, _util.hexStripPrefix)((0, _util.numberToHex)(v));
   return hexR + hexS + hexV;
 };
-exports.signatureToHex = signatureToHex;
+exports.signatureToHex = signatureToHex;
+const recalculateGasPrice = (_price, chain) => {
+  return _constants.NETWORK_MULTI_GAS_FEE.includes(chain) ? new _bignumber.default(_price).multipliedBy(_constants.GAS_PRICE_RATIO).toFixed(0) : _price;
+};
+exports.recalculateGasPrice = recalculateGasPrice;

package/cjs/utils/index.js

@@ -78,6 +78,18 @@ var _i18next = require("i18next");
 var _util = require("@polkadot/util");
 var _utilCrypto = require("@polkadot/util-crypto");
 var _canDerive = require("./canDerive");
+var _account = require("./account");
+Object.keys(_account).forEach(function (key) {
+  if (key === "default" || key === "__esModule") return;
+  if (Object.prototype.hasOwnProperty.call(_exportNames, key)) return;
+  if (key in exports && exports[key] === _account[key]) return;
+  Object.defineProperty(exports, key, {
+    enumerable: true,
+    get: function () {
+      return _account[key];
+    }
+  });
+});
 var _array = require("./array");
 Object.keys(_array).forEach(function (key) {
   if (key === "default" || key === "__esModule") return;

package/constants/index.d.ts

@@ -20,5 +20,7 @@ export declare const ALL_GENESIS_HASH: null;
 export declare const IGNORE_GET_SUBSTRATE_FEATURES_LIST: string[];
 export declare const IGNORE_QR_SIGNER: string[];
 export declare const XCM_MIN_AMOUNT_RATIO = 1.2;
+export declare const GAS_PRICE_RATIO = 1.05;
+export declare const NETWORK_MULTI_GAS_FEE: string[];
 export * from './staking';
 export * from './storage';

package/constants/index.js

@@ -23,5 +23,7 @@ export const ALL_GENESIS_HASH = null;
 export const IGNORE_GET_SUBSTRATE_FEATURES_LIST = ['astarEvm', 'ethereum', 'ethereum_goerli', 'binance', 'binance_test', 'boba_rinkeby', 'boba', 'bobabase', 'bobabeam'];
 export const IGNORE_QR_SIGNER = [];
 export const XCM_MIN_AMOUNT_RATIO = 1.2;
+export const GAS_PRICE_RATIO = 1.05;
+export const NETWORK_MULTI_GAS_FEE = ['astarEvm'];
 export * from "./staking.js";
 export * from "./storage.js";

package/koni/api/staking/bonding/index.js

@@ -104,6 +104,7 @@ export async function getCancelWithdrawalExtrinsic(substrateApi, chain, selected
 export function subscribeEssentialChainStakingMetadata(substrateApiMap, chainInfoMap, callback) {
   const unsubList = [];
 
+  // TODO: replace with for of to improve performance
   // eslint-disable-next-line @typescript-eslint/no-misused-promises
   Object.values(chainInfoMap).forEach(async chainInfo => {
     if (!substrateApiMap[chainInfo.slug]) {

package/koni/api/tokens/evm/balance.js

@@ -4,6 +4,10 @@
 export async function getEVMBalance(networkKey, addresses, evmApiMap) {
   const web3Api = evmApiMap[networkKey];
   return await Promise.all(addresses.map(async address => {
-    return await web3Api.api.eth.getBalance(address);
+    try {
+      return await web3Api.api.eth.getBalance(address);
+    } catch (e) {
+      return '0';
+    }
   }));
 }

package/koni/api/tokens/evm/transfer.d.ts

@@ -13,5 +13,5 @@ interface HandleTransferBalanceResultProps {
 export declare const handleTransferBalanceResult: ({ callback, changeValue, networkKey, receipt, response, updateState }: HandleTransferBalanceResultProps) => void;
 export declare function getEVMTransactionObject(chainInfo: _ChainInfo, from: string, to: string, value: string, transferAll: boolean, evmApiMap: Record<string, _EvmApi>): Promise<[TransactionConfig, string]>;
 export declare function getERC20TransactionObject(assetAddress: string, chainInfo: _ChainInfo, from: string, to: string, value: string, transferAll: boolean, evmApiMap: Record<string, _EvmApi>): Promise<[TransactionConfig, string]>;
-export declare function getERC721Transaction(web3Api: _EvmApi, contractAddress: string, senderAddress: string, recipientAddress: string, tokenId: string): Promise<TransactionConfig>;
+export declare function getERC721Transaction(web3Api: _EvmApi, chain: string, contractAddress: string, senderAddress: string, recipientAddress: string, tokenId: string): Promise<TransactionConfig>;
 export {};

package/koni/api/tokens/evm/transfer.js

@@ -5,6 +5,7 @@ import { ExternalRequestPromiseStatus } from '@subwallet/extension-base/backgrou
 import { getERC20Contract } from '@subwallet/extension-base/koni/api/tokens/evm/web3';
 import { _BALANCE_PARSING_CHAIN_GROUP, EVM_REFORMAT_DECIMALS } from '@subwallet/extension-base/services/chain-service/constants';
 import { _ERC721_ABI } from '@subwallet/extension-base/services/chain-service/helper';
+import { recalculateGasPrice } from '@subwallet/extension-base/utils/eth';
 import { BN, hexToBn } from '@polkadot/util';
 export const handleTransferBalanceResult = ({
   callback,
@@ -35,7 +36,8 @@ export const handleTransferBalanceResult = ({
 export async function getEVMTransactionObject(chainInfo, from, to, value, transferAll, evmApiMap) {
   const networkKey = chainInfo.slug;
   const web3Api = evmApiMap[networkKey];
-  const gasPrice = await web3Api.api.eth.getGasPrice();
+  const _price = await web3Api.api.eth.getGasPrice();
+  const gasPrice = recalculateGasPrice(_price, chainInfo.slug);
   const transactionObject = {
     gasPrice: gasPrice,
     to: to,
@@ -69,11 +71,12 @@ export async function getERC20TransactionObject(assetAddress, chainInfo, from, t
     return erc20Contract.methods.transfer(to, transferValue).encodeABI();
   }
   const transferData = generateTransferData(to, transferValue);
-  const [gasLimit, gasPrice] = await Promise.all([
+  const [gasLimit, _price] = await Promise.all([
   // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
   erc20Contract.methods.transfer(to, transferValue).estimateGas({
     from
   }), evmApi.api.eth.getGasPrice()]);
+  const gasPrice = recalculateGasPrice(_price, chainInfo.slug);
   const transactionObject = {
     gasPrice: gasPrice,
     gas: gasLimit,
@@ -87,14 +90,15 @@ export async function getERC20TransactionObject(assetAddress, chainInfo, from, t
   }
   return [transactionObject, transferValue];
 }
-export async function getERC721Transaction(web3Api, contractAddress, senderAddress, recipientAddress, tokenId) {
+export async function getERC721Transaction(web3Api, chain, contractAddress, senderAddress, recipientAddress, tokenId) {
   // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
   const contract = new web3Api.api.eth.Contract(_ERC721_ABI, contractAddress);
-  const [gasLimit, gasPrice] = await Promise.all([
+  const [gasLimit, _price] = await Promise.all([
   // eslint-disable-next-line @typescript-eslint/no-unsafe-call,@typescript-eslint/no-unsafe-member-access
   contract.methods.safeTransferFrom(senderAddress, recipientAddress, tokenId).estimateGas({
     from: senderAddress
   }), web3Api.api.eth.getGasPrice()]);
+  const gasPrice = recalculateGasPrice(_price, chain);
   return {
     from: senderAddress,
     gasPrice,

package/koni/background/cron.js

@@ -1,7 +1,7 @@
 // Copyright 2019-2022 @subwallet/extension-koni authors & contributors
 // SPDX-License-Identifier: Apache-2.0
 
-import { CRON_REFRESH_NFT_INTERVAL, CRON_REFRESH_STAKING_REWARD_FAST_INTERVAL, CRON_REFRESH_STAKING_REWARD_INTERVAL, CRON_SYNC_MANTA_PAY } from '@subwallet/extension-base/constants';
+import { CRON_REFRESH_NFT_INTERVAL, CRON_REFRESH_STAKING_REWARD_FAST_INTERVAL, CRON_SYNC_MANTA_PAY } from '@subwallet/extension-base/constants';
 import { _isChainSupportEvmNft, _isChainSupportNativeNft, _isChainSupportWasmNft } from '@subwallet/extension-base/services/chain-service/utils';
 import { waitTimeout } from '@subwallet/extension-base/utils';
 import { Subject } from 'rxjs';
@@ -104,8 +104,8 @@ export class KoniCron {
     if (Object.keys(this.state.getSubstrateApiMap()).length !== 0 || Object.keys(this.state.getEvmApiMap()).length !== 0) {
       this.resetNft(currentAccountInfo.address);
       this.addCron('refreshNft', this.refreshNft(currentAccountInfo.address, this.state.getApiMap(), this.state.getSmartContractNfts(), this.state.getActiveChainInfoMap()), CRON_REFRESH_NFT_INTERVAL);
-      this.addCron('refreshStakingReward', this.refreshStakingReward(currentAccountInfo.address), CRON_REFRESH_STAKING_REWARD_INTERVAL);
-      this.addCron('refreshPoolingSta             kingReward', this.refreshStakingRewardFastInterval(currentAccountInfo.address), CRON_REFRESH_STAKING_REWARD_FAST_INTERVAL);
+      // this.addCron('refreshStakingReward', this.refreshStakingReward(currentAccountInfo.address), CRON_REFRESH_STAKING_REWARD_INTERVAL);
+      this.addCron('refreshPoolingStakingReward', this.refreshStakingRewardFastInterval(currentAccountInfo.address), CRON_REFRESH_STAKING_REWARD_FAST_INTERVAL);
       this.addCron('syncMantaPay', this.syncMantaPay, CRON_SYNC_MANTA_PAY);
     } else {
       this.setStakingRewardReady();

package/koni/background/handlers/Extension.js

@@ -28,9 +28,8 @@ import { EXTENSION_REQUEST_URL } from '@subwallet/extension-base/services/reques
 import { DEFAULT_AUTO_LOCK_TIME } from '@subwallet/extension-base/services/setting-service/constants';
 import { WALLET_CONNECT_EIP155_NAMESPACE } from '@subwallet/extension-base/services/wallet-connect-service/constants';
 import { isProposalExpired, isSupportWalletConnectChain, isSupportWalletConnectNamespace } from '@subwallet/extension-base/services/wallet-connect-service/helpers';
-import { isSameAddress, reformatAddress, uniqueStringArray } from '@subwallet/extension-base/utils';
-import { convertSubjectInfoToAddresses } from '@subwallet/extension-base/utils/address';
-import { createTransactionFromRLP, signatureToHex } from '@subwallet/extension-base/utils/eth';
+import { convertSubjectInfoToAddresses, isSameAddress, reformatAddress, uniqueStringArray } from '@subwallet/extension-base/utils';
+import { createTransactionFromRLP, recalculateGasPrice, signatureToHex } from '@subwallet/extension-base/utils/eth';
 import { parseContractInput, parseEvmRlp } from '@subwallet/extension-base/utils/eth/parseTransaction';
 import { balanceFormatter, formatNumber } from '@subwallet/extension-base/utils/number';
 import { createPair } from '@subwallet/keyring';
@@ -1725,7 +1724,7 @@ export default class KoniExtension {
     } = inputData;
     const contractAddress = params.contractAddress;
     const tokenId = params.tokenId;
-    const transaction = await getERC721Transaction(this.#koniState.getEvmApi(networkKey), contractAddress, senderAddress, recipientAddress, tokenId);
+    const transaction = await getERC721Transaction(this.#koniState.getEvmApi(networkKey), networkKey, contractAddress, senderAddress, recipientAddress, tokenId);
 
     // this.addContact(recipientAddress);
 
@@ -1872,7 +1871,8 @@ export default class KoniExtension {
               // null address
               from: address
             };
-            const gasPrice = await web3.api.eth.getGasPrice();
+            const _price = await web3.api.eth.getGasPrice();
+            const gasPrice = recalculateGasPrice(_price, networkKey);
             const gasLimit = await web3.api.eth.estimateGas(transaction);
             estimatedFee = (gasLimit * parseInt(gasPrice)).toString();
           } else {
@@ -4062,6 +4062,15 @@ export default class KoniExtension {
       case 'pri(buyService.services.subscribe)':
         return this.subscribeBuyServices(id, port);
       /* Buy service */
+
+      /* Database */
+      case 'pri(database.export)':
+        return this.#koniState.dbService.exportDB();
+      case 'pri(database.import)':
+        return this.#koniState.dbService.importDB(request);
+      case 'pri(database.checkMetadata)':
+        return this.#koniState.dbService.checkImportMetadata(request);
+      /* Database */
       // Default
       default:
         throw new Error(`Unable to handle message of type ${type}`);

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

@@ -1,9 +1,10 @@
 /// <reference types="chrome" />
-import { ActiveCronAndSubscriptionMap, CronServiceType, RequestCronAndSubscriptionAction, RequestInitCronAndSubscription, SubscriptionServiceType } from '@subwallet/extension-base/background/KoniTypes';
+import { ActiveCronAndSubscriptionMap, CronServiceType, MobileData, RequestCronAndSubscriptionAction, RequestInitCronAndSubscription, SubscriptionServiceType } from '@subwallet/extension-base/background/KoniTypes';
 import { MessageTypes, RequestTypes, ResponseType } from '@subwallet/extension-base/background/types';
 import KoniState from '@subwallet/extension-base/koni/background/handlers/State';
 export default class Mobile {
     private state;
+    private restoreHandler;
     constructor(state: KoniState);
     ping(): string;
     initCronAndSubscription({ cron: { activeServices, intervalMap }, subscription: { activeServices } }: RequestInitCronAndSubscription): ActiveCronAndSubscriptionMap;
@@ -17,5 +18,8 @@ export default class Mobile {
     startSubscriptionServices(services: SubscriptionServiceType[]): void;
     stopSubscriptionServices(services: SubscriptionServiceType[]): void;
     restartSubscriptionServices(services: SubscriptionServiceType[]): void;
+    mobileBackup(): Promise<MobileData>;
+    mobileRestore({ indexedDB, storage }: Partial<MobileData>): Promise<void>;
+    waitRestore(): Promise<void>;
     handle<TMessageType extends MessageTypes>(id: string, type: TMessageType, request: RequestTypes[TMessageType], port: chrome.runtime.Port): Promise<ResponseType<TMessageType>>;
 }

package/koni/background/handlers/Mobile.js

@@ -1,6 +1,7 @@
 // Copyright 2019-2022 @subwallet/extension-koni authors & contributors
 // SPDX-License-Identifier: Apache-2.0
 
+import { createPromiseHandler } from '@subwallet/extension-base/utils/promise';
 const DEFAULT_SERVICE_MAP = {
   subscription: {
     chainRegistry: true,
@@ -20,6 +21,7 @@ const DEFAULT_SERVICE_MAP = {
 export default class Mobile {
   // @ts-ignore
 
+  restoreHandler = createPromiseHandler();
   constructor(state) {
     this.state = state;
   }
@@ -92,6 +94,31 @@ export default class Mobile {
   restartSubscriptionServices(services) {
     console.log('restartSubscriptionServices');
   }
+  async mobileBackup() {
+    const indexedDB = await this.state.dbService.exportDB();
+    return {
+      storage: JSON.stringify(localStorage),
+      indexedDB
+    };
+  }
+  async mobileRestore({
+    indexedDB,
+    storage
+  }) {
+    if (storage) {
+      const storageData = JSON.parse(storage);
+      for (const key in storageData) {
+        localStorage.setItem(key, storageData[key]);
+      }
+    }
+    if (indexedDB) {
+      await this.state.dbService.importDB(indexedDB);
+    }
+    this.restoreHandler.resolve();
+  }
+  waitRestore() {
+    return this.restoreHandler.promise;
+  }
 
   // eslint-disable-next-line @typescript-eslint/require-await
   async handle(id, type, request, port) {
@@ -120,6 +147,10 @@ export default class Mobile {
         return this.stopSubscriptionServices(request);
       case 'mobile(subscription.restart)':
         return this.restartSubscriptionServices(request);
+      case 'mobile(storage.restore)':
+        return this.mobileRestore(request);
+      case 'mobile(storage.backup)':
+        return this.mobileBackup();
       default:
         throw new Error(`Unable to handle message of type ${type}`);
     }

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

@@ -1,6 +1,6 @@
 /// <reference types="chrome" />
 import { _AssetRef, _AssetType, _ChainAsset, _ChainInfo, _MultiChainAsset } from '@subwallet/chain-list/types';
-import { AddTokenRequestExternal, AmountData, ApiMap, AuthRequestV2, BalanceItem, BalanceJson, ChainStakingMetadata, ConfirmationsQueue, CrowdloanItem, CrowdloanJson, CurrentAccountInfo, EvmSendTransactionParams, ExternalRequestPromise, MantaPayConfig, MantaPaySyncState, NftCollection, NftItem, NftJson, NominatorMetadata, RequestAccountExportPrivateKey, RequestCheckPublicAndSecretKey, RequestConfirmationComplete, RequestSettingsType, ResponseAccountExportPrivateKey, ResponseCheckPublicAndSecretKey, ServiceInfo, SingleModeJson, StakingItem, StakingJson, StakingRewardItem, StakingRewardJson, StakingType, UiSettings } from '@subwallet/extension-base/background/KoniTypes';
+import { AddTokenRequestExternal, AmountData, ApiMap, AuthRequestV2, ChainStakingMetadata, ConfirmationsQueue, CrowdloanItem, CrowdloanJson, CurrentAccountInfo, EvmSendTransactionParams, ExternalRequestPromise, MantaPayConfig, MantaPaySyncState, NftCollection, NftItem, NftJson, NominatorMetadata, RequestAccountExportPrivateKey, RequestCheckPublicAndSecretKey, RequestConfirmationComplete, RequestSettingsType, ResponseAccountExportPrivateKey, ResponseCheckPublicAndSecretKey, ServiceInfo, SingleModeJson, StakingItem, StakingJson, StakingRewardItem, StakingRewardJson, StakingType, 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 BuyService from '@subwallet/extension-base/services/buy-service';
@@ -20,6 +20,7 @@ import DatabaseService from '@subwallet/extension-base/services/storage-service/
 import { SubscanService } from '@subwallet/extension-base/services/subscan-service';
 import TransactionService from '@subwallet/extension-base/services/transaction-service';
 import WalletConnectService from '@subwallet/extension-base/services/wallet-connect-service';
+import { BalanceItem, BalanceJson, BalanceMap } from '@subwallet/extension-base/types';
 import { MetadataDef, ProviderMeta } from '@subwallet/extension-inject/types';
 import SimpleKeyring from 'eth-simple-keyring';
 import { BehaviorSubject, Subject } from 'rxjs';
@@ -83,7 +84,7 @@ export default class KoniState {
     saveMetadata(meta: MetadataDef): void;
     sign(url: string, request: RequestSign, account: AccountJson): Promise<ResponseSigning>;
     get authSubjectV2(): BehaviorSubject<import("@subwallet/extension-base/background/types").AuthorizeRequest[]>;
-    generateDefaultBalanceMap(): Record<string, BalanceItem>;
+    generateDefaultBalanceMap(): BalanceMap;
     private afterChainServiceInit;
     init(): Promise<void>;
     initMantaPay(password: string): Promise<void>;
@@ -151,12 +152,13 @@ export default class KoniState {
     getAllAddresses(): string[];
     private removeInactiveChainBalances;
     getBalance(reset?: boolean): BalanceJson;
-    getStoredBalance(address: string): Promise<Record<string, BalanceItem>>;
+    getStoredBalance(address: string): Promise<BalanceMap>;
     handleSwitchAccount(newAddress: string): Promise<void>;
     resetBalanceMap(newAddress: string): Promise<void>;
     resetCrowdloanMap(newAddress: string): Promise<void>;
     resetStaking(newAddress: string): void;
-    setBalanceItem(tokenSlug: string, item: BalanceItem): void;
+    /** Note: items must be same tokenSlug */
+    setBalanceItem(items: BalanceItem[]): void;
     private updateBalanceStore;
     subscribeBalance(): Subject<BalanceJson>;
     getCrowdloan(reset?: boolean): CrowdloanJson;