2024-07-09 09:43:34 +08:00
|
|
|
import { pinyin } from 'pinyin-pro'
|
2024-07-25 11:21:51 +08:00
|
|
|
import type { FC, RefObject } from 'react'
|
2024-07-09 09:43:34 +08:00
|
|
|
|
2024-07-25 11:21:51 +08:00
|
|
|
export const groupItems = (items: Array<any>, getFirstChar: (item: string) => string) => {
|
2024-07-09 09:43:34 +08:00
|
|
|
const groups = items.reduce((acc, item) => {
|
|
|
|
const firstChar = getFirstChar(item)
|
|
|
|
if (!firstChar || firstChar.length === 0)
|
|
|
|
return acc
|
|
|
|
|
|
|
|
let letter
|
|
|
|
|
|
|
|
// transform Chinese to pinyin
|
|
|
|
if (/[\u4E00-\u9FA5]/.test(firstChar))
|
|
|
|
letter = pinyin(firstChar, { pattern: 'first', toneType: 'none' })[0].toUpperCase()
|
|
|
|
else
|
|
|
|
letter = firstChar.toUpperCase()
|
|
|
|
|
|
|
|
if (!/[A-Z]/.test(letter))
|
|
|
|
letter = '#'
|
|
|
|
|
|
|
|
if (!acc[letter])
|
|
|
|
acc[letter] = []
|
|
|
|
|
|
|
|
acc[letter].push(item)
|
|
|
|
return acc
|
|
|
|
}, {})
|
|
|
|
|
|
|
|
const letters = Object.keys(groups).sort()
|
|
|
|
// move '#' to the end
|
|
|
|
const hashIndex = letters.indexOf('#')
|
|
|
|
if (hashIndex !== -1) {
|
|
|
|
letters.splice(hashIndex, 1)
|
|
|
|
letters.push('#')
|
|
|
|
}
|
|
|
|
return { letters, groups }
|
|
|
|
}
|
|
|
|
|
2024-07-25 11:21:51 +08:00
|
|
|
type IndexBarProps = {
|
|
|
|
letters: string[]
|
|
|
|
itemRefs: RefObject<{ [key: string]: HTMLElement | null }>
|
|
|
|
}
|
|
|
|
|
|
|
|
const IndexBar: FC<IndexBarProps> = ({ letters, itemRefs }) => {
|
|
|
|
const handleIndexClick = (letter: string) => {
|
|
|
|
const element = itemRefs.current?.[letter]
|
2024-07-09 09:43:34 +08:00
|
|
|
if (element)
|
|
|
|
element.scrollIntoView({ behavior: 'smooth' })
|
|
|
|
}
|
|
|
|
return (
|
|
|
|
<div className="index-bar fixed right-4 top-36 flex flex-col items-center text-xs font-medium text-gray-500">
|
|
|
|
{letters.map(letter => (
|
|
|
|
<div className="hover:text-gray-900 cursor-pointer" key={letter} onClick={() => handleIndexClick(letter)}>
|
|
|
|
{letter}
|
|
|
|
</div>
|
|
|
|
))}
|
|
|
|
</div>
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
|
|
|
export default IndexBar
|