♻️ refactor: 优化 out 的节点逻辑

arvinxx · Mar 19, 2021 · 7a3c171 · 7a3c171
1 parent 4e14640
commit 7a3c171
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 deletions.
diff --git a/packages/mindflow/src/components/Port/OutPort.tsx b/packages/mindflow/src/components/Port/OutPort.tsx
@@ -8,17 +8,16 @@ import CollapseIcon from '../CollapseIcon';
 interface OutPortProps {
   color: string;
   collapsed: boolean;
-  visible: boolean;
   id: string;
 }
 
 export const OutPort: FC<OutPortProps> = ({
   color,
   collapsed,
-  visible,
+
   id,
 }) => {
-  return !visible ? null : (
+  return (
     <div className="mind-port-out-port">
       <CollapseIcon id={id} color={color} collapsed={collapsed} />
     </div>

diff --git a/packages/mindflow/src/definition/port.tsx b/packages/mindflow/src/definition/port.tsx
@@ -22,14 +22,9 @@ export const port = (args: Hook.OnPortRenderedArgs) => {
     ReactDOM.render(<InPort color={color} />, container as HTMLElement);
   }
 
-  if (port.id === 'out') {
+  if (port.id === 'out' && !leaf) {
     ReactDOM.render(
-      <OutPort
-        visible={!leaf}
-        color={color}
-        collapsed={collapsed}
-        id={node.id}
-      />,
+      <OutPort color={color} collapsed={collapsed} id={node.id} />,
       container as HTMLElement,
     );
   }