refactor: app modules, main.ts (#8156)

server/src/main.ts

@@ -1,6 +1,75 @@
-import { bootstrapApi } from 'src/apps/api.main';
-import { bootstrapImmichAdmin } from 'src/apps/immich-admin.main';
-import { bootstrapMicroservices } from 'src/apps/microservices.main';
+import { NestFactory } from '@nestjs/core';
+import { NestExpressApplication } from '@nestjs/platform-express';
+import { json } from 'body-parser';
+import cookieParser from 'cookie-parser';
+import { CommandFactory } from 'nest-commander';
+import { existsSync } from 'node:fs';
+import sirv from 'sirv';
+import { ApiModule, ImmichAdminModule, MicroservicesModule } from 'src/app.module';
+import { WEB_ROOT, envName, excludePaths, isDev, serverVersion } from 'src/constants';
+import { LogLevel } from 'src/entities/system-config.entity';
+import { WebSocketAdapter } from 'src/middleware/websocket.adapter';
+import { ApiService } from 'src/services/api.service';
+import { otelSDK } from 'src/utils/instrumentation';
+import { ImmichLogger } from 'src/utils/logger';
+import { useSwagger } from 'src/utils/misc';
+
+async function bootstrapMicroservices() {
+  const logger = new ImmichLogger('ImmichMicroservice');
+  const port = Number(process.env.MICROSERVICES_PORT) || 3002;
+
+  otelSDK.start();
+  const app = await NestFactory.create(MicroservicesModule, { bufferLogs: true });
+  app.useLogger(app.get(ImmichLogger));
+  app.useWebSocketAdapter(new WebSocketAdapter(app));
+
+  await app.listen(port);
+
+  logger.log(`Immich Microservices is listening on ${await app.getUrl()} [v${serverVersion}] [${envName}] `);
+}
+
+async function bootstrapApi() {
+  const logger = new ImmichLogger('ImmichServer');
+  const port = Number(process.env.SERVER_PORT) || 3001;
+
+  otelSDK.start();
+  const app = await NestFactory.create<NestExpressApplication>(ApiModule, { bufferLogs: true });
+
+  app.useLogger(app.get(ImmichLogger));
+  app.set('trust proxy', ['loopback', 'linklocal', 'uniquelocal']);
+  app.set('etag', 'strong');
+  app.use(cookieParser());
+  app.use(json({ limit: '10mb' }));
+  if (isDev) {
+    app.enableCors();
+  }
+  app.useWebSocketAdapter(new WebSocketAdapter(app));
+  useSwagger(app, isDev);
+
+  app.setGlobalPrefix('api', { exclude: excludePaths });
+  if (existsSync(WEB_ROOT)) {
+    // copied from https://github.com/sveltejs/kit/blob/679b5989fe62e3964b9a73b712d7b41831aa1f07/packages/adapter-node/src/handler.js#L46
+    // provides serving of precompressed assets and caching of immutable assets
+    app.use(
+      sirv(WEB_ROOT, {
+        etag: true,
+        gzip: true,
+        brotli: true,
+        setHeaders: (res, pathname) => {
+          if (pathname.startsWith(`/_app/immutable`) && res.statusCode === 200) {
+            res.setHeader('cache-control', 'public,max-age=31536000,immutable');
+          }
+        },
+      }),
+    );
+  }
+  app.use(app.get(ApiService).ssr(excludePaths));
+
+  const server = await app.listen(port);
+  server.requestTimeout = 30 * 60 * 1000;
+
+  logger.log(`Immich Server is listening on ${await app.getUrl()} [v${serverVersion}] [${envName}] `);
+}
 
 const immichApp = process.argv[2] || process.env.IMMICH_APP;
 
@@ -8,6 +77,11 @@ if (process.argv[2] === immichApp) {
   process.argv.splice(2, 1);
 }
 
+async function bootstrapImmichAdmin() {
+  process.env.LOG_LEVEL = LogLevel.WARN;
+  await CommandFactory.run(ImmichAdminModule);
+}
+
 function bootstrap() {
   switch (immichApp) {
     case 'immich': {
@@ -23,8 +97,9 @@ function bootstrap() {
       return bootstrapImmichAdmin();
     }
     default: {
-      throw new Error(`Invalid app name: ${immichApp}. Expected one of immich|microservices|cli`);
+      throw new Error(`Invalid app name: ${immichApp}. Expected one of immich|microservices|immich-admin`);
     }
   }
 }
+
 void bootstrap();