node-gamedig/lib/QueryRunner.js

import GameResolver from "./GameResolver.js";
import {getProtocol} from './ProtocolResolver.js';
import GlobalUdpSocket from "./GlobalUdpSocket.js";

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

export default class QueryRunner {
    constructor(runnerOpts = {}) {
        this.udpSocket = new GlobalUdpSocket({
            port: runnerOpts.listenUdpPort
        });
        this.gameResolver = new GameResolver();
    }

    async run(userOptions) {
        for (const key of Object.keys(userOptions)) {
            const value = userOptions[key];
            if (['port', 'ipFamily'].includes(key)) {
                userOptions[key] = parseInt(value);
            }
        }

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

        const optionsCollection = {
            ...defaultOptions,
            ...gameOptions,
            ...userOptions
        };

        const addAttemptWithPort = port => {
            attempts.push({
                ...optionsCollection,
                port
            });
        }

        if (userOptions.port) {
            if(!userOptions.givenPortOnly) {
                if (gameQueryPortOffset)
                    addAttemptWithPort(userOptions.port + gameQueryPortOffset);

                if (userOptions.port === gameOptions.port && gameQueryPort)
                    addAttemptWithPort(gameQueryPort);
            }

            attempts.push(optionsCollection);
        } else if (gameQueryPort) {
            addAttemptWithPort(gameQueryPort);
        } else if (gameOptions.port) {
            addAttemptWithPort(gameOptions.port + (gameQueryPortOffset || 0));
        } else {
            // Hopefully the request doesn't need a port. If it does, it'll fail when making the request.
            attempts.push(optionsCollection);
        }

        const numRetries = userOptions.maxAttempts || gameOptions.maxAttempts || defaultOptions.maxAttempts;

        let attemptNum = 0;
        const errors = [];
        for (const attempt of attempts) {
            for (let retry = 0; retry < numRetries; retry++) {
                attemptNum++;
                try {
                    return await this._attempt(attempt);
                } catch (e) {
                    e.stack = 'Attempt #' + attemptNum + ' - Port=' + attempt.port + ' Retry=' + (retry) + ':\n' + e.stack;
                    errors.push(e);
                }
            }
        }

        const err = new Error('Failed all ' + errors.length + ' attempts');
        for (const e of errors) {
            err.stack += '\n' + e.stack;
        }

        throw err;
    }

    async _attempt(options) {
        const core = getProtocol(options.protocol);
        core.options = options;
        core.udpSocket = this.udpSocket;
        return await core.runOnceSafe();
    }
}