From 1a9941b38e2f353db6d1f97af44e5967092a39c9 Mon Sep 17 00:00:00 2001
From: Alex Freska <alex.freska@gmail.com>
Date: Fri, 18 Oct 2024 11:44:18 -0400
Subject: [PATCH] refactor: refine multiselect API

---
 .../Keys/KeysBatchMenu/KeysBatchDelete.tsx    | 16 ++---
 apps/renterd/contexts/keys/columns.tsx        | 13 ++--
 apps/renterd/contexts/keys/index.tsx          | 24 ++-----
 apps/renterd/contexts/keys/types.ts           |  7 +-
 .../src/components/Table/TableRow.tsx         | 13 +++-
 .../src/components/Table/index.tsx            |  4 +-
 libs/design-system/src/core/Tooltip.tsx       |  4 +-
 .../src/multi/MultiSelectionMenu.tsx          | 38 ++++++-----
 .../src/multi/useMultiSelect.tsx              | 64 ++++++++++++++++---
 9 files changed, 111 insertions(+), 72 deletions(-)

diff --git a/apps/renterd/components/Keys/KeysBatchMenu/KeysBatchDelete.tsx b/apps/renterd/components/Keys/KeysBatchMenu/KeysBatchDelete.tsx
index 6d0433cec..1fd8c5bb9 100644
--- a/apps/renterd/components/Keys/KeysBatchMenu/KeysBatchDelete.tsx
+++ b/apps/renterd/components/Keys/KeysBatchMenu/KeysBatchDelete.tsx
@@ -15,15 +15,11 @@ import { useDialog } from '../../../contexts/dialog'
 import { useKeys } from '../../../contexts/keys'
 
 export function KeysBatchDelete() {
-  const { selectionMap, deselect } = useKeys()
+  const { multiSelect } = useKeys()
 
-  const ids = useMemo(
-    () => Object.entries(selectionMap).map(([_, item]) => item.id),
-    [selectionMap]
-  )
   const keys = useMemo(
-    () => Object.entries(selectionMap).map(([_, item]) => item.key),
-    [selectionMap]
+    () => Object.entries(multiSelect.selectionMap).map(([_, item]) => item.key),
+    [multiSelect.selectionMap]
   )
   const { openConfirmDialog } = useDialog()
   const settingsS3 = useSettingsS3()
@@ -43,13 +39,13 @@ export function KeysBatchDelete() {
         },
       },
     })
-    deselect(ids)
+    multiSelect.deselectAll()
     if (response.error) {
       triggerErrorToast({ title: 'Error deleting keys', body: response.error })
     } else {
       triggerSuccessToast({ title: `Keys deleted` })
     }
-  }, [settingsS3.data, settingsS3Update, deselect, keys, ids])
+  }, [settingsS3.data, settingsS3Update, multiSelect, keys])
 
   return (
     <Button
@@ -64,7 +60,7 @@ export function KeysBatchDelete() {
             <div className="flex flex-col gap-1">
               <Paragraph size="14">
                 Are you sure you would like to delete the{' '}
-                {ids.length.toLocaleString()} selected keys?
+                {multiSelect.selectionCount.toLocaleString()} selected keys?
               </Paragraph>
             </div>
           ),
diff --git a/apps/renterd/contexts/keys/columns.tsx b/apps/renterd/contexts/keys/columns.tsx
index 31017edfd..00ec06ab7 100644
--- a/apps/renterd/contexts/keys/columns.tsx
+++ b/apps/renterd/contexts/keys/columns.tsx
@@ -19,17 +19,20 @@ export const columns: KeysTableColumn[] = [
     fixed: true,
     contentClassName: '!pl-3 !pr-4',
     cellClassName: 'w-[20px] !pl-0 !pr-0',
-    heading: ({ context: { onSelectPage, isPageAllSelected } }) => (
+    heading: ({ context: { multiSelect } }) => (
       <ControlGroup className="flex h-4">
-        <Checkbox onClick={onSelectPage} checked={isPageAllSelected} />
+        <Checkbox
+          onClick={multiSelect.onSelectPage}
+          checked={multiSelect.isPageAllSelected}
+        />
       </ControlGroup>
     ),
-    render: ({ data: { id, key }, context: { selectionMap, onSelect } }) => (
+    render: ({ data: { id, key }, context: { multiSelect } }) => (
       <ControlGroup className="flex h-4">
         <Checkbox
           aria-label="select key"
-          onClick={(e) => onSelect(id, e)}
-          checked={!!selectionMap[id]}
+          onClick={(e) => multiSelect.onSelect(id, e)}
+          checked={!!multiSelect.selectionMap[id]}
         />
         <KeyContextMenu s3Key={key} />
       </ControlGroup>
diff --git a/apps/renterd/contexts/keys/index.tsx b/apps/renterd/contexts/keys/index.tsx
index 4fa91c7a9..005d25655 100644
--- a/apps/renterd/contexts/keys/index.tsx
+++ b/apps/renterd/contexts/keys/index.tsx
@@ -101,25 +101,14 @@ function useKeysMain() {
     filters
   )
 
-  const {
-    onSelect,
-    deselect,
-    deselectAll,
-    selectionMap,
-    selectionCount,
-    onSelectPage,
-    isPageAllSelected,
-  } = useMultiSelect(dataset)
+  const multiSelect = useMultiSelect(dataset)
 
   const cellContext = useMemo(
     () =>
       ({
-        selectionMap,
-        onSelect,
-        onSelectPage,
-        isPageAllSelected,
+        multiSelect,
       } as CellContext),
-    [selectionMap, onSelect, onSelectPage, isPageAllSelected]
+    [multiSelect]
   )
 
   return {
@@ -132,12 +121,7 @@ function useKeysMain() {
     datasetCount: dataset?.length || 0,
     datasetFilteredCount: datasetFiltered?.length || 0,
     columns: filteredTableColumns,
-    selectionMap,
-    selectionCount,
-    onSelectPage,
-    isPageAllSelected,
-    deselect,
-    deselectAll,
+    multiSelect,
     cellContext,
     dataset,
     datasetPage,
diff --git a/apps/renterd/contexts/keys/types.ts b/apps/renterd/contexts/keys/types.ts
index b6e51c16a..0a3929965 100644
--- a/apps/renterd/contexts/keys/types.ts
+++ b/apps/renterd/contexts/keys/types.ts
@@ -1,4 +1,4 @@
-import { MouseEvent } from 'react'
+import { MultiSelect } from '@siafoundation/design-system'
 
 export type KeyData = {
   id: string
@@ -7,10 +7,7 @@ export type KeyData = {
 }
 
 export type CellContext = {
-  selectionMap: Record<string, KeyData>
-  onSelect: (id: string, e: MouseEvent<HTMLButtonElement>) => void
-  onSelectPage: () => void
-  isPageAllSelected: boolean | 'indeterminate'
+  multiSelect: MultiSelect
 }
 
 export type TableColumnId = 'selection' | 'actions' | 'key' | 'secret'
diff --git a/libs/design-system/src/components/Table/TableRow.tsx b/libs/design-system/src/components/Table/TableRow.tsx
index b1b21acfa..518d59bb8 100644
--- a/libs/design-system/src/components/Table/TableRow.tsx
+++ b/libs/design-system/src/components/Table/TableRow.tsx
@@ -1,6 +1,6 @@
 'use client'
 
-import { CSSProperties, forwardRef, useMemo } from 'react'
+import { CSSProperties, forwardRef, MouseEvent, useMemo } from 'react'
 import { cx } from 'class-variance-authority'
 import { useDroppable, useDraggable } from '@dnd-kit/core'
 import {
@@ -13,7 +13,8 @@ type Data = {
   isDraggable?: boolean
   isDroppable?: boolean
   className?: string
-  onClick?: () => void
+  onClick?: (e: MouseEvent<HTMLTableRowElement>) => void
+  isSelected?: boolean
 }
 
 export type Row<Data, Context> = {
@@ -91,7 +92,13 @@ export function createTableRow<
           data-testid={data.id}
           onClick={data.onClick}
           className={cx(
-            'border-b border-gray-200/50 dark:border-graydark-100',
+            'border-b',
+            data.isSelected
+              ? [
+                  'bg-blue-400 border-blue-500/30',
+                  'dark:bg-blue-600/50 dark:border-blue-600/20',
+                ]
+              : 'border-gray-200/50 dark:border-graydark-100',
             data.onClick ? 'cursor-pointer' : '',
             data.className,
             className
diff --git a/libs/design-system/src/components/Table/index.tsx b/libs/design-system/src/components/Table/index.tsx
index 851888dca..216446edd 100644
--- a/libs/design-system/src/components/Table/index.tsx
+++ b/libs/design-system/src/components/Table/index.tsx
@@ -3,7 +3,7 @@
 import { Tooltip } from '../../core/Tooltip'
 import { Panel } from '../../core/Panel'
 import { Text } from '../../core/Text'
-import { useCallback, useMemo } from 'react'
+import { MouseEvent, useCallback, useMemo } from 'react'
 import { cx } from 'class-variance-authority'
 import { ChevronDown16, ChevronUp16 } from '@siafoundation/react-icons'
 import { times } from '@technically/lodash'
@@ -30,7 +30,7 @@ type Data = {
   id: string
   isDraggable?: boolean
   isDroppable?: boolean
-  onClick?: () => void
+  onClick?: (e: MouseEvent<HTMLTableRowElement>) => void
 }
 
 export type Row<Data, Context> = {
diff --git a/libs/design-system/src/core/Tooltip.tsx b/libs/design-system/src/core/Tooltip.tsx
index d8e9d47b3..d146777ea 100644
--- a/libs/design-system/src/core/Tooltip.tsx
+++ b/libs/design-system/src/core/Tooltip.tsx
@@ -93,7 +93,9 @@ export function Tooltip({
                     panelStyles()
                   )}
                 >
-                  {typeof content === 'string' || Array.isArray(content) ? (
+                  {typeof content === 'string' ||
+                  (React.isValidElement(content) &&
+                    content?.type === React.Fragment) ? (
                     <Paragraph size="12">{content}</Paragraph>
                   ) : (
                     content
diff --git a/libs/design-system/src/multi/MultiSelectionMenu.tsx b/libs/design-system/src/multi/MultiSelectionMenu.tsx
index fd6b1f5e6..43b91e13b 100644
--- a/libs/design-system/src/multi/MultiSelectionMenu.tsx
+++ b/libs/design-system/src/multi/MultiSelectionMenu.tsx
@@ -6,26 +6,25 @@ import { Panel } from '../core/Panel'
 import { Text } from '../core/Text'
 import { pluralize } from '@siafoundation/units'
 import { Close16 } from '@siafoundation/react-icons'
+import { MultiSelect } from './useMultiSelect'
 
 export function MultiSelectionMenu({
-  isVisible,
-  selectionCount,
-  isPageAllSelected,
-  deselectAll,
-  pageCount,
+  multiSelect: {
+    selectionCount,
+    deselectAll,
+    someSelectedItemsOutsideCurrentPage,
+    someSelectedOnCurrentPage,
+  },
   children,
   entityWord,
   entityWordPlural,
 }: {
-  isVisible: boolean
-  selectionCount: number
-  isPageAllSelected: boolean | 'indeterminate'
-  pageCount: number
+  multiSelect: MultiSelect
   children: React.ReactNode
-  deselectAll: () => void
   entityWord: string
   entityWordPlural?: string
 }) {
+  const isVisible = selectionCount > 0
   return (
     <div className="z-20 fixed bottom-5 left-0 right-0 flex justify-center dark pointer-events-none">
       <AnimatePresence>
@@ -42,15 +41,22 @@ export function MultiSelectionMenu({
             >
               {!!selectionCount && (
                 <Text size="14">
-                  {pluralize(selectionCount, entityWord, {
+                  {`${pluralize(selectionCount, entityWord, {
                     plural: entityWordPlural,
-                  })}{' '}
-                  selected
+                  })} selected${
+                    someSelectedItemsOutsideCurrentPage &&
+                    someSelectedOnCurrentPage
+                      ? ' on this and other pages'
+                      : !someSelectedItemsOutsideCurrentPage &&
+                        someSelectedOnCurrentPage
+                      ? ''
+                      : someSelectedItemsOutsideCurrentPage &&
+                        !someSelectedOnCurrentPage
+                      ? ' on other pages'
+                      : ''
+                  }`}
                 </Text>
               )}
-              {isPageAllSelected && selectionCount > pageCount && (
-                <Text>across multiple pages</Text>
-              )}
               <div className="flex-1" />
               {children}
               <Button tip="Deselect all" onClick={deselectAll} size="small">
diff --git a/libs/design-system/src/multi/useMultiSelect.tsx b/libs/design-system/src/multi/useMultiSelect.tsx
index 54ba4a182..98e4e5cb5 100644
--- a/libs/design-system/src/multi/useMultiSelect.tsx
+++ b/libs/design-system/src/multi/useMultiSelect.tsx
@@ -2,6 +2,8 @@
 
 import { MouseEvent, useCallback, useMemo, useState } from 'react'
 
+export type MultiSelect = ReturnType<typeof useMultiSelect>
+
 export function useMultiSelect<Item extends { id: string }>(dataset?: Item[]) {
   const [selectionMap, setSelectionMap] = useState<Record<string, Item>>({})
   const [, setLastSelectedItem] = useState<{
@@ -10,7 +12,7 @@ export function useMultiSelect<Item extends { id: string }>(dataset?: Item[]) {
   }>()
 
   const onSelect = useCallback(
-    (id: string, e: MouseEvent<HTMLButtonElement>) => {
+    (id: string, e: MouseEvent<HTMLButtonElement | HTMLDivElement>) => {
       if (!dataset) {
         return
       }
@@ -61,6 +63,31 @@ export function useMultiSelect<Item extends { id: string }>(dataset?: Item[]) {
     return getIsPageAllSelected({ dataset, selectionMap })
   }, [dataset, selectionMap])
 
+  const selectedIds = useMemo(
+    () =>
+      Object.entries(selectionMap)
+        .filter(([_, item]) => !!item)
+        .map(([id]) => id),
+    [selectionMap]
+  )
+
+  const someSelectedItemsOutsideCurrentPage = useMemo(() => {
+    if (!dataset) {
+      if (selectedIds.length === 0) {
+        return false
+      }
+      return true
+    }
+    return selectedIds.some((id) => !dataset.some((datum) => datum.id === id))
+  }, [dataset, selectedIds])
+
+  const someSelectedOnCurrentPage = useMemo(() => {
+    if (!dataset) {
+      return false
+    }
+    return dataset.some((datum) => selectionMap[datum.id])
+  }, [dataset, selectionMap])
+
   const onSelectPage = useCallback(() => {
     if (!dataset) {
       return
@@ -112,15 +139,32 @@ export function useMultiSelect<Item extends { id: string }>(dataset?: Item[]) {
     [selectionMap]
   )
 
-  return {
-    onSelect,
-    onSelectPage,
-    selectionMap,
-    isPageAllSelected,
-    selectionCount,
-    deselect,
-    deselectAll,
-  }
+  return useMemo(
+    () => ({
+      onSelect,
+      onSelectPage,
+      selectionMap,
+      selectedIds,
+      isPageAllSelected,
+      selectionCount,
+      someSelectedItemsOutsideCurrentPage,
+      someSelectedOnCurrentPage,
+      deselect,
+      deselectAll,
+    }),
+    [
+      onSelect,
+      onSelectPage,
+      selectionMap,
+      selectedIds,
+      isPageAllSelected,
+      selectionCount,
+      someSelectedItemsOutsideCurrentPage,
+      someSelectedOnCurrentPage,
+      deselect,
+      deselectAll,
+    ]
+  )
 }
 
 function getIsPageAllSelected<Item extends { id: string }>({