import { Dispatch, SetStateAction } from 'react'
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, FiEye, FiLink2, FiTrash } from 'react-icons/fi'
import { useRouter } from 'next/navigation'

import tableTheme from '@/lib/themes/table'
import { useTable } from '@/lib/hooks/useTable'
import { cn } from '@/lib/utils/tw-merge'
import { formatToMYT } from '@/lib/utils/date'
import { capitalizeWords } from '@/lib/utils/string'
import { AffiliatePartner } from '@/lib/types/partner'

interface DataTableProps {
  affiliatePartners: AffiliatePartner[]
  isLoading?: boolean
  setModal: Dispatch<SetStateAction<ModalState<AffiliatePartner>>>
  setAffiliatePartner: Dispatch<SetStateAction<AffiliatePartner | null>>
}

export default function DataTable({
  affiliatePartners,
  isLoading = false,
  setModal,
  setAffiliatePartner,
}: DataTableProps) {
  const router = useRouter()
  const { selectAllRows, selectedRows, toggleRowSelection, isRowSelected } =
    useTable<AffiliatePartner>()

  function handleDelete(affiliatePartner: AffiliatePartner) {
    setModal({ type: 'delete', isOpened: true })
    setAffiliatePartner(affiliatePartner)
  }

  function handleEdit(affiliatePartner: AffiliatePartner) {
    router.push(`/affiliate-partners/${affiliatePartner.id}/edit`)
  }

  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 === affiliatePartners?.length}
                onChange={() => {
                  selectAllRows(affiliatePartners)
                }}
              />
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">Name</TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">Email</TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Affiliate Level
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Guest Market
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Affiliate Category
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Affiliate Experience
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Referral Code
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Total Referral Links
            </TableHeadCell>
            <TableHeadCell className="text-nowrap p-4">
              Merchant Codes
            </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: 14 }).map((_, cellIdx) => (
                  <TableCell key={cellIdx} className="p-4">
                    <div className="h-4 bg-slate-200 rounded w-full" />
                  </TableCell>
                ))}
              </TableRow>
            ))
          ) : affiliatePartners?.length > 0 ? (
            affiliatePartners.map((affiliatePartner) => (
              <TableRow
                key={affiliatePartner.id}
                className={cn('bg-white', {
                  'bg-blue-50': isRowSelected(affiliatePartner),
                })}
              >
                <TableCell className="text-center p-4">
                  <Checkbox
                    color="blue"
                    checked={isRowSelected(affiliatePartner)}
                    onChange={() => toggleRowSelection(affiliatePartner)}
                  />
                </TableCell>

                {/* Name */}
                <TableCell className="text-nowrap p-4 text-black">
                  {affiliatePartner.affiliate?.name}
                </TableCell>

                {/* Email */}
                <TableCell className="p-4 text-nowrap">
                  {affiliatePartner.affiliate?.email}
                </TableCell>

                {/* Affiliate Level */}
                <TableCell className="p-4 text-nowrap">
                  <Badge color="indigo" size="xs" className="w-fit">
                    {affiliatePartner.affiliate?.affiliate_level?.name}
                  </Badge>
                </TableCell>

                {/* Guest Market */}
                <TableCell className="p-4 text-nowrap">
                  {affiliatePartner.guest_market}
                </TableCell>

                {/* Affiliate Category */}
                <TableCell className="p-4 text-nowrap">
                  <Badge color="green" size="xs" className="w-fit">
                    {affiliatePartner.affiliate?.affiliate_category?.name}
                  </Badge>
                </TableCell>

                {/* Affiliate Experience */}
                <TableCell className="p-4 text-nowrap">
                  {affiliatePartner.affiliate?.affiliate_experience
                    ? capitalizeWords(
                        affiliatePartner.affiliate.affiliate_experience,
                      )
                    : 'N/A'}
                </TableCell>

                {/* Referral Code */}
                <TableCell className="p-4 text-nowrap font-mono">
                  {affiliatePartner.referral_code}
                </TableCell>

                {/* Total Referral Links */}
                <TableCell className="p-4 text-nowrap">
                  <Badge
                    color={
                      affiliatePartner.referral_links.length > 0
                        ? 'blue'
                        : 'gray'
                    }
                    size="xs"
                    className="w-fit px-2"
                    icon={FiLink2}
                  >
                    {affiliatePartner.referral_links.length || 0}{' '}
                    {affiliatePartner.referral_links.length <= 1
                      ? 'Link'
                      : 'Links'}
                  </Badge>
                </TableCell>

                {/* Merchant Codes */}
                <TableCell className="p-4">
                  <div className="flex flex-wrap gap-1">
                    {affiliatePartner.merchant_codes?.length > 0
                      ? affiliatePartner.merchant_codes.map((code) => (
                          <Badge
                            key={code}
                            color="light"
                            size="xs"
                            className="w-fit px-2 mr-1 mb-1"
                          >
                            {code}
                          </Badge>
                        ))
                      : 'N/A'}
                  </div>
                </TableCell>

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

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

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

                {/* Updated By */}
                <TableCell className="p-4 text-nowrap">
                  {affiliatePartner.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={FiTrash}
                      onClick={() => {
                        handleDelete(affiliatePartner)
                      }}
                    >
                      Delete
                    </DropdownItem>
                    <DropdownDivider />
                    <DropdownItem
                      icon={FiEdit}
                      onClick={() => handleEdit(affiliatePartner)}
                    >
                      Edit
                    </DropdownItem>
                    <DropdownDivider />
                    <DropdownItem
                      icon={FiEye}
                      onClick={() =>
                        router.push(
                          `/affiliate-partners/${affiliatePartner.id}`,
                        )
                      }
                    >
                      View Details
                    </DropdownItem>
                  </Dropdown>
                </TableCell>
              </TableRow>
            ))
          ) : (
            <TableRow>
              <TableCell colSpan={15} className="text-center p-8">
                No data available
              </TableCell>
            </TableRow>
          )}
        </TableBody>
      </Table>
    </div>
  )
}