nest-arango
Version:
ArangoDB driver module for NestJS with a built-in CLI tool for creating and running migration scripts
146 lines • 6.12 kB
JavaScript
;
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __param = (this && this.__param) || function (paramIndex, decorator) {
return function (target, key) { decorator(target, key, paramIndex); }
};
var ArangoCoreModule_1;
Object.defineProperty(exports, "__esModule", { value: true });
exports.ArangoCoreModule = void 0;
const common_1 = require("@nestjs/common");
const arangojs_1 = require("arangojs");
const arango_constants_1 = require("./arango.constants");
const arango_util_1 = require("./common/arango.util");
const arango_manager_1 = require("./manager/arango.manager");
let ArangoCoreModule = ArangoCoreModule_1 = class ArangoCoreModule {
connectionName;
moduleRef;
constructor(connectionName, moduleRef) {
this.connectionName = connectionName;
this.moduleRef = moduleRef;
}
static forRoot(options) {
const { config, debug, connectionName, connectionFactory } = options;
const arangoConnectionFactory = connectionFactory || ((config) => new arangojs_1.Database(config));
const arangoConnectionName = (0, arango_util_1.getConnectionToken)(connectionName);
const arangoConnectionNameProvider = {
provide: arango_constants_1.ARANGO_CONNECTION_NAME,
useValue: arangoConnectionName,
};
const connectionProvider = {
provide: arangoConnectionName,
useFactory: async () => {
try {
return arangoConnectionFactory(config);
}
catch (error) {
throw error;
}
},
};
const managerProvider = {
provide: (0, arango_util_1.getManagerToken)(connectionName),
useFactory: async (database) => {
return new arango_manager_1.ArangoManager(database, debug);
},
inject: [arangoConnectionName],
};
return {
module: ArangoCoreModule_1,
providers: [
connectionProvider,
managerProvider,
arangoConnectionNameProvider,
],
exports: [connectionProvider, managerProvider],
};
}
static forRootAsync(options) {
const arangoConnectionName = (0, arango_util_1.getConnectionToken)(options.connectionName);
const arangoManagerName = (0, arango_util_1.getManagerToken)(options.connectionName);
const arangoConnectionNameProvider = {
provide: arango_constants_1.ARANGO_CONNECTION_NAME,
useValue: arangoConnectionName,
};
const connectionProvider = {
provide: arangoConnectionName,
useFactory: async (arangoModuleOptions) => {
const { config, connectionFactory } = arangoModuleOptions;
const arangoConnectionFactory = connectionFactory || ((config) => new arangojs_1.Database(config));
try {
return arangoConnectionFactory(config);
}
catch (error) {
throw error;
}
},
inject: [arango_constants_1.ARANGO_MODULE_OPTIONS],
};
const managerProvider = {
provide: arangoManagerName,
useFactory: async (database, arangoModuleOptions) => {
const { debug } = arangoModuleOptions;
return new arango_manager_1.ArangoManager(database, debug);
},
inject: [arangoConnectionName, arango_constants_1.ARANGO_MODULE_OPTIONS],
};
const asyncProviders = this.createAsyncProviders(options);
return {
module: ArangoCoreModule_1,
imports: options.imports,
providers: [
...asyncProviders,
connectionProvider,
managerProvider,
arangoConnectionNameProvider,
],
exports: [connectionProvider, managerProvider],
};
}
async onApplicationShutdown() {
const connection = this.moduleRef?.get(this.connectionName);
connection && connection.close();
}
static createAsyncProviders(options) {
if (options.useExisting || options.useFactory) {
return [this.createAsyncOptionsProvider(options)];
}
const useClass = options.useClass;
return [
this.createAsyncOptionsProvider(options),
{
provide: useClass,
useClass,
},
];
}
static createAsyncOptionsProvider(options) {
if (options.useFactory) {
return {
provide: arango_constants_1.ARANGO_MODULE_OPTIONS,
useFactory: options.useFactory,
inject: options.inject || [],
};
}
// `as Type<ArangoOptionsFactory>` is a workaround for microsoft/TypeScript#31603
const inject = [
(options.useClass || options.useExisting),
];
return {
provide: arango_constants_1.ARANGO_MODULE_OPTIONS,
useFactory: async (optionsFactory) => await optionsFactory.createArangoOptions(),
inject,
};
}
};
exports.ArangoCoreModule = ArangoCoreModule;
exports.ArangoCoreModule = ArangoCoreModule = ArangoCoreModule_1 = __decorate([
(0, common_1.Global)(),
(0, common_1.Module)({}),
__param(0, (0, common_1.Inject)(arango_constants_1.ARANGO_CONNECTION_NAME))
], ArangoCoreModule);
//# sourceMappingURL=arango-core.module.js.map