122 lines
3.7 KiB
TypeScript
122 lines
3.7 KiB
TypeScript
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 };
|
|
}
|