tjdify/web/app/components/workflow/block-selector/tools.tsx

import {
  memo,
  useCallback,
  useMemo,
} from 'react'
import { useTranslation } from 'react-i18next'
import BlockIcon from '../block-icon'
import { BlockEnum } from '../types'
import type { ToolWithProvider } from '../types'
import { useStore } from '../store'
import type { ToolDefaultValue } from './types'
import Tooltip from '@/app/components/base/tooltip'
import { useGetLanguage } from '@/context/i18n'

type ToolsProps = {
  isCustom?: boolean
  onSelect: (type: BlockEnum, tool?: ToolDefaultValue) => void
  searchText: string
}
const Blocks = ({
  isCustom,
  searchText,
  onSelect,
}: ToolsProps) => {
  const { t } = useTranslation()
  const language = useGetLanguage()
  const buildInTools = useStore(s => s.buildInTools)
  const customTools = useStore(s => s.customTools)

  const tools = useMemo(() => {
    const currentTools = isCustom ? customTools : buildInTools

    return currentTools.filter((toolWithProvider) => {
      return toolWithProvider.tools.some((tool) => {
        return tool.label[language].toLowerCase().includes(searchText.toLowerCase())
      })
    })
  }, [isCustom, customTools, buildInTools, searchText, language])

  const renderGroup = useCallback((toolWithProvider: ToolWithProvider) => {
    const list = toolWithProvider.tools

    return (
      <div
        key={toolWithProvider.id}
        className='mb-1 last-of-type:mb-0'
      >
        <div className='flex items-start px-3 h-[22px] text-xs font-medium text-gray-500'>
          {toolWithProvider.label[language]}
        </div>
        {
          list.map(tool => (
            <Tooltip
              key={tool.name}
              selector={`workflow-block-tool-${tool.name}`}
              position='right'
              className='!p-0 !px-3 !py-2.5 !w-[200px] !leading-[18px] !text-xs !text-gray-700 !border-[0.5px] !border-black/5 !bg-transparent !rounded-xl !shadow-lg'
              htmlContent={(
                <div>
                  <div className='flex items-center mb-2'>
                    <BlockIcon
                      size='md'
                      className='mr-2'
                      type={BlockEnum.Tool}
                      toolIcon={toolWithProvider.icon}
                    />
                    <div className='text-sm text-gray-900'>{tool.label[language]}</div>
                  </div>
                  {tool.description[language]}
                </div>
              )}
              noArrow
            >
              <div
                className='flex items-center px-3 w-full h-8 rounded-lg hover:bg-gray-50 cursor-pointer'
                onClick={() => onSelect(BlockEnum.Tool, {
                  provider_id: toolWithProvider.id,
                  provider_type: toolWithProvider.type,
                  provider_name: toolWithProvider.name,
                  tool_name: tool.name,
                  tool_label: tool.label[language],
                  title: tool.label[language],
                })}
              >
                <BlockIcon
                  className='mr-2'
                  type={BlockEnum.Tool}
                  toolIcon={toolWithProvider.icon}
                />
                <div className='text-sm text-gray-900'>{tool.label[language]}</div>
              </div>
            </Tooltip>
          ))
        }
      </div>
    )
  }, [onSelect, language])

  return (
    <div className='p-1 max-h-[464px] overflow-y-auto'>
      {
        !tools.length && (
          <div className='flex items-center px-3 h-[22px] text-xs font-medium text-gray-500'>{t('workflow.tabs.noResult')}</div>
        )
      }
      {
        !!tools.length && tools.map(renderGroup)
      }
    </div>
  )
}

export default memo(Blocks)