Adds relation fields to objects

src/Schema.js

@@ -634,6 +634,24 @@ class Schema {
     }
     return false;
   };
+
+  getRelationFields(className) {
+    if (this.data && this.data[className]) {
+      let classData = this.data[className];
+      return Object.keys(classData).filter((field) => {
+        return classData[field].startsWith('relation');
+      }).reduce((memo, field) => {
+        let type = classData[field];
+        let className = type.slice('relation<'.length, type.length - 1);
+        memo[field] = {
+          __type: 'Relation',
+          className: className
+        };
+        return memo;
+      }, {});
+    }
+    return {};
+  }
 }
 
 // Returns a promise for a new Schema.

src/transform.js

@@ -732,6 +732,11 @@ function untransformObject(schema, className, mongoObject, isNestedObject = fals
                                             mongoObject[key], true);
       }
     }
+
+    if (!isNestedObject) {
+      let relationFields = schema.getRelationFields(className);
+      Object.assign(restObject, relationFields);
+    }
     return restObject;
   default:
     throw 'unknown js type';