@nestjs/core
Version:
Nest - modern, fast, powerful node.js web framework (@core)
57 lines (56 loc) • 2.36 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.InstanceLinksHost = void 0;
const shared_utils_1 = require("@nestjs/common/utils/shared.utils");
const unknown_element_exception_1 = require("../errors/exceptions/unknown-element.exception");
class InstanceLinksHost {
constructor(container) {
this.container = container;
this.instanceLinks = new Map();
this.initialize();
}
get(token, options = {}) {
const instanceLinksForGivenToken = this.instanceLinks.get(token);
if (!instanceLinksForGivenToken) {
throw new unknown_element_exception_1.UnknownElementException(this.getInstanceNameByToken(token));
}
if (options.each) {
return instanceLinksForGivenToken;
}
const instanceLink = options.moduleId
? instanceLinksForGivenToken.find(item => item.moduleId === options.moduleId)
: instanceLinksForGivenToken[instanceLinksForGivenToken.length - 1];
if (!instanceLink) {
throw new unknown_element_exception_1.UnknownElementException(this.getInstanceNameByToken(token));
}
return instanceLink;
}
initialize() {
const modules = this.container.getModules();
modules.forEach(moduleRef => {
const { providers, injectables, controllers } = moduleRef;
providers.forEach((wrapper, token) => this.addLink(wrapper, token, moduleRef, 'providers'));
injectables.forEach((wrapper, token) => this.addLink(wrapper, token, moduleRef, 'injectables'));
controllers.forEach((wrapper, token) => this.addLink(wrapper, token, moduleRef, 'controllers'));
});
}
addLink(wrapper, token, moduleRef, collectionName) {
const instanceLink = {
moduleId: moduleRef.id,
wrapperRef: wrapper,
collection: moduleRef[collectionName],
token,
};
const existingLinks = this.instanceLinks.get(token);
if (!existingLinks) {
this.instanceLinks.set(token, [instanceLink]);
}
else {
existingLinks.push(instanceLink);
}
}
getInstanceNameByToken(token) {
return (0, shared_utils_1.isFunction)(token) ? token?.name : token;
}
}
exports.InstanceLinksHost = InstanceLinksHost;