Merge pull request #248 from digitalsadhu/fix_and_refactor_updating_of_relationships

Fix and refactor updating of relationships

Author: digitalsadhu

lib/create.js

@@ -1,3 +1,5 @@
+'use strict'
+
 var url = require('url')
 var utils = require('./utils')
 var statusCodes = require('http-status-codes')

lib/delete.js

@@ -1,3 +1,5 @@
+'use strict'
+
 var utils = require('./utils')
 var statusCodes = require('http-status-codes')
 

lib/deserialize.js

@@ -1,3 +1,5 @@
+'use strict'
+
 /* global require,module */
 var deserializer = require('./deserializer')
 var RelUtils = require('./utilities/relationship-utils')

lib/deserializer.js

@@ -1,3 +1,5 @@
+'use strict'
+
 var _ = require('lodash')
 
 function defaultBeforeDeserialize (options, cb) {

lib/errors.js

@@ -1,3 +1,5 @@
+'use strict'
+
 var debug
 var statusCodes = require('http-status-codes')
 

lib/headers.js

@@ -1,4 +1,5 @@
 'use strict'
+
 var is = require('type-is')
 var _ = require('lodash')
 var utils = require('./utils')

lib/patch.js

@@ -1,3 +1,5 @@
+'use strict'
+
 var statusCodes = require('http-status-codes')
 
 module.exports = function (app, options) {

lib/querystring.js

@@ -1,3 +1,5 @@
+'use strict'
+
 var utils = require('./utils')
 
 module.exports = function (app, options) {

lib/relationships.js

@@ -2,7 +2,9 @@
 
 var _ = require('lodash')
 var utils = require('./utils')
-var debug = require('debug')('loopback-component-jsonapi')
+const linkRelatedModels = require(
+  './utilities/relationship-utils'
+).linkRelatedModels
 
 module.exports = function (app, options) {
   // get remote methods.
@@ -24,9 +26,7 @@ module.exports = function (app, options) {
     data = options.data
     model = utils.getModelFromContext(ctx, app)
 
-    relationships(id, data, model)
-
-    next()
+    relationships(model, id, data).then(() => next()).catch(err => next(err))
   })
 
   // for create
@@ -41,144 +41,42 @@ module.exports = function (app, options) {
       id = ctx.result.data.id
       data = options.data
       model = utils.getModelFromContext(ctx, app)
-      relationships(id, data, model)
+      return relationships(model, id, data)
+        .then(() => next())
+        .catch(err => next(err))
     }
 
     next()
   })
 }
 
-function relationships (id, data, model) {
-  if (!data || !data.data || !id || !model || !data.data.relationships) {
-    return
+function extractIdsFromResource (resource) {
+  if (_.isArray(resource)) {
+    return _.map(resource, 'id')
   }
-
-  _.each(data.data.relationships, function (relationship, name) {
-    var serverRelation = model.relations[name]
-    if (!serverRelation) return
-    var type = serverRelation.type
-
-    // don't handle belongsTo in relationships function
-    if (type === 'belongsTo') return
-
-    var modelTo = serverRelation.modelTo
-
-    var fkName = serverRelation.keyTo
-
-    if (type === 'belongsTo') {
-      fkName = serverRelation.keyFrom
-      modelTo = serverRelation.modelFrom
-    }
-
-    if (!modelTo) {
-      return false
-    }
-
-    var setTo = {}
-    setTo[fkName] = null
-    var where = {}
-    where[fkName] = id
-
-    // remove all relations to the model (eg .: post/1)
-    if (type !== 'belongsTo') {
-      modelTo.updateAll(where, setTo, function (err, info) {
-        if (err) console.log(err)
-      })
-    }
-
-    var idToFind = null
-
-    if (_.isArray(relationship.data)) {
-      // find all instance from the relation data eg
-      // [{type: "comments", id: 1}, {type: "comments", id: 2}]
-      _.each(relationship.data, function (item) {
-        idToFind = item.id
-
-        if (type === 'belongsTo') {
-          where[fkName] = item.id
-          idToFind = id
-        }
-
-        updateRelation(modelTo, idToFind, where)
-      })
-
-      if (serverRelation.modelThrough) {
-        var modelThrough = serverRelation.modelThrough
-        var key = keyByModel(modelThrough, modelTo)
-        var data = {}
-        data[fkName] = id
-        var stringIds = false
-
-        var payloadIds = _.map(relationship.data, function (item) {
-          if (typeof item.id === 'string') {
-            stringIds = true
-          }
-          return item.id
-        })
-
-        modelThrough.find({ where: data, fields: key }, function (
-          err,
-          instances
-        ) {
-          if (err) return
-          var serverIds = _.map(instances, function (instance) {
-            return stringIds ? instance[key].toString() : instance[key]
-          })
-          // to delete
-          var toDelete = _.difference(serverIds, payloadIds)
-          _.each(toDelete, function (id) {
-            data[key] = id
-            modelThrough.destroyAll(data)
-          })
-          // new
-          var newAssocs = _.difference(payloadIds, serverIds)
-          _.each(newAssocs, function (id) {
-            data[key] = id
-            modelThrough.create(data)
-          })
-        })
-      }
-    } else {
-      if (relationship.data === null) {
-        where[fkName] = null
-        updateRelation(model, id, where)
-        return
-      }
-
-      idToFind = relationship.data.id
-
-      if (type === 'belongsTo') {
-        idToFind = id
-        where[fkName] = relationship.data.id
-      }
-      // relationship: {data: {type: "comments": id: 1}}
-      updateRelation(modelTo, idToFind, where)
-    }
-  })
+  return _.get(resource, 'id', null)
 }
 
-// if the instance exist, then update it (create relationship),
-// according to JSON API spec we MUST NOT create new ones
-function updateRelation (model, id, data) {
-  model.findById(id, function (err, instance) {
-    if (err) console.log(err)
+function relationships (model, id, payload) {
+  if (!id || !model) return
+  const relationships = _.get(payload, 'data.relationships', {})
 
-    if (instance) {
-      instance.updateAttributes(data)
-    }
-  })
-}
+  return Promise.all(
+    Object.keys(relationships).map(relationName => {
+      const relationship = relationships[relationName]
+      const relationDefn = model.relations[relationName]
+      if (!relationDefn) return
 
-function keyByModel (assocModel, model) {
-  var key = null
-  _.each(assocModel.relations, function (relation) {
-    if (relation.modelTo.modelName === model.modelName) {
-      key = relation.keyFrom
-    }
-  })
+      const type = relationDefn.type
+      const modelTo = relationDefn.modelTo
 
-  if (key === null) {
-    debug('Can not find relation for ' + model.modelName)
-  }
-  return key
+      // don't handle belongsTo in relationships function
+      if (!modelTo || type === 'belongsTo') return
+
+      const data = extractIdsFromResource(relationship.data)
+      const from = { model, id }
+      const to = { model: modelTo, data }
+      return linkRelatedModels(relationName, from, to)
+    })
+  )
 }

lib/removeRemoteMethods.js

@@ -1,3 +1,5 @@
+'use strict'
+
 module.exports = function (app, options) {
   var models = app.models