[feat] Automatically provide context based on INQUIRER

Resolves #532
iamolegga · Oct 12, 2021 · 184bfde · 184bfde
1 parent e20a777
commit 184bfde
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/__tests__/no-context.spec.ts b/__tests__/no-context.spec.ts
@@ -55,7 +55,7 @@ describe('no context', () => {
 
         const ctrlLog = logs.find((v) => v.msg === msg);
         expect(ctrlLog).toBeTruthy();
-        expect(ctrlLog).not.toHaveProperty('context');
+        expect(ctrlLog?.context).toEqual(TestController.name);
       });
     });
   }

diff --git a/__tests__/utils/test-case.ts b/__tests__/utils/test-case.ts
@@ -83,6 +83,7 @@ export class TestCase {
     __resetSingletons();
 
     const app = await NestFactory.create(this.module, this.adapter, {
+      abortOnError: false,
       bufferLogs: true,
     });
     app.useLogger(app.get(Logger));

diff --git a/src/InjectPinoLogger.ts b/src/InjectPinoLogger.ts
@@ -1,16 +1,33 @@
-import { Inject, Provider } from '@nestjs/common';
+import { Inject, Provider, Scope } from '@nestjs/common';
+import { INQUIRER } from '@nestjs/core';
 import { PinoLogger } from './PinoLogger';
 
 const decoratedTokenPrefix = 'PinoLogger:';
 
 const decoratedLoggers = new Set<string>();
+const transientToken = getLoggerToken('');
 
 export function InjectPinoLogger(context = '') {
   decoratedLoggers.add(context);
   return Inject(getLoggerToken(context));
 }
 
 function createDecoratedLoggerProvider(context: string): Provider<PinoLogger> {
+  if (context === '') {
+    return {
+      provide: transientToken,
+      useFactory: (logger: PinoLogger, inquirer: any) => {
+        if (typeof inquirer === 'string') {
+          logger.setContext(inquirer);
+        } else if (typeof inquirer === 'object') {
+          logger.setContext(inquirer.constructor.name);
+        }
+        return logger;
+      },
+      inject: [PinoLogger, INQUIRER],
+      scope: Scope.TRANSIENT,
+    };
+  }
   return {
     provide: getLoggerToken(context),
     useFactory: (logger: PinoLogger) => {