git.delta.rocks / unique-network / refs/commits / 7c2f408f0bda

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    const maxBlocksChecked = await this.helper.arrange.isDevNode() ? 50 : 5;160    // checkBalances retry up to 5-50 blocks161    for (let index = 0; index < maxBlocksChecked; index++) {162      accountsCreated = await checkBalances();163      if(accountsCreated) break;164      await wait.newBlocks(1);165    }166167    if (!accountsCreated) throw Error('Accounts generation failed');168    //#endregion169170    return accounts;171  };172173  // TODO combine this method and createAccounts into one174  createCrowd = async (accountsToCreate: number, withBalance: bigint, donor: IKeyringPair): Promise<IKeyringPair[]> => {  175    const createAsManyAsCan = async () => {176      let transactions: any = [];177      const accounts: IKeyringPair[] = [];178      let nonce = await this.helper.chain.getNonce(donor.address);179      const tokenNominal = this.helper.balance.getOneTokenNominal();180      for (let i = 0; i < accountsToCreate; i++) {181        if (i === 500) { // if there are too many accounts to create182          await Promise.allSettled(transactions); // wait while first 500 (should be 100 for devnode) tx will be settled 183          transactions = []; //184          nonce = await this.helper.chain.getNonce(donor.address); // update nonce 185        }186        const recepient = this.helper.util.fromSeed(mnemonicGenerate());187        accounts.push(recepient);188        if (withBalance !== 0n) {189          const tx = this.helper.constructApiCall('api.tx.balances.transfer', [{Id: recepient.address}, withBalance * tokenNominal]);190          transactions.push(this.helper.signTransaction(donor, tx, {nonce}, 'account generation'));191          nonce++;192        }193      }194      195      const fullfilledAccounts = [];196      await Promise.allSettled(transactions);197      for (const account of accounts) {198        const accountBalance = await this.helper.balance.getSubstrate(account.address);199        if (accountBalance === withBalance * tokenNominal) {200          fullfilledAccounts.push(account);201        }202      }203      return fullfilledAccounts;204    };205206    207    const crowd: IKeyringPair[] = [];208    // do up to 5 retries209    for (let index = 0; index < 5 && accountsToCreate !== 0; index++) {210      const asManyAsCan = await createAsManyAsCan();211      crowd.push(...asManyAsCan);212      accountsToCreate -= asManyAsCan.length;213    }214215    if (accountsToCreate !== 0) throw Error(`Crowd generation failed: ${accountsToCreate} accounts left`);216217    return crowd;218  };219220  isDevNode = async () => {221    const block1 = await this.helper.callRpc('api.rpc.chain.getBlock', [await this.helper.callRpc('api.rpc.chain.getBlockHash', [1])]);222    const block2 = await this.helper.callRpc('api.rpc.chain.getBlock', [await this.helper.callRpc('api.rpc.chain.getBlockHash', [2])]);223    const findCreationDate = async (block: any) => {224      const humanBlock = block.toHuman();225      let date;226      humanBlock.block.extrinsics.forEach((ext: any) => {227        if(ext.method.section === 'timestamp') {228          date = Number(ext.method.args.now.replaceAll(',', ''));229        }230      });231      return date;232    };233    const block1date = await findCreationDate(block1);234    const block2date = await findCreationDate(block2);235    if(block2date! - block1date! < 9000) return true;236  };237  238  async calculcateFee(payer: ICrossAccountId, promise: () => Promise<any>): Promise<bigint> {239    const address = payer.Substrate ? payer.Substrate : await this.helper.address.ethToSubstrate(payer.Ethereum!);240    let balance = await this.helper.balance.getSubstrate(address); 241    242    await promise();243    244    balance -= await this.helper.balance.getSubstrate(address);245    246    return balance;247  }248249  calculatePalletAddress(palletId: any) {250    const address = stringToU8a(('modl' + palletId).padEnd(32, '\0'));251    return encodeAddress(address);252  }253}254255class WaitGroup {256  helper: DevUniqueHelper;257258  constructor(helper: DevUniqueHelper) {259    this.helper = helper;260  }261262  /**263   * Wait for specified number of blocks264   * @param blocksCount number of blocks to wait265   * @returns 266   */267  async newBlocks(blocksCount = 1): Promise<void> {268    // eslint-disable-next-line no-async-promise-executor269    const promise = new Promise<void>(async (resolve) => {270      const unsubscribe = await this.helper.getApi().rpc.chain.subscribeNewHeads(() => {271        if (blocksCount > 0) {272          blocksCount--;273        } else {274          unsubscribe();275          resolve();276        }277      });278    });279    return promise;280  }281282  async forParachainBlockNumber(blockNumber: bigint) {283    // eslint-disable-next-line no-async-promise-executor284    return new Promise<void>(async (resolve) => {285      const unsubscribe = await this.helper.getApi().rpc.chain.subscribeNewHeads(async (data: any) => {286        if (data.number.toNumber() >= blockNumber) {287          unsubscribe();288          resolve();289        }290      });291    });292  }293  294  async forRelayBlockNumber(blockNumber: bigint) {295    // eslint-disable-next-line no-async-promise-executor296    return new Promise<void>(async (resolve) => {297      const unsubscribe = await this.helper.getApi().query.parachainSystem.validationData(async (data: any) => {298        if (data.value.relayParentNumber.toNumber() >= blockNumber) {299          // @ts-ignore300          unsubscribe();301          resolve();302        }303      });304    });305  }306}307308class AdminGroup {309  helper: UniqueHelper;310311  constructor(helper: UniqueHelper) {312    this.helper = helper;313  }314315  async payoutStakers(signer: IKeyringPair, stakersToPayout: number) {316    const payoutResult = await this.helper.executeExtrinsic(signer, 'api.tx.appPromotion.payoutStakers', [stakersToPayout], true);317    return payoutResult.result.events.filter(e => e.event.method === 'StakingRecalculation').map(e => {318      return {319        staker: e.event.data[0].toString(),320        stake: e.event.data[1].toBigInt(),321        payout: e.event.data[2].toBigInt(),322      };323    });324  }325}