feat: LeftPane now shows the number of tables currently being displayed

liam-hq · Dec 16, 2024 · bfcbb3a · bfcbb3a
1 parent 009ad4f
commit bfcbb3a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/frontend/.changeset/few-socks-jam.md b/frontend/.changeset/few-socks-jam.md
@@ -0,0 +1,6 @@
+---
+"@liam-hq/erd-core": patch
+"@liam-hq/cli": patch
+---
+
+LeftPane now shows the number of tables currently being displayed
diff --git a/frontend/packages/erd-core/src/components/ERDRenderer/LeftPane/TableCounter/TableCounter.tsx b/frontend/packages/erd-core/src/components/ERDRenderer/LeftPane/TableCounter/TableCounter.tsx
@@ -1,13 +1,12 @@
 import { Table2 } from '@liam-hq/ui'
+import { useNodes } from '@xyflow/react'
 import type { FC } from 'react'
-import { useDBStructureStore } from '../../../../stores'
 import styles from './TableCounter.module.css'
 
 export const TableCounter: FC = () => {
-  const { tables } = useDBStructureStore()
-  const allCount = Object.keys(tables).length
-  // TODO: Implement filtering
-  const visibleCount = allCount
+  const nodes = useNodes()
+  const allCount = nodes.length
+  const visibleCount = nodes.filter((node) => !node.hidden).length
 
   return (
     <div className={styles.wrapper}>