refactor: extract service installation logic to custom hook and optimize related component handling

This commit is contained in:
wonfen
2025-05-17 19:09:50 +08:00
parent 7556758284
commit e1d6c74e4f
19 changed files with 129 additions and 126 deletions

View File

@@ -0,0 +1,80 @@
import { useTranslation } from "react-i18next";
import { useLockFn } from "ahooks";
import { showNotice } from "@/services/noticeService";
import { installService, isServiceAvailable, restartCore } from "@/services/cmds";
import { useSystemState } from "@/hooks/use-system-state";
import { mutate } from "swr";
export function useServiceInstaller() {
const { t } = useTranslation();
const { mutateRunningMode } = useSystemState();
const installServiceAndRestartCore = useLockFn(async () => {
try {
showNotice('info', t("Installing Service..."));
await installService();
showNotice('success', t("Service Installed Successfully"));
showNotice('info', t("Waiting for service to be ready..."));
let serviceReady = false;
for (let i = 0; i < 5; i++) {
try {
// 等待1秒再检查
await new Promise(resolve => setTimeout(resolve, 1000));
const isAvailable = await isServiceAvailable();
if (isAvailable) {
serviceReady = true;
mutate("isServiceAvailable", true, false);
break;
}
// 最后一次尝试不显示重试信息
if (i < 4) {
showNotice('info', t("Service not ready, retrying attempt {count}/{total}...", { count: i + 1, total: 5 }));
}
} catch (error) {
console.error(t("Error checking service status:"), error);
if (i < 4) {
showNotice('error', t("Failed to check service status, retrying attempt {count}/{total}...", { count: i + 1, total: 5 }));
}
}
}
if (!serviceReady) {
showNotice('info', t("Service did not become ready after attempts. Proceeding with core restart."));
}
showNotice('info', t("Restarting Core..."));
await restartCore();
// 核心重启后,再次确认并更新相关状态
await mutateRunningMode();
const finalServiceStatus = await isServiceAvailable();
mutate("isServiceAvailable", finalServiceStatus, false);
if (serviceReady && finalServiceStatus) {
showNotice('success', t("Service is ready and core restarted"));
} else if (finalServiceStatus) {
showNotice('success', t("Core restarted. Service is now available."));
} else if (serviceReady) {
showNotice('info', t("Service was ready, but core restart might have issues or service became unavailable. Please check."));
} else {
showNotice('error', t("Service installation or core restart encountered issues. Service might not be available. Please check system logs."));
}
return finalServiceStatus;
} catch (err: any) {
showNotice('error', err.message || err.toString());
// 尝试性回退或最终操作
try {
showNotice('info', t("Attempting to restart core as a fallback..."));
await restartCore();
await mutateRunningMode();
await isServiceAvailable().then(status => mutate("isServiceAvailable", status, false));
} catch (recoveryError: any) {
showNotice('error', t("Fallback core restart also failed: {message}", { message: recoveryError.message }));
}
return false;
}
});
return { installServiceAndRestartCore };
}