📦 socketio / engine.io-parser

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67import {
  ERROR_PACKET,
  PACKET_TYPES_REVERSE,
  Packet,
  BinaryType,
  RawData,
} from "./commons.js";

export const decodePacket = (
  encodedPacket: RawData,
  binaryType?: BinaryType,
): Packet => {
  if (typeof encodedPacket !== "string") {
    return {
      type: "message",
      data: mapBinary(encodedPacket, binaryType),
    };
  }
  const type = encodedPacket.charAt(0);
  if (type === "b") {
    const buffer = Buffer.from(encodedPacket.substring(1), "base64");
    return {
      type: "message",
      data: mapBinary(buffer, binaryType),
    };
  }
  if (!PACKET_TYPES_REVERSE[type]) {
    return ERROR_PACKET;
  }
  return encodedPacket.length > 1
    ? {
        type: PACKET_TYPES_REVERSE[type],
        data: encodedPacket.substring(1),
      }
    : {
        type: PACKET_TYPES_REVERSE[type],
      };
};

const mapBinary = (data: RawData, binaryType?: BinaryType) => {
  switch (binaryType) {
    case "arraybuffer":
      if (data instanceof ArrayBuffer) {
        // from WebSocket & binaryType "arraybuffer"
        return data;
      } else if (Buffer.isBuffer(data)) {
        // from HTTP long-polling
        return data.buffer.slice(
          data.byteOffset,
          data.byteOffset + data.byteLength,
        );
      } else {
        // from WebTransport (Uint8Array)
        return data.buffer;
      }
    case "nodebuffer":
    default:
      if (Buffer.isBuffer(data)) {
        // from HTTP long-polling or WebSocket & binaryType "nodebuffer" (default)
        return data;
      } else {
        // from WebTransport (Uint8Array)
        return Buffer.from(data);
      }
  }
};