ForestAdmin/forest-express-sequelize

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((a) => 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;