next
Version:
The React Framework
48 lines (47 loc) • 1.47 kB
JavaScript
import { useCallback, useContext, useEffect, useRef } from "react";
import { GlobalLayoutRouterContext } from "../../../../../shared/lib/app-router-context";
import { getSocketUrl } from "./get-socket-url";
export function useWebsocket(assetPrefix) {
const webSocketRef = useRef();
useEffect(()=>{
if (webSocketRef.current) {
return;
}
const url = getSocketUrl(assetPrefix);
webSocketRef.current = new window.WebSocket("" + url + "/_next/webpack-hmr");
}, [
assetPrefix
]);
return webSocketRef;
}
export function useSendMessage(webSocketRef) {
const sendMessage = useCallback((data)=>{
const socket = webSocketRef.current;
if (!socket || socket.readyState !== socket.OPEN) {
return;
}
return socket.send(data);
}, [
webSocketRef
]);
return sendMessage;
}
export function useWebsocketPing(websocketRef) {
const sendMessage = useSendMessage(websocketRef);
const { tree } = useContext(GlobalLayoutRouterContext);
useEffect(()=>{
// Taken from on-demand-entries-client.js
const interval = setInterval(()=>{
sendMessage(JSON.stringify({
event: "ping",
tree,
appDirRoute: true
}));
}, 2500);
return ()=>clearInterval(interval);
}, [
tree,
sendMessage
]);
}
//# sourceMappingURL=use-websocket.js.map