vitamin/src/relations/belongs-to.js

Data Mapper library for Node.js applications

import Relation from './base'
import mixin from './mixins/one-to-one'

// exports
export default class extends mixin(Relation) {
  
  /**
   * BelongsToRelation constructor
   * 
   * @param {Model} parent model instance
   * @param {Model} target model instance
   * @param {String} fk parent model foreign key
   * @param {String} pk target model primary key
   * @constructor
   */
  constructor(parent, target, fk, pk) {
    super(parent, target)
    
    this.localKey = fk
    this.otherKey = pk
  }
  
  /**
   * Associate the model instance to the current model
   * 
   * @param {Model} model
   * @return parent model
   */
  associate(model) {
    return this.model.set(this.localKey, model.get(this.otherKey))
  }
  
  /**
   * Dissociate the parent model from the current model
   * 
   * @return parent model
   */
  dissociate() {
    return this.model.set(this.localKey, null)
  }
  
}