puter/paseo
1
0
Fork 0
mirror of https://github.com/getpaseo/paseo.git synced 2026-07-29 12:01:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: simplify relay ws message normalization

Browse Source
This commit is contained in:
Mohamed Boudra
2026-02-03 14:24:04 +07:00
parent 434955eb93
commit 6e5cfa1eb4
1 changed files with 44 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
packages/relay/src/node-adapter.ts
View File

@@ -76,32 +76,7 @@ export function createRelayServer(config: NodeRelayServerConfig): RelayServer {
relay.addConnection(sessionId, role, connection);
ws.on("message", (data, isBinary) => {
if (isBinary) {
const message =
data instanceof ArrayBuffer
? data
: bufferToArrayBuffer(
data instanceof Buffer
? data
: ArrayBuffer.isView(data)
? Buffer.from(data.buffer, data.byteOffset, data.byteLength)
: Buffer.from(String(data), "utf8")
);
relay.forward(sessionId, role, message);
return;
}
const text =
typeof data === "string"
? data
: data instanceof Buffer
? data.toString("utf8")
: data instanceof ArrayBuffer
? Buffer.from(data).toString("utf8")
: ArrayBuffer.isView(data)
? Buffer.from(data.buffer, data.byteOffset, data.byteLength).toString("utf8")
: String(data);
relay.forward(sessionId, role, text);
relay.forward(sessionId, role, normalizeWsMessageForRelay(data, isBinary));
});
ws.on("close", () => {
@@ -160,3 +135,46 @@ function bufferToArrayBuffer(buffer: Buffer): ArrayBuffer {
out.set(buffer);
return out.buffer;
}
function normalizeWsMessageForRelay(data: unknown, isBinary: boolean): string | ArrayBuffer {
if (isBinary) {
return normalizeWsBinaryMessage(data);
}
return normalizeWsTextMessage(data);
}
function normalizeWsBinaryMessage(data: unknown): ArrayBuffer {
if (data instanceof ArrayBuffer) {
return data;
}
return bufferToArrayBuffer(bufferFromWsData(data));
}
function normalizeWsTextMessage(data: unknown): string {
if (typeof data === "string") {
return data;
}
return bufferFromWsData(data).toString("utf8");
}
function bufferFromWsData(data: unknown): Buffer {
if (Buffer.isBuffer(data)) return data;
if (Array.isArray(data)) {
return Buffer.concat(data.map(bufferFromWsData));
}
if (data instanceof ArrayBuffer) {
return Buffer.from(data);
}
if (ArrayBuffer.isView(data)) {
return Buffer.from(data.buffer, data.byteOffset, data.byteLength);
}
if (typeof data === "string") {
return Buffer.from(data, "utf8");
}
return Buffer.from(String(data), "utf8");
}