From 4acf7193a42b2b2d4567b3f1de2243181c07c43a Mon Sep 17 00:00:00 2001
From: tbphp <admin@tbphp.net>
Date: Sat, 12 Jul 2025 21:38:12 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E4=BB=BB=E5=8A=A1?=
 =?UTF-8?q?=E7=8A=B6=E6=80=81=E6=A3=80=E6=9F=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 web/src/components/GlobalTaskProgressBar.vue | 10 +++++++++-
 web/src/components/keys/KeyTable.vue         |  8 +++++---
 web/src/utils/app-state.ts                   |  2 ++
 3 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/web/src/components/GlobalTaskProgressBar.vue b/web/src/components/GlobalTaskProgressBar.vue
index 3c8748c..0e0159d 100644
--- a/web/src/components/GlobalTaskProgressBar.vue
+++ b/web/src/components/GlobalTaskProgressBar.vue
@@ -1,8 +1,9 @@
 <script setup lang="ts">
 import { keysApi } from "@/api/keys";
 import type { TaskInfo } from "@/types/models";
+import { appState } from "@/utils/app-state";
 import { NButton, NCard, NProgress, NText, useMessage } from "naive-ui";
-import { onBeforeUnmount, onMounted, ref } from "vue";
+import { onBeforeUnmount, onMounted, ref, watch } from "vue";
 
 const taskInfo = ref<TaskInfo>({ is_running: false });
 const visible = ref(false);
@@ -14,6 +15,13 @@ onMounted(() => {
   startPolling();
 });
 
+watch(
+  () => appState.taskPollingTrigger,
+  () => {
+    startPolling();
+  }
+);
+
 onBeforeUnmount(() => {
   stopPolling();
 });
diff --git a/web/src/components/keys/KeyTable.vue b/web/src/components/keys/KeyTable.vue
index 3a641eb..e16136e 100644
--- a/web/src/components/keys/KeyTable.vue
+++ b/web/src/components/keys/KeyTable.vue
@@ -1,6 +1,7 @@
 <script setup lang="ts">
 import { keysApi } from "@/api/keys";
 import type { APIKey, Group, KeyStatus } from "@/types/models";
+import { appState } from "@/utils/app-state";
 import { getGroupDisplayName, maskKey } from "@/utils/display";
 import {
   AddCircleOutline,
@@ -57,9 +58,9 @@ const statusOptions = [
 
 // 更多操作下拉菜单选项
 const moreOptions = [
-  { label: "导出所有密钥", key: "copyAll" },
-  { label: "导出有效密钥", key: "copyValid" },
-  { label: "导出无效密钥", key: "copyInvalid" },
+  // { label: "导出所有密钥", key: "copyAll" },
+  // { label: "导出有效密钥", key: "copyValid" },
+  // { label: "导出无效密钥", key: "copyInvalid" },
   { type: "divider" },
   { label: "恢复所有无效密钥", key: "restoreAll" },
   { label: "清空所有无效密钥", key: "clearInvalid", props: { style: { color: "#d03050" } } },
@@ -349,6 +350,7 @@ async function validateAllKeys() {
   try {
     await keysApi.validateGroupKeys(props.selectedGroup.id);
     localStorage.removeItem("last_closed_task");
+    appState.taskPollingTrigger++;
   } catch (_error) {
     console.error("测试失败");
   } finally {
diff --git a/web/src/utils/app-state.ts b/web/src/utils/app-state.ts
index e3f8cb2..41748a4 100644
--- a/web/src/utils/app-state.ts
+++ b/web/src/utils/app-state.ts
@@ -2,8 +2,10 @@ import { reactive } from "vue";
 
 interface AppState {
   loading: boolean;
+  taskPollingTrigger: number;
 }
 
 export const appState = reactive<AppState>({
   loading: false,
+  taskPollingTrigger: 0,
 });