♻️ Replace hidden update logic with a shared function.

liam-hq · Feb 14, 2025 · 5ce3616 · 5ce3616
1 parent b746784
commit 5ce3616
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/.changeset/lucky-flowers-reply.md b/.changeset/lucky-flowers-reply.md
@@ -0,0 +1,6 @@
+---
+"@liam-hq/erd-core": patch
+"@liam-hq/cli": patch
+---
+
+♻️ Replace hidden update logic with a shared function.
diff --git a/frontend/packages/erd-core/src/components/ERDRenderer/ERDContent/useInitialAutoLayout.ts b/frontend/packages/erd-core/src/components/ERDRenderer/ERDContent/useInitialAutoLayout.ts
@@ -13,7 +13,9 @@ import type { Node } from '@xyflow/react'
 import { useCallback, useEffect, useMemo, useState } from 'react'
 import { useERDContentContext } from './ERDContentContext'
 import { computeAutoLayout } from './computeAutoLayout'
+import { hasNonRelatedChildNodes } from './hasNonRelatedChildNodes'
 import { highlightNodesAndEdges } from './highlightNodesAndEdges'
+import { updateNodesHiddenState } from './updateNodesHiddenState'
 
 export const useInitialAutoLayout = (
   nodes: Node[],
@@ -49,10 +51,12 @@ export const useInitialAutoLayout = (
 
     if (tableNodesInitialized) {
       setLoading(true)
-      const updatedNodes = nodes.map((node) => ({
-        ...node,
-        hidden: hiddenNodeIds.includes(node.id),
-      }))
+
+      const updatedNodes = updateNodesHiddenState({
+        nodes,
+        hiddenNodeIds,
+        shouldHideGroupNodeId: !hasNonRelatedChildNodes(nodes),
+      })
 
       const { nodes: highlightedNodes, edges: highlightedEdges } =
         highlightNodesAndEdges(updatedNodes, getEdges(), {