From 95afadbec5be46ffd963585594bb15e2e6ff1f57 Mon Sep 17 00:00:00 2001
From: johnathan <952508490@qq.com>
Date: Thu, 26 Jun 2025 16:19:40 +0800
Subject: [PATCH] Refactor: Make page worker as a top module

---
 src/components/AmazonDetailDescription.vue    |   5 +-
 src/composables/useCloudCommit.ts             |   0
 src/logic/page-worker/index.ts                |   4 -
 src/options/views/AmazonResultTable.vue       |   3 +-
 src/options/views/HomedepotResultTable.vue    |   2 +-
 src/{logic => }/page-worker/amazon.ts         |   2 +-
 src/{logic => }/page-worker/base.ts           |   0
 src/{logic => page-worker}/error-handler.ts   |   0
 src/{logic => }/page-worker/homedepot.ts      |   4 +-
 .../usePageWorker.ts => page-worker/index.ts} | 122 ++++++++++++++++--
 src/{logic => }/page-worker/types.ts          |   0
 .../views/AmazonEntries/DetailPageEntry.vue   |   2 +-
 .../views/AmazonEntries/ReviewPageEntry.vue   |   2 +-
 .../views/AmazonEntries/SearchPageEntry.vue   |   2 +-
 src/sidepanel/views/AmazonSidepanel.vue       |   2 +-
 src/sidepanel/views/HomedepotSidepanel.vue    |  77 ++++++-----
 16 files changed, 172 insertions(+), 55 deletions(-)
 create mode 100644 src/composables/useCloudCommit.ts
 delete mode 100644 src/logic/page-worker/index.ts
 rename src/{logic => }/page-worker/amazon.ts (99%)
 rename src/{logic => }/page-worker/base.ts (100%)
 rename src/{logic => page-worker}/error-handler.ts (100%)
 rename src/{logic => }/page-worker/homedepot.ts (96%)
 rename src/{composables/usePageWorker.ts => page-worker/index.ts} (60%)
 rename src/{logic => }/page-worker/types.ts (100%)

diff --git a/src/components/AmazonDetailDescription.vue b/src/components/AmazonDetailDescription.vue
index e832f4b..a79893a 100644
--- a/src/components/AmazonDetailDescription.vue
+++ b/src/components/AmazonDetailDescription.vue
@@ -5,9 +5,12 @@ defineProps<{ model: AmazonDetailItem }>();
 <template>
   <div class="detail-description">
     <n-descriptions label-placement="left" bordered :column="4" label-style="min-width: 100px">
-      <n-descriptions-item label="ASIN" :span="2">
+      <n-descriptions-item label="ASIN">
         {{ model.asin }}
       </n-descriptions-item>
+      <n-descriptions-item label="获取日期">
+        {{ model.timestamp }}
+      </n-descriptions-item>
       <n-descriptions-item label="评价">
         {{ model.rating || '-' }}
       </n-descriptions-item>
diff --git a/src/composables/useCloudCommit.ts b/src/composables/useCloudCommit.ts
new file mode 100644
index 0000000..e69de29
diff --git a/src/logic/page-worker/index.ts b/src/logic/page-worker/index.ts
deleted file mode 100644
index 579e91a..0000000
--- a/src/logic/page-worker/index.ts
+++ /dev/null
@@ -1,4 +0,0 @@
-import amazon from './amazon';
-import homedepot from './homedepot';
-
-export { amazon, homedepot };
diff --git a/src/options/views/AmazonResultTable.vue b/src/options/views/AmazonResultTable.vue
index 579c411..cdd6b4e 100644
--- a/src/options/views/AmazonResultTable.vue
+++ b/src/options/views/AmazonResultTable.vue
@@ -84,7 +84,7 @@ const columns: TableColumn[] = [
     hidden: true,
   },
   {
-    title: '创建日期',
+    title: '获取日期',
     key: 'createTime',
     minWidth: 160,
   },
@@ -146,6 +146,7 @@ const extraHeaders: Header<AmazonItem>[] = [
   { prop: 'category1.rank', label: '大类排行' },
   { prop: 'category2.name', label: '小类' },
   { prop: 'category2.rank', label: '小类排行' },
+  { prop: 'timestamp', label: '获取日期（详情页）' },
   {
     prop: 'imageUrls',
     label: '商品图片链接',
diff --git a/src/options/views/HomedepotResultTable.vue b/src/options/views/HomedepotResultTable.vue
index b0179d5..62784dd 100644
--- a/src/options/views/HomedepotResultTable.vue
+++ b/src/options/views/HomedepotResultTable.vue
@@ -50,7 +50,7 @@ const columns: TableColumn[] = [
   },
   {
     title: '操作',
-    key: 'action',
+    key: 'actions',
     render(row: (typeof allItems.value)[0]) {
       return (
         <n-space>
diff --git a/src/logic/page-worker/amazon.ts b/src/page-worker/amazon.ts
similarity index 99%
rename from src/logic/page-worker/amazon.ts
rename to src/page-worker/amazon.ts
index 84ad3f0..68c2407 100644
--- a/src/logic/page-worker/amazon.ts
+++ b/src/page-worker/amazon.ts
@@ -1,6 +1,6 @@
 import type { AmazonPageWorker, AmazonPageWorkerEvents, LanchTaskBaseOptions } from './types';
 import type { Tabs } from 'webextension-polyfill';
-import { withErrorHandling } from '../error-handler';
+import { withErrorHandling } from './error-handler';
 import {
   AmazonDetailPageInjector,
   AmazonReviewPageInjector,
diff --git a/src/logic/page-worker/base.ts b/src/page-worker/base.ts
similarity index 100%
rename from src/logic/page-worker/base.ts
rename to src/page-worker/base.ts
diff --git a/src/logic/error-handler.ts b/src/page-worker/error-handler.ts
similarity index 100%
rename from src/logic/error-handler.ts
rename to src/page-worker/error-handler.ts
diff --git a/src/logic/page-worker/homedepot.ts b/src/page-worker/homedepot.ts
similarity index 96%
rename from src/logic/page-worker/homedepot.ts
rename to src/page-worker/homedepot.ts
index c267acc..7f8a334 100644
--- a/src/logic/page-worker/homedepot.ts
+++ b/src/page-worker/homedepot.ts
@@ -1,6 +1,6 @@
 import type { HomedepotEvents, HomedepotWorker, LanchTaskBaseOptions } from './types';
 import { Tabs } from 'webextension-polyfill';
-import { withErrorHandling } from '../error-handler';
+import { withErrorHandling } from './error-handler';
 import { HomedepotDetailPageInjector } from '~/logic/web-injectors/homedepot';
 import { BaseWorker } from './base';
 
@@ -58,7 +58,7 @@ class HomedepotWorkerImpl
 }
 
 export default {
-  useHomedepotWorker() {
+  getHomedepotWorker() {
     return HomedepotWorkerImpl.getInstance();
   },
 };
diff --git a/src/composables/usePageWorker.ts b/src/page-worker/index.ts
similarity index 60%
rename from src/composables/usePageWorker.ts
rename to src/page-worker/index.ts
index 6f0f3c1..670e4dd 100644
--- a/src/composables/usePageWorker.ts
+++ b/src/page-worker/index.ts
@@ -1,6 +1,7 @@
-import { amazon } from '~/logic/page-worker';
+import amazon from './amazon';
+import homedepot from './homedepot';
 import { uploadImage } from '~/logic/upload';
-import { useLongTask } from './useLongTask';
+import { useLongTask } from '~/composables/useLongTask';
 
 export interface AmazonPageWorkerSettings {
   searchItems?: Ref<AmazonSearchItem[]>;
@@ -9,7 +10,7 @@ export interface AmazonPageWorkerSettings {
   commitChangeIngerval?: number;
 }
 
-class PageWorkerFactory {
+class AmazonPageWorkerFactory {
   public amazonWorker: ReturnType<typeof this.buildAmazonPageWorker> | null = null;
 
   public amazonWorkerSettings: AmazonPageWorkerSettings = {};
@@ -160,11 +161,112 @@ class PageWorkerFactory {
   }
 }
 
-const facotry = new PageWorkerFactory();
-
-export function usePageWorker(type: 'amazon', settings?: AmazonPageWorkerSettings) {
-  if (type === 'amazon') {
-    return facotry.loadAmazonPageWorker(settings);
-  }
-  throw new Error(`Unsupported page worker type: ${type}`);
+export interface HomedepotWorkerSettings {
+  detailItems?: Ref<Map<string, HomedepotDetailItem>>;
+  commitChangeIngerval?: number;
+}
+
+class HomedepotWorkerFactory {
+  public homedepotWorkerSettings: HomedepotWorkerSettings = {};
+
+  public homedepotWorker: ReturnType<typeof this.buildHomedepotWorker> | null = null;
+
+  public buildHomedepotWorker() {
+    const worker = homedepot.getHomedepotWorker();
+    const { isRunning, startTask } = useLongTask();
+
+    const detailCache = new Map<string, HomedepotDetailItem>();
+
+    const unsubscribeFuncs = [] as (() => void)[];
+
+    onMounted(() => {
+      unsubscribeFuncs.push(
+        ...[
+          worker.on('error', () => {
+            worker.stop();
+          }),
+          worker.on('detail-item-collected', (ev) => {
+            const { item } = ev;
+            if (detailCache.has(item.OSMID)) {
+              const origin = detailCache.get(item.OSMID);
+              detailCache.set(item.OSMID, { ...origin, ...item });
+            } else {
+              detailCache.set(item.OSMID, item);
+            }
+          }),
+        ],
+      );
+    });
+
+    onUnmounted(() => {
+      unsubscribeFuncs.forEach((unsubscribe) => unsubscribe());
+      unsubscribeFuncs.splice(0, unsubscribeFuncs.length);
+    });
+
+    const commitChange = () => {
+      const { detailItems } = this.homedepotWorkerSettings;
+      if (typeof detailItems !== 'undefined') {
+        for (const [k, v] of detailCache.entries()) {
+          detailItems.value.delete(k); // Trigger update
+          detailItems.value.set(k, v);
+        }
+        detailCache.clear();
+      }
+    };
+
+    const taskWrapper = <T extends (...params: any) => any>(func: T) => {
+      const { commitChangeIngerval = 1500 } = this.homedepotWorkerSettings;
+      return (...params: Parameters<T>) =>
+        startTask(async () => {
+          const interval = setInterval(() => commitChange(), commitChangeIngerval);
+          await func(...params);
+          clearInterval(interval);
+          commitChange();
+        });
+    };
+
+    const runDetailPageTask = taskWrapper(worker.runDetailPageTask.bind(worker));
+
+    return {
+      isRunning,
+      runDetailPageTask,
+      on: worker.on.bind(worker),
+      off: worker.off.bind(worker),
+      once: worker.once.bind(worker),
+      stop: worker.stop.bind(worker),
+    };
+  }
+
+  loadHomedepotWorker(settings?: HomedepotWorkerSettings) {
+    if (settings) {
+      this.homedepotWorkerSettings = { ...this.homedepotWorkerSettings, ...settings };
+    }
+    if (!this.homedepotWorker) {
+      this.homedepotWorker = this.buildHomedepotWorker();
+    }
+    return this.homedepotWorker;
+  }
+}
+
+const amazonfacotry = new AmazonPageWorkerFactory();
+
+const homedepotfactory = new HomedepotWorkerFactory();
+
+export function usePageWorker(
+  type: 'amazon',
+  settings?: AmazonPageWorkerSettings,
+): ReturnType<typeof amazonfacotry.loadAmazonPageWorker>;
+export function usePageWorker(
+  type: 'homedepot',
+  settings?: HomedepotWorkerSettings,
+): ReturnType<typeof homedepotfactory.loadHomedepotWorker>;
+export function usePageWorker(type: 'amazon' | 'homedepot', settings?: any) {
+  switch (type) {
+    case 'amazon':
+      return amazonfacotry.loadAmazonPageWorker(settings);
+    case 'homedepot':
+      return homedepotfactory.loadHomedepotWorker(settings);
+    default:
+      throw new Error(`Unsupported page worker type: ${type}`);
+  }
 }
diff --git a/src/logic/page-worker/types.ts b/src/page-worker/types.ts
similarity index 100%
rename from src/logic/page-worker/types.ts
rename to src/page-worker/types.ts
diff --git a/src/sidepanel/views/AmazonEntries/DetailPageEntry.vue b/src/sidepanel/views/AmazonEntries/DetailPageEntry.vue
index dc86cdc..43aa14a 100644
--- a/src/sidepanel/views/AmazonEntries/DetailPageEntry.vue
+++ b/src/sidepanel/views/AmazonEntries/DetailPageEntry.vue
@@ -1,6 +1,6 @@
 <script setup lang="ts">
 import type { Timeline } from '~/components/ProgressReport.vue';
-import { usePageWorker } from '~/composables/usePageWorker';
+import { usePageWorker } from '~/page-worker';
 import { detailAsinInput, detailItems, detailWorkerSettings } from '~/logic/storages/amazon';
 
 const message = useMessage();
diff --git a/src/sidepanel/views/AmazonEntries/ReviewPageEntry.vue b/src/sidepanel/views/AmazonEntries/ReviewPageEntry.vue
index cc40760..e1cb26d 100644
--- a/src/sidepanel/views/AmazonEntries/ReviewPageEntry.vue
+++ b/src/sidepanel/views/AmazonEntries/ReviewPageEntry.vue
@@ -1,6 +1,6 @@
 <script lang="ts" setup>
 import type { Timeline } from '~/components/ProgressReport.vue';
-import { usePageWorker } from '~/composables/usePageWorker';
+import { usePageWorker } from '~/page-worker';
 import { reviewAsinInput, reviewItems } from '~/logic/storages/amazon';
 
 const worker = usePageWorker('amazon', { reviewItems });
diff --git a/src/sidepanel/views/AmazonEntries/SearchPageEntry.vue b/src/sidepanel/views/AmazonEntries/SearchPageEntry.vue
index 6b53905..dda27f8 100644
--- a/src/sidepanel/views/AmazonEntries/SearchPageEntry.vue
+++ b/src/sidepanel/views/AmazonEntries/SearchPageEntry.vue
@@ -2,7 +2,7 @@
 import { keywordsList } from '~/logic/storages/amazon';
 import { searchItems } from '~/logic/storages/amazon';
 import type { Timeline } from '~/components/ProgressReport.vue';
-import { usePageWorker } from '~/composables/usePageWorker';
+import { usePageWorker } from '~/page-worker';
 
 const message = useMessage();
 
diff --git a/src/sidepanel/views/AmazonSidepanel.vue b/src/sidepanel/views/AmazonSidepanel.vue
index 4cc7beb..1ffac2e 100644
--- a/src/sidepanel/views/AmazonSidepanel.vue
+++ b/src/sidepanel/views/AmazonSidepanel.vue
@@ -2,7 +2,7 @@
 import DetailPageEntry from './AmazonEntries/DetailPageEntry.vue';
 import SearchPageEntry from './AmazonEntries/SearchPageEntry.vue';
 import ReviewPageEntry from './AmazonEntries/ReviewPageEntry.vue';
-import { usePageWorker } from '~/composables/usePageWorker';
+import { usePageWorker } from '~/page-worker';
 
 const tabs = [
   {
diff --git a/src/sidepanel/views/HomedepotSidepanel.vue b/src/sidepanel/views/HomedepotSidepanel.vue
index 14cf7c7..82abc1e 100644
--- a/src/sidepanel/views/HomedepotSidepanel.vue
+++ b/src/sidepanel/views/HomedepotSidepanel.vue
@@ -1,48 +1,57 @@
 <script setup lang="ts">
 import type { Timeline } from '~/components/ProgressReport.vue';
-import { useLongTask } from '~/composables/useLongTask';
-import { homedepot } from '~/logic/page-worker';
+import { usePageWorker } from '~/page-worker';
 import { detailItems } from '~/logic/storages/homedepot';
 
 const inputText = ref('');
-const { isRunning, startTask } = useLongTask();
+const idInputRef = useTemplateRef('id-input');
 
-const worker = homedepot.useHomedepotWorker();
-worker.channel.on('detail-item-collected', ({ item }) => {
+const worker = usePageWorker('homedepot', { detailItems });
+worker.on('detail-item-collected', ({ item }) => {
   timelines.value.push({
     type: 'success',
     title: `成功`,
     content: `成功获取到${item.OSMID}的商品信息`,
     time: new Date().toLocaleString(),
   });
-  detailItems.value.set(item.OSMID, item);
 });
 
 const timelines = ref<Timeline[]>([]);
 
-const handleStart = () =>
-  startTask(async () => {
-    timelines.value.push({
-      type: 'info',
-      title: `开始`,
-      content: '任务开始',
-      time: new Date().toLocaleString(),
-    });
-    await worker.runDetailPageTask(
-      inputText.value.split('\n').filter((id) => /\d+/.exec(id)),
-      {
-        progress: (remains) => {
-          inputText.value = remains.join('\n');
+const handleStart = async () => {
+  idInputRef.value?.validate().then(async (success) => {
+    if (success) {
+      const ids = inputText.value.split(/\n|\s|,|;/).filter((item) => item.length > 0);
+      timelines.value = [
+        {
+          type: 'info',
+          title: '开始',
+          time: new Date().toLocaleString(),
+          content: `开始采集OSMID: ${ids.join(', ')}`,
         },
-      },
-    );
-    timelines.value.push({
-      type: 'info',
-      title: `结束`,
-      content: '任务完成',
-      time: new Date().toLocaleString(),
-    });
+      ];
+      await worker.runDetailPageTask(ids, {
+        progress: (remains) => {
+          if (remains.length > 0) {
+            timelines.value.push({
+              type: 'info',
+              title: '继续',
+              time: new Date().toLocaleString(),
+              content: `继续采集OSMID: ${remains.join(', ')}`,
+            });
+            inputText.value = remains.join('\n');
+          }
+        },
+      });
+      timelines.value.push({
+        type: 'info',
+        title: '结束',
+        time: new Date().toLocaleString(),
+        content: `数据采集完成`,
+      });
+    }
   });
+};
 
 const handleInterrupt = () => {
   worker.stop();
@@ -55,13 +64,19 @@ const handleInterrupt = () => {
     <div class="interative-section">
       <ids-input
         v-model="inputText"
-        :disabled="isRunning"
-        ref="asin-input"
+        :disabled="worker.isRunning.value"
+        ref="id-input"
         :match-pattern="/^\d+(\n\d+)*\n?$/g"
         placeholder="输入OSMID"
         validate-message="请输入格式正确的OSMID"
       />
-      <n-button v-if="!isRunning" round size="large" type="primary" @click="handleStart">
+      <n-button
+        v-if="!worker.isRunning.value"
+        round
+        size="large"
+        type="primary"
+        @click="handleStart"
+      >
         <template #icon>
           <ant-design-thunderbolt-outlined />
         </template>
@@ -74,7 +89,7 @@ const handleInterrupt = () => {
         停止
       </n-button>
     </div>
-    <div v-if="isRunning" class="running-tip-section">
+    <div v-if="worker.isRunning.value" class="running-tip-section">
       <n-alert title="Warning" type="warning"> 警告，在插件运行期间请勿与浏览器交互。 </n-alert>
     </div>
     <progress-report class="progress-report" :timelines="timelines" />