import {
  EdrContext,
  GENERIC_CHAIN_TYPE,
  genericChainProviderFactory,
  L1_CHAIN_TYPE,
  l1ProviderFactory,
  l1SolidityTestRunnerFactory,
  OP_CHAIN_TYPE,
  opProviderFactory,
  opSolidityTestRunnerFactory,
} from "@nomicfoundation/edr";

let _globalEdrContext: EdrContext | undefined;

export async function getGlobalEdrContext(): Promise<EdrContext> {
  if (_globalEdrContext === undefined) {
    _globalEdrContext = new EdrContext();
    await _globalEdrContext.registerProviderFactory(
      GENERIC_CHAIN_TYPE,
      genericChainProviderFactory(),
    );
    await _globalEdrContext.registerProviderFactory(
      L1_CHAIN_TYPE,
      l1ProviderFactory(),
    );
    await _globalEdrContext.registerProviderFactory(
      OP_CHAIN_TYPE,
      opProviderFactory(),
    );
    await _globalEdrContext.registerSolidityTestRunnerFactory(
      L1_CHAIN_TYPE,
      l1SolidityTestRunnerFactory(),
    );
    await _globalEdrContext.registerSolidityTestRunnerFactory(
      OP_CHAIN_TYPE,
      opSolidityTestRunnerFactory(),
    );
  }

  return _globalEdrContext;
}