Files
protocol/packages/sol-resolver/src/resolvers/spy_resolver.ts
2018-12-19 15:41:49 +01:00

26 lines
889 B
TypeScript

import * as _ from 'lodash';
import { ContractSource } from '../types';
import { Resolver } from './resolver';
/**
* This resolver is a passthrough proxy to any resolver that records all the resolved contracts sources.
* You can access them later using the `resolvedContractSources` public field.
*/
export class SpyResolver extends Resolver {
public resolvedContractSources: ContractSource[] = [];
private readonly _resolver: Resolver;
constructor(resolver: Resolver) {
super();
this._resolver = resolver;
}
public resolveIfExists(importPath: string): ContractSource | undefined {
const contractSourceIfExists = this._resolver.resolveIfExists(importPath);
if (!_.isUndefined(contractSourceIfExists)) {
this.resolvedContractSources.push(contractSourceIfExists);
}
return contractSourceIfExists;
}
}