xinyu-pro 0.21.0

package/lib/plugin-resource-tracker.ts

@@ -0,0 +1,175 @@
+// lib/plugin-resource-tracker.ts - 插件资源追踪器
+// 负责追踪每个插件创建的所有资源，在插件卸载时自动清理
+
+/** 单个事件监听器记录 */
+interface TrackedEventListener {
+  element: HTMLElement | Window | Document;
+  event: string;
+  handler: EventListener;
+}
+
+/** 单个插件的资源集合 */
+interface PluginResources {
+  pluginId: string;
+  elements: Map<string, HTMLElement>;
+  styleTags: Map<string, HTMLStyleElement>;
+  eventListeners: TrackedEventListener[];
+  slotRegistrationIds: string[];
+  modalIds: string[];
+  intervals: ReturnType<typeof setInterval>[];
+  timeouts: ReturnType<typeof setTimeout>[];
+  hostEventHandlers: Array<{ eventName: string; handler: (...args: unknown[]) => unknown }>;
+}
+
+/** 插件资源追踪器 */
+export class PluginResourceTracker {
+  private resources: Map<string, PluginResources> = new Map();
+
+  /** 创建或获取插件的资源追踪 */
+  getOrCreate(pluginId: string): PluginResources {
+    if (!this.resources.has(pluginId)) {
+      this.resources.set(pluginId, {
+        pluginId,
+        elements: new Map(),
+        styleTags: new Map(),
+        eventListeners: [],
+        slotRegistrationIds: [],
+        modalIds: [],
+        intervals: [],
+        timeouts: [],
+        hostEventHandlers: [],
+      });
+    }
+    return this.resources.get(pluginId)!;
+  }
+
+  /** 获取插件的资源追踪（不存在返回 undefined） */
+  get(pluginId: string): PluginResources | undefined {
+    return this.resources.get(pluginId);
+  }
+
+  // ---- DOM 元素追踪 ----
+
+  trackElement(pluginId: string, id: string, element: HTMLElement): void {
+    const res = this.getOrCreate(pluginId);
+    res.elements.set(id, element);
+  }
+
+  untrackElement(pluginId: string, id: string): void {
+    const res = this.resources.get(pluginId);
+    if (res) {
+      res.elements.delete(id);
+    }
+  }
+
+  getElement(pluginId: string, id: string): HTMLElement | undefined {
+    return this.resources.get(pluginId)?.elements.get(id);
+  }
+
+  // ---- 样式标签追踪 ----
+
+  trackStyle(pluginId: string, id: string, styleElement: HTMLStyleElement): void {
+    const res = this.getOrCreate(pluginId);
+    res.styleTags.set(id, styleElement);
+  }
+
+  untrackStyle(pluginId: string, id: string): void {
+    const res = this.resources.get(pluginId);
+    if (res) {
+      res.styleTags.delete(id);
+    }
+  }
+
+  // ---- 事件监听器追踪 ----
+
+  trackEventListener(pluginId: string, listener: TrackedEventListener): void {
+    const res = this.getOrCreate(pluginId);
+    res.eventListeners.push(listener);
+  }
+
+  // ---- 插槽注册追踪 ----
+
+  trackSlotRegistration(pluginId: string, registrationId: string): void {
+    const res = this.getOrCreate(pluginId);
+    res.slotRegistrationIds.push(registrationId);
+  }
+
+  // ---- 模态框追踪 ----
+
+  trackModal(pluginId: string, modalId: string): void {
+    const res = this.getOrCreate(pluginId);
+    res.modalIds.push(modalId);
+  }
+
+  // ---- 定时器追踪 ----
+
+  trackInterval(pluginId: string, id: ReturnType<typeof setInterval>): void {
+    const res = this.getOrCreate(pluginId);
+    res.intervals.push(id);
+  }
+
+  trackTimeout(pluginId: string, id: ReturnType<typeof setTimeout>): void {
+    const res = this.getOrCreate(pluginId);
+    res.timeouts.push(id);
+  }
+
+  // ---- 宿主事件追踪 ----
+
+  trackHostEventHandler(pluginId: string, eventName: string, handler: (...args: unknown[]) => unknown): void {
+    const res = this.getOrCreate(pluginId);
+    res.hostEventHandlers.push({ eventName, handler });
+  }
+
+  // ---- 清理 ----
+
+  /** 清理插件的所有资源 */
+  cleanup(pluginId: string): void {
+    const res = this.resources.get(pluginId);
+    if (!res) return;
+
+    // 1. 移除所有 DOM 元素
+    res.elements.forEach((el) => {
+      try {
+        if (el.parentNode) {
+          el.parentNode.removeChild(el);
+        }
+      } catch (e) {
+        console.warn(`[ResourceTracker] 移除元素失败 (${pluginId}):`, e);
+      }
+    });
+
+    // 2. 移除所有样式标签
+    res.styleTags.forEach((tag) => {
+      try {
+        if (tag.parentNode) {
+          tag.parentNode.removeChild(tag);
+        }
+      } catch (e) {
+        console.warn(`[ResourceTracker] 移除样式失败 (${pluginId}):`, e);
+      }
+    });
+
+    // 3. 移除所有事件监听
+    res.eventListeners.forEach(({ element, event, handler }) => {
+      try {
+        element.removeEventListener(event, handler);
+      } catch (e) {
+        console.warn(`[ResourceTracker] 移除事件监听失败 (${pluginId}):`, e);
+      }
+    });
+
+    // 4. 清除所有定时器
+    res.intervals.forEach((id) => clearInterval(id));
+    res.timeouts.forEach((id) => clearTimeout(id));
+
+    // 5. 删除追踪记录
+    this.resources.delete(pluginId);
+  }
+
+  /** 重置所有资源 */
+  reset(): void {
+    this.resources.forEach((_, pluginId) => {
+      this.cleanup(pluginId);
+    });
+  }
+}