import { Image } from '@heroui/image'
import qface from 'qface'
import { FaReply } from 'react-icons/fa6'

import { OB11Segment } from '@/types/onebot'

export const renderMessageContent = (
  segments: OB11Segment[],
  small = false
): React.ReactElement[] => {
  return segments.map((segment, index) => {
    switch (segment.type) {
      case 'text':
        return <span key={index}>{segment.data.text}</span>
      case 'face':
        return (
          <Image
            removeWrapper
            classNames={{
              img: 'w-6 h-6 inline !text-[0px] m-0 -mt-1.5 !p-0'
            }}
            key={index}
            src={qface.getUrl(segment.data.id)}
            alt={`face-${segment.data.id}`}
          />
        )
      case 'image':
        return (
          <Image
            classNames={{
              wrapper: 'block !text-[0px] !m-0 !p-0',
              img: 'block'
            }}
            radius="sm"
            className={
              small
                ? 'max-h-16 object-cover'
                : 'max-w-64 max-h-96 h-auto object-cover'
            }
            key={index}
            src={segment.data.url || segment.data.file}
            alt="image"
            referrerPolicy="no-referrer"
          />
        )
      case 'record':
        return (
          <audio
            key={index}
            controls
            src={segment.data.url || segment.data.file}
          />
        )
      case 'video':
        return (
          <video
            key={index}
            controls
            src={segment.data.url || segment.data.file}
          />
        )
      case 'at':
        return (
          <span key={index} className="text-blue-500">
            @
            {segment.data.qq === 'all' ? (
              '所有人'
            ) : (
              <span>
                {segment.data.name}({segment.data.qq})
              </span>
            )}
          </span>
        )
      case 'rps':
        return <span key={index}>[猜拳]</span>
      case 'dice':
        return <span key={index}>[掷骰子]</span>
      case 'shake':
        return <span key={index}>[窗口抖动]</span>
      case 'poke':
        return (
          <span key={index}>
            [戳一戳: {segment.data.name || segment.data.id}]
          </span>
        )
      case 'anonymous':
        return <span key={index}>[匿名消息]</span>
      case 'share':
        return (
          <a
            key={index}
            href={segment.data.url}
            target="_blank"
            rel="noopener noreferrer"
          >
            {segment.data.title}
          </a>
        )
      case 'contact':
        return (
          <span key={index}>
            [推荐{segment.data.type === 'qq' ? '好友' : '群'}: {segment.data.id}
            ]
          </span>
        )
      case 'location':
        return <span key={index}>[位置: {segment.data.title || '未知'}]</span>
      case 'music':
        if (segment.data.type === 'custom') {
          return (
            <a
              key={index}
              href={segment.data.url}
              target="_blank"
              rel="noopener noreferrer"
            >
              {segment.data.title}
            </a>
          )
        }
        return (
          <span key={index}>
            [音乐: {segment.data.type} - {segment.data.id}]
          </span>
        )
      case 'reply':
        return (
          <div
            key={index}
            className="bg-content3 py-1 px-2 rounded-md flex items-center gap-1"
          >
            <FaReply className="text-default-500" />
            回复消息ID: {segment.data.id}
          </div>
        )
      case 'forward':
        return <span key={index}>[合并转发: {segment.data.id}]</span>
      case 'node':
        return <span key={index}>[消息节点]</span>
      case 'xml':
        return <pre key={index}>{segment.data.data}</pre>
      case 'json':
        return (
          <pre key={index} className="break-all whitespace-break-spaces">
            {segment.data.data}
          </pre>
        )
      case 'file':
        return (
          <a
            key={index}
            href={segment.data.file}
            target="_blank"
            rel="noopener noreferrer"
          >
            [文件]
          </a>
        )
      default:
        return <span key={index}>[不支持的消息类型]</span>
    }
  })
}