From 9a4d562d47d31c9e9b585152b7fd4b48b35d32ea Mon Sep 17 00:00:00 2001
From: Ione Ione <102559775+Psionyc@users.noreply.github.com>
Date: Fri, 12 Aug 2022 23:15:06 +0100
Subject: [PATCH] intellisense through partials

---
 src/database/entity.repository.ts | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/database/entity.repository.ts b/src/database/entity.repository.ts
index 829919b..d935d3e 100644
--- a/src/database/entity.repository.ts
+++ b/src/database/entity.repository.ts
@@ -1,6 +1,6 @@
-import { Document, FilterQuery, Model, ModelUpdateOptions, QueryOptions, UpdateQuery } from 'mongoose';
+import { Document, FilterQuery, Model,  UpdateQuery } from 'mongoose';
 
-export abstract class EntityRepository<T extends Document> {
+export abstract class EntityRepository<T extends Document, D> {
   constructor(protected readonly entityModel: Model<T>) {}
 
   async findOne(
@@ -20,14 +20,14 @@ export abstract class EntityRepository<T extends Document> {
     return this.entityModel.find(entityFilterQuery);
   }
 
-  async create(createEntityData: unknown): Promise<T> {
+  async create(createEntityData: Partial<D>): Promise<T> {
     const entity = new this.entityModel(createEntityData);
     return entity.save()
   }
 
   async findOneAndUpdate(
     entityFilterQuery: FilterQuery<T>,
-    updateEntityData: UpdateQuery<unknown>
+    updateEntityData: UpdateQuery<Partial<D>>
   ): Promise<T | null> {
     return this.entityModel.findOneAndUpdate(
       entityFilterQuery,