mirror of
https://github.com/socketio/socket.io.git
synced 2026-01-13 08:57:59 -05:00
67 lines
1.6 KiB
TypeScript
67 lines
1.6 KiB
TypeScript
import {
|
|
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.slice().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);
|
|
}
|
|
}
|
|
};
|