squid-js/src/keeper/ContractHandler.ts

import Contract from "web3-eth-contract"
import Logger from "../utils/Logger"
import Keeper from "./Keeper"
import Web3Provider from "./Web3Provider"

const contracts: Map<string, Contract> = new Map<string, Contract>()

export default class ContractHandler {

    public static async get(what: string): Contract {
        const where = (await (await Keeper.getInstance()).getNetworkName()).toLowerCase()
        try {
            return contracts.get(what) || await ContractHandler.load(what, where)
        } catch (err) {
            Logger.error("Failed to load", what, "from", where, err)
            throw err
        }
    }

    public static async deployContracts() {
        Logger.log("Trying to deploy contracts")

        const web3 = Web3Provider.getWeb3()

        const deployerAddress = (await web3.eth.getAccounts())[0]

        // deploy libs
        /* not part of trilobite
        const dll = await ContractHandler.deployContract("DLL", deployerAddress)
        const attributeStore = await ContractHandler.deployContract("AttributeStore", deployerAddress)
        */
        // deploy contracts
        const token = await ContractHandler.deployContract("OceanToken", deployerAddress)
        /* not part of trilobite
        const plcrVoting = await ContractHandler.deployContract("PLCRVoting", deployerAddress, {
            args: [token.options.address],
            tokens: [
                {
                    name: "DLL", address: dll.options.address,
                }, {
                    name: "AttributeStore", address: attributeStore.options.address,
                },
            ],
        })
        /* not part of trilobite
        const registry = await ContractHandler.deployContract("OceanRegistry", deployerAddress, {
            args: [token.options.address, plcrVoting.options.address],
        })
        */
        const market = await ContractHandler.deployContract("OceanMarket", deployerAddress, {
            args: [token.options.address],
        })

        const sa = await ContractHandler.deployContract("ServiceAgreement", deployerAddress, {
            args: [],

        })
        await ContractHandler.deployContract("AccessConditions", deployerAddress, {
            args: [sa.options.address],
        })

        await ContractHandler.deployContract("PaymentConditions", deployerAddress, {
            args: [sa.options.address, token.options.address],
        })

        await ContractHandler.deployContract("DIDRegistry", deployerAddress, {})
        /* not part of trilobite
        const dispute = await ContractHandler.deployContract("OceanDispute", deployerAddress, {
            args: [market.options.address, registry.options.address, plcrVoting.options.address],
        })
        */
        await ContractHandler.deployContract("OceanAuth", deployerAddress, {
            args: [market.options.address],
        })
    }

    private static async load(what: string, where: string): Promise<Contract> {
        const web3 = Web3Provider.getWeb3()
        // Logger.log("Loading", what, "from", where)
        const artifact = require(`@oceanprotocol/keeper-contracts/artifacts/${what}.${where}`)
        // Logger.log('Loaded artifact', artifact)
        const code = await web3.eth.getCode(artifact.address)
        if (code === "0x0") {
            // no code in the blockchain dude
            throw new Error(`No code deployed at address ${artifact.address}, sorry.`)
        }
        // Logger.log("Getting instance of", what, "from", where, "at", artifact.address)
        const contract = new web3.eth.Contract(artifact.abi, artifact.address)
        Logger.log("Loaded", what, "from", where)
        contracts.set(what, contract)
        return contracts.get(what)
    }

    // todo: reactivate for tethys
    private static replaceTokens(bytecode: string, tokens: any[]) {

        for (const token of tokens) {

            bytecode = bytecode.replace(
                new RegExp(`_+${token.name}_+`, "g"),
                token.address.replace("0x", ""))
        }
        // Logger.log(bytecode)

        return bytecode.toString()
    }

    private static async deployContract(name: string, from: string, params?): Promise<Contract> {

        // dont redeploy if there is already something loaded
        if (contracts.has(name)) {
            return contracts.get(name)
        }

        const web3 = Web3Provider.getWeb3()

        let contractInstance: Contract
        try {
            Logger.log("Deploying", name)

            const artifact = require(`@oceanprotocol/keeper-contracts/artifacts/${name}.development.json`)
            const tempContract = new web3.eth.Contract(artifact.abi, artifact.address)
            contractInstance = await tempContract.deploy({
                data: params && params.tokens ?
                    ContractHandler.replaceTokens(artifact.bytecode.toString(), params.tokens) :
                    artifact.bytecode,
                arguments: params && params.args ? params.args : null,
            }).send({
                from,
                gas: 3000000,
                gasPrice: 10000000000,
            })
            contracts.set(name, contractInstance)
            // Logger.log("Deployed", name, "at", contractInstance.options.address);
        } catch (err) {
            Logger.error("Deployment failed for", name, "with params", JSON.stringify(params, null, 2), err.message)
            throw err
        }

        return contractInstance
    }
}