@subwallet/extension-base 0.8.4-0 → 1.0.2-1

package/cjs/koni/background/cron.js

@@ -0,0 +1,207 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.KoniCron = void 0;
+var _constants = require("@subwallet/extension-base/constants");
+var _rxjs = require("rxjs");
+var _util = require("@polkadot/util");
+// Copyright 2019-2022 @subwallet/extension-koni authors & contributors
+// SPDX-License-Identifier: Apache-2.0
+
+class KoniCron {
+  status = 'pending';
+  constructor(state, subscriptions, dbService) {
+    this.subscriptions = subscriptions;
+    this.dbService = dbService;
+    this.state = state;
+    this.logger = (0, _util.logger)('Cron');
+    // this.init();
+  }
+
+  cronMap = {};
+  subjectMap = {};
+  getCron = name => {
+    return this.cronMap[name];
+  };
+  getSubjectMap = name => {
+    return this.subjectMap[name];
+  };
+  addCron = (() => {
+    var _this = this;
+    return function (name, callback, interval) {
+      let runFirst = arguments.length > 3 && arguments[3] !== undefined ? arguments[3] : true;
+      if (runFirst) {
+        callback();
+      }
+      _this.cronMap[name] = setInterval(callback, interval);
+    };
+  })();
+  addSubscribeCron = (name, callback, interval) => {
+    const sb = new _rxjs.Subject();
+    callback(sb);
+    this.subjectMap[name] = sb;
+    this.cronMap[name] = setInterval(callback, interval);
+  };
+  removeCron = name => {
+    const interval = this.cronMap[name];
+    if (interval) {
+      clearInterval(interval);
+      delete this.cronMap[name];
+    }
+  };
+  removeAllCrons = () => {
+    Object.entries(this.cronMap).forEach(_ref => {
+      let [key, interval] = _ref;
+      clearInterval(interval);
+      delete this.cronMap[key];
+    });
+  };
+  init = () => {
+    const currentAccountInfo = this.state.keyringService.currentAccount;
+    if (!(currentAccountInfo !== null && currentAccountInfo !== void 0 && currentAccountInfo.address)) {
+      return;
+    }
+    if (Object.keys(this.state.getSubstrateApiMap()).length !== 0 || Object.keys(this.state.getEvmApiMap()).length !== 0) {
+      this.refreshNft(currentAccountInfo.address, this.state.getApiMap(), this.state.getSmartContractNfts(), this.state.getActiveChainInfoMap());
+      this.refreshStakingReward(currentAccountInfo.address);
+      this.refreshStakingRewardFastInterval(currentAccountInfo.address);
+      this.updateChainStakingMetadata(this.state.getChainInfoMap(), this.state.getChainStateMap(), this.state.getSubstrateApiMap());
+      this.updateNominatorMetadata(currentAccountInfo.address, this.state.getChainInfoMap(), this.state.getChainStateMap(), this.state.getSubstrateApiMap());
+    } else {
+      this.setStakingRewardReady();
+    }
+  };
+  start = () => {
+    if (this.status === 'running') {
+      return;
+    }
+    this.logger.log('Starting cron jobs');
+    const currentAccountInfo = this.state.keyringService.currentAccount;
+    if (!(currentAccountInfo !== null && currentAccountInfo !== void 0 && currentAccountInfo.address)) {
+      return;
+    }
+    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()), _constants.CRON_REFRESH_NFT_INTERVAL);
+      this.addCron('recoverApiMap', this.recoverApiMap, _constants.CRON_AUTO_RECOVER_DOTSAMA_INTERVAL, false);
+      this.addCron('refreshStakingReward', this.refreshStakingReward(currentAccountInfo.address), _constants.CRON_REFRESH_STAKING_REWARD_INTERVAL);
+      this.addCron('refreshPoolingStakingReward', this.refreshStakingRewardFastInterval(currentAccountInfo.address), _constants.CRON_REFRESH_STAKING_REWARD_FAST_INTERVAL);
+      this.addCron('updateChainStakingMetadata', this.updateChainStakingMetadata(this.state.getChainInfoMap(), this.state.getChainStateMap(), this.state.getSubstrateApiMap()), _constants.CRON_REFRESH_CHAIN_STAKING_METADATA);
+      this.addCron('updateNominatorMetadata', this.updateNominatorMetadata(currentAccountInfo.address, this.state.getChainInfoMap(), this.state.getChainStateMap(), this.state.getSubstrateApiMap()), _constants.CRON_REFRESH_CHAIN_NOMINATOR_METADATA);
+    } else {
+      this.setStakingRewardReady();
+    }
+    const reloadEvents = ['account.add', 'account.remove', 'account.updateCurrent', 'chain.add', 'chain.update', 'chain.enable', 'asset.update', 'asset.enable', 'transaction.done', 'transaction.failed'];
+    this.state.eventService.onLazy((events, eventTypes) => {
+      var _serviceInfo$currentA;
+      const serviceInfo = this.state.getServiceInfo();
+      const needReload = eventTypes.some(eT => reloadEvents.includes(eT));
+      if (!needReload) {
+        return;
+      }
+      this.logger.log('ServiceInfo updated, restarting...');
+      const address = (_serviceInfo$currentA = serviceInfo.currentAccountInfo) === null || _serviceInfo$currentA === void 0 ? void 0 : _serviceInfo$currentA.address;
+      if (!address) {
+        return;
+      }
+      this.resetStakingReward();
+      this.resetNft(address);
+      this.removeCron('refreshNft');
+      this.removeCron('refreshStakingReward');
+      this.removeCron('refreshPoolingStakingReward');
+      this.removeCron('checkStatusApiMap');
+      this.removeCron('recoverApiMap');
+      this.removeCron('updateChainStakingMetadata');
+      this.removeCron('updateNominatorMetadata');
+      if (this.checkNetworkAvailable(serviceInfo)) {
+        // only add cron job if there's at least 1 active network
+        this.addCron('refreshNft', this.refreshNft(address, serviceInfo.chainApiMap, this.state.getSmartContractNfts(), this.state.getActiveChainInfoMap()), _constants.CRON_REFRESH_NFT_INTERVAL);
+        this.addCron('recoverApiMap', this.recoverApiMap, _constants.CRON_AUTO_RECOVER_DOTSAMA_INTERVAL, false);
+        this.addCron('refreshStakingReward', this.refreshStakingReward(address), _constants.CRON_REFRESH_STAKING_REWARD_INTERVAL);
+        this.addCron('refreshPoolingStakingReward', this.refreshStakingRewardFastInterval(address), _constants.CRON_REFRESH_STAKING_REWARD_FAST_INTERVAL);
+        this.addCron('updateChainStakingMetadata', this.updateChainStakingMetadata(serviceInfo.chainInfoMap, serviceInfo.chainStateMap, serviceInfo.chainApiMap.substrate), _constants.CRON_REFRESH_CHAIN_STAKING_METADATA);
+        this.addCron('updateNominatorMetadata', this.updateNominatorMetadata(address, serviceInfo.chainInfoMap, serviceInfo.chainStateMap, serviceInfo.chainApiMap.substrate), _constants.CRON_REFRESH_CHAIN_NOMINATOR_METADATA);
+      } else {
+        this.setStakingRewardReady();
+      }
+    });
+    this.status = 'running';
+  };
+  stop = () => {
+    if (this.status === 'stopped') {
+      return;
+    }
+    if (this.serviceSubscription) {
+      this.serviceSubscription.unsubscribe();
+      this.serviceSubscription = undefined;
+    }
+    this.logger.log('Stopping cron jobs');
+    this.removeAllCrons();
+    this.status = 'stopped';
+  };
+  recoverApiMap = () => {
+    var _this$subscriptions;
+    const apiMap = this.state.getApiMap();
+    for (const [networkKey, apiProp] of Object.entries(apiMap.substrate)) {
+      if (!apiProp.isApiConnected) {
+        this.state.refreshSubstrateApi(networkKey);
+      }
+    }
+    for (const [key, evmApi] of Object.entries(apiMap.evm)) {
+      evmApi.api.eth.net.isListening().catch(() => {
+        this.state.refreshWeb3Api(key);
+      });
+    }
+    const {
+      address
+    } = this.state.keyringService.currentAccount;
+    ((_this$subscriptions = this.subscriptions) === null || _this$subscriptions === void 0 ? void 0 : _this$subscriptions.subscribeBalancesAndCrowdloans) && this.subscriptions.subscribeBalancesAndCrowdloans(address, this.state.getChainInfoMap(), this.state.getChainStateMap(), this.state.getSubstrateApiMap(), this.state.getEvmApiMap());
+  };
+  refreshNft = (address, apiMap, smartContractNfts, chainInfoMap) => {
+    return () => {
+      this.logger.log('Refresh Nft state');
+      this.subscriptions.subscribeNft(address, apiMap.substrate, apiMap.evm, smartContractNfts, chainInfoMap);
+    };
+  };
+  resetNft = newAddress => {
+    this.logger.log('Reset Nft state');
+    this.state.resetNft(newAddress);
+  };
+  resetStakingReward = () => {
+    this.logger.log('Reset Staking Reward State');
+    this.state.resetStakingReward();
+  };
+  refreshStakingReward = address => {
+    return () => {
+      this.logger.log('Fetching staking reward data');
+      this.subscriptions.subscribeStakingReward(address).then(() => this.logger.log('Refresh staking reward state')).catch(this.logger.error);
+    };
+  };
+  refreshStakingRewardFastInterval = address => {
+    return () => {
+      this.logger.log('Fetching staking reward data with fast interval');
+      this.subscriptions.subscribeStakingRewardFastInterval(address).then(() => this.logger.log('Refresh staking reward state with fast interval')).catch(this.logger.error);
+    };
+  };
+  setStakingRewardReady = () => {
+    this.state.updateStakingRewardReady(true);
+  };
+  checkNetworkAvailable = serviceInfo => {
+    return Object.keys(serviceInfo.chainApiMap.substrate).length > 0 || Object.keys(serviceInfo.chainApiMap.evm).length > 0;
+  };
+  updateChainStakingMetadata = (chainInfoMap, chainStateMap, substrateApiMap) => {
+    return () => {
+      this.logger.log('Fetching chain staking metadata');
+      this.subscriptions.fetchChainStakingMetadata(chainInfoMap, chainStateMap, substrateApiMap).then(() => this.logger.log('Updated chain staking metadata')).catch(this.logger.error);
+    };
+  };
+  updateNominatorMetadata = (address, chainInfoMap, chainStateMap, substrateApiMap) => {
+    return () => {
+      this.logger.log('Fetching nominator data', address);
+      this.subscriptions.fetchNominatorMetadata(address, chainInfoMap, chainStateMap, substrateApiMap).then(() => this.logger.log('Updated nominator data', address)).catch(this.logger.error);
+    };
+  };
+}
+exports.KoniCron = KoniCron;

package/cjs/koni/background/events.js

@@ -0,0 +1,14 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.onExtensionInstall = void 0;
+var _handlers = require("@subwallet/extension-base/koni/background/handlers");
+// Copyright 2019-2022 @subwallet/extension-koni authors & contributors
+// SPDX-License-Identifier: Apache-2.0
+
+const onExtensionInstall = () => {
+  _handlers.state.onInstall();
+};
+exports.onExtensionInstall = onExtensionInstall;