import type { FC } from 'react'
import type { SegmentListContextValue } from '..'
import { cn } from '@langgenius/dify-ui/cn'
import * as React from 'react'
import { Markdown } from '@/app/components/base/markdown'
import { useSegmentListContext } from '..'

type ChunkContentProps = {
  detail: {
    answer?: string
    content: string
    sign_content: string
  }
  isFullDocMode: boolean
  className?: string
}

const selectIsCollapsed = (s: SegmentListContextValue) => s.isCollapsed

const ChunkContent: FC<ChunkContentProps> = ({
  detail,
  isFullDocMode,
  className,
}) => {
  const { answer, content, sign_content } = detail
  const isCollapsed = useSegmentListContext(selectIsCollapsed)

  if (answer) {
    return (
      <div className={className}>
        <div className="flex gap-x-1">
          <div className="w-4 shrink-0 text-[13px] leading-[20px] font-medium text-text-tertiary">Q</div>
          <Markdown
            className={cn('body-md-regular text-text-secondary', isCollapsed ? 'line-clamp-2' : 'line-clamp-20')}
            content={content}
            customDisallowedElements={['input']}
          />
        </div>
        <div className="flex gap-x-1">
          <div className="w-4 shrink-0 text-[13px] leading-[20px] font-medium text-text-tertiary">A</div>
          <Markdown
            className={cn('body-md-regular text-text-secondary', isCollapsed ? 'line-clamp-2' : 'line-clamp-20')}
            content={answer}
            customDisallowedElements={['input']}
          />
        </div>
      </div>
    )
  }
  return (
    <Markdown
      className={cn('mt-0.5! text-text-secondary!', isFullDocMode ? 'line-clamp-3' : isCollapsed ? 'line-clamp-2' : 'line-clamp-20', className)}
      content={sign_content || content || ''}
      customDisallowedElements={['input']}
    />
  )
}

export default React.memo(ChunkContent)