From 0488442652fdac10f68587f36fd9244351501226 Mon Sep 17 00:00:00 2001
From: Gautam-Hegde <gautamatjob@protonmail.com>
Date: Wed, 13 Mar 2024 13:45:10 +0530
Subject: [PATCH] fix: pagination discrepancy

---
 packages/ui/primitives/data-table-pagination.tsx | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/packages/ui/primitives/data-table-pagination.tsx b/packages/ui/primitives/data-table-pagination.tsx
index feebf6c54..a143960b4 100644
--- a/packages/ui/primitives/data-table-pagination.tsx
+++ b/packages/ui/primitives/data-table-pagination.tsx
@@ -20,6 +20,9 @@ export function DataTablePagination<TData>({
   table,
   additionalInformation = 'VisibleCount',
 }: DataTablePaginationProps<TData>) {
+  const pageCount = table.getPageCount();
+  const isEmptyTable = pageCount === 0;
+
   return (
     <div className="flex flex-wrap items-center justify-between gap-x-4 gap-y-4 px-2">
       <div className="text-muted-foreground flex-1 text-sm">
@@ -65,7 +68,9 @@ export function DataTablePagination<TData>({
       </div>
       <div className="flex flex-wrap items-center gap-x-6 gap-y-4 lg:gap-x-8">
         <div className="flex items-center text-sm font-medium md:justify-center">
-          Page {table.getState().pagination.pageIndex + 1} of {table.getPageCount()}
+          {isEmptyTable
+            ? 'Page 1 of 1'
+            : `Page ${table.getState().pagination.pageIndex + 1} of ${pageCount}`}
         </div>
 
         <div className="flex items-center gap-x-2">
@@ -99,7 +104,7 @@ export function DataTablePagination<TData>({
           <Button
             variant="outline"
             className="hidden h-8 w-8 p-0 lg:flex"
-            onClick={() => table.setPageIndex(table.getPageCount() - 1)}
+            onClick={() => table.setPageIndex(pageCount - 1)}
             disabled={!table.getCanNextPage()}
           >
             <span className="sr-only">Go to last page</span>