mango-ui-v3/hooks/useSortableData.tsx

import { useMemo, useState } from 'react'

type Direction = 'ascending' | 'descending'

interface Config {
  key: string
  direction: Direction
}

export function useSortableData<T>(
  items: T[],
  config = null
): { items: T[]; requestSort: any; sortConfig: any } {
  const [sortConfig, setSortConfig] = useState<Config | null>(config)

  const sortedItems = useMemo(() => {
    const sortableItems = items ? [...items] : []
    if (sortConfig !== null) {
      sortableItems.sort((a, b) => {
        if (!isNaN(a[sortConfig.key])) {
          return sortConfig.direction === 'ascending'
            ? a[sortConfig.key] - b[sortConfig.key]
            : b[sortConfig.key] - a[sortConfig.key]
        }
        if (a[sortConfig.key] < b[sortConfig.key]) {
          return sortConfig.direction === 'ascending' ? -1 : 1
        }
        if (a[sortConfig.key] > b[sortConfig.key]) {
          return sortConfig.direction === 'ascending' ? 1 : -1
        }
        return 0
      })
    }
    return sortableItems
  }, [items, sortConfig])

  const requestSort = (key) => {
    let direction: Direction = 'ascending'
    if (
      sortConfig &&
      sortConfig.key === key &&
      sortConfig.direction === 'ascending'
    ) {
      direction = 'descending'
    }
    setSortConfig({ key, direction })
  }

  return { items: sortedItems, requestSort, sortConfig }
}