revanced-bots/client/index.js

import { createConnection } from 'node:net';
import { serialize, deserialize } from 'bson';
import EventEmitter from 'node:events';

class HelperClient extends EventEmitter {
    constructor({ server }) {
        super();
        if (!server?.port) throw new Error('You did not specify the server port.');
        this.server = server;
    };

    connect() {
        this.client = createConnection(this.server.port, this.server.host ? this.server.host : 'localhost', () => {
            this.emit('connect');
        });

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

            switch (eventData.op) {
                case 2: {
                    // The 'aiResponse' event.

                    this.emit('aiResponse', eventData);
                };

                case 6: {
                    // The 'ocrResponse' event.

                    this.emit('ocrResponse', eventData);
                };
            };
        });
    };

    sendData(data) {
        this.client.write(serialize(data));
        return;
    };

    scanText(text, id) {
        this.sendData({
            op: 1,
            id, 
            text
        });

        return;
    };

    scanImage(url, id) {
        this.sendData({
            op: 5,
            id, 
            url
        });

        return;
    };

    sendTrainData(text, label) {
        this.sendData({
            op: 3,
            label, 
            text
        });

        return;
    };

    trainAI() {
        this.sendData({ op: 4 });

        return;
    };
};

export default HelperClient;