57 lines
1.4 KiB
JavaScript
57 lines
1.4 KiB
JavaScript
import { kissLog } from "./log";
|
|
|
|
/**
|
|
* @class ShadowRootMonitor
|
|
* @description 通过覆写 Element.prototype.attachShadow 来监控页面上所有新创建的 Shadow DOM
|
|
*/
|
|
export class ShadowRootMonitor {
|
|
/**
|
|
* @param {function(ShadowRoot): void} callback - 当一个新的 shadowRoot 被创建时调用的回调函数。
|
|
*/
|
|
constructor(callback) {
|
|
if (typeof callback !== "function") {
|
|
throw new Error("Callback must be a function.");
|
|
}
|
|
|
|
this.callback = callback;
|
|
this.isMonitoring = false;
|
|
this.originalAttachShadow = Element.prototype.attachShadow;
|
|
}
|
|
|
|
/**
|
|
* 开始监控 shadowRoot 的创建。
|
|
*/
|
|
start() {
|
|
if (this.isMonitoring) {
|
|
return;
|
|
}
|
|
const monitorInstance = this;
|
|
|
|
Element.prototype.attachShadow = function (...args) {
|
|
const shadowRoot = monitorInstance.originalAttachShadow.apply(this, args);
|
|
if (shadowRoot) {
|
|
try {
|
|
monitorInstance.callback(shadowRoot);
|
|
} catch (error) {
|
|
kissLog("Error in ShadowRootMonitor callback", error);
|
|
}
|
|
}
|
|
return shadowRoot;
|
|
};
|
|
|
|
this.isMonitoring = true;
|
|
}
|
|
|
|
/**
|
|
* 停止监控，并恢复原始的 attachShadow 方法。
|
|
*/
|
|
stop() {
|
|
if (!this.isMonitoring) {
|
|
return;
|
|
}
|
|
|
|
Element.prototype.attachShadow = this.originalAttachShadow;
|
|
this.isMonitoring = false;
|
|
}
|
|
}
|