dify-web/app/(commonLayout)/database/DatabaseCard.tsx

// DatasetCard.js
import React, { useContext, useCallback, useEffect, useState } from 'react';
import { useTranslation } from 'react-i18next';
import { RiMoreFill } from '@remixicon/react';
import cn from 'classnames';
import Confirm from '@/app/components/base/confirm';
import { ToastContext } from '@/app/components/base/toast';
import { checkIsUsedInApp, deleteDataset } from '@/service/datasets';
import type { DataSet } from '@/models/datasets';
import Tooltip from '@/app/components/base/tooltip';
import { Folder } from '@/app/components/base/icons/src/vender/solid/files';
import CustomPopover from '@/app/components/base/popover';
import Divider from '@/app/components/base/divider';
import RenameDatasetModal from '@/app/components/datasets/rename-modal';
import TagSelector from '@/app/components/base/tag-management/selector';
import CornerLabel from '@/app/components/base/corner-label';
import { useAppContext } from '@/context/app-context';
import { useRouter } from 'next/navigation';

export type DatasetCardProps = {
  dataset: DataSet,
  onSuccess?: () => void,
};

const DatasetCard = ({ dataset, onSuccess }: DatasetCardProps) => {
  const { t } = useTranslation();
  const { notify } = useContext(ToastContext);
  const { push } = useRouter();
  const EXTERNAL_PROVIDER = 'external' as const;

  const { isCurrentWorkspaceDatasetOperator } = useAppContext();
  const [tags, setTags] = useState<Tag[]>(dataset.tags);

  const [showRenameModal, setShowRenameModal] = useState(false);
  const [showConfirmDelete, setShowConfirmDelete] = useState(false);
  const [confirmMessage, setConfirmMessage] = useState('');

  const isExternalProvider = (provider: string): boolean => provider === EXTERNAL_PROVIDER;

  const detectIsUsedByApp = useCallback(async () => {
    try {
      const { is_using: isUsedByApp } = await checkIsUsedInApp(dataset.id);
      setConfirmMessage(isUsedByApp ? t('dataset.datasetUsedByApp')! : t('dataset.deleteDatasetConfirmContent')!);
    } catch (e: any) {
      const res = await e.json();
      notify({ type: 'error', message: res?.message || 'Unknown error' });
    }
    setShowConfirmDelete(true);
  }, [dataset.id, notify, t]);

  const onConfirmDelete = useCallback(async () => {
    try {
      await deleteDataset(dataset.id);
      notify({ type: 'success', message: t('dataset.datasetDeleted') });
      if (onSuccess) onSuccess();
    } catch (e: any) {
      // Handle errors here
    }
    setShowConfirmDelete(false);
  }, [dataset.id, notify, onSuccess, t]);

  const Operations = ({ showDelete, ...props }) => {
    const onMouseLeave = async () => {
      props.onClose?.();
    };

    const onClickRename = (e) => {
      e.stopPropagation();
      props.onClick?.();
      e.preventDefault();
      setShowRenameModal(true);
    };

    const onClickDelete = (e) => {
      e.stopPropagation();
      props.onClick?.();
      e.preventDefault();
      detectIsUsedByApp();
    };

    return (
      <div className="relative w-full py-1" onMouseLeave={onMouseLeave}>
        <div
          className='h-8 py-[6px] px-3 mx-1 flex items-center gap-2 hover:bg-gray-100 rounded-lg cursor-pointer'
          onClick={onClickRename}
        >
          <span className='text-gray-700 text-sm'>{t('common.operation.settings')}</span>
        </div>
        {showDelete && (
          <>
            <Divider className="!my-1" />
            <div
              className='group h-8 py-[6px] px-3 mx-1 flex items-center gap-2 hover:bg-red-50 rounded-lg cursor-pointer'
              onClick={onClickDelete}
            >
              <span className={cn('text-gray-700 text-sm', 'group-hover:text-red-500')}>
                {t('common.operation.delete')}
              </span>
            </div>
          </>
        )}
      </div>
    );
  };

  useEffect(() => {
    setTags(dataset.tags);
  }, [dataset]);

  return (
    <>
      <div
        className='group relative col-span-1 bg-white border-[0.5px] border-solid border-transparent rounded-xl shadow-sm min-h-[160px] flex flex-col transition-all duration-200 ease-in-out cursor-pointer hover:shadow-lg'
        data-disable-nprogress={true}
        onClick={(e) => {
          e.preventDefault();
          isExternalProvider(dataset.provider)
            ? push(`/database/${dataset.id}/hitTesting`)
            : push(`/database/${dataset.id}/documents`);
        }}
      >
        {isExternalProvider(dataset.provider) && (
          <CornerLabel label='External' className='absolute right-0' labelClassName='rounded-tr-xl' />
        )}
        <div className='flex pt-[14px] px-[14px] pb-3 h-[66px] items-center gap-3 grow-0 shrink-0'>
          <div className={cn(
            'shrink-0 flex items-center justify-center p-2.5 bg-[#F5F8FF] rounded-md border-[0.5px] border-[#E0EAFF]',
            !dataset.embedding_available && 'opacity-50 hover:opacity-100',
          )}>
            <Folder className='w-5 h-5 text-[#444CE7]' />
          </div>
          <div className='grow w-0 py-[1px]'>
            <div className='flex items-center text-sm leading-5 font-semibold text-gray-800'>
              <div className={cn('truncate', !dataset.embedding_available && 'opacity-50 hover:opacity-100')} title={dataset.name}>{dataset.name}</div>
              {!dataset.embedding_available && (
                <Tooltip popupContent={t('dataset.unavailableTip')}>
                  <span className='shrink-0 inline-flex w-max ml-1 px-1 border border-gray-200 rounded-md text-gray-500 text-xs font-normal leading-[18px]'>{t('dataset.unavailable')}</span>
                </Tooltip>
              )}
            </div>
          </div>
        </div>
        <div
          className={cn(
            'grow mb-2 px-[14px] max-h-[72px] text-xs leading-normal text-gray-500 group-hover:line-clamp-2 group-hover:max-h-[36px]',
            tags.length ? 'line-clamp-2' : 'line-clamp-4',
            !dataset.embedding_available && 'opacity-50 hover:opacity-100',
          )}
          title={dataset.description}
        >
          {dataset.description}
        </div>
        <div className={cn(
          'items-center shrink-0 mt-1 pt-1 pl-[14px] pr-[6px] pb-[6px] h-[42px]',
          tags.length ? 'flex' : '!hidden group-hover:!flex',
        )}>
          <div className={cn('grow flex items-center gap-1 w-0', !dataset.embedding_available && 'opacity-50 hover:opacity-100')} onClick={(e) => {
            e.stopPropagation();
            e.preventDefault();
          }}>
            <div className={cn(
              'group-hover:!block group-hover:!mr-0 mr-[41px] grow w-full',
              tags.length ? '!block' : '!hidden',
            )}>
              <TagSelector
                position='bl'
                type='knowledge'
                targetID={dataset.id}
                value={tags.map(tag => tag.id)}
                selectedTags={tags}
                onCacheUpdate={setTags}
                onChange={onSuccess}
              />
            </div>
          </div>
          <div className='!hidden group-hover:!flex shrink-0 mx-1 w-[1px] h-[14px] bg-gray-200' />
          <div className='!hidden group-hover:!flex shrink-0'>
            <CustomPopover
              htmlContent={<Operations showDelete={!isCurrentWorkspaceDatasetOperator} />}
              position="br"
              trigger="click"
              btnElement={
                <div className='flex items-center justify-center w-8 h-8 cursor-pointer rounded-md'>
                  <RiMoreFill className='w-4 h-4 text-gray-700' />
                </div>
              }
              btnClassName={open =>
                cn(
                  open ? '!bg-black/5 !shadow-none' : '!bg-transparent',
                  'h-8 w-8 !p-2 rounded-md border-none hover:!bg-black/5',
                )
              }
              className={'!w-[128px] h-fit !z-20'}
            />
          </div>
        </div>
      </div>
      {showRenameModal && (
        <RenameDatasetModal
          show={showRenameModal}
          dataset={dataset}
          onClose={() => setShowRenameModal(false)}
          onSuccess={onSuccess}
        />
      )}
      {showConfirmDelete && (
        <Confirm
          title={t('dataset.deleteDatasetConfirmTitle')}
          content={confirmMessage}
          isShow={showConfirmDelete}
          onConfirm={onConfirmDelete}
          onCancel={() => setShowConfirmDelete(false)}
        />
      )}
    </>
  );
};

export default DatasetCard;