git.delta.rocks / unique-network / refs/commits / 51bdbddb0c24

1//2// This file is subject to the terms and conditions defined in3// file 'LICENSE', which is part of this source code package.4//56import { ApiPromise } from '@polkadot/api';7import { addressToEvm, evmToAddress } from '@polkadot/util-crypto';8import Web3 from 'web3';9import usingApi, { submitTransactionAsync } from '../../substrate/substrate-api';10import { IKeyringPair } from '@polkadot/types/types';11import { expect } from 'chai';12import { getGenericResult } from '../../util/helpers';1314let web3Connected = false;15export async function usingWeb3<T>(cb: (web3: Web3) => Promise<T> | T): Promise<T> {16  if (web3Connected) throw new Error('do not nest usingWeb3 calls');17  web3Connected = true;1819  const provider = new Web3.providers.WebsocketProvider(config.substrateUrl);20  const web3 = new Web3(provider);2122  try {23    return await cb(web3);24  } finally {25    // provider.disconnect(3000, 'normal disconnect');26    provider.connection.close();27    web3Connected = false;28  }29}3031export function collectionIdToAddress(address: number): string {32  if (address >= 0xffffffff || address < 0) throw new Error('id overflow');33  const buf = Buffer.from([0x17, 0xc4, 0xe6, 0x45, 0x3c, 0xc4, 0x9a, 0xaa, 0xae, 0xac, 0xa8, 0x94, 0xe6, 0xd9, 0x68, 0x3e,34    address >> 24,35    (address >> 16) & 0xff,36    (address >> 8) & 0xff,37    address & 0xff,38  ]);39  return Web3.utils.toChecksumAddress('0x' + buf.toString('hex'));40}4142export function createEthAccount(web3: Web3) {43  const account = web3.eth.accounts.create();44  web3.eth.accounts.wallet.add(account.privateKey);45  return account.address;46}4748export async function createEthAccountWithBalance(api: ApiPromise, web3: Web3) {49  const alice = privateKey('//Alice');50  const account = createEthAccount(web3);51  await transferBalanceToEth(api, alice, account);5253  return account;54}5556export async function transferBalanceToEth(api: ApiPromise, source: IKeyringPair, target: string, amount = 999999999999999) {57  const tx = api.tx.balances.transfer(evmToAddress(target), amount);58  const events = await submitTransactionAsync(source, tx);59  const result = getGenericResult(events);60  expect(result.success).to.be.true;61}6263export async function itWeb3(name: string, cb: (apis: { web3: Web3, api: ApiPromise }) => any, opts: { only?: boolean, skip?: boolean } = {}) {64  let i: any = it;65  if (opts.only) i = i.only;66  else if (opts.skip) i = i.skip;67  i(name, async () => {68    await usingApi(async api => {69      await usingWeb3(async web3 => {70        await cb({ api, web3 });71      });72    });73  });74}75itWeb3.only = (name: string, cb: (apis: { web3: Web3, api: ApiPromise }) => any) => itWeb3(name, cb, { only: true });76itWeb3.skip = (name: string, cb: (apis: { web3: Web3, api: ApiPromise }) => any) => itWeb3(name, cb, { skip: true });7778export async function generateSubstrateEthPair(web3: Web3) {79  const account = web3.eth.accounts.create();80  evmToAddress(account.address);81}8283type NormalizedEvent = {84    address: string,85    event: string,86    args: { [key: string]: string }87};8889export function normalizeEvents(events: any): NormalizedEvent[] {90  const output = [];91  for (const key of Object.keys(events)) {92    if (key.match(/^[0-9]+$/)) {93      output.push(events[key]);94    } else if (Array.isArray(events[key])) {95      output.push(...events[key]);96    } else {97      output.push(events[key]);98    }99  }100  output.sort((a, b) => a.logIndex - b.logIndex);101  return output.map(({ address, event, returnValues }) => {102    const args: { [key: string]: string } = {};103    for (const key of Object.keys(returnValues)) {104      if (!key.match(/^[0-9]+$/)) {105        args[key] = returnValues[key];106      }107    }108    return {109      address,110      event,111      args,112    };113  });114}115116export async function recordEvents(contract: any, action: () => Promise<void>): Promise<NormalizedEvent[]> {117  const out: any = [];118  contract.events.allEvents((_: any, event: any) => {119    out.push(event);120  });121  await action();122  return normalizeEvents(out);123}124125export function subToEth(eth: string): string {126  const bytes = addressToEvm(eth);127  const string = '0x' + Buffer.from(bytes).toString('hex');128  return Web3.utils.toChecksumAddress(string);129}