index-bar.tsx 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. import { pinyin } from 'pinyin-pro'
  2. export const groupItems = (items, getFirstChar) => {
  3. const groups = items.reduce((acc, item) => {
  4. const firstChar = getFirstChar(item)
  5. if (!firstChar || firstChar.length === 0)
  6. return acc
  7. let letter
  8. // transform Chinese to pinyin
  9. if (/[\u4E00-\u9FA5]/.test(firstChar))
  10. letter = pinyin(firstChar, { pattern: 'first', toneType: 'none' })[0].toUpperCase()
  11. else
  12. letter = firstChar.toUpperCase()
  13. if (!/[A-Z]/.test(letter))
  14. letter = '#'
  15. if (!acc[letter])
  16. acc[letter] = []
  17. acc[letter].push(item)
  18. return acc
  19. }, {})
  20. const letters = Object.keys(groups).sort()
  21. // move '#' to the end
  22. const hashIndex = letters.indexOf('#')
  23. if (hashIndex !== -1) {
  24. letters.splice(hashIndex, 1)
  25. letters.push('#')
  26. }
  27. return { letters, groups }
  28. }
  29. const IndexBar = ({ letters, itemRefs }) => {
  30. const handleIndexClick = (letter) => {
  31. const element = itemRefs.current[letter]
  32. if (element)
  33. element.scrollIntoView({ behavior: 'smooth' })
  34. }
  35. return (
  36. <div className="index-bar fixed right-4 top-36 flex flex-col items-center text-xs font-medium text-gray-500">
  37. {letters.map(letter => (
  38. <div className="hover:text-gray-900 cursor-pointer" key={letter} onClick={() => handleIndexClick(letter)}>
  39. {letter}
  40. </div>
  41. ))}
  42. </div>
  43. )
  44. }
  45. export default IndexBar