1234567891011121314151617import '../interfaces/augment-api-events';18import {WsProvider, ApiPromise} from '@polkadot/api';19import {EventRecord} from '@polkadot/types/interfaces/system/types';20import {ExtrinsicStatus} from '@polkadot/types/interfaces/author/types';21import {IKeyringPair} from '@polkadot/types/types';2223import config from '../config';24import promisifySubstrate from './promisify-substrate';25import {ApiOptions, SubmittableExtrinsic, ApiTypes} from '@polkadot/api/types';26import * as defs from '../interfaces/definitions';272829function defaultApiOptions(): ApiOptions {30 const wsProvider = new WsProvider(config.substrateUrl);31 return {32 provider: wsProvider, signedExtensions: {33 ContractHelpers: {34 extrinsic: {},35 payload: {},36 },37 FakeTransactionFinalizer: {38 extrinsic: {},39 payload: {},40 },41 },42 rpc: {43 unique: defs.unique.rpc,44 rmrk: defs.rmrk.rpc,45 eth: {46 feeHistory: {47 description: 'Dummy',48 params: [],49 type: 'u8',50 },51 maxPriorityFeePerGas: {52 description: 'Dummy',53 params: [],54 type: 'u8',55 },56 },57 },58 };59}6061export default async function usingApi<T = void>(action: (api: ApiPromise) => Promise<T>, settings: ApiOptions | undefined = undefined): Promise<T> {62 settings = settings || defaultApiOptions();63 const api: ApiPromise = new ApiPromise(settings);64 let result: T = null as unknown as T;6566 67 68 const consoleErr = console.error;69 const consoleLog = console.log;70 const consoleWarn = console.warn;7172 const outFn = (printer: any) => (...args: any[]) => {73 for (const arg of args) {74 if (typeof arg !== 'string')75 continue;76 if (arg.includes('1000:: Normal connection closure' || arg === 'Normal connection closure'))77 return;78 }79 printer(...args);80 };8182 console.error = outFn(consoleErr.bind(console));83 console.log = outFn(consoleLog.bind(console));84 console.warn = outFn(consoleWarn.bind(console));8586 try {87 await promisifySubstrate(api, async () => {88 if (api) {89 await api.isReadyOrError;90 result = await action(api);91 }92 })();93 } finally {94 await api.disconnect();95 console.error = consoleErr;96 console.log = consoleLog;97 console.warn = consoleWarn;98 }99 return result as T;100}101102enum TransactionStatus {103 Success,104 Fail,105 NotReady106}107108function getTransactionStatus(events: EventRecord[], status: ExtrinsicStatus): TransactionStatus {109 if (status.isReady) {110 return TransactionStatus.NotReady;111 }112 if (status.isBroadcast) {113 return TransactionStatus.NotReady;114 }115 if (status.isInBlock || status.isFinalized) {116 if(events.filter(e => e.event.data.method === 'ExtrinsicFailed').length > 0) {117 return TransactionStatus.Fail;118 }119 if(events.filter(e => e.event.data.method === 'ExtrinsicSuccess').length > 0) {120 return TransactionStatus.Success;121 }122 }123124 return TransactionStatus.Fail;125}126127export function executeTransaction(api: ApiPromise, sender: IKeyringPair, transaction: SubmittableExtrinsic<'promise'>): Promise<EventRecord[]> {128 return new Promise(async (res, rej) => {129 try {130 await transaction.signAndSend(sender, ({events, status}) => {131 if (!status.isInBlock && !status.isFinalized) return;132 for (const {event} of events) {133 if (api.events.system.ExtrinsicSuccess.is(event)) {134 res(events);135 } else if (api.events.system.ExtrinsicFailed.is(event)) {136 const {data: [error]} = event;137 if (error.isModule) {138 const decoded = api.registry.findMetaError(error.asModule);139 const {method, section} = decoded;140 rej(new Error(`${section}.${method}`));141 } else {142 rej(new Error(error.toString()));143 }144 }145 }146 });147 } catch (e) {148 rej(e);149 }150 });151}152153export function154submitTransactionAsync(sender: IKeyringPair, transaction: SubmittableExtrinsic<ApiTypes>): Promise<EventRecord[]> {155 156 return new Promise(async (resolve, reject) => {157 try {158 await transaction.signAndSend(sender, ({events = [], status}) => {159 const transactionStatus = getTransactionStatus(events, status);160161 if (transactionStatus === TransactionStatus.Success) {162 resolve(events);163 } else if (transactionStatus === TransactionStatus.Fail) {164 console.log(`Something went wrong with transaction. Status: ${status}`);165 reject(events);166 }167 });168 } catch (e) {169 console.log('Error: ', e);170 reject(e);171 }172 });173}174175export function submitTransactionExpectFailAsync(sender: IKeyringPair, transaction: SubmittableExtrinsic<ApiTypes>): Promise<EventRecord[]> {176 console.error = () => {};177 console.log = () => {};178179 180 return new Promise<EventRecord[]>(async function(res, rej) {181 const resolve = (rec: EventRecord[]) => {182 setTimeout(() => {183 res(rec);184 });185 };186 const reject = (errror: any) => {187 setTimeout(() => {188 rej(errror);189 });190 };191 try {192 await transaction.signAndSend(sender, ({events = [], status}) => {193 const transactionStatus = getTransactionStatus(events, status);194195 196197 if (transactionStatus === TransactionStatus.Success) {198 resolve(events);199 } else if (transactionStatus === TransactionStatus.Fail) {200 reject(events);201 }202 });203 } catch (e) {204 reject(e);205 }206 });207}