63 lines
1.6 KiB
JavaScript
63 lines
1.6 KiB
JavaScript
'use strict';
|
|
|
|
const birpc = require('birpc');
|
|
|
|
function createRPCServer(name, ws, functions, options = {}) {
|
|
const event = `${name}:rpc`;
|
|
const group = birpc.createBirpcGroup(
|
|
functions,
|
|
() => birpc.cachedMap(
|
|
Array.from(ws?.clients || []),
|
|
(channel) => {
|
|
if (channel.socket.readyState === channel.socket.CLOSED)
|
|
return void 0;
|
|
return {
|
|
on: (fn) => {
|
|
function handler(data, source) {
|
|
if (!source.socket)
|
|
throw new Error("source.socket is undefined");
|
|
if (channel.socket === source.socket)
|
|
fn(data, source);
|
|
}
|
|
ws.on(event, handler);
|
|
channel.socket.on("close", () => {
|
|
ws.off(event, handler);
|
|
});
|
|
},
|
|
post: (data) => {
|
|
channel.send(event, data);
|
|
}
|
|
};
|
|
}
|
|
).filter((c) => !!c),
|
|
options
|
|
);
|
|
ws.on("connection", () => {
|
|
group.updateChannels();
|
|
});
|
|
return group.broadcast;
|
|
}
|
|
function createRPCClient(name, hot, functions = {}, options = {}) {
|
|
const event = `${name}:rpc`;
|
|
const promise = Promise.resolve(hot).then((r) => {
|
|
if (!r)
|
|
console.warn("[vite-hot-client] Received undefined hot context, RPC calls are ignored");
|
|
return r;
|
|
});
|
|
return birpc.createBirpc(
|
|
functions,
|
|
{
|
|
...options,
|
|
on: async (fn) => {
|
|
(await promise)?.on(event, fn);
|
|
},
|
|
post: async (data) => {
|
|
(await promise)?.send(event, data);
|
|
}
|
|
}
|
|
);
|
|
}
|
|
|
|
exports.createRPCClient = createRPCClient;
|
|
exports.createRPCServer = createRPCServer;
|