/*!
 * Copyright (c) Microsoft. All rights reserved.
 * Licensed under the MIT license. See LICENSE file in the project.
 */
import { SortDirection } from '@datashaper/schema'
import { desc } from 'arquero'
import type ColumnTable from 'arquero/dist/types/table/column-table'
import { useMemo } from 'react'

export function useSortedTable(
	table: ColumnTable,
	column?: string,
	sort?: SortDirection,
): ColumnTable {
	return useMemo(() => {
		let columns: string[] = []
		const isColumnOnTable = table.columnNames().includes(column || '')
		if (!sort) {
			return table
		}

		const isNotSortable =
			(column == null && !table.isGrouped) || !isColumnOnTable

		if (isNotSortable) {
			return table.unorder()
		} else if (column) {
			columns.push(column)
		}

		if (table.isGrouped()) {
			columns = [...table.groups().names, ...columns]
		}
		return table.orderby(
			sort === SortDirection.Descending
				? columns.map((col) => desc(col))
				: columns,
		)
	}, [table, column, sort])
}