From b4f5560ffbc6644101566d5c316a130fec3d1e5a Mon Sep 17 00:00:00 2001
From: 1ilit <1ilit@proton.me>
Date: Sun, 2 Nov 2025 20:03:08 +0400
Subject: [PATCH] Show/hide tables

---
 src/components/EditorCanvas/Relationship.jsx  |  5 +-
 src/components/EditorCanvas/Table.jsx         |  2 +
 .../EditorSidePanel/TablesTab/TablesTab.jsx   | 49 +++++++++++++++++--
 src/components/SortableList/DragHandle.jsx    |  2 +-
 src/data/schemas.js                           |  1 +
 5 files changed, 52 insertions(+), 7 deletions(-)

diff --git a/src/components/EditorCanvas/Relationship.jsx b/src/components/EditorCanvas/Relationship.jsx
index 72457de98..a39f89f2c 100644
--- a/src/components/EditorCanvas/Relationship.jsx
+++ b/src/components/EditorCanvas/Relationship.jsx
@@ -19,7 +19,8 @@ export default function Relationship({ data }) {
     const startTable = tables.find((t) => t.id === data.startTableId);
     const endTable = tables.find((t) => t.id === data.endTableId);
 
-    if (!startTable || !endTable) return null;
+    if (!startTable || !endTable || startTable.hidden || endTable.hidden)
+      return null;
 
     return {
       startFieldIndex: startTable.fields.findIndex(
@@ -110,6 +111,8 @@ export default function Relationship({ data }) {
     }
   };
 
+  if (!pathValues) return null;
+
   return (
     <>
       <g className="select-none group" onDoubleClick={edit}>
diff --git a/src/components/EditorCanvas/Table.jsx b/src/components/EditorCanvas/Table.jsx
index 1bf471ed9..9cd5c472f 100644
--- a/src/components/EditorCanvas/Table.jsx
+++ b/src/components/EditorCanvas/Table.jsx
@@ -129,6 +129,8 @@ export default function Table({
     }
   };
 
+  if (tableData.hidden) return null;
+
   return (
     <>
       <foreignObject
diff --git a/src/components/EditorSidePanel/TablesTab/TablesTab.jsx b/src/components/EditorSidePanel/TablesTab/TablesTab.jsx
index 42e7815bd..b1b2fa152 100644
--- a/src/components/EditorSidePanel/TablesTab/TablesTab.jsx
+++ b/src/components/EditorSidePanel/TablesTab/TablesTab.jsx
@@ -1,7 +1,14 @@
 import { Collapse, Button } from "@douyinfe/semi-ui";
+import { IconEyeOpened, IconEyeClosed } from "@douyinfe/semi-icons";
 import { IconPlus } from "@douyinfe/semi-icons";
-import { useSelect, useDiagram, useSaveState, useLayout } from "../../../hooks";
-import { ObjectType, State } from "../../../data/constants";
+import {
+  useSelect,
+  useDiagram,
+  useSaveState,
+  useLayout,
+  useUndoRedo,
+} from "../../../hooks";
+import { Action, ObjectType, State } from "../../../data/constants";
 import { useTranslation } from "react-i18next";
 import { DragHandle } from "../../SortableList/DragHandle";
 import { SortableList } from "../../SortableList/SortableList";
@@ -67,24 +74,56 @@ export default function TablesTab() {
 
 function TableListItem({ table }) {
   const { layout } = useLayout();
+  const { updateTable } = useDiagram();
+  const { setUndoStack, setRedoStack } = useUndoRedo();
+  const { t } = useTranslation();
+
+  const toggleTableVisibility = (e) => {
+    e.stopPropagation();
+    setUndoStack((prev) => [
+      ...prev,
+      {
+        action: Action.EDIT,
+        element: ObjectType.TABLE,
+        component: "self",
+        tid: table.id,
+        undo: { hidden: table.hidden },
+        redo: { hidden: !table.hidden },
+        message: t("edit_table", {
+          tableName: table.name,
+          extra: "[hidden]",
+        }),
+      },
+    ]);
+    setRedoStack([]);
+    updateTable(table.id, { hidden: !table.hidden });
+  };
 
   return (
     <div id={`scroll_table_${table.id}`}>
       <Collapse.Panel
         className="relative"
         header={
-          <>
-            <div className="flex items-center gap-2">
+          <div className="flex items-center justify-between w-full">
+            <div className="flex items-center gap-2 flex-1">
               <DragHandle readOnly={layout.readOnly} id={table.id} />
               <div className="overflow-hidden text-ellipsis whitespace-nowrap">
                 {table.name}
               </div>
             </div>
+            <Button
+              size="small"
+              theme="borderless"
+              type="tertiary"
+              onClick={toggleTableVisibility}
+              icon={table.hidden ? <IconEyeClosed /> : <IconEyeOpened />}
+              className="me-2"
+            />
             <div
               className="w-1 h-full absolute top-0 left-0 bottom-0"
               style={{ backgroundColor: table.color }}
             />
-          </>
+          </div>
         }
         itemKey={`${table.id}`}
       >
diff --git a/src/components/SortableList/DragHandle.jsx b/src/components/SortableList/DragHandle.jsx
index dc5b15a63..38cd2b454 100644
--- a/src/components/SortableList/DragHandle.jsx
+++ b/src/components/SortableList/DragHandle.jsx
@@ -6,7 +6,7 @@ export function DragHandle({ id, readOnly }) {
 
   return (
     <div
-      className={`opacity-50 mt-0.5 ${readOnly ? "cursor-not-allowed" : "cursor-move"}`}
+      className={`opacity-50 mt-1.5 ${readOnly ? "cursor-not-allowed" : "cursor-move"}`}
       {...(!readOnly && listeners)}
     >
       <IconHandle />
diff --git a/src/data/schemas.js b/src/data/schemas.js
index 2c28a4b43..4955821da 100644
--- a/src/data/schemas.js
+++ b/src/data/schemas.js
@@ -39,6 +39,7 @@ export const tableSchema = {
     },
     comment: { type: "string" },
     locked: { type: "boolean" },
+    hidden: { type: "boolean" },
     indices: {
       type: "array",
       items: {