trufflesuite/truffle

import type Resolver from "@truffle/resolver";

export class ResolverIntercept {
  public resolver: Resolver;
  public cache: object[];

  constructor(resolver: Resolver) {
    this.resolver = resolver;
    this.cache = [];
  }

  require(contractName: string) {
    // remove file extension if present on name
    const sanitizedContractName = contractName
      .replace(/^\.\//, "")
      .replace(/\.sol$/i, "");

    // there may be more than one contract of the same name which will be
    // problematic - only return the first one found in the cache for now
    for (const contract of this.cache) {
      // @ts-ignore
      if (contract.contract_name === sanitizedContractName) {
        return contract;
      }
    }

    // Note, will error if nothing is found.
    const resolved = this.resolver.require(sanitizedContractName);

    this.cache.push(resolved);

    return resolved;
  }

  contracts() {
    return this.cache;
  }
}