feat: add dokploy server gpu setup

apps/dokploy/components/dashboard/settings/servers/actions/show-dokploy-actions.tsx

@@ -13,6 +13,7 @@ import {
 import { api } from "@/utils/api";
 import { toast } from "sonner";
 import { ShowModalLogs } from "../../web-server/show-modal-logs";
+import { GPUSupportModal } from "../gpu-support-modal";
 
 export const ShowDokployActions = () => {
 	const { mutateAsync: reloadServer, isLoading } =
@@ -45,6 +46,7 @@ export const ShowDokployActions = () => {
 					<ShowModalLogs appName="dokploy">
 						<span>Watch logs</span>
 					</ShowModalLogs>
+					<GPUSupportModal />
 				</DropdownMenuGroup>
 			</DropdownMenuContent>
 		</DropdownMenu>

apps/dokploy/components/dashboard/settings/servers/gpu-support-modal.tsx

@@ -0,0 +1,36 @@
+import {
+	Dialog,
+	DialogContent,
+	DialogHeader,
+	DialogTitle,
+	DialogTrigger,
+} from "@/components/ui/dialog";
+import { DropdownMenuItem } from "@/components/ui/dropdown-menu";
+import { useState } from "react";
+import { GPUSupport } from "./gpu-support";
+
+export const GPUSupportModal = () => {
+	const [isOpen, setIsOpen] = useState(false);
+
+	return (
+		<Dialog open={isOpen} onOpenChange={setIsOpen}>
+			<DialogTrigger asChild>
+				<DropdownMenuItem
+					className="w-full cursor-pointer"
+					onSelect={(e) => e.preventDefault()}
+				>
+					<span>GPU Setup</span>
+				</DropdownMenuItem>
+			</DialogTrigger>
+			<DialogContent className="sm:max-w-4xl overflow-y-auto max-h-screen">
+				<DialogHeader>
+					<DialogTitle className="flex items-center gap-2">
+						Dokploy Server GPU Setup
+					</DialogTitle>
+				</DialogHeader>
+
+				<GPUSupport serverId="" />
+			</DialogContent>
+		</Dialog>
+	);
+};

apps/dokploy/components/dashboard/settings/servers/gpu-support.tsx

@@ -26,7 +26,7 @@ export function GPUSupport({ serverId }: GPUSupportProps) {
 		api.settings.checkGPUStatus.useQuery(
 			{ serverId },
 			{
-				enabled: !!serverId,
+				enabled: serverId !== undefined,
 				refetchInterval: 5000,
 			},
 		);
@@ -38,17 +38,20 @@ export function GPUSupport({ serverId }: GPUSupportProps) {
 		onSuccess: async () => {
 			toast.success("GPU support enabled successfully");
 			setIsLoading(false);
-
-			await Promise.all([
-				utils.settings.checkGPUStatus.invalidate({ serverId }),
-				utils.server.invalidate(),
-			]);
+			await utils.settings.checkGPUStatus.invalidate({ serverId });
 		},
 		onError: (error) => {
 			if (error instanceof TRPCClientError) {
 				const errorMessage = error.message;
-				if (errorMessage.includes("permission denied")) {
-					toast.error("Permission denied. Please ensure proper sudo access.");
+				if (
+					errorMessage.includes(
+						"Permission denied. Please ensure proper sudo access.",
+					) ||
+					errorMessage.includes("sudo access required")
+				) {
+					toast.error(
+						"Administrator privileges required. Please enter your password when prompted.",
+					);
 				} else if (errorMessage.includes("Failed to configure GPU")) {
 					toast.error(
 						"GPU configuration failed. Please check system requirements.",
@@ -59,13 +62,12 @@ export function GPUSupport({ serverId }: GPUSupportProps) {
 			} else {
 				toast.error("Failed to enable GPU support. Please check server logs.");
 			}
-
 			setIsLoading(false);
 		},
 	});
 
 	const handleEnableGPU = async () => {
-		if (!serverId) {
+		if (serverId === undefined) {
 			toast.error("No server selected");
 			return;
 		}
@@ -99,7 +101,7 @@ export function GPUSupport({ serverId }: GPUSupportProps) {
 							>
 								<Button
 									isLoading={isLoading}
-									disabled={isLoading || !serverId || isChecking}
+									disabled={isLoading || serverId === undefined || isChecking}
 								>
 									{isLoading
 										? "Enabling GPU..."
@@ -227,7 +229,7 @@ interface StatusRowProps {
 	showIcon?: boolean;
 }
 
-function StatusRow({
+export function StatusRow({
 	label,
 	isEnabled,
 	description,

apps/dokploy/server/api/routers/settings.ts

@@ -661,33 +661,20 @@ export const settingsRouter = createTRPCRouter({
 	setupGPU: adminProcedure
 		.input(
 			z.object({
-				serverId: z.string(),
+				serverId: z.string().optional(),
 			}),
 		)
 		.mutation(async ({ input }) => {
+			if (IS_CLOUD) {
+				throw new Error("GPU setup is not available in cloud mode");
+			}
+
 			try {
-				if (IS_CLOUD) {
-					return { success: true };
-				}
-
-				if (!input.serverId) {
-					throw new TRPCError({
-						code: "BAD_REQUEST",
-						message: "Server ID is required",
-					});
-				}
-
 				await setupGPUSupport(input.serverId);
 				return { success: true };
 			} catch (error) {
-				throw new TRPCError({
-					code: "INTERNAL_SERVER_ERROR",
-					message:
-						error instanceof Error
-							? error.message
-							: "Failed to enable GPU support",
-					cause: error,
-				});
+				console.error("GPU Setup Error:", error);
+				throw error;
 			}
 		}),
 	checkGPUStatus: adminProcedure
@@ -712,7 +699,15 @@ export const settingsRouter = createTRPCRouter({
 					gpuResources: 0,
 				};
 			}
-			return await checkGPUStatus(input.serverId);
+
+			try {
+				const status = await checkGPUStatus(input.serverId || "");
+				console.log("GPU Status Check Result:", status);
+				return status;
+			} catch (error) {
+				console.error("GPU Status Check Error:", error);
+				throw new Error("Failed to check GPU status");
+			}
 		}),
 });
 // {