scan modules recursively in flatMap

This commit is contained in:
Viktor 2021-02-22 13:20:13 +03:00
parent ec69dff7ee
commit 2718e899e9

View File

@ -26,12 +26,28 @@ export class BaseExplorerService {
modules: Module[],
callback: (instance: InstanceWrapper, moduleRef: Module) => T | T[],
): T[] {
const invokeMap = () => {
return modules.map((moduleRef) => {
const visitedModules = new Set<Module>();
const unwrap = (moduleRef: Module) => {
// protection from circular recursion
if (visitedModules.has(moduleRef)) {
return [];
} else {
visitedModules.add(moduleRef);
}
const providers = [...moduleRef.providers.values()];
return providers.map((wrapper) => callback(wrapper, moduleRef));
});
const defined = providers.map((wrapper) => callback(wrapper, moduleRef));
const imported: (T | T[])[] = moduleRef.imports?.size
? [...moduleRef.imports.values()].reduce((prev, cur) => {
return [...prev, ...unwrap(cur)];
}, [])
: [];
return [...defined, ...imported];
};
return flattenDeep(invokeMap()).filter(identity);
return flattenDeep(modules.map(unwrap)).filter(identity);
}
}