Commit generated files

dist/classes/Packet.js

@@ -0,0 +1,120 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const PacketFlags_1 = require("../enums/PacketFlags");
+const Zerocoder_1 = require("./Zerocoder");
+const MessageClasses_1 = require("./MessageClasses");
+const MessageFlags_1 = require("../enums/MessageFlags");
+const MessageClass = require("./MessageClasses");
+class Packet {
+    constructor() {
+        this.packetFlags = 0;
+        this.sequenceNumber = 0;
+        this.extraHeader = Buffer.allocUnsafe(0);
+    }
+    getSize() {
+        let idSize = 4;
+        if (this.message.messageFlags & MessageFlags_1.MessageFlags.FrequencyHigh) {
+            idSize = 1;
+        }
+        else if (this.message.messageFlags & MessageFlags_1.MessageFlags.FrequencyMedium) {
+            idSize = 2;
+        }
+        return 1 + 4 + 1 + this.extraHeader.length + idSize + this.message.getSize();
+    }
+    writeToBuffer(buf, pos) {
+        if (this.message.messageFlags & MessageFlags_1.MessageFlags.Zerocoded) {
+            this.packetFlags = this.packetFlags | PacketFlags_1.PacketFlags.Zerocoded;
+        }
+        buf.writeUInt8(this.packetFlags, pos++);
+        buf.writeUInt32BE(this.sequenceNumber, pos);
+        pos = pos + 4;
+        buf.writeUInt8(this.extraHeader.length, pos++);
+        if (this.extraHeader.length > 0) {
+            this.extraHeader.copy(buf, pos);
+            pos += this.extraHeader.length;
+        }
+        const bodyStart = pos;
+        if (this.message.messageFlags & MessageFlags_1.MessageFlags.FrequencyHigh) {
+            buf.writeUInt8(this.message.id, pos++);
+        }
+        else if (this.message.messageFlags & MessageFlags_1.MessageFlags.FrequencyMedium) {
+            buf.writeUInt16BE(this.message.id, pos);
+            pos += 2;
+        }
+        else {
+            buf.writeUInt32BE(this.message.id, pos);
+            pos += 4;
+        }
+        const expectedLength = this.message.getSize();
+        const actualLength = this.message.writeToBuffer(buf, pos);
+        if (actualLength !== expectedLength) {
+            console.error('WARNING: Bytes written does not match expected message data length');
+        }
+        pos += actualLength;
+        if (pos < buf.length) {
+            console.error('WARNING: BUFFER UNDERFLOW: Finished writing but we are not at the end of the buffer');
+        }
+        if (this.packetFlags & PacketFlags_1.PacketFlags.Zerocoded) {
+            buf = Zerocoder_1.Zerocoder.Encode(buf, bodyStart, pos);
+        }
+        return buf;
+    }
+    readFromBuffer(buf, pos, ackReceived, sendAck) {
+        this.packetFlags = buf.readUInt8(pos++);
+        this.sequenceNumber = buf.readUInt32BE(pos);
+        if (this.packetFlags & PacketFlags_1.PacketFlags.Reliable) {
+            sendAck(this.sequenceNumber);
+        }
+        pos = pos + 4;
+        const extraBytes = buf.readUInt8(pos++);
+        if (extraBytes > 0) {
+            this.extraHeader = buf.slice(pos, pos + extraBytes);
+            pos += extraBytes;
+        }
+        else {
+            this.extraHeader = Buffer.allocUnsafe(0);
+        }
+        let appendedAcks = 0;
+        if (this.packetFlags & PacketFlags_1.PacketFlags.Zerocoded) {
+            let tail = 0;
+            if (this.packetFlags & PacketFlags_1.PacketFlags.Ack) {
+                tail = 1;
+                appendedAcks = buf.readUInt8(buf.length - 1);
+                if (appendedAcks > 0) {
+                    tail += appendedAcks * 4;
+                }
+            }
+            buf = Zerocoder_1.Zerocoder.Decode(buf, pos, buf.length - 1, tail);
+        }
+        let messageID = buf.readUInt8(pos);
+        if (messageID === 0xFF) {
+            messageID = buf.readUInt16BE(pos);
+            if (messageID === 0xFFFF) {
+                messageID = buf.readUInt32BE(pos);
+                pos += 4;
+            }
+            else {
+                pos += 2;
+            }
+        }
+        else {
+            pos++;
+        }
+        this.message = new MessageClass[MessageClasses_1.nameFromID(messageID)]();
+        const readLength = this.message.readFromBuffer(buf, pos);
+        pos += readLength;
+        if (this.packetFlags & PacketFlags_1.PacketFlags.Ack) {
+            for (let i = 0; i < appendedAcks; i++) {
+                const ackID = buf.readUInt32BE(pos);
+                ackReceived(ackID);
+                pos += 4;
+            }
+            pos++;
+        }
+        if (pos < buf.length) {
+            console.error('WARNING: Finished reading ' + MessageClasses_1.nameFromID(messageID) + ' but we\'re not at the end of the packet (' + pos + ' < ' + buf.length + ', seq ' + this.sequenceNumber + ')');
+        }
+    }
+}
+exports.Packet = Packet;
+//# sourceMappingURL=Packet.js.map