From 093f3698fd6f0f407c7084d445016588cc04d0d9 Mon Sep 17 00:00:00 2001
From: Zack Cerza <zack@redhat.com>
Date: Thu, 28 Mar 2024 22:08:29 -0500
Subject: [PATCH] Node: Fix job list pagination

This fixes a regression introuced with the MRT port.

Signed-off-by: Zack Cerza <zack@redhat.com>
---
 src/components/JobList/index.tsx | 55 ++++++++++++++++++++++++++------
 src/pages/Node/index.tsx         | 12 +++++--
 2 files changed, 55 insertions(+), 12 deletions(-)

diff --git a/src/components/JobList/index.tsx b/src/components/JobList/index.tsx
index 8fba7d1..388a6bb 100644
--- a/src/components/JobList/index.tsx
+++ b/src/components/JobList/index.tsx
@@ -2,11 +2,18 @@ import { ReactNode } from "react";
 import DescriptionIcon from "@mui/icons-material/Description";
 import Typography from "@mui/material/Typography";
 import Box from "@mui/material/Box";
+import type {
+  DecodedValueMap,
+  QueryParamConfigMap,
+  SetQuery,
+} from "use-query-params";
 import {
   useMaterialReactTable,
   MaterialReactTable,
   type MRT_ColumnDef,
   type MRT_Row,
+  type MRT_PaginationState,
+  type MRT_Updater,
 } from 'material-react-table';
 import type { UseQueryResult } from "@tanstack/react-query";
 import { type Theme } from "@mui/material/styles";
@@ -21,6 +28,8 @@ import useDefaultTableOptions from "../../lib/table";
 import sentryIcon from "./assets/sentry.svg";
 
 
+const DEFAULT_PAGE_SIZE = 25;
+
 const columns: MRT_ColumnDef<Job>[] = [
   {
     header: "status",
@@ -196,18 +205,53 @@ function JobDetailPanel(props: JobDetailPanelProps): ReactNode {
 };
 
 type JobListProps = {
+  params: DecodedValueMap<QueryParamConfigMap>;
+  setter: SetQuery<QueryParamConfigMap>;
+  manualPagination: Boolean;
   query: UseQueryResult<Run> | UseQueryResult<NodeJobs>;
   sortMode?: "time" | "id";
 }
 
-export default function JobList({ query, sortMode }: JobListProps) {
+export default function JobList({
+  query,
+  params,
+  setter,
+  manualPagination,
+  sortMode,
+}: JobListProps) {
   const options = useDefaultTableOptions<Job>();
   const data = query.data?.jobs || [];
+  const onPaginationChange = (updater: MRT_Updater<MRT_PaginationState>) => {
+    if (!(updater instanceof Function)) return;
+    const newParams = updater({
+      pageSize: params.pageSize,
+      pageIndex: params.page,
+    });
+    setter({
+      page: newParams.pageIndex || 0,
+      pageSize: newParams.pageSize || DEFAULT_PAGE_SIZE,
+    });
+  };
+  options.state = {};
+  if (manualPagination === true) {
+    options.manualPagination = true;
+    options.rowCount = Infinity;
+    options.onPaginationChange = onPaginationChange;
+    options.state.pagination = {
+      pageIndex: params.page || 0,
+      pageSize: params.pageSize || DEFAULT_PAGE_SIZE,
+    };
+  } else {
+    options.initialState = {
+      pagination: { pageIndex: 0, pageSize: DEFAULT_PAGE_SIZE },
+    };
+  }
+  options.state.isLoading = query.isLoading || query.isFetching;
   const table = useMaterialReactTable({
+    rowCount: data.length,
     ...options,
     columns,
     data: data,
-    rowCount: data.length,
     enableFacetedValues: true,
     initialState: {
       ...options.initialState,
@@ -217,10 +261,6 @@ export default function JobList({ query, sortMode }: JobListProps) {
         duration: false,
         waiting: false,
       },
-      pagination: {
-        pageIndex: 0,
-        pageSize: 25,
-      },
       sorting: [
         {
           id: sortMode === "time"? "started" : "job_id",
@@ -228,9 +268,6 @@ export default function JobList({ query, sortMode }: JobListProps) {
         },
       ],
     },
-    state: {
-      isLoading: query.isLoading || query.isFetching,
-    },
     renderDetailPanel: JobDetailPanel,
     muiTableBodyRowProps: ({row, isDetailPanel}) => {
       if ( isDetailPanel ) {
diff --git a/src/pages/Node/index.tsx b/src/pages/Node/index.tsx
index 89fcee9..d90884c 100644
--- a/src/pages/Node/index.tsx
+++ b/src/pages/Node/index.tsx
@@ -10,7 +10,7 @@ import type { NodeParams } from "../../lib/paddles.d";
 import { useNode, useNodeJobs } from "../../lib/paddles";
 
 export default function Node() {
-  const [params, _] = useQueryParams({
+  const [params, setParams] = useQueryParams({
     page: NumberParam,
     pageSize: NumberParam,
   });
@@ -23,7 +23,7 @@ export default function Node() {
 
   if (jobsQuery === null) return <Typography>404</Typography>;
   if (jobsQuery.isError) return null;
-  
+
   return (
     <div>
       <Helmet>
@@ -36,7 +36,13 @@ export default function Node() {
       <div>
         <div>
           <NodeList query={detailsQuery} />
-          <JobList query={jobsQuery} sortMode={"time"} />
+          <JobList
+            query={jobsQuery}
+            params={params}
+            setter={setParams}
+            manualPagination={true}
+            sortMode={"time"}
+          />
         </div>
       </div>
     </div>