git.delta.rocks / unique-network / refs/commits / 3833b2da5ea7

1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// SPDX-License-Identifier: Apache-2.034import {stringToU8a} from '@polkadot/util';5import {encodeAddress, mnemonicGenerate} from '@polkadot/util-crypto';6import {UniqueHelper} from './unique';7import {ApiPromise, WsProvider} from '@polkadot/api';8import * as defs from '../../interfaces/definitions';9import {IKeyringPair} from '@polkadot/types/types';10import {ICrossAccountId} from './types';1112export class SilentLogger {13  log(_msg: any, _level: any): void { }14  level = {15    ERROR: 'ERROR' as const,16    WARNING: 'WARNING' as const,17    INFO: 'INFO' as const,18  };19}2021export class SilentConsole {22  // TODO: Remove, this is temporary: Filter unneeded API output23  // (Jaco promised it will be removed in the next version)24  consoleErr: any;25  consoleLog: any;26  consoleWarn: any;2728  constructor() {29    this.consoleErr = console.error;30    this.consoleLog = console.log;31    this.consoleWarn = console.warn;32  }3334  enable() {  35    const outFn = (printer: any) => (...args: any[]) => {36      for (const arg of args) {37        if (typeof arg !== 'string')38          continue;39        if (arg.includes('1000:: Normal connection closure') || arg.includes('Not decorating unknown runtime apis:') || arg.includes('RPC methods not decorated:') || arg === 'Normal connection closure')40          return;41      }42      printer(...args);43    };44  45    console.error = outFn(this.consoleErr.bind(console));46    console.log = outFn(this.consoleLog.bind(console));47    console.warn = outFn(this.consoleWarn.bind(console));48  }4950  disable() {51    console.error = this.consoleErr;52    console.log = this.consoleLog;53    console.warn = this.consoleWarn;54  }55}565758export class DevUniqueHelper extends UniqueHelper {59  /**60   * Arrange methods for tests61   */62  arrange: ArrangeGroup;63  wait: WaitGroup;64  admin: AdminGroup;6566  constructor(logger: { log: (msg: any, level: any) => void, level: any }, options: {[key: string]: any} = {}) {67    options.helperBase = options.helperBase ?? DevUniqueHelper;6869    super(logger, options);70    this.arrange = new ArrangeGroup(this);71    this.wait = new WaitGroup(this);72    this.admin = new AdminGroup(this);73  }7475  async connect(wsEndpoint: string, _listeners?: any): Promise<void> {76    const wsProvider = new WsProvider(wsEndpoint);77    this.api = new ApiPromise({78      provider: wsProvider,79      signedExtensions: {80        ContractHelpers: {81          extrinsic: {},82          payload: {},83        },84        FakeTransactionFinalizer: {85          extrinsic: {},86          payload: {},87        },88      },89      rpc: {90        unique: defs.unique.rpc,91        appPromotion: defs.appPromotion.rpc,92        rmrk: defs.rmrk.rpc,93        eth: {94          feeHistory: {95            description: 'Dummy',96            params: [],97            type: 'u8',98          },99          maxPriorityFeePerGas: {100            description: 'Dummy',101            params: [],102            type: 'u8',103          },104        },105      },106    });107    await this.api.isReadyOrError;108    this.network = await UniqueHelper.detectNetwork(this.api);109  }110}111112class ArrangeGroup {113  helper: DevUniqueHelper;114115  constructor(helper: DevUniqueHelper) {116    this.helper = helper;117  }118119  /**120   * Generates accounts with the specified UNQ token balance 121   * @param balances balances for generated accounts. Each balance will be multiplied by the token nominal.122   * @param donor donor account for balances123   * @returns array of newly created accounts124   * @example const [acc1, acc2, acc3] = await createAccounts([0n, 10n, 20n], donor); 125   */126  createAccounts = async (balances: bigint[], donor: IKeyringPair): Promise<IKeyringPair[]> => {127    let nonce = await this.helper.chain.getNonce(donor.address);128    const wait = new WaitGroup(this.helper);129    const ss58Format = this.helper.chain.getChainProperties().ss58Format;130    const tokenNominal = this.helper.balance.getOneTokenNominal();131    const transactions = [];132    const accounts: IKeyringPair[] = [];133    for (const balance of balances) {134      const recipient = this.helper.util.fromSeed(mnemonicGenerate(), ss58Format);135      accounts.push(recipient);136      if (balance !== 0n) {137        const tx = this.helper.constructApiCall('api.tx.balances.transfer', [{Id: recipient.address}, balance * tokenNominal]);138        transactions.push(this.helper.signTransaction(donor, tx, {nonce}, 'account generation'));139        nonce++;140      }141    }142143    await Promise.all(transactions).catch(_e => {});144    145    //#region TODO remove this region, when nonce problem will be solved146    const checkBalances = async () => {147      let isSuccess = true;148      for (let i = 0; i < balances.length; i++) {149        const balance = await this.helper.balance.getSubstrate(accounts[i].address);150        if (balance !== balances[i] * tokenNominal) {151          isSuccess = false;152          break;153        }154      }155      return isSuccess;156    };157158    let accountsCreated = false;159    // checkBalances retry up to 5 blocks160    for (let index = 0; index < 5; index++) {161      accountsCreated = await checkBalances();162      if(accountsCreated) break;163      await wait.newBlocks(1);164    }165166    if (!accountsCreated) throw Error('Accounts generation failed');167    //#endregion168169    return accounts;170  };171172  // TODO combine this method and createAccounts into one173  createCrowd = async (accountsToCreate: number, withBalance: bigint, donor: IKeyringPair): Promise<IKeyringPair[]> => {  174    const createAsManyAsCan = async () => {175      let transactions: any = [];176      const accounts: IKeyringPair[] = [];177      let nonce = await this.helper.chain.getNonce(donor.address);178      const tokenNominal = this.helper.balance.getOneTokenNominal();179      for (let i = 0; i < accountsToCreate; i++) {180        if (i === 500) { // if there are too many accounts to create181          await Promise.allSettled(transactions); // wait while first 500 (should be 100 for devnode) tx will be settled 182          transactions = []; //183          nonce = await this.helper.chain.getNonce(donor.address); // update nonce 184        }185        const recepient = this.helper.util.fromSeed(mnemonicGenerate());186        accounts.push(recepient);187        if (withBalance !== 0n) {188          const tx = this.helper.constructApiCall('api.tx.balances.transfer', [{Id: recepient.address}, withBalance * tokenNominal]);189          transactions.push(this.helper.signTransaction(donor, tx, {nonce}, 'account generation'));190          nonce++;191        }192      }193      194      const fullfilledAccounts = [];195      await Promise.allSettled(transactions);196      for (const account of accounts) {197        const accountBalance = await this.helper.balance.getSubstrate(account.address);198        if (accountBalance === withBalance * tokenNominal) {199          fullfilledAccounts.push(account);200        }201      }202      return fullfilledAccounts;203    };204205    206    const crowd: IKeyringPair[] = [];207    // do up to 5 retries208    for (let index = 0; index < 5 && accountsToCreate !== 0; index++) {209      const asManyAsCan = await createAsManyAsCan();210      crowd.push(...asManyAsCan);211      accountsToCreate -= asManyAsCan.length;212    }213214    if (accountsToCreate !== 0) throw Error(`Crowd generation failed: ${accountsToCreate} accounts left`);215216    return crowd;217  };218219  isDevNode = async () => {220    const block1 = await this.helper.callRpc('api.rpc.chain.getBlock', [await this.helper.callRpc('api.rpc.chain.getBlockHash', [1])]);221    const block2 = await this.helper.callRpc('api.rpc.chain.getBlock', [await this.helper.callRpc('api.rpc.chain.getBlockHash', [2])]);222    const findCreationDate = async (block: any) => {223      const humanBlock = block.toHuman();224      let date;225      humanBlock.block.extrinsics.forEach((ext: any) => {226        if(ext.method.section === 'timestamp') {227          date = Number(ext.method.args.now.replaceAll(',', ''));228        }229      });230      return date;231    };232    const block1date = await findCreationDate(block1);233    const block2date = await findCreationDate(block2);234    if(block2date! - block1date! < 9000) return true;235  };236  237  async calculcateFee(payer: ICrossAccountId, promise: () => Promise<any>): Promise<bigint> {238    const address = payer.Substrate ? payer.Substrate : await this.helper.address.ethToSubstrate(payer.Ethereum!);239    let balance = await this.helper.balance.getSubstrate(address); 240    241    await promise();242    243    balance -= await this.helper.balance.getSubstrate(address);244    245    return balance;246  }247248  calculatePalleteAddress(palletId: any) {249    const address = stringToU8a(('modl' + palletId).padEnd(32, '\0'));250    return encodeAddress(address);251  }252}253254class WaitGroup {255  helper: DevUniqueHelper;256257  constructor(helper: DevUniqueHelper) {258    this.helper = helper;259  }260261  /**262   * Wait for specified number of blocks263   * @param blocksCount number of blocks to wait264   * @returns 265   */266  async newBlocks(blocksCount = 1): Promise<void> {267    // eslint-disable-next-line no-async-promise-executor268    const promise = new Promise<void>(async (resolve) => {269      const unsubscribe = await this.helper.getApi().rpc.chain.subscribeNewHeads(() => {270        if (blocksCount > 0) {271          blocksCount--;272        } else {273          unsubscribe();274          resolve();275        }276      });277    });278    return promise;279  }280281  async forParachainBlockNumber(blockNumber: bigint) {282    // eslint-disable-next-line no-async-promise-executor283    return new Promise<void>(async (resolve) => {284      const unsubscribe = await this.helper.getApi().rpc.chain.subscribeNewHeads(async (data: any) => {285        if (data.number.toNumber() >= blockNumber) {286          unsubscribe();287          resolve();288        }289      });290    });291  }292  293  async forRelayBlockNumber(blockNumber: bigint) {294    // eslint-disable-next-line no-async-promise-executor295    return new Promise<void>(async (resolve) => {296      const unsubscribe = await this.helper.getApi().query.parachainSystem.validationData(async (data: any) => {297        if (data.value.relayParentNumber.toNumber() >= blockNumber) {298          // @ts-ignore299          unsubscribe();300          resolve();301        }302      });303    });304  }305}306307class AdminGroup {308  helper: UniqueHelper;309310  constructor(helper: UniqueHelper) {311    this.helper = helper;312  }313314  async payoutStakers(signer: IKeyringPair, stakersToPayout: number) {315    const payoutResult = await this.helper.executeExtrinsic(signer, 'api.tx.appPromotion.payoutStakers', [stakersToPayout], true);316    return payoutResult.result.events.filter(e => e.event.method === 'StakingRecalculation').map(e => {317      return {318        staker: e.event.data[0].toString(),319        stake: e.event.data[1].toBigInt(),320        payout: e.event.data[2].toBigInt(),321      };322    });323  }324}