From 56259adaf70a08da8ca4c7ab1e56679e45c24416 Mon Sep 17 00:00:00 2001
From: Oumar Barry <oumar.barry1202@gmail.com>
Date: Sun, 2 Jul 2023 08:55:49 +0000
Subject: [PATCH] feat: add possibility to type mongoose model (#8)

Co-authored-by: Arash <38922203+arashsheyda@users.noreply.github.com>
---
 src/runtime/server/services/mongoose.ts | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/runtime/server/services/mongoose.ts b/src/runtime/server/services/mongoose.ts
index b6fd073..fc8bb6b 100644
--- a/src/runtime/server/services/mongoose.ts
+++ b/src/runtime/server/services/mongoose.ts
@@ -18,7 +18,7 @@ export async function defineMongooseConnection({ uri, options }: { uri?: string;
   }
 }
 
-export function defineMongooseModel(nameOrOptions: string | { name: string; schema: SchemaDefinition; options?: SchemaOptions }, schema?: SchemaDefinition, options?: SchemaOptions): Model<any> {
+export function defineMongooseModel<T>(nameOrOptions: string | { name: string; schema: SchemaDefinition; options?: SchemaOptions }, schema?: SchemaDefinition, options?: SchemaOptions): Model<T> {
   let name: string
   if (typeof nameOrOptions === 'string') {
     name = nameOrOptions
@@ -29,8 +29,8 @@ export function defineMongooseModel(nameOrOptions: string | { name: string; sche
     options = nameOrOptions.options
   }
 
-  const newSchema = new mongoose.Schema({
+  const newSchema = new mongoose.Schema<T>({
     ...schema,
   }, { ...options })
-  return mongoose.model(name, newSchema)
+  return mongoose.model<T>(name, newSchema)
 }