diff --git a/server/src/app.module.ts b/server/src/app.module.ts index 7ae60ffeb4..f0006d0442 100644 --- a/server/src/app.module.ts +++ b/server/src/app.module.ts @@ -48,7 +48,7 @@ const imports = [ KyselyModule.forRoot(getKyselyConfig(database.config)), ]; -class BaseModule implements OnModuleInit, OnModuleDestroy { +export class BaseModule implements OnModuleInit, OnModuleDestroy { constructor( @Inject(IWorker) private worker: ImmichWorker, logger: LoggingRepository, diff --git a/server/src/workers/api.ts b/server/src/workers/api.ts index 01b8041635..e8d3480189 100644 --- a/server/src/workers/api.ts +++ b/server/src/workers/api.ts @@ -14,15 +14,16 @@ import { bootstrapTelemetry } from 'src/repositories/telemetry.repository'; import { ApiService } from 'src/services/api.service'; import { MaintenanceService } from 'src/services/maintenance.service'; import { isStartUpError, useSwagger } from 'src/utils/misc'; -async function bootstrap() { - process.title = 'immich-api'; + +export async function bootstrapApi(title: string = 'immich-api', module = ApiModule) { + process.title = title; const { telemetry, network } = new ConfigRepository().getEnv(); if (telemetry.metrics.size > 0) { bootstrapTelemetry(telemetry.apiPort); } - const app = await NestFactory.create(ApiModule, { bufferLogs: true }); + const app = await NestFactory.create(module, { bufferLogs: true }); const logger = await app.resolve(LoggingRepository); const configRepository = app.get(ConfigRepository); app.get(MaintenanceService).nestApplication = app; @@ -68,10 +69,12 @@ async function bootstrap() { logger.log(`Immich Server is listening on ${await app.getUrl()} [v${serverVersion}] [${environment}] `); } -bootstrap().catch((error) => { - if (!isStartUpError(error)) { - console.error(error); - } - // eslint-disable-next-line unicorn/no-process-exit - process.exit(1); -}); +if (require.main === module) { + bootstrapApi().catch((error) => { + if (!isStartUpError(error)) { + console.error(error); + } + // eslint-disable-next-line unicorn/no-process-exit + process.exit(1); + }); +} diff --git a/server/src/workers/maintenance.ts b/server/src/workers/maintenance.ts index d1f7620cc1..289338c633 100644 --- a/server/src/workers/maintenance.ts +++ b/server/src/workers/maintenance.ts @@ -1,77 +1,13 @@ -import { NestFactory } from '@nestjs/core'; -import { NestExpressApplication } from '@nestjs/platform-express'; -import { json } from 'body-parser'; -import compression from 'compression'; -import cookieParser from 'cookie-parser'; -import { existsSync } from 'node:fs'; -import sirv from 'sirv'; import { MaintenanceModule } from 'src/app.module'; -import { excludePaths, serverVersion } from 'src/constants'; -import { WebSocketAdapter } from 'src/middleware/websocket.adapter'; -import { ConfigRepository } from 'src/repositories/config.repository'; -import { LoggingRepository } from 'src/repositories/logging.repository'; -import { bootstrapTelemetry } from 'src/repositories/telemetry.repository'; -import { ApiService } from 'src/services/api.service'; -import { MaintenanceService } from 'src/services/maintenance.service'; -import { isStartUpError, useSwagger } from 'src/utils/misc'; -async function bootstrap() { - process.title = 'immich-api'; +import { isStartUpError } from 'src/utils/misc'; +import { bootstrapApi } from 'src/workers/api'; - const { telemetry, network } = new ConfigRepository().getEnv(); - if (telemetry.metrics.size > 0) { - bootstrapTelemetry(telemetry.apiPort); - } - - const app = await NestFactory.create(MaintenanceModule, { bufferLogs: true }); - const logger = await app.resolve(LoggingRepository); - const configRepository = app.get(ConfigRepository); - app.get(MaintenanceService).nestApplication = app; - - const { environment, host, port, resourcePaths } = configRepository.getEnv(); - - logger.setContext('Bootstrap'); - app.useLogger(logger); - app.set('trust proxy', ['loopback', ...network.trustedProxies]); - app.set('etag', 'strong'); - app.use(cookieParser()); - app.use(json({ limit: '10mb' })); - if (configRepository.isDev()) { - app.enableCors(); - } - app.useWebSocketAdapter(new WebSocketAdapter(app)); - useSwagger(app, { write: configRepository.isDev() }); - - app.setGlobalPrefix('api', { exclude: excludePaths }); - if (existsSync(resourcePaths.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(resourcePaths.web.root, { - etag: true, - gzip: true, - brotli: true, - extensions: [], - 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)); - app.use(compression()); - - const server = await (host ? app.listen(port, host) : app.listen(port)); - server.requestTimeout = 24 * 60 * 60 * 1000; - - logger.log(`Immich Server is listening on ${await app.getUrl()} [v${serverVersion}] [${environment}] `); +if (require.main === module) { + bootstrapApi('immich-maintenance', MaintenanceModule).catch((error) => { + if (!isStartUpError(error)) { + console.error(error); + } + // eslint-disable-next-line unicorn/no-process-exit + process.exit(1); + }); } - -bootstrap().catch((error) => { - if (!isStartUpError(error)) { - console.error(error); - } - // eslint-disable-next-line unicorn/no-process-exit - process.exit(1); -});