Sort list of strings alphabetically about it-tools HOT 1 OPEN

export type SortOptions = { dir: Dir } & (NaiveSortOptions | CollateSortOptions)

type Dir = 'asc' | 'desc'

type NaiveSortOptions = {
    kind: 'naive'
}

type CollateSortOptions = {
    kind: 'collate'
    locale: string | Intl.Locale
} & Intl.CollatorOptions

export function sort(list: readonly string[], options: SortOptions) {
    const sorted = options.kind === 'naive' ? naiveSort(list) : collateSort(list, options)
    return options.dir === 'desc' ? sorted.reverse() : sorted
}

function naiveSort(list: readonly string[]) {
    return [...list].sort((a, b) => a > b ? 1 : a < b ? -1 : 0)
}

function collateSort(list: readonly string[], { locale, ...options }: CollateSortOptions) {
    return [...list].sort((a, b) => a.localeCompare(b, locale, options))
}

/* Usage */

const strings = ['a', 'A', 'b', 'B', 'á', '1', '2', '10', '一', '阿']

sort(strings, { kind: 'naive', dir: 'asc' })
// ['1', '10', '2', 'A', 'B', 'a', 'b', 'á', '一', '阿']

sort(strings, { kind: 'naive', dir: 'desc' })
// ['阿', '一', 'á', 'b', 'a', 'B', 'A', '2', '10', '1']

sort(strings, { kind: 'collate', dir: 'asc', locale: 'en-US' })
// ['1', '10', '2', 'a', 'A', 'á', 'b', 'B', '一', '阿']

sort(strings, { kind: 'collate', dir: 'asc', locale: 'en-US', numeric: true, caseFirst: 'upper' })
// ['1', '2', '10', 'A', 'a', 'á', 'B', 'b', '一', '阿']

sort(strings, { kind: 'collate', dir: 'asc', locale: 'zh-CN', numeric: true })
// ['1', '2', '10', '阿', '一', 'á', 'a', 'A', 'b', 'B']