dify/web/app/components/datasets/create/file-preview/index.tsx

'use client'
import type { CustomFile as File } from '@/models/datasets'
import { XMarkIcon } from '@heroicons/react/20/solid'
import { cn } from '@langgenius/dify-ui/cn'
import * as React from 'react'
import { useEffect, useState } from 'react'
import { useTranslation } from 'react-i18next'
import Loading from '@/app/components/base/loading'
import { fetchFilePreview } from '@/service/common'
import s from './index.module.css'

type IProps = {
  file?: File
  hidePreview: () => void
}

const FilePreview = ({ file, hidePreview }: IProps) => {
  const { t } = useTranslation()
  const [previewContent, setPreviewContent] = useState('')
  const [loading, setLoading] = useState(true)

  const getPreviewContent = async (fileID: string) => {
    try {
      const res = await fetchFilePreview({ fileID })
      setPreviewContent(res.content)
      setLoading(false)
    } catch {}
  }

  const getFileName = (currentFile?: File) => {
    if (!currentFile) return ''
    const arr = currentFile.name.split('.')
    return arr.slice(0, -1).join()
  }

  useEffect(() => {
    if (file?.id) {
      setLoading(true)
      getPreviewContent(file.id)
    }
  }, [file])

  return (
    <div className={cn(s.filePreview, 'h-full')}>
      <div className={cn(s.previewHeader)}>
        <div className={cn(s.title, 'title-md-semi-bold')}>
          <span>{t(($) => $['stepOne.filePreview'], { ns: 'datasetCreation' })}</span>
          <button
            type="button"
            className="flex size-6 cursor-pointer items-center justify-center border-none bg-transparent p-0 focus-visible:ring-1 focus-visible:ring-components-input-border-active focus-visible:outline-hidden"
            aria-label={t(($) => $['operation.close'], { ns: 'common' })}
            onClick={hidePreview}
          >
            <XMarkIcon className="size-4" aria-hidden="true"></XMarkIcon>
          </button>
        </div>
        <div className={cn(s.fileName, 'system-xs-medium')}>
          <span>{getFileName(file)}</span>
          <span className={cn(s.filetype)}>.{file?.extension}</span>
        </div>
      </div>
      <div className={cn(s.previewContent)}>
        {loading && <Loading type="area" />}
        {!loading && <div className={cn(s.fileContent, 'body-md-regular')}>{previewContent}</div>}
      </div>
    </div>
  )
}

export default FilePreview