import type { HardhatRuntimeEnvironmentHooks } from "../../../../types/hooks.js";
import type { CoverageManagerImplementation as CoverageManagerImplementationT } from "../coverage-manager.js";

import { getCoveragePath, setCoverageManager } from "../helpers/accessors.js";

let CoverageManagerImplementation:
  | typeof CoverageManagerImplementationT
  | undefined;

export default async (): Promise<Partial<HardhatRuntimeEnvironmentHooks>> => ({
  created: async (context, hre) => {
    if (!context.globalOptions.coverage) {
      return;
    }

    if (CoverageManagerImplementation === undefined) {
      ({ CoverageManagerImplementation } = await import(
        "../coverage-manager.js"
      ));
    }

    const coveragePath = getCoveragePath(hre.config.paths.root);
    const coverageManager = new CoverageManagerImplementation(coveragePath);

    setCoverageManager(hre, coverageManager);

    // NOTE: We register this hook dynamically because we use the information about
    // the existence of onCoverageData hook handlers to determine whether coverage
    // is enabled or not.
    hre.hooks.registerHandlers("network", {
      onCoverageData: async (_context, coverageData) => {
        await coverageManager.addData(coverageData);
      },
    });
  },
});