node-gamedig/protocols/armagetron.js

class Armagetron extends require('./core') {
    constructor() {
        super();
        this.encoding = 'latin1';
        this.byteorder = 'be';
    }

    run(state) {
        const b = Buffer.from([0,0x35,0,0,0,0,0,0x11]);

        this.udpSend(b,(buffer) => {
            const reader = this.reader(buffer);

            reader.skip(6);

            state.raw.port = this.readUInt(reader);
            state.raw.hostname = this.readString(reader);
            state.name = this.stripColorCodes(this.readString(reader));
            state.raw.numplayers = this.readUInt(reader);
            state.raw.versionmin = this.readUInt(reader);
            state.raw.versionmax = this.readUInt(reader);
            state.raw.version = this.readString(reader);
            state.maxplayers = this.readUInt(reader);

            const players = this.readString(reader);
            const list = players.split('\n');
            for(const name of list) {
                if(!name) continue;
                state.players.push({
                    name: this.stripColorCodes(name)
                });
            }

            state.raw.options = this.stripColorCodes(this.readString(reader));
            state.raw.uri = this.readString(reader);
            state.raw.globalids = this.readString(reader);
            this.finish(state);
            return true;
        });
    }

    readUInt(reader) {
        const a = reader.uint(2);
        const b = reader.uint(2);
        return (b<<16) + a;
    }
    readString(reader) {
        const len = reader.uint(2);
        if(!len) return '';

        let out = '';
        for(let i = 0; i < len; i += 2) {
            const hi = reader.uint(1);
            const lo = reader.uint(1);
            if(i+1<len) out += String.fromCharCode(lo);
            if(i+2<len) out += String.fromCharCode(hi);
        }

        return out;
    }
    stripColorCodes(str) {
        return str.replace(/0x[0-9a-f]{6}/g,'');
    }
}

module.exports = Armagetron;