50 lines
1.3 KiB
TypeScript
50 lines
1.3 KiB
TypeScript
|
import {
|
||
|
|
useMemo,
|
||
|
|
} from 'react'
|
||
|
|
import type { NodeSelectorProps } from './main'
|
||
|
|
import NodeSelector from './main'
|
||
|
|
import { useHooksStore } from '@/app/components/workflow/hooks-store/store'
|
||
|
|
import { BlockEnum } from '@/app/components/workflow/types'
|
||
|
|
import { useStore } from '../store'
|
||
|
|
|
||
|
|
const NodeSelectorWrapper = (props: NodeSelectorProps) => {
|
||
|
|
const availableNodesMetaData = useHooksStore(s => s.availableNodesMetaData)
|
||
|
|
const dataSourceList = useStore(s => s.dataSourceList)
|
||
|
|
|
||
|
|
const blocks = useMemo(() => {
|
||
|
|
const result = availableNodesMetaData?.nodes || []
|
||
|
|
|
||
|
|
return result.filter((block) => {
|
||
|
|
if (block.metaData.type === BlockEnum.Start)
|
||
|
|
return false
|
||
|
|
|
||
|
|
if (block.metaData.type === BlockEnum.DataSource)
|
||
|
|
return false
|
||
|
|
|
||
|
|
if (block.metaData.type === BlockEnum.Tool)
|
||
|
|
return false
|
||
|
|
|
||
|
|
if (block.metaData.type === BlockEnum.IterationStart)
|
||
|
|
return false
|
||
|
|
|
||
|
|
if (block.metaData.type === BlockEnum.LoopStart)
|
||
|
|
return false
|
||
|
|
|
||
|
|
if (block.metaData.type === BlockEnum.DataSourceEmpty)
|
||
|
|
return false
|
||
|
|
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
}, [availableNodesMetaData?.nodes])
|
||
|
|
|
||
|
|
return (
|
||
|
|
<NodeSelector
|
||
|
|
{...props}
|
||
|
|
blocks={props.blocks || blocks}
|
||
|
|
dataSources={props.dataSources || dataSourceList || []}
|
||
|
|
/>
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
export default NodeSelectorWrapper
|