refactor(server): app module (#13193)

server/src/repositories/event.repository.ts

@@ -1,5 +1,5 @@
 import { Inject, Injectable } from '@nestjs/common';
-import { ModuleRef } from '@nestjs/core';
+import { ModuleRef, Reflector } from '@nestjs/core';
 import {
   OnGatewayConnection,
   OnGatewayDisconnect,
@@ -7,11 +7,16 @@ import {
   WebSocketGateway,
   WebSocketServer,
 } from '@nestjs/websockets';
+import { ClassConstructor } from 'class-transformer';
+import _ from 'lodash';
 import { Server, Socket } from 'socket.io';
+import { EventConfig } from 'src/decorators';
+import { MetadataKey } from 'src/enum';
 import {
   ArgsOf,
   ClientEventMap,
   EmitEvent,
+  EmitHandler,
   EventItem,
   IEventRepository,
   serverEvents,
@@ -24,6 +29,14 @@ import { handlePromiseError } from 'src/utils/misc';
 
 type EmitHandlers = Partial<{ [T in EmitEvent]: Array<EventItem<T>> }>;
 
+type Item<T extends EmitEvent> = {
+  event: T;
+  handler: EmitHandler<T>;
+  priority: number;
+  server: boolean;
+  label: string;
+};
+
 @Instrumentation()
 @WebSocketGateway({
   cors: true,
@@ -44,6 +57,49 @@ export class EventRepository implements OnGatewayConnection, OnGatewayDisconnect
     this.logger.setContext(EventRepository.name);
   }
 
+  setup({ services }: { services: ClassConstructor<unknown>[] }) {
+    const reflector = this.moduleRef.get(Reflector, { strict: false });
+    const repository = this.moduleRef.get<IEventRepository>(IEventRepository);
+    const items: Item<EmitEvent>[] = [];
+
+    // discovery
+    for (const Service of services) {
+      const instance = this.moduleRef.get<any>(Service);
+      const ctx = Object.getPrototypeOf(instance);
+      for (const property of Object.getOwnPropertyNames(ctx)) {
+        const descriptor = Object.getOwnPropertyDescriptor(ctx, property);
+        if (!descriptor || descriptor.get || descriptor.set) {
+          continue;
+        }
+
+        const handler = instance[property];
+        if (typeof handler !== 'function') {
+          continue;
+        }
+
+        const event = reflector.get<EventConfig>(MetadataKey.EVENT_CONFIG, handler);
+        if (!event) {
+          continue;
+        }
+
+        items.push({
+          event: event.name,
+          priority: event.priority || 0,
+          server: event.server ?? false,
+          handler: handler.bind(instance),
+          label: `${Service.name}.${handler.name}`,
+        });
+      }
+    }
+
+    const handlers = _.orderBy(items, ['priority'], ['asc']);
+
+    // register by priority
+    for (const handler of handlers) {
+      repository.on(handler);
+    }
+  }
+
   afterInit(server: Server) {
     this.logger.log('Initialized websocket server');
 

server/src/repositories/logger.repository.spec.ts

@@ -1,4 +1,5 @@
 import { ClsService } from 'nestjs-cls';
+import { ImmichWorker } from 'src/enum';
 import { IConfigRepository } from 'src/interfaces/config.interface';
 import { LoggerRepository } from 'src/repositories/logger.repository';
 import { mockEnvData, newConfigRepositoryMock } from 'test/repositories/config.repository.mock';
@@ -22,18 +23,18 @@ describe(LoggerRepository.name, () => {
       configMock.getEnv.mockReturnValue(mockEnvData({ noColor: false }));
 
       sut = new LoggerRepository(clsMock, configMock);
-      sut.setAppName('api');
+      sut.setAppName(ImmichWorker.API);
 
-      expect(sut['formatContext']('context')).toBe('\u001B[33m[api:context]\u001B[39m ');
+      expect(sut['formatContext']('context')).toBe('\u001B[33m[Api:context]\u001B[39m ');
     });
 
     it('should not use colors when noColor is true', () => {
       configMock.getEnv.mockReturnValue(mockEnvData({ noColor: true }));
 
       sut = new LoggerRepository(clsMock, configMock);
-      sut.setAppName('api');
+      sut.setAppName(ImmichWorker.API);
 
-      expect(sut['formatContext']('context')).toBe('[api:context] ');
+      expect(sut['formatContext']('context')).toBe('[Api:context] ');
     });
   });
 });

server/src/repositories/logger.repository.ts

@@ -34,7 +34,7 @@ export class LoggerRepository extends ConsoleLogger implements ILoggerRepository
   private static appName?: string = undefined;
 
   setAppName(name: string): void {
-    LoggerRepository.appName = name;
+    LoggerRepository.appName = name.charAt(0).toUpperCase() + name.slice(1);
   }
 
   isLevelEnabled(level: LogLevel) {