const shell = require('shelljs'); const globby = require('globby'); const pluginUtils = require("./plugin.utils"); const path = require('path'); const DataCollector = require("./../../lib/collector") const util = require('util') // ============================= // Nomiclabs Plugin Utils // ============================= /** * Returns a list of test files to pass to mocha. * @param {String} files file or glob * @return {String[]} list of files to pass to mocha */ function getTestFilePaths(files){ const target = globby.sync([files]) // Buidler/Hardhat supports js & ts const testregex = /.*\.(js|ts)$/; return target.filter(f => f.match(testregex) != null); } /** * Normalizes Buidler/Hardhat paths / logging for use by the plugin utilities and * attaches them to the config * @param {Buidler/HardhatConfig} config * @return {Buidler/HardhatConfig} updated config */ function normalizeConfig(config, args={}){ config.workingDir = config.paths.root; config.contractsDir = config.paths.sources; config.testDir = config.paths.tests; config.artifactsDir = config.paths.artifacts; config.logger = config.logger ? config.logger : {log: null}; config.solcoverjs = args.solcoverjs config.gasReporter = { enabled: false } return config; } function setupBuidlerNetwork(env, api, ui){ const { createProvider } = require("@nomiclabs/buidler/internal/core/providers/construction"); let networkConfig = {}; let networkName = (env.buidlerArguments.network !== 'buidlerevm') ? env.buidlerArguments.network : api.defaultNetworkName; if (networkName !== api.defaultNetworkName){ networkConfig = env.config.networks[networkName]; configureHttpProvider(networkConfig, api, ui) } else { networkConfig.url = `http://${api.host}:${api.port}` } const provider = createProvider(networkName, networkConfig); return configureNetworkEnv( env, networkName, networkConfig, provider ) } function setupHardhatNetwork(env, api, ui){ const { createProvider } = require("hardhat/internal/core/providers/construction"); const { HARDHAT_NETWORK_NAME } = require("hardhat/plugins") let provider, networkName, networkConfig; let isHardhatEVM = false; networkName = env.hardhatArguments.network || HARDHAT_NETWORK_NAME; // HardhatEVM if (networkName === HARDHAT_NETWORK_NAME){ isHardhatEVM = true; api.collector = new DataCollector(api.instrumenter.instrumentationData); networkConfig = env.network.config; configureHardhatEVMGas(networkConfig, api); provider = createProvider( networkName, networkConfig, env.config.paths, env.artifacts, [api.hardhatTraceHandler.bind(api)] ) // HttpProvider } else { if (!(env.config.networks && env.config.networks[networkName])){ throw new Error(ui.generate('network-fail', [networkName])) } networkConfig = env.config.networks[networkName] configureNetworkGas(networkConfig, api); configureHttpProvider(networkConfig, api, ui) provider = createProvider(networkName, networkConfig); } return configureNetworkEnv( env, networkName, networkConfig, provider, isHardhatEVM ) } function configureNetworkGas(networkConfig, api){ networkConfig.gas = api.gasLimit; networkConfig.gasPrice = api.gasPrice; } function configureHardhatEVMGas(networkConfig, api){ networkConfig.allowUnlimitedContractSize = true; networkConfig.blockGasLimit = api.gasLimit networkConfig.gas = api.gasLimit; networkConfig.gasPrice = api.gasPrice; } function configureNetworkEnv(env, networkName, networkConfig, provider, isHardhatEVM){ env.config.networks[networkName] = networkConfig; env.config.defaultNetwork = networkName; env.network = { name: networkName, config: networkConfig, provider: provider, isHardhatEVM: isHardhatEVM } env.ethereum = provider; // Return a reference so we can set the from account return env.network; } /** * Extracts port from url / sets network.url * @param {Object} networkConfig * @param {SolidityCoverage} api */ function configureHttpProvider(networkConfig, api, ui){ const configPort = networkConfig.url.split(':')[2]; // Warn: port conflicts if (api.port !== api.defaultPort && api.port !== configPort){ ui.report('port-clash', [ configPort ]) } // Prefer network port api.port = parseInt(configPort); networkConfig.url = `http://${api.host}:${api.port}`; } async function getAccounts(provider){ return provider.send("eth_accounts", []) } async function getNodeInfo(provider){ return provider.send("web3_clientVersion", []) } /** * Sets the default `from` account field in the network that will be used. * This needs to be done after accounts are fetched from the launched client. * @param {env} config * @param {Array} accounts */ function setNetworkFrom(networkConfig, accounts){ if (!networkConfig.from){ networkConfig.from = accounts[0]; } } // TODO: Hardhat cacheing?? /** * Generates a path to a temporary compilation cache directory * @param {BuidlerConfig} config * @return {String} .../.coverage_cache */ function tempCacheDir(config){ return path.join(config.paths.root, '.coverage_cache'); } /** * Silently removes temporary folders and calls api.finish to shut server down * @param {Buidler/HardhatConfig} config * @param {SolidityCoverage} api * @return {Promise} */ async function finish(config, api){ const { tempContractsDir, tempArtifactsDir } = pluginUtils.getTempLocations(config); shell.config.silent = true; shell.rm('-Rf', tempContractsDir); shell.rm('-Rf', tempArtifactsDir); shell.rm('-Rf', path.join(config.paths.root, '.coverage_cache')); shell.config.silent = false; if (api) await api.finish(); } module.exports = { normalizeConfig: normalizeConfig, finish: finish, tempCacheDir: tempCacheDir, setupBuidlerNetwork: setupBuidlerNetwork, setupHardhatNetwork: setupHardhatNetwork, getTestFilePaths: getTestFilePaths, getAccounts: getAccounts, getNodeInfo: getNodeInfo, setNetworkFrom: setNetworkFrom }