mirror of
https://github.com/langgenius/dify.git
synced 2025-07-18 14:52:32 +00:00
40 lines
1.3 KiB
TypeScript
40 lines
1.3 KiB
TypeScript
import {
|
|
useMemo,
|
|
} from 'react'
|
|
import type {
|
|
Node,
|
|
} from '../types'
|
|
import {
|
|
BlockEnum,
|
|
} from '../types'
|
|
import {
|
|
useStore,
|
|
} from '../store'
|
|
import { useDataSources } from '../block-selector/hooks'
|
|
import { CollectionType } from '@/app/components/tools/types'
|
|
import { canFindTool } from '@/utils'
|
|
|
|
export const useToolIcon = (data: Node['data']) => {
|
|
const buildInTools = useStore(s => s.buildInTools)
|
|
const customTools = useStore(s => s.customTools)
|
|
const workflowTools = useStore(s => s.workflowTools)
|
|
const dataSourceList = useDataSources()
|
|
// const a = useStore(s => s.data)
|
|
const toolIcon = useMemo(() => {
|
|
if (data.type === BlockEnum.Tool) {
|
|
let targetTools = buildInTools
|
|
if (data.provider_type === CollectionType.builtIn)
|
|
targetTools = buildInTools
|
|
else if (data.provider_type === CollectionType.custom)
|
|
targetTools = customTools
|
|
else
|
|
targetTools = workflowTools
|
|
return targetTools.find(toolWithProvider => canFindTool(toolWithProvider.id, data.provider_id))?.icon
|
|
}
|
|
if (data.type === BlockEnum.DataSource)
|
|
return dataSourceList.find(toolWithProvider => canFindTool(toolWithProvider.id, data.provider_id))?.icon
|
|
}, [data, buildInTools, customTools, workflowTools, dataSourceList])
|
|
|
|
return toolIcon
|
|
}
|