forest-express-sequelize
Version:
Official Express/Sequelize Liana for Forest
56 lines (53 loc) • 2.01 kB
JavaScript
;
require("core-js/modules/es.promise.js");
const _ = require('lodash');
const orm = require('../utils/orm');
const associationRecord = require('../utils/association-record');
class BelongsToUpdater {
constructor(model, assoc, opts, params, data) {
this.model = model;
this.assoc = assoc;
this.opts = opts;
this.params = params;
this.data = data;
}
// WORKAROUND: Make the hasOne associations update work while waiting
// for the Sequelize 4 release with the fix of the following
// issue: https://github.com/sequelize/sequelize/issues/6069
async _getTargetKey(association) {
const pk = this.data.data.id;
const targetKeyIsPrimaryKey = association.targetKey === association.target.primaryKeyAttribute;
let targetKey = pk;
if (association.associationType === 'HasOne' || !targetKeyIsPrimaryKey) {
const record = await associationRecord.get(association.target, pk);
if (association.associationType === 'HasOne') {
targetKey = record;
} else if (!targetKeyIsPrimaryKey) {
// NOTICE: special use case with foreign key non pointing to a primary key
targetKey = record[association.targetKey];
}
}
return targetKey;
}
async perform() {
const {
associationName,
recordId
} = this.params;
const record = await orm.findRecord(this.model, recordId);
const association = Object.values(this.model.associations).find(function (a) {
return a.associationAccessor === associationName;
});
const setterName = `set${_.upperFirst(associationName)}`;
const options = {
fields: null
};
if (association && this.data.data) {
const targetKey = await this._getTargetKey(association);
// NOTICE: Enable model hooks to change fields values during an association update.
return record[setterName](targetKey, options);
}
return record[setterName](null, options);
}
}
module.exports = BelongsToUpdater;