Merge pull request #287 from caorushizi/feature

feat: ✨  video sniffing logic optimization

packages/main/package.json

@@ -72,6 +72,7 @@
     "inversify": "^6.0.2",
     "lint-staged": "^15.2.5",
     "lodash": "^4.17.21",
+    "lru-cache": "^11.0.1",
     "mime-types": "^2.1.35",
     "nanoid": "^5.0.7",
     "node-fetch": "^3.3.2",

packages/main/src/helper/cache.ts

@@ -0,0 +1,10 @@
+import { LRUCache } from "lru-cache";
+
+const options: LRUCache.OptionsMaxLimit<string, boolean, unknown> = {
+  max: 5000,
+
+  // how long to live in ms
+  ttl: 1000 * 60 * 5,
+};
+
+export const urlCache = new LRUCache(options);

packages/main/src/helper/index.ts

@@ -41,3 +41,4 @@ export { convertToAudio } from "./ffmpeg.ts";
 export { fetchWrapper as fetch };
 export { fileExists } from "./file.ts";
 export { videoPattern } from "./video.ts";
+export { urlCache } from "./cache.ts";

packages/main/src/services/SniffingHelperService.ts

@@ -8,6 +8,7 @@ import {
   PERSIST_WEBVIEW,
   PRIVACY_WEBVIEW,
   formatHeaders,
+  urlCache,
 } from "../helper/index.ts";
 
 export interface SourceParams {
@@ -48,8 +49,7 @@ const filterList: SourceFilter[] = [
 @injectable()
 export class SniffingHelper extends EventEmitter {
   private pageInfo: PageInfo = { title: "", url: "" };
-  private ready = false;
-  private queue: SourceParams[] = [];
+  private readonly prepareDelay = 1000;
 
   constructor(
     @inject(TYPES.ElectronLogger)
@@ -59,38 +59,38 @@ export class SniffingHelper extends EventEmitter {
   }
 
   pluginReady() {
-    this.ready = true;
-    this.queue.forEach((item) => {
-      this.emit("source", item);
-    });
+    // empty
   }
 
   update(pageInfo: PageInfo) {
     this.pageInfo = pageInfo;
   }
 
-  reset(pageInfo: PageInfo) {
-    this.pageInfo = pageInfo;
-    this.ready = false;
-    this.queue = [];
-
-    listLoop: for (const filter of filterList) {
-      if (filter.hosts) {
-        for (const host of filter.hosts) {
-          if (!host.test(pageInfo.url)) {
-            continue;
+  checkPageInfo() {
+    // 发送 page 相关的信息
+    const sendPageInfo = () => {
+      listLoop: for (const filter of filterList) {
+        if (filter.hosts) {
+          for (const host of filter.hosts) {
+            if (!host.test(this.pageInfo.url)) {
+              continue;
+            }
+
+            this.send({
+              url: this.pageInfo.url,
+              documentURL: this.pageInfo.url,
+              name: this.pageInfo.title,
+              type: filter.type,
+            });
+            break listLoop;
           }
-
-          this.send({
-            url: pageInfo.url,
-            documentURL: pageInfo.url,
-            name: pageInfo.title,
-            type: filter.type,
-          });
-          break listLoop;
         }
       }
-    }
+    };
+
+    setTimeout(() => {
+      sendPageInfo();
+    }, this.prepareDelay);
   }
 
   start(privacy: boolean = false) {
@@ -100,13 +100,16 @@ export class SniffingHelper extends EventEmitter {
   }
 
   send = (item: SourceParams) => {
+    const urlCacheKey = `${item.url}_${item.name}`;
+    const cacheUrl = urlCache.get(urlCacheKey);
+    if (cacheUrl) {
+      return;
+    }
+
     this.logger.info(`[SniffingHelper] send: ${item.url}`);
-    // 等待 DOM 中浮窗加载完成
-    // if (this.ready) {
-    // } else {
-    //   this.queue.push(item);
-    // }
     this.emit("source", item);
+
+    urlCache.set(urlCacheKey, true);
   };
 
   private onSendHeaders = (details: OnSendHeadersListenerDetails): void => {

packages/main/src/services/WebviewService.ts

@@ -85,7 +85,7 @@ export default class WebviewService {
   onDomReady = () => {
     if (!this.view) return;
     const pageInfo = this.getPageInfo();
-    this.sniffingHelper.reset(pageInfo);
+    this.sniffingHelper.update(pageInfo);
     this.window.webContents.send("webview-dom-ready", pageInfo);
   };
 
@@ -107,6 +107,8 @@ export default class WebviewService {
     } catch (err) {
       // empty
     }
+
+    this.sniffingHelper.checkPageInfo();
   };
 
   onDidFailLoad = (e: Event, code: number, desc: string) => {