Fixed some namespace pollution when using the ObjectID decorator

SierraSoftworks · Jun 29, 2015 · 486696d · 486696d
1 parent 0b0e8da
commit 486696d
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 13 deletions.
diff --git a/dist/lib/Decorators.js b/dist/lib/Decorators.js
diff --git a/dist/lib/Decorators.js.map b/dist/lib/Decorators.js.map
diff --git a/lib/Decorators.ts b/lib/Decorators.ts
@@ -60,10 +60,9 @@ 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 && value._bsontype == 'ObjectID' ? new MongoDB.ObjectID(value.id).toHexString() : value,
-        toDB: value => value && typeof value === 'string' ? new MongoDB.ObjectID(value) : value
-    };
+	Property(MongoDB.ObjectID)(target, name);
+	Transform(
+		value => value && value._bsontype == 'ObjectID' ? new MongoDB.ObjectID(value.id).toHexString() : value,
+		value => value && typeof value === 'string' ? new MongoDB.ObjectID(value) : value
+	)(target, name);
 }