steam-server-query/lib/promiseSocket.js

Module which implements the Master Server Query Protocol and Game Server Queries.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.PromiseSocket = void 0;
const dgram_1 = require("dgram");
class PromiseSocket {
    constructor(_attempts, _timeout) {
        this._attempts = _attempts;
        this._timeout = _timeout;
        if (Array.isArray(this._timeout) &&
            this._attempts !== this._timeout.length) {
            throw new Error(`Number of attempts (${this._attempts}) does not match the length of the timeout array (${this._timeout.length})`);
        }
        this._socket = (0, dgram_1.createSocket)('udp4');
    }
    async send(buffer, host, port) {
        return new Promise(async (resolve, reject) => {
            for (let i = 0; i < this._attempts; i++) {
                let timeout;
                if (Array.isArray(this._timeout)) {
                    timeout = this._timeout[i];
                }
                else {
                    timeout = this._timeout;
                }
                try {
                    const messageBuffer = await this._socketSend(buffer, host, port, timeout);
                    return resolve(messageBuffer);
                }
                catch (err) {
                    if (i === this._attempts - 1) {
                        return reject(err);
                    }
                }
            }
        });
    }
    closeSocket() {
        this._socket.close();
    }
    _socketSend(buffer, host, port, timeout) {
        return new Promise((resolve, reject) => {
            this._socket.send(buffer, port, host, (err) => {
                if (err)
                    return reject(typeof err == 'string' ? new Error(err) : err);
                const messageListener = (buffer) => {
                    this._socket.removeListener('message', messageListener);
                    this._socket.removeListener('error', errorListener);
                    clearTimeout(timeoutFnc);
                    return resolve(buffer);
                };
                const errorListener = (err) => {
                    clearTimeout(timeoutFnc);
                    return reject(err);
                };
                const timeoutFnc = setTimeout(() => {
                    this._socket.removeListener('message', messageListener);
                    this._socket.removeListener('error', errorListener);
                    return reject('Timeout reached. Possible reasons: You are being rate limited; Timeout too short; Wrong server host configured;');
                }, timeout);
                this._socket.on('message', messageListener);
                this._socket.on('error', errorListener);
            });
        });
    }
}
exports.PromiseSocket = PromiseSocket;