From b9d3beaa7f2fd624deaee28990e7399931438beb Mon Sep 17 00:00:00 2001
From: Keith Grant <keithjgrant@redhat.com>
Date: Mon, 23 Nov 2020 16:32:16 -0800
Subject: [PATCH] add HeaderRow component with table sorting

---
 .../components/PaginatedTable/HeaderRow.jsx   | 73 +++++++++++++++++++
 .../src/components/PaginatedTable/index.js    |  1 +
 .../OrganizationList/OrganizationList.jsx     | 64 ++++++++++++++--
 3 files changed, 133 insertions(+), 5 deletions(-)
 create mode 100644 awx/ui_next/src/components/PaginatedTable/HeaderRow.jsx

diff --git a/awx/ui_next/src/components/PaginatedTable/HeaderRow.jsx b/awx/ui_next/src/components/PaginatedTable/HeaderRow.jsx
new file mode 100644
index 0000000000..8fd0fef003
--- /dev/null
+++ b/awx/ui_next/src/components/PaginatedTable/HeaderRow.jsx
@@ -0,0 +1,73 @@
+import React from 'react';
+import { useLocation, useHistory } from 'react-router-dom';
+import { Thead, Tr, Th } from '@patternfly/react-table';
+import {
+  encodeNonDefaultQueryString,
+  parseQueryString,
+  replaceParams,
+} from '../../util/qs';
+
+export default function HeaderRow({
+  handleSelectAll,
+  isAllSelected,
+  qsConfig,
+  defaultSortKey,
+  children,
+}) {
+  const location = useLocation();
+  const history = useHistory();
+
+  const params = parseQueryString(qsConfig, location.search);
+
+  // TODO: asc vs desc -- correct for both alpha & numeric sorting?
+  const onSort = (key, order) => {
+    console.log({ key, order });
+    const newParams = replaceParams(params, {
+      order_by: order === 'desc' ? key : `-${key}`,
+      page: null,
+    });
+    const encodedParams = encodeNonDefaultQueryString(qsConfig, newParams);
+    history.push(
+      encodedParams
+        ? `${location.pathname}?${encodedParams}`
+        : location.pathname
+    );
+  };
+
+  const sortKey = params.order_by?.replace('-', '');
+  const sortBy = {
+    index: sortKey || defaultSortKey,
+    direction: params.order_by?.startsWith('-') ? 'asc' : 'desc',
+  };
+
+  return (
+    <Thead>
+      <Tr>
+        <Th
+          select={{
+            onSelect: handleSelectAll,
+            isSelected: isAllSelected,
+          }}
+        />
+        {React.Children.map(children, child =>
+          React.cloneElement(child, {
+            onSort,
+            sortBy,
+            columnIndex: child.props.sortKey,
+          })
+        )}
+      </Tr>
+    </Thead>
+  );
+}
+
+export function HeaderCell({ sortKey, onSort, sortBy, columnIndex, children }) {
+  const sort = sortKey
+    ? {
+        onSort: (event, key, order) => onSort(sortKey, order),
+        sortBy,
+        columnIndex,
+      }
+    : null;
+  return <Th sort={sort}>{children}</Th>;
+}
diff --git a/awx/ui_next/src/components/PaginatedTable/index.js b/awx/ui_next/src/components/PaginatedTable/index.js
index 3604417d95..c588cf47eb 100644
--- a/awx/ui_next/src/components/PaginatedTable/index.js
+++ b/awx/ui_next/src/components/PaginatedTable/index.js
@@ -1,5 +1,6 @@
 export { default } from './PaginatedTable';
 export { default as PaginatedTableRow } from './PaginatedTableRow';
 export { default as ActionsTd } from './ActionsTd';
+export { default as HeaderRow, HeaderCell } from './HeaderRow';
 // export { default as ToolbarDeleteButton } from './ToolbarDeleteButton';
 // export { default as ToolbarAddButton } from './ToolbarAddButton';
diff --git a/awx/ui_next/src/screens/Organization/OrganizationList/OrganizationList.jsx b/awx/ui_next/src/screens/Organization/OrganizationList/OrganizationList.jsx
index 6ddfe41470..2f647be8fc 100644
--- a/awx/ui_next/src/screens/Organization/OrganizationList/OrganizationList.jsx
+++ b/awx/ui_next/src/screens/Organization/OrganizationList/OrganizationList.jsx
@@ -14,7 +14,10 @@ import {
   ToolbarAddButton,
   ToolbarDeleteButton,
 } from '../../../components/PaginatedDataList';
-import PaginatedTable from '../../../components/PaginatedTable';
+import PaginatedTable, {
+  HeaderRow,
+  HeaderCell,
+} from '../../../components/PaginatedTable';
 import { getQSConfig, parseQueryString } from '../../../util/qs';
 import OrganizationListItem from './OrganizationListItem';
 
@@ -115,6 +118,10 @@ function OrganizationsList({ i18n }) {
     }
   };
 
+  const onSort = (e, index, direction) => {
+    console.log(index, direction);
+  };
+
   return (
     <>
       <PageSection>
@@ -125,7 +132,7 @@ function OrganizationsList({ i18n }) {
             items={organizations}
             itemCount={organizationCount}
             pluralizedItemName={i18n._(t`Organizations`)}
-            qsConfig={QS_CONFIG}
+            qsConfig={QS_CONFIG} // TODO: still used?
             onRowClick={handleSelect}
             toolbarSearchColumns={[
               {
@@ -155,6 +162,20 @@ function OrganizationsList({ i18n }) {
             toolbarSearchableKeys={searchableKeys}
             toolbarRelatedSearchableKeys={relatedSearchableKeys}
             headerRow={
+              // TODO: move selectAll logic into HeaderRow?
+              <HeaderRow
+                handleSelectAll={handleSelectAll}
+                isAllSelected={isAllSelected}
+                defaultSortKey="name"
+                qsConfig={QS_CONFIG}
+              >
+                <HeaderCell sortKey="name">{i18n._(t`Name`)}</HeaderCell>
+                <HeaderCell>{i18n._(t`Members`)}</HeaderCell>
+                <HeaderCell>{i18n._(t`Teams`)}</HeaderCell>
+              </HeaderRow>
+            }
+            _headerRow={
+              // TODO: move sorting into <PaginatedTableHeader> w/ friendly API
               <Thead>
                 <Tr>
                   <Th
@@ -163,9 +184,42 @@ function OrganizationsList({ i18n }) {
                       isSelected: isAllSelected,
                     }}
                   />
-                  <Th>{i18n._(t`Name`)}</Th>
-                  <Th>{i18n._(t`Members`)}</Th>
-                  <Th>{i18n._(t`Teams`)}</Th>
+                  <Th
+                    sort={{
+                      onSort,
+                      sortBy: {
+                        index: 'name',
+                        direction: 'asc',
+                      },
+                      columnIndex: 'name',
+                    }}
+                  >
+                    {i18n._(t`Name`)}
+                  </Th>
+                  <Th
+                    sort={{
+                      onSort,
+                      sortBy: {
+                        index: 'name',
+                        direction: 'asc',
+                      },
+                      columnIndex: 'members',
+                    }}
+                  >
+                    {i18n._(t`Members`)}
+                  </Th>
+                  <Th
+                    sort={{
+                      onSort,
+                      sortBy: {
+                        index: 'name',
+                        direction: 'asc',
+                      },
+                      columnIndex: 'teams',
+                    }}
+                  >
+                    {i18n._(t`Teams`)}
+                  </Th>
                 </Tr>
               </Thead>
             }