helper-functions.ts

import { ethers, network, run } from "hardhat"
import { networkConfig } from "./helper-hardhat-config"
import { SignerWithAddress } from "@nomiclabs/hardhat-ethers/signers"
import { BigNumber, constants } from "ethers"
import { LinkToken, LinkToken__factory } from "./typechain"

export const autoFundCheck = async (
    contractAddr: string,
    networkName: string,
    linkTokenAddress: string,
    additionalMessage: string
) => {
    const chainId: number | undefined = network.config.chainId

    console.log("Checking to see if contract can be auto-funded with LINK:")

    if (!chainId) return
    const amount: BigNumber = networkConfig[chainId].fundAmount

    const accounts: SignerWithAddress[] = await ethers.getSigners()
    const signer: SignerWithAddress = accounts[0]

    const linkTokenContract: LinkToken = LinkToken__factory.connect(linkTokenAddress, signer)

    const balance: BigNumber = await linkTokenContract.balanceOf(signer.address)
    const contractBalance: BigNumber = await linkTokenContract.balanceOf(contractAddr)

    if (balance > amount && amount > constants.Zero && contractBalance < amount) {
        //user has enough LINK to auto-fund and the contract isn't already funded
        return true
    } else {
        //user doesn't have enough LINK, print a warning
        console.warn(
            `Account doesn't have enough LINK to fund contracts, or you're deploying to a network where auto funding isnt' done by default\n`,
            `Please obtain LINK via the faucet at https://faucets.chain.link/, then run the following command to fund contract with LINK:\n`,
            `yarn hardhat fund-link --contract ${contractAddr} --network ${networkName} ${additionalMessage}`
        )

        return false
    }
}

export const verify = async (contractAddress: string, args: any[]) => {
    console.log("Verifying contract...")
    try {
        await run("verify:verify", {
            address: contractAddress,
            constructorArguments: args,
        })
    } catch (e: any) {
        if (e.message.toLowerCase().includes("already verified")) {
            console.log("Already verified!")
        } else {
            console.log(e)
        }
    }
}