From d7f033bd4667e958a73b79c0159e7fc939710f9f Mon Sep 17 00:00:00 2001
From: huzibaca <masterorigin91@gmail.com>
Date: Wed, 25 Sep 2024 21:20:36 +0800
Subject: [PATCH] chore: update

---
 src/components/setting/mods/update-viewer.tsx | 6 ++++--
 src/components/test/test-item.tsx             | 2 +-
 src/pages/_layout.tsx                         | 1 -
 src/pages/profiles.tsx                        | 6 +++---
 4 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/src/components/setting/mods/update-viewer.tsx b/src/components/setting/mods/update-viewer.tsx
index d0887e6f..ca600ff6 100644
--- a/src/components/setting/mods/update-viewer.tsx
+++ b/src/components/setting/mods/update-viewer.tsx
@@ -7,10 +7,11 @@ import { relaunch } from "@tauri-apps/plugin-process";
 import { check as checkUpdate } from "@tauri-apps/plugin-updater";
 import { BaseDialog, DialogRef, Notice } from "@/components/base";
 import { useUpdateState, useSetUpdateState } from "@/services/states";
-import { listen, Event, UnlistenFn } from "@tauri-apps/api/event";
+import { Event, UnlistenFn } from "@tauri-apps/api/event";
 import { portableFlag } from "@/pages/_layout";
 import { open as openUrl } from "@tauri-apps/plugin-shell";
 import ReactMarkdown from "react-markdown";
+import { useListen } from "@/hooks/use-listen";
 
 let eventListener: UnlistenFn | null = null;
 
@@ -21,6 +22,7 @@ export const UpdateViewer = forwardRef<DialogRef>((props, ref) => {
 
   const updateState = useUpdateState();
   const setUpdateState = useSetUpdateState();
+  const { addListener } = useListen();
 
   const { data: updateInfo } = useSWR("checkUpdate", checkUpdate, {
     errorRetryCount: 2,
@@ -66,7 +68,7 @@ export const UpdateViewer = forwardRef<DialogRef>((props, ref) => {
     if (eventListener !== null) {
       eventListener();
     }
-    eventListener = await listen(
+    eventListener = await addListener(
       "tauri://update-download-progress",
       (e: Event<any>) => {
         setTotal(e.payload.contentLength);
diff --git a/src/components/test/test-item.tsx b/src/components/test/test-item.tsx
index a63e4c17..dfb75315 100644
--- a/src/components/test/test-item.tsx
+++ b/src/components/test/test-item.tsx
@@ -18,7 +18,7 @@ import { Notice } from "@/components/base";
 import { TestBox } from "./test-box";
 import delayManager from "@/services/delay";
 import { cmdTestDelay, downloadIconCache } from "@/services/cmds";
-import { listen, UnlistenFn } from "@tauri-apps/api/event";
+import { UnlistenFn } from "@tauri-apps/api/event";
 import { convertFileSrc } from "@tauri-apps/api/core";
 import { useListen } from "@/hooks/use-listen";
 interface Props {
diff --git a/src/pages/_layout.tsx b/src/pages/_layout.tsx
index c62855ae..19d4f8c0 100644
--- a/src/pages/_layout.tsx
+++ b/src/pages/_layout.tsx
@@ -6,7 +6,6 @@ import { useEffect } from "react";
 import { useTranslation } from "react-i18next";
 import { useLocation, useRoutes, useNavigate } from "react-router-dom";
 import { List, Paper, ThemeProvider, SvgIcon } from "@mui/material";
-import { listen } from "@tauri-apps/api/event";
 import { getCurrentWebviewWindow } from "@tauri-apps/api/webviewWindow";
 import { routers } from "./_routers";
 import { getAxios } from "@/services/api";
diff --git a/src/pages/profiles.tsx b/src/pages/profiles.tsx
index f67c65a8..32731415 100644
--- a/src/pages/profiles.tsx
+++ b/src/pages/profiles.tsx
@@ -47,15 +47,15 @@ import { useProfiles } from "@/hooks/use-profiles";
 import { ConfigViewer } from "@/components/setting/mods/config-viewer";
 import { throttle } from "lodash-es";
 import { BaseStyledTextField } from "@/components/base/base-styled-text-field";
-import { listen } from "@tauri-apps/api/event";
 import { readTextFile } from "@tauri-apps/plugin-fs";
 import { readText } from "@tauri-apps/plugin-clipboard-manager";
 import { useLocation } from "react-router-dom";
+import { useListen } from "@/hooks/use-listen";
 
 const ProfilePage = () => {
   const { t } = useTranslation();
   const location = useLocation();
-
+  const { addListener } = useListen();
   const [url, setUrl] = useState("");
   const [disabled, setDisabled] = useState(false);
   const [activatings, setActivatings] = useState<string[]>([]);
@@ -69,7 +69,7 @@ const ProfilePage = () => {
   const { current } = location.state || {};
 
   useEffect(() => {
-    const unlisten = listen("tauri://file-drop", async (event) => {
+    const unlisten = addListener("tauri://file-drop", async (event) => {
       const fileList = event.payload as string[];
       for (let file of fileList) {
         if (!file.endsWith(".yaml") && !file.endsWith(".yml")) {