UNPKG

sequelize-typescript

Version:

Decorators and some other extras for sequelize (v3 + v4)

125 lines 6.24 kB
"use strict"; var __extends = (this && this.__extends) || (function () { var extendStatics = Object.setPrototypeOf || ({ __proto__: [] } instanceof Array && function (d, b) { d.__proto__ = b; }) || function (d, b) { for (var p in b) if (b.hasOwnProperty(p)) d[p] = b[p]; }; return function (d, b) { extendStatics(d, b); function __() { this.constructor = d; } d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __()); }; })(); 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; }; Object.defineProperty(exports, "__esModule", { value: true }); require("reflect-metadata"); var SequelizeOrigin = require("sequelize"); var Model_1 = require("../Model"); var models_1 = require("../../services/models"); var models_2 = require("../../services/models"); var BaseSequelize_1 = require("../BaseSequelize"); var Table_1 = require("../../annotations/Table"); var Sequelize = /** @class */ (function (_super) { __extends(Sequelize, _super); function Sequelize(config) { var _this = this; if (typeof config === "string") { _this = _super.call(this, config, BaseSequelize_1.BaseSequelize.prepareConfig({ url: config })) || this; } else if (BaseSequelize_1.BaseSequelize.isISequelizeUriConfig(config)) { _this = _super.call(this, config.url, BaseSequelize_1.BaseSequelize.prepareConfig(config)) || this; } else { _this = _super.call(this, BaseSequelize_1.BaseSequelize.prepareConfig(config)) || this; } _this.throughMap = {}; _this._ = {}; _this.Model = Function; if (typeof config !== "string") { _this.init(config); } return _this; } Sequelize.prototype.getThroughModel = function (through) { // tslint:disable:max-classes-per-file var Through = /** @class */ (function (_super) { __extends(Through, _super); function Through() { return _super !== null && _super.apply(this, arguments) || this; } Through = __decorate([ Table_1.Table({ tableName: through, modelName: through }) ], Through); return Through; }(Model_1.Model)); return Through; }; /** * The association needs to be adjusted. So that throughModel properties * referencing a original sequelize Model instance */ Sequelize.prototype.adjustAssociation = function (model, association) { var options = association.getSequelizeOptions(); // The associations has to be adjusted var internalAssociation = model['associations'][options.as]; // String based through's need adjustment if (internalAssociation.oneFromSource && internalAssociation.oneFromSource.as === 'Through') { var belongsToManyOptions = options; var tableName = belongsToManyOptions.through.model.getTableName(); // as and associationAccessor values referring to string "Through" internalAssociation.oneFromSource.as = tableName; internalAssociation.oneFromSource.options.as = tableName; internalAssociation.oneFromSource.associationAccessor = tableName; internalAssociation.oneFromTarget.as = tableName; internalAssociation.oneFromTarget.options.as = tableName; internalAssociation.oneFromTarget.associationAccessor = tableName; } if (internalAssociation.throughModel && internalAssociation.throughModel.Model) { var seqThroughModel_1 = internalAssociation.throughModel.Model; var throughModel_1 = internalAssociation.throughModel; Object.keys(seqThroughModel_1).forEach(function (key) { if (key !== 'name') throughModel_1[key] = seqThroughModel_1[key]; }); internalAssociation.throughModel = internalAssociation.through.model = internalAssociation.throughModel.Model; } }; /** * Creates sequelize models and registers these models * in the registry */ Sequelize.prototype.defineModels = function (classes) { var _this = this; classes.forEach(function (_class) { var modelName = models_1.getModelName(_class.prototype); var attributes = models_1.getAttributes(_class.prototype); var options = models_1.getOptions(_class.prototype); if (!options) throw new Error("@Table annotation is missing on class \"" + _class['name'] + "\""); var model = _this.define(modelName, attributes, options); // replace Instance model with the original model model.Instance = _class; model.Instance.prototype.Model = _class; model.Instance.prototype.$Model = _class; // this initializes some stuff for Instance model['refreshAttributes'](); // copy own static fields to class Object.keys(model).forEach(function (key) { return key !== 'name' && (_class[key] = model[key]); }); // the class needs to know its sequelize model _class['Model'] = model; _class.prototype['Model'] = _class.prototype['$Model'] = model; // model needs to know its original class model[models_2.PROPERTY_LINK_TO_ORIG] = _class; // to fix "$1" called with something that's not an instance of Sequelize.Model _class['sequelize'] = _this; }); }; return Sequelize; }(SequelizeOrigin)); exports.Sequelize = Sequelize; //# sourceMappingURL=Sequelize.js.map