33 lines
1.1 KiB
JavaScript
33 lines
1.1 KiB
JavaScript
|
/* global __webpack_dev_server_client__ */
|
||
|
|
|
||
|
|
const getSocketUrlParts = require('./utils/getSocketUrlParts.js');
|
||
|
|
const getUrlFromParts = require('./utils/getUrlFromParts');
|
||
|
|
const getWDSMetadata = require('./utils/getWDSMetadata');
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Initializes a socket server for HMR for webpack-dev-server.
|
||
|
|
* @param {function(*): void} messageHandler A handler to consume Webpack compilation messages.
|
||
|
|
* @param {string} [resourceQuery] Webpack's `__resourceQuery` string.
|
||
|
|
* @returns {void}
|
||
|
|
*/
|
||
|
|
function initWDSSocket(messageHandler, resourceQuery) {
|
||
|
|
if (typeof __webpack_dev_server_client__ !== 'undefined') {
|
||
|
|
let SocketClient = __webpack_dev_server_client__;
|
||
|
|
if (typeof __webpack_dev_server_client__.default !== 'undefined') {
|
||
|
|
SocketClient = __webpack_dev_server_client__.default;
|
||
|
|
}
|
||
|
|
|
||
|
|
const wdsMeta = getWDSMetadata(SocketClient);
|
||
|
|
const urlParts = getSocketUrlParts(resourceQuery, wdsMeta);
|
||
|
|
|
||
|
|
const connection = new SocketClient(getUrlFromParts(urlParts, wdsMeta));
|
||
|
|
|
||
|
|
connection.onMessage(function onSocketMessage(data) {
|
||
|
|
const message = JSON.parse(data);
|
||
|
|
messageHandler(message);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = { init: initWDSSocket };
|