1 files changed, 177 insertions, 0 deletions
diff --git a/packages/backend/src/server/ServerService.ts b/packages/backend/src/server/ServerService.ts
new file mode 100644
index 0000000000..2ee7135c2a
--- /dev/null
+++ b/packages/backend/src/server/ServerService.ts
@@ -0,0 +1,177 @@
+import cluster from 'node:cluster';
+import * as fs from 'node:fs';
+import * as http from 'node:http';
+import { Inject, Injectable } from '@nestjs/common';
+import Koa from 'koa';
+import Router from '@koa/router';
+import mount from 'koa-mount';
+import koaLogger from 'koa-logger';
+import * as slow from 'koa-slow';
+import { IsNull } from 'typeorm';
+import { GlobalEventService } from '@/core/GlobalEventService.js';
+import { Config } from '@/config.js';
+import { UserProfilesRepository, UsersRepository } from '@/models/index.js';
+import { DI } from '@/di-symbols.js';
+import Logger from '@/logger.js';
+import { envOption } from '@/env.js';
+import * as Acct from '@/misc/acct.js';
+import { genIdenticon } from '@/misc/gen-identicon.js';
+import { createTemp } from '@/misc/create-temp.js';
+import { UserEntityService } from '@/core/entities/UserEntityService.js';
+import { ActivityPubServerService } from './ActivityPubServerService.js';
+import { NodeinfoServerService } from './NodeinfoServerService.js';
+import { ApiServerService } from './api/ApiServerService.js';
+import { StreamingApiServerService } from './api/StreamingApiServerService.js';
+import { WellKnownServerService } from './WellKnownServerService.js';
+import { MediaProxyServerService } from './MediaProxyServerService.js';
+import { FileServerService } from './FileServerService.js';
+import { ClientServerService } from './web/ClientServerService.js';
+
+const serverLogger = new Logger('server', 'gray', false);
+
+@Injectable()
+export class ServerService {
+	constructor(
+		@Inject(DI.config)
+		private config: Config,
+
+		@Inject(DI.usersRepository)
+		private usersRepository: UsersRepository,
+
+		@Inject(DI.userProfilesRepository)
+		private userProfilesRepository: UserProfilesRepository,
+
+		private userEntityService: UserEntityService,
+		private apiServerService: ApiServerService,
+		private streamingApiServerService: StreamingApiServerService,
+		private activityPubServerService: ActivityPubServerService,
+		private wellKnownServerService: WellKnownServerService,
+		private nodeinfoServerService: NodeinfoServerService,
+		private fileServerService: FileServerService,
+		private mediaProxyServerService: MediaProxyServerService,
+		private clientServerService: ClientServerService,
+		private globalEventService: GlobalEventService,
+	) {
+	}
+
+	public launch() {
+		// Init app
+		const koa = new Koa();
+		koa.proxy = true;
+
+		if (!['production', 'test'].includes(process.env.NODE_ENV ?? '')) {
+		// Logger
+			koa.use(koaLogger(str => {
+				serverLogger.info(str);
+			}));
+
+			// Delay
+			if (envOption.slow) {
+				koa.use(slow({
+					delay: 3000,
+				}));
+			}
+		}
+
+		// HSTS
+		// 6months (15552000sec)
+		if (this.config.url.startsWith('https') && !this.config.disableHsts) {
+			koa.use(async (ctx, next) => {
+				ctx.set('strict-transport-security', 'max-age=15552000; preload');
+				await next();
+			});
+		}
+
+		koa.use(mount('/api', this.apiServerService.createApiServer(koa)));
+		koa.use(mount('/files', this.fileServerService.createServer()));
+		koa.use(mount('/proxy', this.mediaProxyServerService.createServer()));
+
+		// Init router
+		const router = new Router();
+
+		// Routing
+		router.use(this.activityPubServerService.createRouter().routes());
+		router.use(this.nodeinfoServerService.createRouter().routes());
+		router.use(this.wellKnownServerService.createRouter().routes());
+
+		router.get('/avatar/@:acct', async ctx => {
+			const { username, host } = Acct.parse(ctx.params.acct);
+			const user = await this.usersRepository.findOne({
+				where: {
+					usernameLower: username.toLowerCase(),
+					host: (host == null) || (host === this.config.host) ? IsNull() : host,
+					isSuspended: false,
+				},
+				relations: ['avatar'],
+			});
+
+			if (user) {
+				ctx.redirect(this.userEntityService.getAvatarUrlSync(user));
+			} else {
+				ctx.redirect('/static-assets/user-unknown.png');
+			}
+		});
+
+		router.get('/identicon/:x', async ctx => {
+			const [temp, cleanup] = await createTemp();
+			await genIdenticon(ctx.params.x, fs.createWriteStream(temp));
+			ctx.set('Content-Type', 'image/png');
+			ctx.body = fs.createReadStream(temp).on('close', () => cleanup());
+		});
+
+		router.get('/verify-email/:code', async ctx => {
+			const profile = await this.userProfilesRepository.findOneBy({
+				emailVerifyCode: ctx.params.code,
+			});
+
+			if (profile != null) {
+				ctx.body = 'Verify succeeded!';
+				ctx.status = 200;
+
+				await this.userProfilesRepository.update({ userId: profile.userId }, {
+					emailVerified: true,
+					emailVerifyCode: null,
+				});
+
+				this.globalEventService.publishMainStream(profile.userId, 'meUpdated', await this.userEntityService.pack(profile.userId, { id: profile.userId }, {
+					detail: true,
+					includeSecrets: true,
+				}));
+			} else {
+				ctx.status = 404;
+			}
+		});
+
+		// Register router
+		koa.use(router.routes());
+
+		koa.use(mount(this.clientServerService.createApp()));
+
+		const server = http.createServer(koa.callback());
+
+		this.streamingApiServerService.attachStreamingApi(server);
+
+		server.on('error', e => {
+			switch ((e as any).code) {
+				case 'EACCES':
+					serverLogger.error(`You do not have permission to listen on port ${this.config.port}.`);
+					break;
+				case 'EADDRINUSE':
+					serverLogger.error(`Port ${this.config.port} is already in use by another process.`);
+					break;
+				default:
+					serverLogger.error(e);
+					break;
+			}
+
+			if (cluster.isWorker) {
+			process.send!('listenFailed');
+			} else {
+			// disableClustering
+				process.exit(1);
+			}
+		});
+
+		server.listen(this.config.port);
+	}
+}