youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 01:48:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Refactor useDatasourceIcon hook and enhance dataset node rendering with AppIcon component

Browse Source
This commit is contained in:
twwu
2025-07-02 13:48:11 +08:00
parent 5d7a533ada
commit 0284e7556e
2 changed files with 18 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
web/app/components/datasets/documents/create-from-pipeline/data-source-options/hooks.tsx
View File

@ -1,31 +1,23 @@
import { useCallback, useMemo, useState } from 'react'
import { useMemo } from 'react'
import type { DataSourceNodeType } from '@/app/components/workflow/nodes/data-source/types'
import type { DataSourceItem } from '@/app/components/workflow/block-selector/types'
import { basePath } from '@/utils/var'
import { useDataSourceList } from '@/service/use-pipeline'
import { useDatasetDetailContextWithSelector } from '@/context/dataset-detail'
import type { ToolWithProvider } from '@/app/components/workflow/types'
import { transformDataSourceToTool } from '@/app/components/workflow/block-selector/utils'
export const useDatasourceIcon = (data: DataSourceNodeType) => {
const pipelineId = useDatasetDetailContextWithSelector(s => s.dataset?.pipeline_id)
const [dataSourceList, setDataSourceList] = useState<ToolWithProvider[]>([])
const { data: dataSourceListData, isSuccess } = useDataSourceList(true)
const handleUpdateDataSourceList = useCallback((dataSourceList: DataSourceItem[]) => {
const datasourceIcon = useMemo(() => {
if (!isSuccess) return
const dataSourceList = [...(dataSourceListData || [])]
dataSourceList.forEach((item) => {
const icon = item.declaration.identity.icon
if (typeof icon == 'string' && !icon.includes(basePath))
item.declaration.identity.icon = `${basePath}${icon}`
})
const formattedDataSourceList = dataSourceList.map(item => transformDataSourceToTool(item))
setDataSourceList!(formattedDataSourceList)
}, [])
useDataSourceList(!!pipelineId, handleUpdateDataSourceList)
const datasourceIcon = useMemo(() => {
return dataSourceList?.find(toolWithProvider => toolWithProvider.plugin_id === data.plugin_id)?.icon
}, [data, dataSourceList])
return formattedDataSourceList?.find(toolWithProvider => toolWithProvider.plugin_id === data.plugin_id)?.icon
}, [data.plugin_id, dataSourceListData, isSuccess])
return datasourceIcon
}