rank of array

rank.ts

/**
 * rank({
 *   array: [
 *     {name: 'a', age: 20},
 *     {name: 'b', age: 5},
 *     {name: 'c', age: 20},
 *     {name: 'd', age: 75},
 *     {name: 'e', age: 21},
 *   ],
 *   column: 'age',
 *   ascOrDesc: 'asc'
 * })
 * => [
 *   {name: 'b', age: 5, rank: 1},
 *   {name: 'a', age: 20, rank: 2},
 *   {name: 'c', age: 20, rank: 2},
 *   {name: 'e', age: 21, rank: 4},
 *   {name: 'd', age: 75, rank: 5},
 * ]
 */
const rank = <T extends Record<string, unknown>>({
  records,
  column,
  ascOrDesc,
}: {
  records: T[]
  column: keyof T
  ascOrDesc: 'asc' | 'desc'
}): (T & { rank: number })[] => {
  const isAsc = ascOrDesc === 'asc'
  return records
    .map((r) => ({
      ...r,
      rank:
        records.filter((_r) =>
          isAsc ? _r[column] < r[column] : _r[column] > r[column]
        ).length + 1,
    }))
    .sort(({ rank: a }, { rank: b }) => (isAsc ? a - b : a - b))
}