node-gamedig/lib/QueryRunner.js

const GameResolver = require('./GameResolver'),
    ProtocolResolver = require('./ProtocolResolver'),
    GlobalUdpSocket = require('./GlobalUdpSocket'),
    Logger = require('./Logger');

const defaultOptions = {
    socketTimeout: 2000,
    attemptTimeout: 10000,
    maxAttempts: 1
};

class QueryRunner {
    constructor() {
        this.udpSocket = new GlobalUdpSocket();
        this.gameResolver = new GameResolver();
        this.protocolResolver = new ProtocolResolver();
    }
    async run(userOptions) {
        for (const key of Object.keys(userOptions)) {
            const value = userOptions[key];
            if (['port'].includes(key)) {
                userOptions[key] = parseInt(value);
            }
        }

        const {
            port_query: gameQueryPort,
            port_query_offset: gameQueryPortOffset,
            ...gameOptions
        } = this.gameResolver.lookup(userOptions.type);
        const attempts = [];

        if (userOptions.port) {
            if (gameQueryPortOffset) {
                attempts.push({
                    ...defaultOptions,
                    ...gameOptions,
                    ...userOptions,
                    port: userOptions.port + gameQueryPortOffset
                });
            }
            if (userOptions.port === gameOptions.port && gameQueryPort) {
                attempts.push({
                    ...defaultOptions,
                    ...gameOptions,
                    ...userOptions,
                    port: gameQueryPort
                });
            }
            attempts.push({
                ...defaultOptions,
                ...gameOptions,
                ...userOptions
            });
        } else if (gameQueryPort) {
            attempts.push({
                ...defaultOptions,
                ...gameOptions,
                ...userOptions,
                port: gameQueryPort
            });
        } else if (gameOptions.port) {
            attempts.push({
                ...defaultOptions,
                ...gameOptions,
                ...userOptions,
                port: gameOptions.port + (gameQueryPortOffset || 0)
            });
        } else {
            throw new Error("Could not determine port to query. Did you provide a port or gameid?");
        }

        if (attempts.length === 1) {
            return await this._attempt(attempts[0]);
        } else {
            const errors = [];
            for (const attempt of attempts) {
                try {
                    return await this._attempt(attempt);
                } catch(e) {
                    const e2 = new Error('Failed to query port ' + attempt.port);
                    e2.stack += "\nCaused by:\n" + e.stack;
                    errors.push(e2);
                }
            }

            const err = new Error('Failed all port attempts');
            err.stack = errors.map(e => e.stack).join('\n');
            throw err;
        }
    }

    async _attempt(options) {
        if (options.debug) {
            const logger = new Logger();
            logger.debugEnabled = true;
            logger.debug("Running attempt with options:");
            logger.debug(options);
        }
        const core = this.protocolResolver.create(options.protocol);
        core.options = options;
        core.udpSocket = this.udpSocket;
        return await core.runAllAttempts();
    }
}

module.exports = QueryRunner;