ObjectID decorator now populates a transform

SierraSoftworks · Jun 18, 2015 · fcddf13 · fcddf13
1 parent e091734
commit fcddf13
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/dist/lib/Decorators.js b/dist/lib/Decorators.js
diff --git a/lib/Decorators.ts b/lib/Decorators.ts
@@ -62,4 +62,8 @@ export function Transform(fromDB: (value: any) => any, toDB: (value: any) => any
 export function ObjectID(target: { constructor: typeof Instance }, name: string) {
 	target.constructor.schema = <Schema>_.clone(target.constructor.schema || {});
 	target.constructor.schema[name] = MongoDB.ObjectID;
+	target.constructor.transforms[name] = {
+        fromDB: value => value._bsontype == 'ObjectID' ? new MongoDB.ObjectID(value.id).toHexString() : value,
+        toDB: value => value && typeof value === 'string' ? new MongoDB.ObjectID(value) : value
+    };
 }