1import {EthUniqueHelper, usingEthPlaygrounds} from '../../eth/util';2import {readFile} from 'fs/promises';3import {4 ICrossAccountId,5} from '../../util/playgrounds/types';6import {IKeyringPair} from '@polkadot/types/types';7import {UniqueNFTCollection} from '../../util/playgrounds/unique';8import {Contract} from 'web3-eth-contract';9import {createObjectCsvWriter} from 'csv-writer';10import {convertToTokens, createCollectionForBenchmarks, PERMISSIONS, PROPERTIES} from '../utils/common';11import {makeNames} from '../../util';12import {ContractImports} from '../../eth/util/playgrounds/types';13import {RMRKNestableMintable} from './ABIGEN';14import {rm} from 'fs';1516const {dirname} = makeNames(import.meta.url);1718export const CONTRACT_IMPORT: ContractImports[] = [19 {20 fsPath: `${dirname}/../../../node_modules/@rmrk-team/evm-contracts/contracts/RMRK/nestable/RMRKNestable.sol`,21 solPath: '@rmrk-team/evm-contracts/contracts/RMRK/nestable/RMRKNestable.sol',22 },23 {24 fsPath: `${dirname}/../../../node_modules/@rmrk-team/evm-contracts/contracts/RMRK/nestable/IERC6059.sol`,25 solPath: '@rmrk-team/evm-contracts/contracts/RMRK/nestable/IERC6059.sol',26 },27 {28 fsPath: `${dirname}/../../../node_modules/@rmrk-team/evm-contracts/contracts/RMRK/core/RMRKCore.sol`,29 solPath: '@rmrk-team/evm-contracts/contracts/RMRK/core/RMRKCore.sol',30 },31 {32 fsPath: `${dirname}/../../../node_modules/@openzeppelin/contracts/token/ERC721/extensions/IERC721Metadata.sol`,33 solPath: '@openzeppelin/contracts/token/ERC721/extensions/IERC721Metadata.sol',34 },35 {36 fsPath: `${dirname}/../../../node_modules/@openzeppelin/contracts/token/ERC721/IERC721.sol`,37 solPath: '@openzeppelin/contracts/token/ERC721/IERC721.sol',38 },39 {40 fsPath: `${dirname}/../../../node_modules/@openzeppelin/contracts/token/ERC721/IERC721Receiver.sol`,41 solPath: '@openzeppelin/contracts/token/ERC721/IERC721Receiver.sol',42 },43 {44 fsPath: `${dirname}/../../../node_modules/@openzeppelin/contracts/utils/Address.sol`,45 solPath: '@openzeppelin/contracts/utils/Address.sol',46 },47 {48 fsPath: `${dirname}/../../../node_modules/@openzeppelin/contracts/utils/Context.sol`,49 solPath: '@openzeppelin/contracts/utils/Context.sol',50 },51 {52 fsPath: `${dirname}/../../../node_modules/@openzeppelin/contracts/utils/introspection/IERC165.sol`,53 solPath: '@openzeppelin/contracts/utils/introspection/IERC165.sol',54 },55 {56 fsPath: `${dirname}/../../../node_modules/@rmrk-team/evm-contracts/contracts/RMRK/library/RMRKErrors.sol`,57 solPath: '@rmrk-team/evm-contracts/contracts/RMRK/library/RMRKErrors.sol',58 },59 {60 fsPath: `${dirname}/../../../node_modules/@rmrk-team/evm-contracts/contracts/RMRK/core/IRMRKCore.sol`,61 solPath: '@rmrk-team/evm-contracts/contracts/RMRK/core/IRMRKCore.sol',62 },63 {64 fsPath: `${dirname}/RMRKNestableMintable.sol`,65 solPath: 'RMRKNestableMintable.sol',66 },67];686970const main = async () => {7172 await usingEthPlaygrounds(async (helper, privateKey) => {7374 const donor = await privateKey('//Alice'); 7576 const eth = await measureEth(helper, donor);77 const sub = await measureSub(helper, donor);78 const rmrk = await measureRMRK(helper, donor);79 console.table({susbtrate: sub, eth: eth, rmrk: rmrk});80 });81};8283async function measureRMRK(helper: EthUniqueHelper, donor: IKeyringPair) {84 const CONTRACT_SOURCE = (85 await readFile(`${dirname}/RMRKNestableMintable.sol`)86 ).toString();87 const RELAYER_SOURCE = (await readFile(`${dirname}/relayer.sol`)).toString();8889 const ethSigner = await helper.eth.createAccountWithBalance(donor);9091 const contract = await helper.ethContract.deployByCode(92 ethSigner,93 'RMRKNestableMintable',94 CONTRACT_SOURCE,95 CONTRACT_IMPORT,96 5000000,97 );9899 const relayer = await helper.ethContract.deployByCode(100 ethSigner,101 'Relayer',102 RELAYER_SOURCE,103 CONTRACT_IMPORT,104 5000000,105 [contract.options.address],106 );107108 const relayerAddress = relayer.options.address;109110 const rmrk = contract as any as RMRKNestableMintable;111 const createTokenFor = async (receiver: string) => {112 const tokenReceipt = await rmrk.methods.safeMint(receiver).send({from: ethSigner});113 return tokenReceipt.events!['Transfer'].returnValues.tokenId as number;114 };115116117 const nestId = await createTokenFor(ethSigner);118 const outerCollectionNestedId = 10;119 const contractOwnedNestId = await createTokenFor(relayerAddress);120 const nextTokenId = contractOwnedNestId + 1;121122 const nestTransfer = await helper.arrange.calculcateFee({Ethereum: ethSigner}, async () => {123 const addChildData = rmrk.methods.addChild(nestId, outerCollectionNestedId, []).encodeABI();124 await relayer.methods.relay(addChildData).send({from: ethSigner});125 await rmrk.methods.acceptChild(nestId, 0, relayerAddress, outerCollectionNestedId).send({from: ethSigner});126 });127128129 const nestMint = await helper.arrange.calculcateFee({Ethereum: ethSigner}, async () => {130 const nestMintData = rmrk.methods.nestMint(rmrk.options.address, nextTokenId, contractOwnedNestId).encodeABI();131 await relayer.methods.relay(nestMintData).send({from: ethSigner});132 const acceptNestedToken = rmrk.methods.acceptChild(contractOwnedNestId, 0, rmrk.options.address, nextTokenId).encodeABI();133 await relayer.methods.relay(acceptNestedToken).send({from: ethSigner});134 });135136137 const unnestToken = await helper.arrange.calculcateFee({Ethereum: ethSigner}, async () => {138 const unnestData = rmrk.methods.transferChild(contractOwnedNestId, ethSigner, 0, 0, rmrk.options.address, nextTokenId, false, []).encodeABI();139 await relayer.methods.relay(unnestData).send({from: ethSigner});140 });141142 return {mint: convertToTokens(nestMint), transfer: convertToTokens(nestTransfer), unnest: convertToTokens(unnestToken)};143}144145async function measureEth(helper: EthUniqueHelper, donor: IKeyringPair) {146 const owner = await helper.eth.createAccountWithBalance(donor);147 const {collectionId, contract} = await createNestingCollection(helper, owner);148149 150 const mintingTargetNFTTokenIdResult = await contract.methods.mint(owner).send({from: owner});151 const targetNFTTokenId = mintingTargetNFTTokenIdResult.events.Transfer.returnValues.tokenId;152 const targetNftTokenAddress = helper.ethAddress.fromTokenId(collectionId, targetNFTTokenId);153154 155 const nestMint = await helper.arrange.calculcateFee({Ethereum: owner}, async () => {156 await contract.methods.mint(targetNftTokenAddress).send({from: owner});157 });158159 160 const mintingSecondTokenIdResult = await contract.methods.mint(owner).send({from: owner});161 const nestedTokenId = mintingSecondTokenIdResult.events.Transfer.returnValues.tokenId;162163 const nestTransfer = await helper.arrange.calculcateFee({Ethereum: owner}, async () => {164 await contract.methods.transfer(targetNftTokenAddress, nestedTokenId).send({from: owner});165 });166167 const unnestToken = await helper.arrange.calculcateFee({Ethereum: owner}, async () => {168 await contract.methods.transferFrom(targetNftTokenAddress, owner, nestedTokenId).send({from: owner});169 });170 return {mint: convertToTokens(nestMint), transfer: convertToTokens(nestTransfer), unnest: convertToTokens(unnestToken)};171}172173async function measureSub(helper: EthUniqueHelper, donor: IKeyringPair) {174175 const [alice, bob] = await helper.arrange.createAccounts([100n, 100n], donor);176177 const targetNFTCollection = await helper.nft.mintCollection(alice);178 const targetTokenBob = await targetNFTCollection.mintToken(alice, {Substrate: bob.address});179180181 const collectionForNesting = await helper.nft.mintCollection(bob);182 183 await targetNFTCollection.setPermissions(alice, {184 nesting: {tokenOwner: true},185 });186187 const nestMint = await helper.arrange.calculcateFee({Substrate: bob.address}, async () => {188 await (collectionForNesting.mintToken(bob, targetTokenBob.nestingAccount()));189 });190191 const nestedToken2 = await collectionForNesting.mintToken(bob);192 const nestTransfer = await helper.arrange.calculcateFee({Substrate: bob.address}, async () => {193 await nestedToken2.nest(bob, targetTokenBob);194 });195196 const unnestToken = await helper.arrange.calculcateFee({Substrate: bob.address}, async () => {197 await nestedToken2.transferFrom(bob, targetTokenBob.nestingAccount(), {Substrate: bob.address});198 });199200 return {mint: convertToTokens(nestMint), transfer: convertToTokens(nestTransfer), unnest: convertToTokens(unnestToken)};201}202203204const createNestingCollection = async (205 helper: EthUniqueHelper,206 owner: string,207): Promise<{ collectionId: number, collectionAddress: string, contract: Contract }> => {208 const {collectionAddress, collectionId} = await helper.eth.createNFTCollection(owner, 'A', 'B', 'C');209210 const contract = helper.ethNativeContract.collection(collectionAddress, 'nft', owner);211 await contract.methods.setCollectionNesting(true).send({from: owner});212213 return {collectionId, collectionAddress, contract};214};215216main()217 .then(() => process.exit(0))218 .catch((e) => {219 console.log(e);220 process.exit(1);221 });222223224225226227