revanced-bots/server/index.js

import { readFileSync } from 'node:fs';
// Fix __dirname not being defined in ES modules. (https://stackoverflow.com/a/64383997)
import { fileURLToPath } from 'node:url';
import { dirname } from 'node:path';

const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);

const config = JSON.parse(readFileSync('./config.json', 'utf-8'));

import { createServer } from 'node:net';
import { deserialize } from 'bson';
import FastText from 'fasttext.js';
import { runAI, trainAI, runOCR, addTrainData } from './events/index.js';

const ft = new FastText(config.fasttext);

ft.load();

const server = createServer(async (client) => {
	client.on('data', async (data) => {
		const eventData = deserialize(data, {
			allowObjectSmallerThanBufferSize: true
		});

		switch (eventData.op) {
		case 1: {
			runAI(client, eventData, ft);
			break;
		}

		case 3: {
			addTrainData(eventData, __dirname, config);
			break;
		}

		case 4: {
			trainAI(ft, __dirname, config);
			break;
		}

		case 5: {
			runOCR(client, eventData);
			break;
		}
		}
	});
});

server.listen(config.server.port || 3000);