import { swrFetcher } from "@/api/api"
import { Checkbox } from "@/components/ui/checkbox"
import { Table, TableBody, TableCell, TableHead, TableHeader, TableRow } from "@/components/ui/table"
import { ColumnDef, flexRender, getCoreRowModel, useReactTable } from "@tanstack/react-table"
import useSWR from "swr"
import { useEffect } from "react"
import { ActionButtonGroup } from "@/components/action-button-group"
import { HeaderButtonGroup } from "@/components/header-button-group"
import { Skeleton } from "@/components/ui/skeleton"
import { toast } from "sonner"
import { ModelAlertRule, triggerModes } from "@/types"
import { deleteAlertRules } from "@/api/alert-rule"
import { NotificationTab } from "@/components/notification-tab"
import { AlertRuleCard } from "@/components/alert-rule"

export default function AlertRulePage() {
    const { data, mutate, error, isLoading } = useSWR<ModelAlertRule[]>("/api/v1/alert-rule", swrFetcher);

    useEffect(() => {
        if (error)
            toast("Error", {
                description: `Error fetching resource: ${error.message}.`,
            })
    }, [error])

    const columns: ColumnDef<ModelAlertRule>[] = [
        {
            id: "select",
            header: ({ table }) => (
                <Checkbox
                    checked={
                        table.getIsAllPageRowsSelected() ||
                        (table.getIsSomePageRowsSelected() && "indeterminate")
                    }
                    onCheckedChange={(value) => table.toggleAllPageRowsSelected(!!value)}
                    aria-label="Select all"
                />
            ),
            cell: ({ row }) => (
                <Checkbox
                    checked={row.getIsSelected()}
                    onCheckedChange={(value) => row.toggleSelected(!!value)}
                    aria-label="Select row"
                />
            ),
            enableSorting: false,
            enableHiding: false,
        },
        {
            header: "ID",
            accessorKey: "id",
            accessorFn: row => row.id,
        },
        {
            header: "Name",
            accessorKey: "name",
            accessorFn: row => row.name,
            cell: ({ row }) => {
                const s = row.original;
                return (
                    <div className="max-w-32 whitespace-normal break-words">
                        {s.name}
                    </div>
                )
            }
        },
        {
            header: "Notifier Group",
            accessorKey: "ngroup",
            accessorFn: row => row.notification_group_id,
        },
        {
            header: "Trigger Mode",
            accessorKey: "trigger Mode",
            accessorFn: row => triggerModes[row.trigger_mode] || '',
        },
        {
            header: "Rules",
            accessorKey: "rules",
            accessorFn: row => JSON.stringify(row.rules),
        },
        {
            header: "Tasks to trigger on alert",
            accessorKey: "failTriggerTasks",
            accessorFn: row => row.fail_trigger_tasks,
        },
        {
            header: "Tasks to trigger after recovery",
            accessorKey: "recoverTriggerTasks",
            accessorFn: row => row.recover_trigger_tasks,
        },
        {
            header: "Enable",
            accessorKey: "enable",
            accessorFn: row => row.enable,
        },
        {
            id: "actions",
            header: "Actions",
            cell: ({ row }) => {
                const s = row.original
                return (
                    <ActionButtonGroup className="flex gap-2" delete={{
                        fn: deleteAlertRules,
                        id: s.id,
                        mutate: mutate,
                    }}>
                        <AlertRuleCard mutate={mutate} data={s} />
                    </ActionButtonGroup>
                )
            },
        },
    ]

    const table = useReactTable({
        data: data ?? [],
        columns,
        getCoreRowModel: getCoreRowModel(),
    })

    const selectedRows = table.getSelectedRowModel().rows;

    return (
        <div className="px-8">
            <div className="flex mt-6 mb-4">
                <NotificationTab className="flex-1 mr-4 sm:max-w-[40%]" />
                <HeaderButtonGroup className="flex-2 flex gap-2 ml-auto" delete={{
                    fn: deleteAlertRules,
                    id: selectedRows.map(r => r.original.id),
                    mutate: mutate,
                }}>
                    <AlertRuleCard mutate={mutate} />
                </HeaderButtonGroup>
            </div>
            {isLoading ? (
                <div className="flex flex-col items-center space-y-4">
                    <Skeleton className="h-[60px] w-[100%] rounded-lg" />
                    <Skeleton className="h-[60px] w-[100%] rounded-lg" />
                    <Skeleton className="h-[60px] w-[100%] rounded-lg" />
                </div>
            ) : (
                <Table>
                    <TableHeader>
                        {table.getHeaderGroups().map((headerGroup) => (
                            <TableRow key={headerGroup.id}>
                                {headerGroup.headers.map((header) => {
                                    return (
                                        <TableHead key={header.id} className="text-sm">
                                            {header.isPlaceholder
                                                ? null
                                                : flexRender(
                                                    header.column.columnDef.header,
                                                    header.getContext()
                                                )}
                                        </TableHead>
                                    )
                                })}
                            </TableRow>
                        ))}
                    </TableHeader>
                    <TableBody>
                        {table.getRowModel().rows?.length ? (
                            table.getRowModel().rows.map((row) => (
                                <TableRow
                                    key={row.id}
                                    data-state={row.getIsSelected() && "selected"}
                                >
                                    {row.getVisibleCells().map((cell) => (
                                        <TableCell key={cell.id} className="text-xsm">
                                            {flexRender(cell.column.columnDef.cell, cell.getContext())}
                                        </TableCell>
                                    ))}
                                </TableRow>
                            ))
                        ) : (
                            <TableRow>
                                <TableCell colSpan={columns.length} className="h-24 text-center">
                                    No results.
                                </TableCell>
                            </TableRow>
                        )}
                    </TableBody>
                </Table>
            )}
        </div>
    )
}