53 lines
1.8 KiB
Plaintext
53 lines
1.8 KiB
Plaintext
let source;
|
|
const eventCallbacks = [];
|
|
function getSocketProtocol(assetPrefix) {
|
|
let protocol = location.protocol;
|
|
try {
|
|
// assetPrefix is a url
|
|
protocol = new URL(assetPrefix).protocol;
|
|
} catch (e) {}
|
|
return protocol === "http:" ? "ws" : "wss";
|
|
}
|
|
export function addMessageListener(callback) {
|
|
eventCallbacks.push(callback);
|
|
}
|
|
export function sendMessage(data) {
|
|
if (!source || source.readyState !== source.OPEN) return;
|
|
return source.send(data);
|
|
}
|
|
export function connectHMR(options) {
|
|
function init() {
|
|
if (source) source.close();
|
|
function handleOnline() {
|
|
window.console.log("[HMR] connected");
|
|
}
|
|
function handleMessage(event) {
|
|
// Coerce into HMR_ACTION_TYPES as that is the format.
|
|
const msg = JSON.parse(event.data);
|
|
for (const eventCallback of eventCallbacks){
|
|
eventCallback(msg);
|
|
}
|
|
}
|
|
function handleDisconnect() {
|
|
source.onerror = null;
|
|
source.onclose = null;
|
|
source.close();
|
|
init();
|
|
}
|
|
const { hostname, port } = location;
|
|
const protocol = getSocketProtocol(options.assetPrefix || "");
|
|
const assetPrefix = options.assetPrefix.replace(/^\/+/, "");
|
|
let url = protocol + "://" + hostname + ":" + port + (assetPrefix ? "/" + assetPrefix : "");
|
|
if (assetPrefix.startsWith("http")) {
|
|
url = protocol + "://" + assetPrefix.split("://")[1];
|
|
}
|
|
source = new window.WebSocket("" + url + options.path);
|
|
source.onopen = handleOnline;
|
|
source.onerror = handleDisconnect;
|
|
source.onclose = handleDisconnect;
|
|
source.onmessage = handleMessage;
|
|
}
|
|
init();
|
|
}
|
|
|
|
//# sourceMappingURL=websocket.js.map |