35 lines
1.4 KiB
TypeScript
35 lines
1.4 KiB
TypeScript
|
import {patchEventTargetMethods, patchOnProperties} from './utils';
|
||
|
|
|
||
|
|
// we have to patch the instance since the proto is non-configurable
|
||
|
|
export function apply(_global: any) {
|
||
|
|
var WS = (<any>_global).WebSocket;
|
||
|
|
// On Safari window.EventTarget doesn't exist so need to patch WS add/removeEventListener
|
||
|
|
// On older Chrome, no need since EventTarget was already patched
|
||
|
|
if (!(<any>_global).EventTarget) {
|
||
|
|
patchEventTargetMethods(WS.prototype);
|
||
|
|
}
|
||
|
|
(<any>_global).WebSocket = function(a, b) {
|
||
|
|
var socket = arguments.length > 1 ? new WS(a, b) : new WS(a);
|
||
|
|
var proxySocket;
|
||
|
|
|
||
|
|
// Safari 7.0 has non-configurable own 'onmessage' and friends properties on the socket instance
|
||
|
|
var onmessageDesc = Object.getOwnPropertyDescriptor(socket, 'onmessage');
|
||
|
|
if (onmessageDesc && onmessageDesc.configurable === false) {
|
||
|
|
proxySocket = Object.create(socket);
|
||
|
|
['addEventListener', 'removeEventListener', 'send', 'close'].forEach(function(propName) {
|
||
|
|
proxySocket[propName] = function() {
|
||
|
|
return socket[propName].apply(socket, arguments);
|
||
|
|
};
|
||
|
|
});
|
||
|
|
} else {
|
||
|
|
// we can patch the real socket
|
||
|
|
proxySocket = socket;
|
||
|
|
}
|
||
|
|
|
||
|
|
patchOnProperties(proxySocket, ['close', 'error', 'message', 'open']);
|
||
|
|
|
||
|
|
return proxySocket;
|
||
|
|
};
|
||
|
|
(<any>global).WebSocket.prototype = Object.create(WS.prototype, {constructor:{value: WebSocket}});
|
||
|
|
}
|