const { after } = require('../utils/promiseUtils');
const InsertOperation = require('../queryBuilder/operations/InsertOperation');

class RelationInsertOperation extends InsertOperation {
  constructor(name, opt) {
    super(name, opt);

    this.relation = opt.relation;
    this.owner = opt.owner;
    this.assignResultToOwner = true;
  }

  onBefore2(builder, result) {
    const queryContext = builder.context();
    const maybePromise = this.relation.executeBeforeInsert(this.models, queryContext, result);

    return after(maybePromise, result => super.onBefore2(builder, result));
  }
}

module.exports = RelationInsertOperation;