squid-js/test/keeper/TestContractHandler.ts

import Contract from "web3-eth-contract"
import ContractHandler from "../../src/keeper/ContractHandler"
import Web3Provider from "../../src/keeper/Web3Provider"
import Logger from "../../src/utils/Logger"
import config from "../config"

export default class TestContractHandler extends ContractHandler {

    private static networkId: number

    public static async prepareContracts() {
        const web3 = Web3Provider.getWeb3(config)
        const deployerAddress = (await web3.eth.getAccounts())[0]
        this.networkId = await web3.eth.net.getId()

        // deploy contracts
        await TestContractHandler.deployContracts(deployerAddress)
    }

    private static async deployContracts(deployerAddress: string) {
        Logger.log("Trying to deploy contracts")

        // Libraries
        const epochLibrary = await TestContractHandler.deployContract("EpochLibrary", deployerAddress)
        const didRegistryLibrary = await TestContractHandler.deployContract("DIDRegistryLibrary", deployerAddress)

        // Contracts
        const token = await TestContractHandler.deployContract("OceanToken", deployerAddress, [deployerAddress, deployerAddress])

        const dispenser = await TestContractHandler.deployContract("Dispenser", deployerAddress, [token.options.address, deployerAddress])

        // Add dispenser as Token minter
        if (!token.$initialized) {
            await token.methods.addMinter(dispenser.options.address)
                .send({from: deployerAddress})
        }

        const didRegistry = await TestContractHandler.deployContract("DIDRegistry", deployerAddress, [deployerAddress], {
            DIDRegistryLibrary: didRegistryLibrary.options.address,
        })

        // Managers
        const templateStoreManager = await TestContractHandler.deployContract("TemplateStoreManager", deployerAddress, [
            deployerAddress,
        ])
        const conditionStoreManager = await TestContractHandler.deployContract("ConditionStoreManager", deployerAddress, [
            deployerAddress,
        ], {
            EpochLibrary: epochLibrary.options.address,
        })
        const agreementStoreManager = await TestContractHandler.deployContract("AgreementStoreManager", deployerAddress, [
            deployerAddress, conditionStoreManager.options.address, templateStoreManager.options.address, didRegistry.options.address,
        ])

        // Conditions
        const lockRewardCondition = await TestContractHandler.deployContract("LockRewardCondition", deployerAddress, [
            deployerAddress, conditionStoreManager.options.address, token.options.address,
        ])
        const accessSecretStoreCondition = await TestContractHandler.deployContract("AccessSecretStoreCondition", deployerAddress, [
            deployerAddress, conditionStoreManager.options.address, agreementStoreManager.options.address,
        ])

        // Conditions rewards
        const escrowReward = await TestContractHandler.deployContract("EscrowReward", deployerAddress, [
            deployerAddress, conditionStoreManager.options.address, token.options.address,
        ])

        // Templates
        await TestContractHandler.deployContract("EscrowAccessSecretStoreTemplate", deployerAddress, [
            deployerAddress,
            agreementStoreManager.options.address,
            didRegistry.options.address,
            accessSecretStoreCondition.options.address,
            lockRewardCondition.options.address,
            escrowReward.options.address,
        ])
    }

    private static async deployContract(
        name: string,
        from: string,
        args: any[] = [],
        tokens: {[name: string]: string} = {},
    ): Promise<Contract & {$initialized: boolean}> {
        const where = this.networkId

        // dont redeploy if there is already something loaded
        if (TestContractHandler.hasContract(name, where)) {
            const contract = await ContractHandler.getContract(name, where)
            if (contract.testContract) {
                return {...contract, $initialized: true}
            }
        }

        const web3 = Web3Provider.getWeb3(config)

        let contractInstance: Contract
        try {
            Logger.log("Deploying", name)
            const sendConfig = {
                from,
                gas: 3000000,
                gasPrice: 10000000000,
            }
            const artifact = require(`@oceanprotocol/keeper-contracts/artifacts/${name}.development.json`)
            const tempContract = new web3.eth.Contract(artifact.abi, artifact.address)
            const isZos = !!tempContract.methods.initialize

            Logger.debug({
                name, from, isZos, args,
                libraries: artifact.bytecode
                    .replace(/(0x)?[a-f0-9]{8}/gi, "")
                    .replace(/__([^_]+)_*[0-9a-f]{2}/g, "|$1")
                    .split("|")
                    .splice(1),
            })

            contractInstance = await tempContract
                .deploy({
                    data: TestContractHandler.replaceTokens(artifact.bytecode.toString(), tokens),
                    arguments: isZos ? undefined : args,
                })
                .send(sendConfig)
            if (isZos) {
                await contractInstance.methods.initialize(...args).send(sendConfig)
            }
            contractInstance.testContract = true
            ContractHandler.setContract(name, where, contractInstance)
            // Logger.log("Deployed", name, "at", contractInstance.options.address);
        } catch (err) {
            Logger.error("Deployment failed for", name, "with args", JSON.stringify(args, null, 2), err.message)
            throw err
        }

        return contractInstance
    }

    private static replaceTokens(bytecode: string, tokens: {[name: string]: string}): string {
        return Object.entries(tokens)
            .reduce(
                (acc, [token, address]) => acc.replace(new RegExp(`_+${token}_+`, "g"), address.substr(2)),
                bytecode,
            )
    }
}