From e90d0bb6fd2d1a2b3555d7ffad55c83a4ad14ec2 Mon Sep 17 00:00:00 2001
From: Thomas Kaul <4159106+dtslvr@users.noreply.github.com>
Date: Wed, 30 Oct 2024 21:28:13 +0100
Subject: [PATCH] Use LOG_LEVELS for prisma query events

---
 apps/api/src/services/prisma/prisma.module.ts |  5 ++--
 .../api/src/services/prisma/prisma.service.ts | 24 ++++++++++++++++++-
 2 files changed, 26 insertions(+), 3 deletions(-)

diff --git a/apps/api/src/services/prisma/prisma.module.ts b/apps/api/src/services/prisma/prisma.module.ts
index 3875c8cab..24da61047 100644
--- a/apps/api/src/services/prisma/prisma.module.ts
+++ b/apps/api/src/services/prisma/prisma.module.ts
@@ -1,9 +1,10 @@
 import { PrismaService } from '@ghostfolio/api/services/prisma/prisma.service';
 
 import { Module } from '@nestjs/common';
+import { ConfigService } from '@nestjs/config';
 
 @Module({
-  providers: [PrismaService],
-  exports: [PrismaService]
+  exports: [PrismaService],
+  providers: [ConfigService, PrismaService]
 })
 export class PrismaModule {}
diff --git a/apps/api/src/services/prisma/prisma.service.ts b/apps/api/src/services/prisma/prisma.service.ts
index e99d6ecf9..4673cbd19 100644
--- a/apps/api/src/services/prisma/prisma.service.ts
+++ b/apps/api/src/services/prisma/prisma.service.ts
@@ -1,16 +1,38 @@
 import {
   Injectable,
   Logger,
+  LogLevel,
   OnModuleDestroy,
   OnModuleInit
 } from '@nestjs/common';
-import { PrismaClient } from '@prisma/client';
+import { ConfigService } from '@nestjs/config';
+import { Prisma, PrismaClient } from '@prisma/client';
 
 @Injectable()
 export class PrismaService
   extends PrismaClient
   implements OnModuleInit, OnModuleDestroy
 {
+  public constructor(configService: ConfigService) {
+    let customLogLevels: LogLevel[];
+
+    try {
+      customLogLevels = JSON.parse(
+        configService.get<string>('LOG_LEVELS')
+      ) as LogLevel[];
+    } catch {}
+
+    const log: Prisma.LogDefinition[] =
+      customLogLevels?.includes('debug') || customLogLevels?.includes('verbose')
+        ? [{ emit: 'stdout', level: 'query' }]
+        : [];
+
+    super({
+      log,
+      errorFormat: 'colorless'
+    });
+  }
+
   public async onModuleInit() {
     try {
       await this.$connect();