From e7e12c1d2e9906d45be441e1385bfd65c7d2bf69 Mon Sep 17 00:00:00 2001
From: zxhlyh <jasonapring2015@outlook.com>
Date: Tue, 29 Apr 2025 16:26:53 +0800
Subject: [PATCH] fix: node selector

---
 .../field/variable-or-constant-input.tsx      |  2 +-
 .../workflow/block-selector/blocks.tsx        | 24 +++++++++----------
 .../workflow/block-selector/index.tsx         |  9 ++++---
 3 files changed, 19 insertions(+), 16 deletions(-)

diff --git a/web/app/components/base/form/components/field/variable-or-constant-input.tsx b/web/app/components/base/form/components/field/variable-or-constant-input.tsx
index 0f620f7154..7092275519 100644
--- a/web/app/components/base/form/components/field/variable-or-constant-input.tsx
+++ b/web/app/components/base/form/components/field/variable-or-constant-input.tsx
@@ -47,7 +47,7 @@ const VariableOrConstantInputField = ({
         <SegmentedControl
           className='mr-1 shrink-0'
           value={variableType}
-          onChange={handleVariableOrConstantChange}
+          onChange={handleVariableOrConstantChange as any}
           options={options as any}
         />
       </div>
diff --git a/web/app/components/workflow/block-selector/blocks.tsx b/web/app/components/workflow/block-selector/blocks.tsx
index de3992a0b7..64622d7de0 100644
--- a/web/app/components/workflow/block-selector/blocks.tsx
+++ b/web/app/components/workflow/block-selector/blocks.tsx
@@ -29,8 +29,8 @@ const Blocks = ({
 
   const groups = useMemo(() => {
     return BLOCK_CLASSIFICATIONS.reduce((acc, classification) => {
-      const list = groupBy(blocks, 'classification')[classification].filter((block) => {
-        return block.title.toLowerCase().includes(searchText.toLowerCase()) && availableBlocksTypes.includes(block.type)
+      const list = groupBy(blocks, 'metaData.classification')[classification].filter((block) => {
+        return block.metaData.title.toLowerCase().includes(searchText.toLowerCase()) && availableBlocksTypes.includes(block.metaData.type)
       })
 
       return {
@@ -42,7 +42,7 @@ const Blocks = ({
   const isEmpty = Object.values(groups).every(list => !list.length)
 
   const renderGroup = useCallback((classification: string) => {
-    const list = groups[classification].sort((a, b) => a.sort - b.sort)
+    const list = groups[classification].sort((a, b) => a.metaData.sort - b.metaData.sort)
 
     return (
       <div
@@ -59,7 +59,7 @@ const Blocks = ({
         {
           list.map(block => (
             <Tooltip
-              key={block.type}
+              key={block.metaData.type}
               position='right'
               popupClassName='w-[200px]'
               popupContent={(
@@ -67,25 +67,25 @@ const Blocks = ({
                   <BlockIcon
                     size='md'
                     className='mb-2'
-                    type={block.type}
+                    type={block.metaData.type}
                   />
-                  <div className='system-md-medium mb-1 text-text-primary'>{block.title}</div>
-                  <div className='system-xs-regular text-text-tertiary'>{block.description}</div>
+                  <div className='system-md-medium mb-1 text-text-primary'>{block.metaData.title}</div>
+                  <div className='system-xs-regular text-text-tertiary'>{block.metaData.description}</div>
                 </div>
               )}
             >
               <div
-                key={block.type}
+                key={block.metaData.type}
                 className='flex h-8 w-full cursor-pointer items-center rounded-lg px-3 hover:bg-state-base-hover'
-                onClick={() => onSelect(block.type)}
+                onClick={() => onSelect(block.metaData.type)}
               >
                 <BlockIcon
                   className='mr-2 shrink-0'
-                  type={block.type}
+                  type={block.metaData.type}
                 />
-                <div className='grow text-sm text-text-secondary'>{block.title}</div>
+                <div className='grow text-sm text-text-secondary'>{block.metaData.title}</div>
                 {
-                  block.type === BlockEnum.LoopEnd && (
+                  block.metaData.type === BlockEnum.LoopEnd && (
                     <Badge
                       text={t('workflow.nodes.loop.loopNode')}
                       className='ml-2 shrink-0'
diff --git a/web/app/components/workflow/block-selector/index.tsx b/web/app/components/workflow/block-selector/index.tsx
index 283a5d9a41..b2d11288b6 100644
--- a/web/app/components/workflow/block-selector/index.tsx
+++ b/web/app/components/workflow/block-selector/index.tsx
@@ -11,13 +11,16 @@ const NodeSelectorWrapper = (props: NodeSelectorProps) => {
     const result = availableNodesMetaData?.nodes || []
 
     return result.filter((block) => {
-      if (block.type === BlockEnum.Start)
+      if (block.metaData.type === BlockEnum.Start)
         return false
 
-      if (block.type === BlockEnum.IterationStart)
+      if (block.metaData.type === BlockEnum.DataSource)
         return false
 
-      if (block.type === BlockEnum.LoopStart)
+      if (block.metaData.type === BlockEnum.IterationStart)
+        return false
+
+      if (block.metaData.type === BlockEnum.LoopStart)
         return false
 
       return true