"use client"
import { z } from "zod"
import { Button } from "@/components/ui/button"
import { Drawer, DrawerClose, DrawerContent, DrawerFooter, DrawerHeader, DrawerTrigger } from "@/components/ui/drawer"
import { DropdownMenu, DropdownMenuContent, DropdownMenuItem, DropdownMenuLabel, DropdownMenuSeparator, DropdownMenuTrigger } from "@/components/ui/dropdown-menu"
import { Input } from "@/components/ui/input"
import { Label } from "@/components/ui/label"
import { useSortable } from "@dnd-kit/sortable"
import { IconGripVertical } from "@tabler/icons-react"
import { useIsMobile } from "@/hooks/use-mobile"
import { Separator } from "@/components/ui/separator"
import { ColumnDef, flexRender, Row } from "@tanstack/react-table"
import { TableCell, TableRow } from "@/components/ui/table"
import { CSS } from "@dnd-kit/utilities"
import { schema, schemaType } from "./schema"
import { dateToLocaleString } from "@/lib/utils"
import { Pencil, Trash } from "lucide-react"

function DragHandle({ id }: { id: number }) {
    const { attributes, listeners } = useSortable({ id })
    return (
        <Button {...attributes} {...listeners} variant="ghost" size="icon" className="text-muted-foreground size-7 hover:bg-transparent">
            <IconGripVertical className="text-muted-foreground size-3" />
            <span className="sr-only">Drag to reorder</span>
        </Button>
    )
}

export function DraggableRow({ row }: { row: Row<z.infer<typeof schema>> }) {
    const { transform, transition, setNodeRef, isDragging } = useSortable({ id: row.original._id })
    return (
        <TableRow data-state={row.getIsSelected() && "selected"} data-dragging={isDragging} ref={setNodeRef}
            className="relative z-0 data-[dragging=true]:z-10 data-[dragging=true]:opacity-80"
            style={{ transform: CSS.Transform.toString(transform), transition: transition }}
        >
            {row.getVisibleCells().map((cell) => (
                <TableCell key={cell.id}>{flexRender(cell.column.columnDef.cell, cell.getContext())}</TableCell>
            ))}
        </TableRow>
    )
}


function getColumns(router: any, deleteHandler: (id: string) => void): ColumnDef<schemaType>[] {
    return [
        {
            accessorKey: "buildType.token",
            header: "Token",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "collectionToken",
            header: "Collection Token",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.govAddressCode",
            header: "Gov Address Code",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.buildName",
            header: "Build Name",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.buildNo",
            header: "Build No",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.maxFloor",
            header: "Max Floor",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.undergroundFloor",
            header: "Underground Floor",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.buildDate",
            header: "Build Date",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.decisionPeriodDate",
            header: "Decision Period Date",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.taxNo",
            header: "Tax No",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.liftCount",
            header: "Lift Count",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.heatingSystem",
            header: "Heating System",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.coolingSystem",
            header: "Cooling System",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.hotWaterSystem",
            header: "Hot Water System",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.blockServiceManCount",
            header: "Block Service Man Count",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.securityServiceManCount",
            header: "Security Service Man Count",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.garageCount",
            header: "Garage Count",
            cell: ({ getValue }) => getValue(),
        },
        {
            accessorKey: "info.managementRoomId",
            header: "Management Room ID",
            cell: ({ getValue }) => getValue(),
        },
        {
            id: "actions",
            header: "Actions",
            cell: ({ row }) => {
                return (
                    <div className="flex flex-row gap-2">
                        <Button className="bg-amber-400 text-black border-amber-400" variant="outline" size="sm" onClick={() => { router.push(`/builds/update?uuid=${row.original._id}`) }}>
                            <Pencil />
                        </Button>
                        <Button className="bg-red-700 text-white border-red-700" variant="outline" size="sm" onClick={() => { deleteHandler(row.original._id || "") }}>
                            <Trash />
                        </Button>
                    </div>
                );
            },
        }
    ]
}

export { getColumns };