refactor: 优化更新的错误处理

EcoPasteHub · Sep 20, 2024 · 900ee84 · 900ee84
1 parent 9414eff
commit 900ee84
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/components/Update/index.tsx b/src/components/Update/index.tsx
@@ -15,7 +15,7 @@ import styles from "./index.module.scss";
 
 interface State {
 	open?: boolean;
-	downloading?: boolean;
+	loading?: boolean;
 	manifest?: UpdateManifest;
 }
 
@@ -90,6 +90,8 @@ const Update = () => {
 				});
 			}
 		} catch (error: any) {
+			state.loading = false;
+
 			if (!showMessage) return;
 
 			messageApi.open({
@@ -118,7 +120,7 @@ const Update = () => {
 	};
 
 	const handleOk = async () => {
-		state.downloading = true;
+		state.loading = true;
 
 		installUpdate();
 
@@ -129,7 +131,7 @@ const Update = () => {
 				case "DONE":
 					return relaunch();
 				case "ERROR":
-					state.downloading = false;
+					state.loading = false;
 
 					return messageApi.open({
 						key: MESSAGE_KEY,
@@ -159,8 +161,8 @@ const Update = () => {
 				okText={t("component.app_update.button.confirm_update")}
 				cancelText={t("component.app_update.button.cancel_update")}
 				className={styles.modal}
-				confirmLoading={state.downloading}
-				cancelButtonProps={{ disabled: state.downloading }}
+				confirmLoading={state.loading}
+				cancelButtonProps={{ disabled: state.loading }}
 				onOk={handleOk}
 				onCancel={handleCancel}
 			>