Model will provide a default _id type if one is not specified

SierraSoftworks · Jun 18, 2015 · 939193e · 939193e
1 parent a77716b
commit 939193e
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/dist/lib/Model.js b/dist/lib/Model.js
diff --git a/dist/lib/Model.js.map b/dist/lib/Model.js.map
diff --git a/lib/Model.ts b/lib/Model.ts
@@ -61,6 +61,14 @@ export default class Model<TDocument extends { _id?: any }, TInstance> {
         this._transforms = instanceType.transforms || {};
         this._validators = instanceType.validators || [];
         this._indexes = instanceType.indexes || [];
+
+        if(!this._schema._id) this._schema._id = MongoDB.ObjectID;
+
+        if(this._schema._id === MongoDB.ObjectID && !this._transforms['_id'])
+            this._transforms['_id'] = {
+                fromDB: value => value._bsontype == 'ObjectID' ? new MongoDB.ObjectID(value.id).toHexString() : value,
+                toDB: value => value && typeof value === 'string' ? new MongoDB.ObjectID(value) : value
+            };
 
         if ((<Function>instanceType).prototype instanceof Instance)
             this._Instance = ModelSpecificInstance(this, instanceType);