From 2d3a6fb4ca8901f88f6ff63c2dc92969268c5f41 Mon Sep 17 00:00:00 2001
From: Udit Takkar <53316345+Udit-takkar@users.noreply.github.com>
Date: Sat, 27 Aug 2022 00:29:58 +0530
Subject: [PATCH] fix: add team model in prisma middleware (#3947)

* fix: add team model in prisma middleware

* refactor: move to getParsedResults function

* fix: add eventtype

Co-authored-by: Peer Richelsen <peeroke@gmail.com>
Co-authored-by: kodiakhq[bot] <49736102+kodiakhq[bot]@users.noreply.github.com>
---
 .../eventTypeDescriptionParseAndSanitize.ts   | 28 +++++++++++++------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/packages/prisma/middleware/eventTypeDescriptionParseAndSanitize.ts b/packages/prisma/middleware/eventTypeDescriptionParseAndSanitize.ts
index 74d99501c3..da5813d1af 100644
--- a/packages/prisma/middleware/eventTypeDescriptionParseAndSanitize.ts
+++ b/packages/prisma/middleware/eventTypeDescriptionParseAndSanitize.ts
@@ -1,4 +1,4 @@
-import type { PrismaClient } from "@prisma/client";
+import type { PrismaClient, EventType } from "@prisma/client";
 import MarkdownIt from "markdown-it";
 
 const md = new MarkdownIt("zero").enable([
@@ -14,22 +14,32 @@ function parseAndSanitize(description: string) {
   return parsedMarkdown;
 }
 
+function getParsedResults(eventTypes: EventType | EventType[]) {
+  const results = Array.isArray(eventTypes) ? eventTypes : [eventTypes];
+  const parsedResults = results.map((record) => ({
+    ...record,
+    descriptionAsSafeHTML: record.description ? parseAndSanitize(record.description) : null,
+  }));
+  /* If the original result was an array, return the parsed array, otherwise is a single record */
+  return Array.isArray(eventTypes) ? parsedResults : parsedResults[0];
+}
+
 /**
  * Parses event type descriptions and treat them as markdown,
  * then sanitizes the resulting HTML and adds a new `descriptionAsSafeHTML` property.
  */
 function eventTypeDescriptionParseAndSanitize(prisma: PrismaClient) {
   prisma.$use(async (params, next) => {
-    if (params.model === "EventType") {
+    if (params.model === "Team") {
+      const result = await next(params);
+      if (result?.eventTypes) {
+        result.eventTypes = getParsedResults(result.eventTypes);
+      }
+      return result;
+    } else if (params.model === "EventType") {
       const result = await next(params);
       if (result) {
-        const results = Array.isArray(result) ? result : [result];
-        const parsedResults = results.map((record) => ({
-          ...record,
-          descriptionAsSafeHTML: record.description ? parseAndSanitize(record.description) : null,
-        }));
-        /* If the original result was an array, return the parsed array, otherwise is a single record */
-        return Array.isArray(result) ? parsedResults : parsedResults[0];
+        return getParsedResults(result);
       }
       return result;
     }