123456import { 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('http://localhost:9944');20 const web3 = new Web3(provider);2122 try {23 return await cb(web3);24 } finally {25 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 transferBalanceToEth(api: ApiPromise, source: IKeyringPair, target: string, amount: number) {49 const tx = api.tx.balances.transfer(evmToAddress(target), amount);50 const events = await submitTransactionAsync(source, tx);51 const result = getGenericResult(events);52 expect(result.success).to.be.true;53}5455export async function itWeb3(name: string, cb: (apis: { web3: Web3, api: ApiPromise }) => any, opts: { only?: boolean, skip?: boolean } = {}) {56 let i: any = it;57 if (opts.only) i = i.only;58 else if (opts.skip) i = i.skip;59 i(name, async () => {60 await usingApi(async api => {61 await usingWeb3(async web3 => {62 await cb({ api, web3 });63 });64 });65 });66}67itWeb3.only = (name: string, cb: (apis: { web3: Web3, api: ApiPromise }) => any) => itWeb3(name, cb, { only: true });68itWeb3.skip = (name: string, cb: (apis: { web3: Web3, api: ApiPromise }) => any) => itWeb3(name, cb, { skip: true });6970export async function generateSubstrateEthPair(web3: Web3) {71 const account = web3.eth.accounts.create();72 evmToAddress(account.address);73}7475type NormalizedEvent = {76 address: string,77 event: string,78 args: { [key: string]: string }79};8081export function normalizeEvents(events: any): NormalizedEvent[] {82 const output = [];83 for (const key of Object.keys(events)) {84 if (key.match(/^[0-9]+$/)) {85 output.push(events[key]);86 } else if (Array.isArray(events[key])) {87 output.push(...events[key]);88 } else {89 output.push(events[key]);90 }91 }92 output.sort((a, b) => a.logIndex - b.logIndex);93 return output.map(({ address, event, returnValues }) => {94 const args: { [key: string]: string } = {};95 for (const key of Object.keys(returnValues)) {96 if (!key.match(/^[0-9]+$/)) {97 args[key] = returnValues[key];98 }99 }100 return {101 address,102 event,103 args,104 };105 });106}107108export async function recordEvents(contract: any, action: () => Promise<void>): Promise<NormalizedEvent[]> {109 const out: any = [];110 contract.events.allEvents((_: any, event: any) => {111 out.push(event);112 });113 await action();114 return normalizeEvents(out);115}116117export function subToEth(eth: string): string {118 const bytes = addressToEvm(eth);119 const string = '0x' + Buffer.from(bytes).toString('hex');120 return Web3.utils.toChecksumAddress(string);121}