Checkmate/server/src/config/services.ts

import { Mongoose } from "mongoose";
import MongoDB from "../db/MongoDB.js";
import { IDb } from "@/db/IDb.js";
import {
	// Service classes
	NetworkService,
	EmailService,
	BufferService,
	GlobalPingService,
	LessSimpleQueue,
	SuperSimpleQueue,
	QueueHelper,
	NotificationsService,
	TagsService,
	StatusService,
	NotificationMessageBuilder,
	MonitorService,
	StatusPageService,
	UserService,
	CheckService,
	GeoChecksService,
	DiagnosticService,
	InviteService,
	MaintenanceWindowService,
	IncidentService,
	// Notification providers
	WebhookProvider,
	SlackProvider,
	EmailProvider,
	DiscordProvider,
	PagerDutyProvider,
	MatrixProvider,
	TeamsProvider,
	TelegramProvider,
	PushoverProvider,
	TwilioProvider,
	// Interfaces
	INetworkService,
	IEmailService,
	IBufferService,
	IJobQueue,
	INotificationsService,
	ITagsService,
	IStatusService,
	IMonitorService,
	IUserService,
	ICheckService,
	IGeoChecksService,
	IDiagnosticService,
	IInviteService,
	IMaintenanceWindowService,
	IStatusPageService,
	IIncidentService,
	INotificationMessageBuilder,
	ISettingsService,
	SettingsService,
	EnvConfig,
} from "@/service/index.js";

// Network providers
import { PingProvider } from "@/service/infrastructure/network/PingProvider.js";
import { HttpProvider } from "@/service/infrastructure/network/HttpProvider.js";
import { AdvancedMatcher } from "@/service/infrastructure/network/AdvancedMatcher.js";
import { PageSpeedProvider } from "@/service/infrastructure/network/PageSpeedProvider.js";
import { HardwareProvider } from "@/service/infrastructure/network/HardwareProvider.js";
import { DockerProvider } from "@/service/infrastructure/network/DockerProvider.js";
import { PortProvider } from "@/service/infrastructure/network/PortProvider.js";
import { GameProvider } from "@/service/infrastructure/network/GameProvider.js";
import { GrpcProvider } from "@/service/infrastructure/network/GrpcProvider.js";
import { WebSocketProvider } from "@/service/infrastructure/network/WebSocketProvider.js";
import { DNSProvider } from "@/service/infrastructure/network/DNSProvider.js";

// Third-party
import { Resolver } from "dns/promises";
import axios from "axios";
import got from "got";
import ping from "ping";
import Docker from "dockerode";
import net from "net";
import fs from "fs";
import path from "path";
import nodemailer from "nodemailer";
import pkg from "handlebars";
const { compile } = pkg;
import mjml2html from "mjml";
import jwt from "jsonwebtoken";
import crypto from "crypto";
import { games, GameDig } from "gamedig";
import jmespath from "jmespath";
import * as grpc from "@grpc/grpc-js";
import * as protoLoader from "@grpc/proto-loader";
import WebSocket from "ws";

// Repositories
import {
	MongoMonitorsRepository,
	MongoChecksRepository,
	MongoGeoChecksRepository,
	MongoMonitorStatsRepository,
	MongoStatusPagesRepository,
	MongoUsersRepository,
	MongoInvitesRepository,
	MongoRecoveryTokensRepository,
	MongoNotificationsRepository,
	MongoTagsRepository,
	MongoIncidentsRepository,
	MongoTeamsRepository,
	MongoMaintenanceWindowsRepository,
	MongoSettingsRepository,
	IMonitorsRepository,
	IChecksRepository,
	IGeoChecksRepository,
	IMonitorStatsRepository,
	IStatusPagesRepository,
	IUsersRepository,
	IInvitesRepository,
	IRecoveryTokensRepository,
	ISettingsRepository,
	INotificationsRepository,
	ITagsRepository,
	IIncidentsRepository,
	ITeamsRepository,
	IMaintenanceWindowsRepository,
} from "@/repositories/index.js";
import { ILogger } from "@/utils/logger.js";
import { AppError } from "@/utils/AppError.js";
import { QueueModes, type QueueMode } from "@/types/settings.js";
import { env } from "process";

export type InitializedServices = {
	settingsService: ISettingsService;
	db: IDb<Mongoose>;
	networkService: INetworkService;
	emailService: IEmailService;
	bufferService: IBufferService;
	statusService: IStatusService;
	jobQueue: IJobQueue;
	userService: IUserService;
	checkService: ICheckService;
	geoChecksService: IGeoChecksService;
	diagnosticService: IDiagnosticService;
	inviteService: IInviteService;
	maintenanceWindowService: IMaintenanceWindowService;
	monitorService: IMonitorService;
	incidentService: IIncidentService;
	logger: ILogger;
	notificationsService: INotificationsService;
	tagsService: ITagsService;
	statusPageService: IStatusPageService;
	notificationMessageBuilder: INotificationMessageBuilder;

	// Repositories
	monitorsRepository: IMonitorsRepository;
	checksRepository: IChecksRepository;
	geoChecksRepository: IGeoChecksRepository;
	monitorStatsRepository: IMonitorStatsRepository;
	statusPagesRepository: IStatusPagesRepository;
	usersRepository: IUsersRepository;
	invitesRepository: IInvitesRepository;
	recoveryTokensRepository: IRecoveryTokensRepository;
	settingsRepository: ISettingsRepository;
	notificationsRepository: INotificationsRepository;
	tagsRepository: ITagsRepository;
	incidentsRepository: IIncidentsRepository;
	teamsRepository: ITeamsRepository;
	maintenanceWindowsRepository: IMaintenanceWindowsRepository;
};

export const initializeServices = async ({
	logger,
	envSettings,
	settingsService,
	queueMode = "primary",
}: {
	logger: ILogger;
	envSettings: EnvConfig;
	settingsService: ISettingsService;
	queueMode?: QueueMode;
}): Promise<InitializedServices> => {
	// Create DB

	let db: IDb<Mongoose> | null = null;
	db = new MongoDB(logger, envSettings);
	await db.connect();

	// ** NOTE **
	// const dbType = envSettings.dbType;
	// DB type has been fixed to MongoDB for now

	// if (dbType === "mongodb") {
	// 	db = new MongoDB(logger, envSettings);
	// }

	// if (!db) {
	// 	throw new AppError({ message: "Unsupported database type", status: 500 });
	// }

	// await db.connect();

	// Repositories

	// ** NOTE **
	// DB Type fixed to MongoDB for now
	// let monitorsRepository: IMonitorsRepository;
	// let checksRepository: IChecksRepository;
	// let geoChecksRepository: IGeoChecksRepository;
	// let monitorStatsRepository: IMonitorStatsRepository;
	// let statusPagesRepository: IStatusPagesRepository;
	// let usersRepository: IUsersRepository;
	// let invitesRepository: IInvitesRepository;
	// let recoveryTokensRepository: IRecoveryTokensRepository;
	// let settingsRepository: ISettingsRepository;
	// let notificationsRepository: INotificationsRepository;
	// let tagsRepository: ITagsRepository;
	// let incidentsRepository: IIncidentsRepository;
	// let teamsRepository: ITeamsRepository;
	// let maintenanceWindowsRepository: IMaintenanceWindowsRepository;

	// if (dbType === "mongodb") {
	// 	monitorsRepository = new MongoMonitorsRepository();
	// 	checksRepository = new MongoChecksRepository(logger);
	// 	geoChecksRepository = new MongoGeoChecksRepository(logger);
	// 	monitorStatsRepository = new MongoMonitorStatsRepository();
	// 	statusPagesRepository = new MongoStatusPagesRepository();
	// 	usersRepository = new MongoUsersRepository();
	// 	invitesRepository = new MongoInvitesRepository();
	// 	recoveryTokensRepository = new MongoRecoveryTokensRepository();
	// 	settingsRepository = new MongoSettingsRepository();
	// 	notificationsRepository = new MongoNotificationsRepository();
	// 	tagsRepository = new MongoTagsRepository();
	// 	incidentsRepository = new MongoIncidentsRepository();
	// 	teamsRepository = new MongoTeamsRepository();
	// 	maintenanceWindowsRepository = new MongoMaintenanceWindowsRepository();
	// } else {
	// 	throw new AppError({ message: "Unsupported database type", status: 500 });
	// }

	const monitorsRepository = new MongoMonitorsRepository();
	const checksRepository = new MongoChecksRepository(logger);
	const geoChecksRepository = new MongoGeoChecksRepository(logger);
	const monitorStatsRepository = new MongoMonitorStatsRepository();
	const statusPagesRepository = new MongoStatusPagesRepository();
	const usersRepository = new MongoUsersRepository();
	const invitesRepository = new MongoInvitesRepository();
	const recoveryTokensRepository = new MongoRecoveryTokensRepository();
	const settingsRepository = new MongoSettingsRepository();
	const notificationsRepository = new MongoNotificationsRepository();
	const tagsRepository = new MongoTagsRepository();
	const incidentsRepository = new MongoIncidentsRepository();
	const teamsRepository = new MongoTeamsRepository();
	const maintenanceWindowsRepository = new MongoMaintenanceWindowsRepository();

	// Inject settings repository into settings service (now that DB is connected)
	(settingsService as SettingsService).setRepository(settingsRepository);

	// Network providers
	const pingProvider = new PingProvider(ping);
	const httpProvider = new HttpProvider(got, new AdvancedMatcher(jmespath));
	const pageSpeedProvider = new PageSpeedProvider(httpProvider, settingsService, logger);
	const hardwareProvider = new HardwareProvider(httpProvider);
	const dockerProvider = new DockerProvider(logger, Docker);
	const portProvider = new PortProvider(net);
	const gameProvider = new GameProvider(logger, GameDig);
	const grpcProvider = new GrpcProvider(grpc, protoLoader);
	const webSocketProvider = new WebSocketProvider(WebSocket);
	const dnsProvider = new DNSProvider(() => new Resolver());

	const networkService = new NetworkService(axios, logger, [
		pingProvider,
		httpProvider,
		pageSpeedProvider,
		hardwareProvider,
		dockerProvider,
		portProvider,
		gameProvider,
		grpcProvider,
		webSocketProvider,
		dnsProvider,
	]);
	const emailService = new EmailService(settingsService, fs, path, compile, mjml2html, nodemailer, logger);

	const notificationMessageBuilder = new NotificationMessageBuilder();

	const incidentService = new IncidentService(logger, incidentsRepository, monitorsRepository, usersRepository, notificationMessageBuilder);

	const checkService = new CheckService(monitorsRepository, logger, checksRepository);

	const globalPingService = new GlobalPingService(logger);

	const geoChecksService = new GeoChecksService({
		logger,
		geoChecksRepository,
		globalPingService,
		monitorsRepository,
	});

	const bufferService = new BufferService(logger, checkService, geoChecksService, settingsService);

	const statusService = new StatusService(logger, bufferService, monitorsRepository, monitorStatsRepository, checksRepository);

	// Notification providers
	const webhookProvider = new WebhookProvider(logger);
	const slackProvider = new SlackProvider(logger);
	const emailProvider = new EmailProvider(emailService, logger);
	const discordProvider = new DiscordProvider(logger);
	const pagerDutyProvider = new PagerDutyProvider(logger);
	const matrixProvider = new MatrixProvider(logger);
	const teamsProvider = new TeamsProvider(logger);
	const telegramProvider = new TelegramProvider(logger);
	const pushoverProvider = new PushoverProvider(logger);
	const twilioProvider = new TwilioProvider(logger);

	const notificationsService = new NotificationsService(
		notificationsRepository,
		monitorsRepository,
		webhookProvider,
		emailProvider,
		slackProvider,
		discordProvider,
		pagerDutyProvider,
		matrixProvider,
		teamsProvider,
		telegramProvider,
		pushoverProvider,
		twilioProvider,
		settingsService,
		logger,
		notificationMessageBuilder
	);

	const tagsService = new TagsService(tagsRepository, monitorsRepository);

	const queueHelper = new QueueHelper(
		logger,
		networkService,
		statusService,
		notificationsService,
		checkService,
		settingsService,
		bufferService,
		incidentService,
		maintenanceWindowsRepository,
		monitorsRepository,
		teamsRepository,
		monitorStatsRepository,
		checksRepository,
		incidentsRepository,
		geoChecksService,
		geoChecksRepository
	);

	if (queueMode === "worker" && envSettings.queueType !== "lessSimpleQueue") {
		throw new AppError({
			message: `Worker mode requires QUEUE_TYPE="lessSimpleQueue" got ${envSettings.queueType}`,
			status: 500,
			service: "config services",
			method: "initializeServices",
		});
	}

	let jobQueue: IJobQueue;
	switch (envSettings.queueType) {
		case "lessSimpleQueue":
			jobQueue = await LessSimpleQueue.create(logger, queueHelper, monitorsRepository, envSettings, queueMode);
			break;
		default:
			jobQueue = await SuperSimpleQueue.create(logger, queueHelper, monitorsRepository);
	}

	// Business services
	const userService = new UserService({
		crypto,
		emailService,
		settingsService,
		logger,
		jwt,
		jobQueue,
		monitorsRepository,
		usersRepository,
		invitesRepository,
		recoveryTokensRepository,
		settingsRepository,
		teamsRepository,
	});

	const diagnosticService = new DiagnosticService(db);
	const inviteService = new InviteService({
		invitesRepository,
		settingsService,
		emailService,
	});
	const maintenanceWindowService = new MaintenanceWindowService({
		monitorsRepository,
		maintenanceWindowsRepository,
	});
	const monitorService = new MonitorService({
		jobQueue,
		logger,
		games,
		monitorsRepository,
		checksRepository,
		geoChecksRepository,
		monitorStatsRepository,
		statusPagesRepository,
		incidentsRepository,
	});

	const statusPageService = new StatusPageService(statusPagesRepository, settingsService);

	const services = {
		settingsService,
		db,
		networkService,
		emailService,
		bufferService,
		statusService,
		jobQueue,
		userService,
		checkService,
		geoChecksService,
		diagnosticService,
		inviteService,
		maintenanceWindowService,
		monitorService,
		incidentService,
		logger,
		notificationsService,
		tagsService,
		statusPageService,
		notificationMessageBuilder,

		// Repositories
		monitorsRepository,
		checksRepository,
		geoChecksRepository,
		monitorStatsRepository,
		statusPagesRepository,
		usersRepository,
		invitesRepository,
		recoveryTokensRepository,
		settingsRepository,
		notificationsRepository,
		tagsRepository,
		incidentsRepository,
		teamsRepository,
		maintenanceWindowsRepository,
	};

	return services;
};