66 lines
2.4 KiB
TypeScript
66 lines
2.4 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright Google Inc. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by an MIT-style license that can be
|
|
* found in the LICENSE file at https://angular.io/license
|
|
*/
|
|
Zone.__load_patch('mediaQuery', (global: any, Zone: ZoneType, api: _ZonePrivate) => {
|
|
function patchAddListener(proto: any) {
|
|
api.patchMethod(proto, 'addListener', (delegate: Function) => (self: any, args: any[]) => {
|
|
const callback = args.length > 0 ? args[0] : null;
|
|
if (typeof callback === 'function') {
|
|
const wrapperedCallback = Zone.current.wrap(callback, 'MediaQuery');
|
|
callback[api.symbol('mediaQueryCallback')] = wrapperedCallback;
|
|
return delegate.call(self, wrapperedCallback);
|
|
} else {
|
|
return delegate.apply(self, args);
|
|
}
|
|
});
|
|
}
|
|
|
|
function patchRemoveListener(proto: any) {
|
|
api.patchMethod(proto, 'removeListener', (delegate: Function) => (self: any, args: any[]) => {
|
|
const callback = args.length > 0 ? args[0] : null;
|
|
if (typeof callback === 'function') {
|
|
const wrapperedCallback = callback[api.symbol('mediaQueryCallback')];
|
|
if (wrapperedCallback) {
|
|
return delegate.call(self, wrapperedCallback);
|
|
} else {
|
|
return delegate.apply(self, args);
|
|
}
|
|
} else {
|
|
return delegate.apply(self, args);
|
|
}
|
|
});
|
|
}
|
|
|
|
if (global['MediaQueryList']) {
|
|
const proto = global['MediaQueryList'].prototype;
|
|
patchAddListener(proto);
|
|
patchRemoveListener(proto);
|
|
} else if (global['matchMedia']) {
|
|
api.patchMethod(global, 'matchMedia', (delegate: Function) => (self: any, args: any[]) => {
|
|
const mql = delegate.apply(self, args);
|
|
if (mql) {
|
|
// try to patch MediaQueryList.prototype
|
|
const proto = Object.getPrototypeOf(mql);
|
|
if (proto && proto['addListener']) {
|
|
// try to patch proto, don't need to worry about patch
|
|
// multiple times, because, api.patchEventTarget will check it
|
|
patchAddListener(proto);
|
|
patchRemoveListener(proto);
|
|
patchAddListener(mql);
|
|
patchRemoveListener(mql);
|
|
} else if (mql['addListener']) {
|
|
// proto not exists, or proto has no addListener method
|
|
// try to patch mql instance
|
|
patchAddListener(mql);
|
|
patchRemoveListener(mql);
|
|
}
|
|
}
|
|
return mql;
|
|
});
|
|
}
|
|
});
|