hensei-web/components/MentionList/index.tsx

import React, {
  forwardRef,
  useEffect,
  useImperativeHandle,
  useState,
} from 'react'
import { useTranslation } from 'next-i18next'
import { useRouter } from 'next/router'
import { SuggestionProps } from '@tiptap/suggestion'
import classNames from 'classnames'

import styles from './index.module.scss'

type Props = Pick<SuggestionProps, 'items' | 'command'>

export type MentionRef = {
  onKeyDown: (props: { event: KeyboardEvent }) => boolean
}

export type MentionSuggestion = {
  granblue_id: string
  name: {
    [key: string]: string
    en: string
    ja: string
  }
  type: string
  element: number
}

interface MentionProps extends SuggestionProps {
  items: MentionSuggestion[]
}

export const MentionList = forwardRef<MentionRef, Props>(
  ({ items, ...props }: Props, forwardedRef) => {
    const router = useRouter()
    const locale = router.locale || 'en'

    const { t } = useTranslation('common')

    const [selectedIndex, setSelectedIndex] = useState(0)

    const selectItem = (index: number) => {
      const item = items[index]

      if (item) {
        props.command({ id: item })
      }
    }

    const upHandler = () => {
      setSelectedIndex((selectedIndex + items.length - 1) % items.length)
    }

    const downHandler = () => {
      setSelectedIndex((selectedIndex + 1) % items.length)
    }

    const enterHandler = () => {
      selectItem(selectedIndex)
    }

    useEffect(() => setSelectedIndex(0), [items])

    useImperativeHandle(forwardedRef, () => ({
      onKeyDown: ({ event }) => {
        if (event.key === 'ArrowUp') {
          upHandler()
          return true
        }

        if (event.key === 'ArrowDown') {
          downHandler()
          return true
        }

        if (event.key === 'Enter') {
          enterHandler()
          return true
        }

        return false
      },
    }))

    return (
      <div className={styles.items}>
        {items.length ? (
          items.map((item, index) => (
            <button
              className={classNames({
                [styles.item]: true,
                [styles.selected]: index === selectedIndex,
              })}
              key={index}
              onClick={() => selectItem(index)}
            >
              <img
                alt={item.name[locale]}
                src={
                  item.type === 'character'
                    ? `${process.env.NEXT_PUBLIC_SIERO_IMG_URL}/${item.type}-square/${item.granblue_id}_01.jpg`
                    : `${process.env.NEXT_PUBLIC_SIERO_IMG_URL}/${item.type}-square/${item.granblue_id}.jpg`
                }
              />
              <span>{item.name[locale]}</span>
            </button>
          ))
        ) : (
          <div className={styles.noResult}>
            {t('search.errors.no_results_generic')}
          </div>
        )}
      </div>
    )
  }
)