Working on the hidde restore action if already a restauration is waiting.

src/features/backup/columns.tsx

@@ -13,7 +13,12 @@ import {Button} from "@/components/ui/button";
 import {Download, MoreHorizontal, Trash2} from "lucide-react";
 import {ReloadIcon} from "@radix-ui/react-icons";
 import {Backup} from "@prisma/client";
-
+import {getFileUrlPresignedLocal} from "@/features/upload/private/upload.action";
+import {useMutation} from "@tanstack/react-query";
+import {createRestaurationAction} from "@/features/restore/restore.action";
+import {toast} from "sonner";
+import {useRouter} from "next/navigation";
+import {TooltipCustom} from "@/components/wrappers/TooltipCustom/TooltipCustom";
 
 
 export const backupColumns: ColumnDef<Backup>[] = [
@@ -37,7 +42,41 @@ export const backupColumns: ColumnDef<Backup>[] = [
     },
     {
         id: "actions",
-        cell: ({row}) => {
+        cell: ({row, table}) => {
+            const status = row.getValue("status")
+            const rowData: Backup = row.original
+            const fileName = rowData.file
+            // @ts-ignore
+            const {extendedProps} = table.options.meta;
+
+
+            const isRestauring = !!rowData.restaurations.find(restauration => restauration.status === "waiting");
+
+            const router = useRouter()
+
+            const handleDownload = async (fileName: string) => {
+                const url = await getFileUrlPresignedLocal(fileName)
+                window.open(url, '_self');
+            }
+
+
+            const mutationRestore = useMutation({
+                mutationFn: async () => {
+                    const restauration = await createRestaurationAction({
+                        backupId: rowData.id,
+                        databaseId: rowData.databaseId
+                    })
+                    if (restauration.data.success) {
+                        toast.success(restauration.data.actionSuccess?.message || "Restauration created successfully!");
+                        router.refresh()
+                    } else {
+                        toast.error(restauration.serverError || "Failed to create restauration.");
+                    }
+                }})
+
+            const handleRestore = async () => {
+                await mutationRestore.mutateAsync()
+            }
 
             return (
                 <DropdownMenu>
@@ -49,24 +88,28 @@ export const backupColumns: ColumnDef<Backup>[] = [
                     </DropdownMenuTrigger>
                     <DropdownMenuContent align="end">
                         <DropdownMenuLabel>Actions</DropdownMenuLabel>
-
-                        <DropdownMenuItem onClick={() => {
-                        }}>
-                            <ReloadIcon/> Restore
-                        </DropdownMenuItem>
-
-                        <DropdownMenuItem onClick={() => {
-                        }}>
-                            <Download/> Download
-                        </DropdownMenuItem>
-
+                        {status == "success" ?
+                            <>
+                                <TooltipCustom disabled={extendedProps} text="Already a restauration waiting">
+                                    <DropdownMenuItem disabled={mutationRestore.isPending || extendedProps}
+                                                      onClick={async () => {
+                                                          await handleRestore()
+                                                      }}>
+                                        <ReloadIcon/> Restore
+                                    </DropdownMenuItem>
+                                </TooltipCustom>
+                                <DropdownMenuItem onClick={async () => {
+                                    await handleDownload(fileName)
+                                }}>
+                                    <Download/> Download
+                                </DropdownMenuItem>
+                            </>
+                            : null}
                         <DropdownMenuSeparator/>
-
                         <DropdownMenuItem className="text-red-600" onClick={() => {
                         }}>
                             <Trash2/> Delete
                         </DropdownMenuItem>
-
                     </DropdownMenuContent>
                 </DropdownMenu>
             )

src/features/restore/restore.action.ts

@@ -0,0 +1,48 @@
+"use server"
+import {userAction} from "@/safe-actions";
+import {z} from "zod";
+import {prisma} from "@/prisma";
+import {ServerActionResult} from "@/types/action-type";
+import {Restauration} from "@prisma/client";
+
+
+export const createRestaurationAction = userAction
+    .schema(z.object({
+        backupId: z.string(),
+        databaseId: z.string(),
+    }))
+    .action(async ({parsedInput, ctx}): Promise<ServerActionResult<Restauration>> => {
+
+        try {
+            const restauration = await prisma.restauration.create({
+                data: {
+                    databaseId: parsedInput.databaseId,
+                    backupId: parsedInput.backupId,
+                    status: "waiting",
+                },
+            });
+
+            return {
+                success: true,
+                value: restauration,
+                actionSuccess: {
+                    message: "Restauration has been successfully created.",
+                    messageParams: { restaurationId: restauration.id },
+                },
+            };
+        } catch (error) {
+            console.error("Error creating restauration:", error);
+            return {
+                success: false,
+                actionError: {
+                    message: "Failed to create backup.",
+                    status: 500,
+                    cause: error instanceof Error ? error.message : "Unknown error",
+                    messageParams: { message: "Error creating the restauration" },
+                },
+            };
+        }
+
+
+
+    });