import {
  ChevronLeft,
  ChevronRight,
  ChevronsLeft,
  ChevronsRight,
} from "lucide-react";

import { Button } from "@/components/ui/button";
import {
  Select,
  SelectContent,
  SelectItem,
  SelectTrigger,
  SelectValue,
} from "@/components/ui/select";

interface DataTablePaginationProps {
  pageCount: number;
  pageIndex: number;
  pageSize: number;
  enableSelect?: boolean;
  pathname: string;
  searchParams: URLSearchParams;
  replace: (url: string) => void;
}

function PaginationIndicator({
  pageIndex,
  pageCount,
}: {
  pageIndex: number;
  pageCount: number;
}) {
  return (
    <div className="flex w-[100px] items-center justify-center text-sm font-medium">
      Page {pageIndex} of {pageCount}
    </div>
  );
}

export function DataTablePagination({
  pageCount,
  pageIndex,
  pageSize,
  enableSelect = false,
  pathname,
  searchParams,
  replace,
}: DataTablePaginationProps) {
  const handlePageChange = (page: number) => {
    const params = new URLSearchParams(searchParams);
    params.set("page", page.toString());
    replace(`${pathname}?${params.toString()}`);
  };

  const handlePageSizeChange = (size: number) => {
    const params = new URLSearchParams(searchParams);
    params.set("size", size.toString());
    params.set("page", "1");
    replace(`${pathname}?${params.toString()}`);
  };

  return (
    <div
      className={`flex items-center ${
        enableSelect ? "justify-between" : "justify-center"
      } px-2 py-4`}
    >
      <div className="flex flex-col lg:flex-row items-center gap-2 lg:space-x-8">
        <div className="flex items-center space-x-2">
          <p className="text-sm font-medium">Rows per page</p>
          <Select
            value={`${pageSize}`}
            onValueChange={(value) => handlePageSizeChange(Number(value))}
          >
            <SelectTrigger className="h-8 w-[70px]">
              <SelectValue placeholder={pageSize} />
            </SelectTrigger>
            <SelectContent side="top">
              {[10, 20, 30, 40, 50].map((size) => (
                <SelectItem key={size} value={`${size}`}>
                  {size}
                </SelectItem>
              ))}
            </SelectContent>
          </Select>
        </div>
        <div className="hidden lg:block">
          <PaginationIndicator pageIndex={pageIndex} pageCount={pageCount} />
        </div>
        <div className="flex items-center space-x-2">
          <Button
            variant="outline"
            className="h-8 w-8 p-0"
            onClick={() => handlePageChange(1)}
            disabled={pageIndex === 1}
          >
            <ChevronsLeft className="h-4 w-4" />
          </Button>
          <Button
            variant="outline"
            className="h-8 w-8 p-0"
            onClick={() => handlePageChange(pageIndex - 1)}
            disabled={pageIndex === 1}
          >
            <ChevronLeft className="h-4 w-4" />
          </Button>
          <div className="block lg:hidden">
            <PaginationIndicator pageIndex={pageIndex} pageCount={pageCount} />
          </div>
          <Button
            variant="outline"
            className="h-8 w-8 p-0"
            onClick={() => handlePageChange(pageIndex + 1)}
            disabled={pageIndex === pageCount}
          >
            <ChevronRight className="h-4 w-4" />
          </Button>
          <Button
            variant="outline"
            className="h-8 w-8 p-0"
            onClick={() => handlePageChange(pageCount)}
            disabled={pageIndex === pageCount}
          >
            <ChevronsRight className="h-4 w-4" />
          </Button>
        </div>
      </div>
    </div>
  );
}