refactoring some files.

src/features/dashboard/backup/columns.tsx

@@ -0,0 +1,118 @@
+"use client"
+
+import {ColumnDef} from "@tanstack/react-table"
+import {
+    DropdownMenu,
+    DropdownMenuContent,
+    DropdownMenuItem,
+    DropdownMenuLabel, DropdownMenuSeparator,
+    DropdownMenuTrigger
+} from "@/components/ui/dropdown-menu";
+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 {createRestorationAction} from "@/features/dashboard/restore/restore.action";
+import {toast} from "sonner";
+import {useRouter} from "next/navigation";
+import {StatusBadge} from "@/components/wrappers/common/status-badge";
+import {TooltipCustom} from "@/components/wrappers/common/tooltipCustom/TooltipCustom";
+
+
+export const backupColumns: ColumnDef<Backup>[] = [
+    {
+        accessorKey: "id",
+        header: "Reference",
+    },
+    {
+        accessorKey: "createdAt",
+        header: "Created At",
+        cell: ({row}) => {
+            return new Date(row.getValue("createdAt")).toLocaleString("fr-FR");
+        },
+    },
+    {
+        accessorKey: "status",
+        header: "Status",
+        cell: ({row}) => {
+            return <StatusBadge status={row.getValue("status")}/>
+        },
+    },
+    {
+        id: "actions",
+        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 isRestoring = !!rowData.restorations.find(restoration => restoration.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 restoration = await createRestorationAction({
+                        backupId: rowData.id,
+                        databaseId: rowData.databaseId
+                    })
+                    if (restoration.data.success) {
+                        toast.success(restoration.data.actionSuccess?.message || "Restoration created successfully!");
+                        router.refresh()
+                    } else {
+                        toast.error(restoration.serverError || "Failed to create restoration.");
+                    }
+                }})
+
+            const handleRestore = async () => {
+                await mutationRestore.mutateAsync()
+            }
+
+            return (
+                <DropdownMenu>
+                    <DropdownMenuTrigger asChild>
+                        <Button variant="ghost" className="h-8 w-8 p-0">
+                            <span className="sr-only">Open menu</span>
+                            <MoreHorizontal className="h-4 w-4"/>
+                        </Button>
+                    </DropdownMenuTrigger>
+                    <DropdownMenuContent align="end">
+                        <DropdownMenuLabel>Actions</DropdownMenuLabel>
+                        {status == "success" ?
+                            <>
+                                <TooltipCustom disabled={extendedProps} text="Already a restoration 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/dashboard/organization-cookie.ts

@@ -5,10 +5,10 @@ import {cookies} from 'next/headers';
 
 const COOKIE_NAME = 'PORTABASE_ORGANIZATION_SLUG';
 
-export async function getCurrentOrganizationSlug() {
-    return (await cookies()).get(COOKIE_NAME)?.value || "default";
+export async function getCurrentOrganizationId() {
+    return (await cookies()).get(COOKIE_NAME)?.value || "";
 }
 
-export async function setCurrentOrganizationSlug(slug: string) {
+export async function setCurrentOrganizationId(slug: string) {
     return (await cookies()).set(COOKIE_NAME, slug).get(COOKIE_NAME)?.value;
 }

src/features/dashboard/restore/columns.tsx

@@ -0,0 +1,64 @@
+"use client"
+
+import {ColumnDef} from "@tanstack/react-table"
+import {
+    DropdownMenu,
+    DropdownMenuContent,
+    DropdownMenuItem,
+    DropdownMenuLabel,
+    DropdownMenuTrigger
+} from "@/components/ui/dropdown-menu";
+import {Button} from "@/components/ui/button";
+import {MoreHorizontal} from "lucide-react";
+import {ReloadIcon} from "@radix-ui/react-icons";
+import {Restoration} from "@prisma/client";
+import {StatusBadge} from "@/components/wrappers/common/status-badge";
+
+
+export const restoreColumns: ColumnDef<Restoration>[] = [
+    {
+        accessorKey: "id",
+        header: "Reference",
+    },
+    {
+        accessorKey: "createdAt",
+        header: "Created At",
+        cell: ({row}) => {
+            return new Date(row.getValue("createdAt")).toLocaleString("fr-FR");
+        },
+    },
+    {
+        accessorKey: "status",
+        header: "Status",
+        cell: ({row}) => {
+            return <StatusBadge status={row.getValue("status")}/>
+        },
+    },
+    {
+        id: "actions",
+        cell: ({row}) => {
+            const payment = row.original
+
+            return (
+                <DropdownMenu>
+                    <DropdownMenuTrigger asChild>
+                        <Button variant="ghost" className="h-8 w-8 p-0">
+                            <span className="sr-only">Open menu</span>
+                            <MoreHorizontal className="h-4 w-4"/>
+                        </Button>
+                    </DropdownMenuTrigger>
+                    <DropdownMenuContent align="end">
+                        <DropdownMenuLabel>Actions</DropdownMenuLabel>
+
+                        <DropdownMenuItem onClick={() => {
+                        }}>
+                            <ReloadIcon/> Rerun
+                        </DropdownMenuItem>
+
+
+                    </DropdownMenuContent>
+                </DropdownMenu>
+            )
+        },
+    },
+]

src/features/dashboard/restore/restore.action.ts

@@ -0,0 +1,49 @@
+"use server"
+
+import {userAction} from "@/safe-actions";
+import {z} from "zod";
+import {prisma} from "@/prisma";
+import {ServerActionResult} from "@/types/action-type";
+import {Restoration} from "@prisma/client";
+
+
+export const createRestorationAction = userAction
+    .schema(z.object({
+        backupId: z.string(),
+        databaseId: z.string(),
+    }))
+    .action(async ({parsedInput, ctx}): Promise<ServerActionResult<Restoration>> => {
+
+        try {
+            const restoration = await prisma.restoration.create({
+                data: {
+                    databaseId: parsedInput.databaseId,
+                    backupId: parsedInput.backupId,
+                    status: "waiting",
+                },
+            });
+
+            return {
+                success: true,
+                value: restoration,
+                actionSuccess: {
+                    message: "Restoration has been successfully created.",
+                    messageParams: { restorationId: restoration.id },
+                },
+            };
+        } catch (error) {
+            console.error("Error creating restoration:", 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 restoration" },
+                },
+            };
+        }
+
+
+
+    });