feat(mongoose): Added one more parameter to @ref for defining the ref…

…erence type
tsedio · Jun 20, 2019 · e22d38d · e22d38d
1 parent 3d18955
commit e22d38d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/packages/mongoose/src/decorators/ref.ts b/packages/mongoose/src/decorators/ref.ts
@@ -2,6 +2,7 @@ import {Property, Schema} from "@tsed/common";
 import {applyDecorators, Store, StoreFn, StoreMerge} from "@tsed/core";
 import {Schema as MongooseSchema} from "mongoose";
 import {MONGOOSE_MODEL_NAME, MONGOOSE_SCHEMA} from "../constants";
+import {MongooseSchemaTypes} from "../interfaces/MongooseSchemaTypes";
 
 export type Ref<T> = T | string;
 
@@ -27,13 +28,14 @@ export type Ref<T> = T | string;
  * }
  * ```
  *
+ * @param model
  * @param type
  * @returns {Function}
  * @decorator
  * @mongoose
  * @property
  */
-export function Ref(type: string | any) {
+export function Ref(model: string | any, type: MongooseSchemaTypes = MongooseSchemaTypes.OBJECT_ID) {
   return applyDecorators(
     Property({use: String}),
     Schema({
@@ -45,8 +47,8 @@ export function Ref(type: string | any) {
       delete store.get("schema").$ref;
     }),
     StoreMerge(MONGOOSE_SCHEMA, {
-      type: MongooseSchema.Types.ObjectId,
-      ref: typeof type === "string" ? type : Store.from(type).get(MONGOOSE_MODEL_NAME)
+      type: MongooseSchema.Types[type],
+      ref: typeof model === "string" ? model : Store.from(model).get(MONGOOSE_MODEL_NAME)
     })
   );
 }
diff --git a/packages/mongoose/src/interfaces/MongooseSchemaTypes.ts b/packages/mongoose/src/interfaces/MongooseSchemaTypes.ts
@@ -0,0 +1,5 @@
+export enum MongooseSchemaTypes {
+  OBJECT_ID = "ObjectId",
+  STRING = "String",
+  NUMBER = "Number"
+}