dify

dify/web/app/components/workflow/block-selector/index.tsx

@@ -0,0 +1,49 @@
+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