From 7b887e4cdda53eb127c81d668d4cebc485322351 Mon Sep 17 00:00:00 2001
From: MystiPanda <mystipanda@proton.me>
Date: Thu, 13 Jun 2024 12:58:47 +0800
Subject: [PATCH] fix: check service

---
 src/components/setting/mods/service-viewer.tsx    | 5 ++++-
 src/components/setting/mods/stack-mode-switch.tsx | 7 ++++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/components/setting/mods/service-viewer.tsx b/src/components/setting/mods/service-viewer.tsx
index 137c55e8..5c4e97fa 100644
--- a/src/components/setting/mods/service-viewer.tsx
+++ b/src/components/setting/mods/service-viewer.tsx
@@ -41,8 +41,11 @@ export const ServiceViewer = forwardRef<DialogRef, Props>((props, ref) => {
   const onInstall = useLockFn(async () => {
     try {
       await installService();
-      mutateCheck();
+      await mutateCheck();
       setOpen(false);
+      setTimeout(() => {
+        mutateCheck();
+      }, 2000);
       Notice.success(t("Service Installed Successfully"));
     } catch (err: any) {
       mutateCheck();
diff --git a/src/components/setting/mods/stack-mode-switch.tsx b/src/components/setting/mods/stack-mode-switch.tsx
index d671d339..4bb7604f 100644
--- a/src/components/setting/mods/stack-mode-switch.tsx
+++ b/src/components/setting/mods/stack-mode-switch.tsx
@@ -4,6 +4,7 @@ import { checkService } from "@/services/cmds";
 import { useVerge } from "@/hooks/use-verge";
 import getSystem from "@/utils/get-system";
 import useSWR from "swr";
+import { useEffect } from "react";
 
 const isWIN = getSystem() === "windows";
 
@@ -17,7 +18,7 @@ export const StackModeSwitch = (props: Props) => {
   const { verge } = useVerge();
   const { enable_service_mode } = verge ?? {};
   // service mode
-  const { data: serviceStatus } = useSWR(
+  const { data: serviceStatus, mutate: mutateCheck } = useSWR(
     isWIN ? "checkService" : null,
     checkService,
     {
@@ -28,6 +29,10 @@ export const StackModeSwitch = (props: Props) => {
 
   const { t } = useTranslation();
 
+  useEffect(() => {
+    mutateCheck();
+  }, []);
+
   return (
     <Tooltip
       title={