Merge pull request #218 from liam-hq/enhance-the-highlighting-of-rela…

…tionships-when-hovering Enhance the highlighting of relationships when hovering
liam-hq · Dec 11, 2024 · 576073b · 576073b
2 parents 2a6ccb0 + a5b96f9
commit 576073b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/frontend/packages/erd-core/src/components/ERDRenderer/ERDContent/ERDContent.tsx b/frontend/packages/erd-core/src/components/ERDRenderer/ERDContent/ERDContent.tsx
@@ -57,7 +57,7 @@ export const ERDContent: FC<Props> = ({
     (_, { id }) => {
       setEdges((edges) =>
         edges.map((e) =>
-          e.source === id
+          e.source === id || e.target === id
             ? { ...e, animated: true, data: { ...e.data, isHovered: true } }
             : e,
         ),
@@ -70,7 +70,7 @@ export const ERDContent: FC<Props> = ({
     (_, { id }) => {
       setEdges((edges) =>
         edges.map((e) =>
-          e.source === id
+          e.source === id || e.target === id
             ? { ...e, animated: false, data: { ...e.data, isHovered: false } }
             : e,
         ),