From 196603126b24b81d37bb26f8774b0f8546566bf0 Mon Sep 17 00:00:00 2001
From: UndefinedPony <szwabo.dev@gmail.com>
Date: Fri, 20 Dec 2024 17:26:31 +0100
Subject: [PATCH] refactor: move check updates function, use new api

---
 .../settings/web-server/update-server.tsx     | 41 ++++++++++---------
 1 file changed, 21 insertions(+), 20 deletions(-)

diff --git a/apps/dokploy/components/dashboard/settings/web-server/update-server.tsx b/apps/dokploy/components/dashboard/settings/web-server/update-server.tsx
index 8d1ed2e0..14a66749 100644
--- a/apps/dokploy/components/dashboard/settings/web-server/update-server.tsx
+++ b/apps/dokploy/components/dashboard/settings/web-server/update-server.tsx
@@ -20,10 +20,28 @@ export const UpdateServer = () => {
 	const [isUpdateAvailable, setIsUpdateAvailable] = useState<null | boolean>(
 		null,
 	);
-	const { mutateAsync: checkServerUpdates, isLoading } =
-		api.settings.checkServerUpdates.useMutation();
+	const { mutateAsync: checkForUpdate, isLoading } =
+		api.settings.checkForUpdate.useMutation();
 	const [isOpen, setIsOpen] = useState(false);
 
+	const handleCheckUpdates = async () => {
+		try {
+			const updateAvailable = await checkForUpdate();
+			setIsUpdateAvailable(updateAvailable);
+			if (updateAvailable) {
+				toast.success("Update is available!");
+			} else {
+				toast.info("No updates available");
+			}
+		} catch (error) {
+			console.error("Error checking for updates:", error);
+			setIsUpdateAvailable(false);
+			toast.error(
+				"An error occurred while checking for updates, please try again.",
+			);
+		}
+	};
+
 	return (
 		<Dialog open={isOpen} onOpenChange={setIsOpen}>
 			<DialogTrigger asChild>
@@ -76,24 +94,7 @@ export const UpdateServer = () => {
 						) : (
 							<Button
 								className="w-full"
-								onClick={async () => {
-									await checkServerUpdates()
-										.then(async (updateAvailable) => {
-											setIsUpdateAvailable(updateAvailable);
-											toast.info(
-												updateAvailable
-													? "Update is available"
-													: "No updates available",
-											);
-										})
-										.catch((error) => {
-											console.error("Error checking for updates:", error);
-											setIsUpdateAvailable(false);
-											toast.error(
-												"An error occurred while checking for updates, please try again.",
-											);
-										});
-								}}
+								onClick={handleCheckUpdates}
 								isLoading={isLoading}
 							>
 								{isLoading ? "Checking for updates..." : "Check for updates"}