collabvm-1.2.ts/cvmts/src/index.ts

import * as toml from 'toml';
import IConfig from './IConfig.js';
import * as fs from 'fs';
import CollabVMServer from './CollabVMServer.js';

import { QemuVM, QemuVmDefinition } from '@cvmts/qemu';

import * as Shared from '@cvmts/shared';
import AuthManager from './AuthManager.js';
import WSServer from './WebSocket/WSServer.js';
import { User } from './User.js';
import TCPServer from './TCP/TCPServer.js';
import VM from './VM.js';
import VNCVM from './VNCVM/VNCVM.js';
import GeoIPDownloader from './GeoIPDownloader.js';

let logger = new Shared.Logger('CVMTS.Init');

logger.Info('CollabVM Server starting up');

// Parse the config file

let Config: IConfig;

if (!fs.existsSync('config.toml')) {
	logger.Error('Fatal error: Config.toml not found. Please copy config.example.toml and fill out fields');
	process.exit(1);
}
try {
	var configRaw = fs.readFileSync('config.toml').toString();
	Config = toml.parse(configRaw);
} catch (e) {
	logger.Error('Fatal error: Failed to read or parse the config file: {0}', (e as Error).message);
	process.exit(1);
}

let exiting = false;
let VM: VM;

async function stop() {
	if (exiting) return;
	exiting = true;
	await VM.Stop();
	process.exit(0);
}

async function start() {
	let geoipReader = null;
	if (Config.geoip.enabled) {
		let downloader = new GeoIPDownloader(Config.geoip.directory, Config.geoip.accountID, Config.geoip.licenseKey);
		geoipReader = await downloader.getGeoIPReader();
	}
	// Init the auth manager if enabled
	let auth = Config.auth.enabled ? new AuthManager(Config.auth.apiEndpoint, Config.auth.secretKey) : null;
	switch (Config.vm.type) {
		case 'qemu': {
			// Print a warning if qmpSockDir is set
			// and the host OS is Windows, as this
			// configuration will very likely not work.
			if (process.platform === 'win32' && Config.qemu.qmpSockDir !== null) {
				logger.Warning("You appear to have the option 'qmpSockDir' enabled in the config.");
				logger.Warning('This is not supported on Windows, and you will likely run into issues.');
				logger.Warning('To remove this warning, use the qmpHost and qmpPort options instead.');
			}

			// Fire up the VM
			let def: QemuVmDefinition = {
				id: Config.collabvm.node,
				command: Config.qemu.qemuArgs,
				snapshot: Config.qemu.snapshots
			};

			VM = new QemuVM(def);
			break;
		}
		case 'vncvm': {
			VM = new VNCVM(Config.vncvm);
			break;
		}
		default: {
			logger.Error('Invalid VM type in config: {0}', Config.vm.type);
			process.exit(1);
			return;
		}
	}
	process.on('SIGINT', async () => await stop());
	process.on('SIGTERM', async () => await stop());

	await VM.Start();
	// Start up the server
	var CVM = new CollabVMServer(Config, VM, auth, geoipReader);

	var WS = new WSServer(Config);
	WS.on('connect', (client: User) => CVM.addUser(client));
	WS.start();

	if (Config.tcp.enabled) {
		var TCP = new TCPServer(Config);
		TCP.on('connect', (client: User) => CVM.addUser(client));
		TCP.start();
	}
}
start();