'use client'

import {
  Badge,
  Checkbox,
  Dropdown,
  DropdownDivider,
  DropdownItem,
  Table,
  TableBody,
  TableCell,
  TableHead,
  TableHeadCell,
  TableRow,
} from 'flowbite-react'
import { isSameDay } from 'date-fns'
import { HiOutlineDotsHorizontal } from 'react-icons/hi'
import { FiEdit, FiTrash } from 'react-icons/fi'

import tableTheme from '@/lib/themes/table'
import { useTable } from '@/lib/hooks/useTable'
import { cn } from '@/lib/utils/tw-merge'
import {
  formatToMYT,
  getKualaLumpurDate,
  parseKualaLumpurDate,
} from '@/lib/utils/date'
import { ReferralLink } from '../types'

interface DataTableProps {
  referralLinks: ReferralLink[]
  isLoading?: boolean
  onEdit: (referralLink: ReferralLink) => void
  onDelete: (referralLink: ReferralLink) => void
}

export default function DataTable({
  referralLinks,
  isLoading = false,
  onEdit,
  onDelete,
}: DataTableProps) {
  const { selectAllRows, selectedRows, toggleRowSelection, isRowSelected } =
    useTable<ReferralLink>()

  const columnCount = 8

  return (
    <div className="overflow-x-auto mt-10 min-h-48">
      <Table theme={tableTheme.table} hoverable>
        <TableHead>
          <TableRow>
            <TableHeadCell className="text-center p-4">
              <Checkbox
                color="blue"
                checked={selectedRows.length === referralLinks?.length}
                onChange={() => {
                  selectAllRows(referralLinks)
                }}
              />
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">Link Name</TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">Merchant</TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Total Partners
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Created At
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Updated At
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Created By
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Updated By
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4 sticky right-0 w-20">
              Action
            </TableHeadCell>
          </TableRow>
        </TableHead>

        <TableBody>
          {isLoading ? (
            Array.from({ length: 6 }).map((_, idx) => (
              <TableRow key={`sk-${idx}`} className="bg-white animate-pulse">
                <TableCell className="text-center p-4">
                  <div className="w-4 h-4 bg-slate-200 rounded mx-auto" />
                </TableCell>

                {Array.from({ length: columnCount }).map((_, cellIdx) => (
                  <TableCell key={cellIdx} className="p-4">
                    <div className="h-4 bg-slate-200 rounded w-full" />
                  </TableCell>
                ))}
              </TableRow>
            ))
          ) : referralLinks.length > 0 ? (
            referralLinks.map((referralLink) => (
              <TableRow
                key={referralLink.id}
                className={cn('bg-white', {
                  'bg-blue-50': isRowSelected(referralLink),
                })}
              >
                <TableCell className="text-center p-4">
                  <Checkbox
                    color="blue"
                    checked={isRowSelected(referralLink)}
                    onChange={() => toggleRowSelection(referralLink)}
                  />
                </TableCell>

                {/* Link Name */}
                <TableCell className="text-nowrap p-4 text-black font-medium">
                  {referralLink.link_name}
                </TableCell>

                {/* Merchant */}
                <TableCell className="text-nowrap p-4">
                  {referralLink.user ? referralLink.user.name : 'N/A'}
                </TableCell>

                {/* Total Partners */}
                <TableCell className="text-nowrap p-4">
                  <Badge color="blue" className="w-fit mx-auto rounded-full">
                    {referralLink.partners?.length ?? 0} Partners
                  </Badge>
                </TableCell>

                {/* Created At */}
                <TableCell
                  className={`p-4 text-nowrap ${referralLink.created_at && isSameDay(getKualaLumpurDate(new Date(), true), parseKualaLumpurDate(referralLink.created_at)) ? 'text-green-600 font-medium' : ''}`}
                >
                  {formatToMYT(referralLink.created_at)}
                </TableCell>

                {/* Updated At */}
                <TableCell
                  className={`p-4 text-nowrap ${referralLink.updated_at && isSameDay(getKualaLumpurDate(new Date(), true), parseKualaLumpurDate(referralLink.updated_at)) ? 'text-green-600 font-medium' : ''}`}
                >
                  {formatToMYT(referralLink.updated_at)}
                </TableCell>

                {/* Created By */}
                <TableCell className="p-4 text-nowrap">
                  {referralLink.created_by
                    ? referralLink.created_by.name
                    : 'N/A'}
                </TableCell>

                {/* Updated By */}
                <TableCell className="p-4 text-nowrap">
                  {referralLink.updated_by
                    ? referralLink.updated_by.name
                    : 'N/A'}
                </TableCell>

                <TableCell className="text-nowrap sticky right-0 gap-2 p-4 bg-gray-100 z-2 w-fit ml-auto">
                  <Dropdown
                    label={<HiOutlineDotsHorizontal />}
                    placement="left-start"
                    arrowIcon={false}
                    color="alternative"
                    theme={tableTheme.dropdown}
                  >
                    <DropdownItem
                      icon={FiEdit}
                      onClick={() => onEdit(referralLink)}
                    >
                      Edit
                    </DropdownItem>
                    <DropdownDivider />
                    <DropdownItem
                      icon={FiTrash}
                      onClick={() => onDelete(referralLink)}
                    >
                      Delete
                    </DropdownItem>
                  </Dropdown>
                </TableCell>
              </TableRow>
            ))
          ) : (
            <TableRow>
              <TableCell colSpan={columnCount + 1} className="text-center p-8">
                No data available
              </TableCell>
            </TableRow>
          )}
        </TableBody>
      </Table>
    </div>
  )
}