add SSE connection ability

This commit is contained in:
Ashwin Bhat 2024-10-10 09:06:12 -07:00
commit d32e7b6725
No known key found for this signature in database

View file

@ -9,6 +9,9 @@ export default function mcpProxy({
transportToServer: Transport;
onerror: (error: Error) => void;
}) {
let transportToClientClosed = false;
let transportToServerClosed = false;
transportToClient.onmessage = (message) => {
transportToServer.send(message).catch(onerror);
};
@ -18,10 +21,20 @@ export default function mcpProxy({
};
transportToClient.onclose = () => {
if (transportToServerClosed) {
return;
}
transportToClientClosed = true;
transportToServer.close().catch(onerror);
};
transportToServer.onclose = () => {
if (transportToClientClosed) {
return;
}
transportToServerClosed = true;
transportToClient.close().catch(onerror);
};