git.delta.rocks / unique-network / refs/commits / 4f3a2095e1df

1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617import {ApiPromise, WsProvider} from '@polkadot/api';18import {ApiOptions, ApiTypes, SubmittableExtrinsic} from '@polkadot/api/types';19import {ExtrinsicStatus} from '@polkadot/types/interfaces/author/types';20import {EventRecord} from '@polkadot/types/interfaces/system/types';21import {IKeyringPair} from '@polkadot/types/types';22import config from '../config';23import '../interfaces/augment-api-events';24import * as defs from '../interfaces/definitions';25import privateKey from './privateKey';26import promisifySubstrate from './promisify-substrate';27282930function defaultApiOptions(): ApiOptions {31  const wsProvider = new WsProvider(config.substrateUrl);32  return {33    provider: wsProvider, signedExtensions: {34      ContractHelpers: {35        extrinsic: {},36        payload: {},37      },38      FakeTransactionFinalizer: {39        extrinsic: {},40        payload: {},41      },42    },43    rpc: {44      unique: defs.unique.rpc,45      rmrk: defs.rmrk.rpc,46      eth: {47        feeHistory: {48          description: 'Dummy',49          params: [],50          type: 'u8',51        },52        maxPriorityFeePerGas: {53          description: 'Dummy',54          params: [],55          type: 'u8',56        },57      },58    },59  };60}6162export async function getApiConnection(settings: ApiOptions | undefined = undefined): Promise<ApiPromise> {63  settings = settings || defaultApiOptions();64  const api = new ApiPromise(settings);6566  if (api) {67    await api.isReadyOrError;68  }6970  return api;71}7273export default async function usingApi<T = void>(action: (api: ApiPromise, privateKeyWrapper: (account: string) => IKeyringPair) => Promise<T>, settings: ApiOptions | undefined = undefined): Promise<T> {74  settings = settings || defaultApiOptions();75  const api: ApiPromise = new ApiPromise(settings);76  let result: T = null as unknown as T;7778  // TODO: Remove, this is temporary: Filter unneeded API output79  // (Jaco promised it will be removed in the next version)80  const consoleErr = console.error;81  const consoleLog = console.log;82  const consoleWarn = console.warn;8384  const outFn = (printer: any) => (...args: any[]) => {85    for (const arg of args) {86      if (typeof arg !== 'string')87        continue;88      if (arg.includes('1000:: Normal connection closure' || arg === 'Normal connection closure'))89        return;90    }91    printer(...args);92  };9394  console.error = outFn(consoleErr.bind(console));95  console.log = outFn(consoleLog.bind(console));96  console.warn = outFn(consoleWarn.bind(console));9798  try {99    await promisifySubstrate(api, async () => {100      if (api) {101        await api.isReadyOrError;102        const ss58Format = (api.registry.getChainProperties())!.toJSON().ss58Format;103        const privateKeyWrapper = (account: string) => privateKey(account, Number(ss58Format));104        result = await action(api, privateKeyWrapper);105      }106    })();107  } finally {108    await api.disconnect();109    console.error = consoleErr;110    console.log = consoleLog;111    console.warn = consoleWarn;112  }113  return result as T;114}115116enum TransactionStatus {117  Success,118  Fail,119  NotReady120}121122function getTransactionStatus(events: EventRecord[], status: ExtrinsicStatus): TransactionStatus {123  if (status.isReady) {124    return TransactionStatus.NotReady;125  }126  if (status.isBroadcast) {127    return TransactionStatus.NotReady;128  }129  if (status.isInBlock || status.isFinalized) {130    if(events.filter(e => e.event.data.method === 'ExtrinsicFailed').length > 0) {131      return TransactionStatus.Fail;132    }133    if(events.filter(e => e.event.data.method === 'ExtrinsicSuccess').length > 0) {134      return TransactionStatus.Success;135    }136  }137138  return TransactionStatus.Fail;139}140141export function executeTransaction(api: ApiPromise, sender: IKeyringPair, transaction: SubmittableExtrinsic<'promise'>): Promise<EventRecord[]> {142  return new Promise(async (res, rej) => {143    try {144      await transaction.signAndSend(sender, ({events, status}) => {145        if (!status.isInBlock && !status.isFinalized) return;146        for (const {event} of events) {147          if (api.events.system.ExtrinsicSuccess.is(event)) {148            res(events);149          } else if (api.events.system.ExtrinsicFailed.is(event)) {150            const {data: [error]} = event;151            if (error.isModule) {152              const decoded = api.registry.findMetaError(error.asModule);153              const {method, section} = decoded;154              rej(new Error(`${section}.${method}`));155            } else {156              rej(new Error(error.toString()));157            }158          }159        }160      });161    } catch (e) {162      rej(e);163    }164  });165}166167export function168submitTransactionAsync(sender: IKeyringPair, transaction: SubmittableExtrinsic<ApiTypes>): Promise<EventRecord[]> {169  /* eslint no-async-promise-executor: "off" */170  return new Promise(async (resolve, reject) => {171    try {172      await transaction.signAndSend(sender, ({events = [], status}) => {173        const transactionStatus = getTransactionStatus(events, status);174175        if (transactionStatus === TransactionStatus.Success) {176          resolve(events);177        } else if (transactionStatus === TransactionStatus.Fail) {178          console.log(`Something went wrong with transaction. Status: ${status}`);179          reject(events);180        }181      });182    } catch (e) {183      console.log('Error: ', e);184      reject(e);185    }186  });187}188189export function submitTransactionExpectFailAsync(sender: IKeyringPair, transaction: SubmittableExtrinsic<ApiTypes>): Promise<EventRecord[]> {190  console.error = () => {};191  console.log = () => {};192193  /* eslint no-async-promise-executor: "off" */194  return new Promise<EventRecord[]>(async function(res, rej) {195    const resolve = (rec: EventRecord[]) => {196      setTimeout(() => {197        res(rec);198      });199    };200    const reject = (errror: any) => {201      setTimeout(() => {202        rej(errror);203      });204    };205    try {206      await transaction.signAndSend(sender, ({events = [], status}) => {207        const transactionStatus = getTransactionStatus(events, status);208209        // console.log('transactionStatus', transactionStatus, 'events', events);210211        if (transactionStatus === TransactionStatus.Success) {212          resolve(events);213        } else if (transactionStatus === TransactionStatus.Fail) {214          reject(events);215        }216      });217    } catch (e) {218      reject(e);219    }220  });221}