nodester 0.7.14 → 0.7.16

package/lib/body/extract.js

@@ -22,9 +22,7 @@ module.exports = extract;
  * @access public
  * @return {Object} filteredBody
  */
-function extract(body, filter=null, model) {
-
-	const sequelize = model.sequelize;
+function extract(body, filter = null, model) {
 	const modelAttributes = Object.keys(model.tableAttributes);
 	const availableIncludes = Object.keys(model.associations);
 
@@ -87,6 +85,20 @@ function extract(body, filter=null, model) {
 			const association = model.associations[key];
 
 			// Mutliple includes:
+			if (association.associationType === 'BelongsToMany') {
+				const targetPkField = association.target.primaryKeyField;
+
+				if (Array.isArray(value)) {
+					filteredBody[key] = value.map(item => ({ [targetPkField]: item[targetPkField] }));
+				}
+				else {
+					filteredBody[key] = { [targetPkField]: value[targetPkField] };
+				}
+
+				continue;
+			}
+
+			// HasMany / HasOne — recursive extraction:
 			if (Array.isArray(value)) {
 				filteredBody[key] = [];
 
@@ -111,7 +123,7 @@ function extract(body, filter=null, model) {
 
 	// Set all static attributes:
 	const staticAttributeEntries = Object.entries(staticAttributes);
-	for (const [ key, value ] of staticAttributeEntries) {
+	for (const [key, value] of staticAttributeEntries) {
 		filteredBody[key] = staticAttributes[key];
 	}
 

package/lib/models/define.js

@@ -2,7 +2,7 @@
  * nodester
  * MIT Licensed
  */
- 
+
 'use strict';
 
 // CRUD mixins.
@@ -97,32 +97,38 @@ function defineModel(
  *
  * @alias getIncludesTree
  */
-function _getIncludesTree(data=null) {
+function _getIncludesTree(data = null) {
 	const result = [];
 
+	if (!data || typeof data !== 'object') {
+		return result;
+	}
+
 	const associations = this.associations;
 	const associationEntries = Object.entries(associations);
-
-	for (const [ associationName, associationDefinition ] of associationEntries) {
-		const formatted = { association: associationName };
-
-		if (!!data && typeof data === 'object') {
-			// If data (for example during create)
-			// is set, go deeper:
-			const keys = Object.keys( data );
-			if (keys.indexOf(associationName) > 0) {
-				const associationModel = associationDefinition.target;
-
-				if (Object.entries(associationModel.associations).length > 0) {
-					const deepData = data[ associationName ];
-					formatted.include = associationModel.getIncludesTree(
-												Array.isArray(deepData) ? deepData[0] : deepData
-											);
+	const keys = Object.keys(data);
+
+	for (const [associationName, associationDefinition] of associationEntries) {
+		// Only include if present in data:
+		if (keys.indexOf(associationName) > -1) {
+			const formatted = { association: associationName };
+			const associationModel = associationDefinition.target;
+
+			if (Object.entries(associationModel.associations).length > 0) {
+				const deepData = data[associationName];
+				const subData = Array.isArray(deepData) ? deepData[0] : deepData;
+
+				// Only recurse if subData is an object (not just an ID or null):
+				if (!!subData && typeof subData === 'object' && Object.keys(subData).length > 0) {
+					const subInclude = associationModel.getIncludesTree(subData);
+					if (subInclude.length > 0) {
+						formatted.include = subInclude;
+					}
 				}
 			}
-		}
 
-		result.push( formatted );
+			result.push(formatted);
+		}
 	}
 
 	return result;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nodester",
-  "version": "0.7.14",
+  "version": "0.7.16",
   "description": "A versatile REST framework for Node.js",
   "directories": {
     "docs": "docs",