:bug: fix lean not working with virtuals

1 files changed, 16 insertions, 11 deletions

diff --git a/src/models/index.ts b/src/models/index.ts

index bb6024fe..03b9fe70 100644
--- a/src/models/index.ts
+++ b/src/models/index.ts
@@ -1,4 +1,20 @@
 import mongoose from "mongoose";
+import { Schema } from "mongoose";
+
+mongoose.plugin((schema: Schema, opts: any) => {
+	schema.set("toObject", {
+		virtuals: true,
+		versionKey: false,
+		transform(doc: any, ret: any) {
+			delete ret._id;
+			delete ret.__v;
+			const props = schema.get("removeResponse") || [];
+			props.forEach((prop: string) => {
+				delete ret[prop];
+			});
+		},
+	});
+});
 
 export * from "./Ban";
 export * from "./Channel";
@@ -15,14 +31,3 @@ export * from "./Message";
 export * from "./Status";
 export * from "./VoiceState";
 export * from "./Event";
-
-mongoose.plugin((schema: any) => {
-	schema.options.toJSON = {
-		virtuals: true,
-		versionKey: false,
-		transform(doc: any, ret: any) {
-			delete ret._id;
-			delete ret.__v;
-		},
-	};
-});